Add support for custom user models

tos/models.py

@@ -1,8 +1,9 @@
 from django.core.exceptions import ValidationError
 from django.db import models
-from django.contrib.auth.models import User
+from django.contrib.auth import get_user_model
 from django.utils.translation import ugettext_lazy as _
 
+User = get_user_model()
 
 class NoActiveTermsOfService(ValidationError):
     pass

tos/tests/test_models.py

@@ -1,8 +1,10 @@
-from django.contrib.auth.models import User 
+from django.contrib.auth import get_user_model
 from django.core.exceptions import ValidationError 
 from django.test import TestCase 
 from django.db.models.signals import post_syncdb 
 
+User = get_user_model()
+
 from tos.models import TermsOfService, UserAgreement, has_user_agreed_latest_tos
 
 class TestModels(TestCase):

tos/tests/test_views.py

@@ -1,11 +1,13 @@
 from django.conf import settings
-from django.contrib.auth.models import User 
+from django.contrib.auth import get_user_model
 from django.core.exceptions import ValidationError 
 from django.core.urlresolvers import reverse 
 from django.test import TestCase 
 
 from tos.models import TermsOfService, UserAgreement, has_user_agreed_latest_tos
 
+User = get_user_model()
+
 class TestViews(TestCase):
 
     def setUp(self):