mirror of
https://github.com/himool/HimoolERP.git
synced 2024-09-20 14:56:00 +08:00
28 lines
782 B
Python
28 lines
782 B
Python
from rest_framework_simplejwt.authentication import JWTAuthentication
|
|
from rest_framework_simplejwt.exceptions import InvalidToken
|
|
from extensions.exceptions import NotAuthenticated
|
|
from apps.system.models import User
|
|
|
|
|
|
class BaseAuthentication(JWTAuthentication):
|
|
|
|
def authenticate(self, request):
|
|
if (header := self.get_header(request)) is None:
|
|
return None
|
|
|
|
if (raw_token := self.get_raw_token(header)) is None:
|
|
return None
|
|
|
|
try:
|
|
validated_token = self.get_validated_token(raw_token)
|
|
user = User.objects.get(id=validated_token['user_id'])
|
|
except (KeyError, InvalidToken, User.DoesNotExist):
|
|
return None
|
|
|
|
return user, validated_token
|
|
|
|
|
|
__all__ = [
|
|
'BaseAuthentication',
|
|
]
|