fix failing django 1.4 test setup

crccheck · Sep 3, 2014 · f1369aa · f1369aa
1 parent df48a1c
commit f1369aa
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/django_object_actions/factories.py b/django_object_actions/factories.py
@@ -1,4 +1,9 @@
-from django.contrib.auth import get_user_model
+try:
+    from django.contrib.auth import get_user_model
+except ImportError:
+    # Django 1.4
+    from django.contrib.auth.models import User
+    get_user_model = lambda: User
 
 import factory