Allow string model names

Author: jpic

dbdiff/tests/test_utils.py

@@ -0,0 +1,7 @@
+from dbdiff.utils import get_model_names
+
+from django.contrib.auth.models import Group
+
+
+def test_get_model_names():
+    assert get_model_names([Group, 'auth.user']) == ['auth.group', 'auth.user']

dbdiff/utils.py

@@ -6,6 +6,8 @@
 from django.apps import apps
 from django.db import connections
 
+import six
+
 
 def get_tree(dump, exclude=None):
     """Return a tree of model -> pk -> fields."""
@@ -90,8 +92,11 @@ def get_absolute_path(path):
 
 def get_model_names(model_classes):
     """Return model names for model classes."""
-    return ['%s.%s' % (m._meta.app_label, m._meta.model_name)
-            for m in model_classes]
+    return [
+        m if isinstance(m, six.string_types)
+        else '%s.%s' % (m._meta.app_label, m._meta.model_name)
+        for m in model_classes
+    ]
 
 
 def get_models_tables(models):

tox.ini

@@ -33,7 +33,7 @@ passenv = TEST_* DBDIFF_*
 basepython = python2.7
 commands =
     flake8 --show-source --exclude tests --max-complexity=7 --ignore=D203 dbdiff
-    flake8 --show-source --exclude migrations --max-complexity=3 --ignore=D100,D101,D102 dbdiff/tests
+    flake8 --show-source --exclude migrations --max-complexity=3 --ignore=D100,D101,D102,D103 dbdiff/tests
 
 deps =
     flake8