datatracker/tastypie/compat.py
2014-12-18 16:05:15 +00:00

25 lines
782 B
Python

from __future__ import unicode_literals
from django.conf import settings
from django.core.exceptions import ImproperlyConfigured
import django
__all__ = ['User', 'AUTH_USER_MODEL']
AUTH_USER_MODEL = getattr(settings, 'AUTH_USER_MODEL', 'auth.User')
# Django 1.5+ compatibility
if django.VERSION >= (1, 5):
try:
from django.contrib.auth import get_user_model
User = get_user_model()
username_field = User.USERNAME_FIELD
except ImproperlyConfigured:
# The the users model might not be read yet.
# This can happen is when setting up the create_api_key signal, in your
# custom user module.
User = None
username_field = None
else:
from django.contrib.auth.models import User
username_field = 'username'