Use Knox AuthToken as default Token Model if REST_USE_KNOX is True

This commit is contained in:
Daniel Stanton 2017-01-24 12:47:46 +00:00
parent ea9bdec6ca
commit ca0cbd88e6

View File

@ -1,6 +1,9 @@
from django.conf import settings
from rest_framework.authtoken.models import Token as DefaultTokenModel
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