Merge pull request #3 from colinferm/cf-fixing-nonetype

Fixed an issue where sometimes there are models with a model_class() …
LegoStormtroopr · Sep 22, 2015 · a95b168 · a95b168
2 parents 5ee3580 + 50a4651
commit a95b168
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/django_spaghetti/views.py b/django_spaghetti/views.py
@@ -16,6 +16,9 @@ def plate(request):
     nodes = []
     edges = []
     for model in models:
+        if (model.model_class() == None):
+            continue
+
         model.doc  = model.model_class().__doc__
         _id = "%s__%s"%(model.app_label,model.model)
         if _id in excludes: