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))