Tivix/django-rest-auth
1
1
Fork 0
mirror of https://github.com/Tivix/django-rest-auth.git synced 2025-06-11 00:13:05 +03:00
Code Issues Packages Projects Releases Wiki Activity

try/except Knox AuthToken model import

Browse Source
This commit is contained in:
Daniel Stanton 2017-01-24 13:53:06 +00:00
parent 0c9c9b5a5a
commit d76b685c36
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
rest_auth/models.py
View File

@ -1,7 +1,10 @@
from django.conf import settings
if getattr(settings, 'REST_USE_KNOX', False):
from knox.models import AuthToken as DefaultTokenModel
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