fix import user to works in old django

valdergallo · Oct 8, 2016 · 0f77cbc · 0f77cbc
1 parent fd211a3
commit 0f77cbc
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/data_importer/models.py b/data_importer/models.py
@@ -18,8 +18,11 @@
 from django.contrib.contenttypes.models import ContentType
 
 # port settings.AUTH_USER_MODEL
-# from django.contrib.auth import get_user_model
-User = settings.AUTH_USER_MODEL
+try:
+    User = settings.AUTH_USER_MODEL
+except AttributeError:
+    from django.contrib.auth.models import User
+
 
 DJANGO_VERSION = StrictVersion(django.get_version())