django-rest-auth/rest_auth/models.py

14 lines
378 B
Python

from django.conf import settings
if getattr(settings, 'REST_USE_KNOX', False):
from knox.models import AuthToken as DefaultTokenModel
else:
from rest_framework.authtoken.models import Token as DefaultTokenModel
from .utils import import_callable
# Register your models here.
TokenModel = import_callable(
getattr(settings, 'REST_AUTH_TOKEN_MODEL', DefaultTokenModel))