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

Added create_knox_token

Browse Source 
as default_create_token fails
This commit is contained in:
Daniel Stanton 2017-01-24 14:34:06 +00:00
parent e9c4ecbc07
commit 83f4db64a4
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
rest_auth/app_settings.py
View File

@ -2,8 +2,10 @@ from django.conf import settings
if getattr(settings, 'REST_AUTH_TOKEN_APP', False) is 'knox': if getattr(settings, 'REST_AUTH_TOKEN_APP', False) is 'knox':
from rest_auth.serializers import KnoxTokenSerializer as DefaultTokenSerializer from rest_auth.serializers import KnoxTokenSerializer as DefaultTokenSerializer
from .utils import create_knox_token as default_create_token
else: else:
from rest_auth.serializers import TokenSerializer as DefaultTokenSerializer from rest_auth.serializers import TokenSerializer as DefaultTokenSerializer
from .utils import default_create_token
from rest_auth.serializers import ( from rest_auth.serializers import (
JWTSerializer as DefaultJWTSerializer, JWTSerializer as DefaultJWTSerializer,
@ -12,7 +14,7 @@ from rest_auth.serializers import (
PasswordResetSerializer as DefaultPasswordResetSerializer, PasswordResetSerializer as DefaultPasswordResetSerializer,
PasswordResetConfirmSerializer as DefaultPasswordResetConfirmSerializer, PasswordResetConfirmSerializer as DefaultPasswordResetConfirmSerializer,
PasswordChangeSerializer as DefaultPasswordChangeSerializer) PasswordChangeSerializer as DefaultPasswordChangeSerializer)
from .utils import import_callable, default_create_token from .utils import import_callable
create_token = import_callable( create_token = import_callable(
getattr(settings, 'REST_AUTH_TOKEN_CREATOR', default_create_token)) getattr(settings, 'REST_AUTH_TOKEN_CREATOR', default_create_token))

5
rest_auth/utils.py
View File

@ -16,6 +16,11 @@ def default_create_token(token_model, user, serializer):
return token return token
def create_knox_token(token_model, user, serializer):
token = token_model.objects.get_or_create(user=user)
return token
def jwt_encode(user): def jwt_encode(user):
try: try:
from rest_framework_jwt.settings import api_settings from rest_framework_jwt.settings import api_settings