mirror of
https://github.com/Tivix/django-rest-auth.git
synced 2025-06-15 02:13:12 +03:00
17 lines
497 B
Python
17 lines
497 B
Python
from django.conf import settings
|
|
|
|
if getattr(settings, 'REST_USE_KNOX', False):
|
|
try:
|
|
from knox.models import AuthToken as DefaultTokenModel
|
|
except ImportError:
|
|
raise ImportError("Install django-rest-knox to use REST_USE_KNOX = True")
|
|
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))
|