encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2024-11-26 03:23:59 +03:00
Code Issues Packages Projects Releases Wiki Activity

Django 1.5 support, and awareness for AUTH_USER_MODEL

Browse Source
This commit is contained in:
Jonas Liljestrand 2012-11-17 17:46:16 +01:00
parent 346a79b170
commit f0d4232c1d
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
rest_framework/authtoken/models.py
View File

@ -2,6 +2,20 @@ 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):
@ -9,7 +23,7 @@ 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):