/Python/How to load user from jwt token request django

How to load user from jwt token request django

from django.conf import settings
from rest_framework import authentication
from rest_framework import exceptions
from rest_framework.authentication import get_authorization_header
import CustomUser # just import your model here
import jwt

class JWTAuthentication(authentication.BaseAuthentication):
    def authenticate(self, request): # it will return user object
        try:
            token = get_authorization_header(request).decode('utf-8')
            if token is None or token == "null" or token.strip() == "":
                raise exceptions.AuthenticationFailed('Authorization Header or Token is missing on Request Headers')
            print(token)
            decoded = jwt.decode(token, settings.SECRET_KEY)
            username = decoded['username']
            user_obj = CustomUser.objects.get(username=username)
        except jwt.ExpiredSignature :
            raise exceptions.AuthenticationFailed('Token Expired, Please Login')
        except jwt.DecodeError :
            raise exceptions.AuthenticationFailed('Token Modified by thirdparty')
        except jwt.InvalidTokenError:
            raise exceptions.AuthenticationFailed('Invalid Token')
        except Exception as e:
            raise exceptions.AuthenticationFailed(e)
        return (user_obj, None)

    def get_user(self, userid):
        try:
            return CustomUser.objects.get(pk=userid)
        except Exception as e:
            return None

Source: stackoverflow.com

Similar codes

0 vote322 views1 answer
How to replace an element of a list using list comprehension
Python
1 vote338 views3 answers
Check if number is between two numbers python
Python
1 vote341 views1 answer
Django httpresponseredirect
Python
2 votes331 views1 answer
Get n random numbers from x to y python
Python
0 vote349 views1 answer
Pytorch dill model save
Python
0 vote320 views1 answer
How to resolve This typically means that you attempted to use functionality that needed an active HTTP request. Consult the documentation on testing for information about how to avoid this problem. in thread python
Python
1 vote390 views1 answer
Django link home page
Python Django
26 votes349 views3 answers
Check if variable is empty python
Python
1 vote357 views1 answer
Python requests force ipv4
Python
0 vote346 views1 answer
Python add new line from textarea
Python