Django 1.5 support, and awareness for AUTH_USER_MODEL

1 files changed, 15 insertions, 1 deletions

diff --git a/rest_framework/authtoken/models.py b/rest_framework/authtoken/models.py
index 5b3071aa..610de8d8 100644
--- a/rest_framework/authtoken/models.py
+++ b/rest_framework/authtoken/models.py
@@ -2,14 +2,28 @@ import uuid
 import hmac
 from hashlib import sha1
 from django.db import models
+from django import VERSION
 
+try:
+    from django.db.models.auth import User
+        user_model = User
+    except ImportError:
+        raise ImportError
+    else:
+        raise
+
+if VERSION[:2] in ((1, 5,),):
+    from django.conf import settings
+    if hasattr(settings, AUTH_USER_MODEL):
+        user_model = settings.AUTH_USER_MODEL
+    
 
 class Token(models.Model):
     """
     The default authorization token model.
     """
     key = models.CharField(max_length=40, primary_key=True)
-    user = models.OneToOneField('auth.User', related_name='auth_token')
+    user = models.OneToOneField(user_model, related_name='auth_token')
     created = models.DateTimeField(auto_now_add=True)
 
     def save(self, *args, **kwargs):