Needs to support other objects besides django.contrib.auth.models.Use…

…r ref pinax#1

brabeion/models.py

@@ -1,7 +1,14 @@
-from django.contrib.auth.models import User
 from django.db import models
 from django.utils import timezone
 
+try:
+    from django.contrib.auth import get_user_model
+except ImportError:
+    # Prior to Django 1.5, the AUTH_USER_MODEL setting does not exist.
+    from django.contrib.auth.models import User
+else:
+    User = get_user_model()
+
 
 class BadgeAward(models.Model):
     user = models.ForeignKey(User, related_name="badges_earned")

brabeion/tests/models.py

@@ -1,6 +1,13 @@
-from django.contrib.auth.models import User
 from django.db import models
 
+try:
+    from django.contrib.auth import get_user_model
+except ImportError:
+    # Prior to Django 1.5, the AUTH_USER_MODEL setting does not exist.
+    from django.contrib.auth.models import User
+else:
+    User = get_user_model()
+
 
 class PlayerStat(models.Model):
     user = models.OneToOneField(User, related_name="stats")

brabeion/tests/tests.py

@@ -1,12 +1,19 @@
 from django.conf import settings
-from django.contrib.auth.models import User
 from django.db import connection
 from django.test import TestCase
 
 from brabeion import badges
 from brabeion.base import Badge, BadgeAwarded
 from brabeion.tests.models import PlayerStat
 
+try:
+    from django.contrib.auth import get_user_model
+except ImportError:
+    # Prior to Django 1.5, the AUTH_USER_MODEL setting does not exist.
+    from django.contrib.auth.models import User
+else:
+    User = get_user_model()
+
 
 class PointsBadge(Badge):
     slug = "points"