From cb86fda8497b26177c4b7bd5ce9d8435800ca60b Mon Sep 17 00:00:00 2001 From: Abbas Toof Date: Mon, 8 Jul 2024 23:18:45 +0300 Subject: [PATCH] fix: Correct return type in CustomTokenObtainPairSerializer.get_token method --- .../auth_service/user_auth/serializers.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/Backend/auth_service/auth_service/user_auth/serializers.py b/Backend/auth_service/auth_service/user_auth/serializers.py index 166a525..2acffd2 100644 --- a/Backend/auth_service/auth_service/user_auth/serializers.py +++ b/Backend/auth_service/auth_service/user_auth/serializers.py @@ -1,22 +1,24 @@ from rest_framework_simplejwt.serializers import TokenObtainPairSerializer -import asyncio +from rest_framework_simplejwt.tokens import Token + class CustomTokenObtainPairSerializer(TokenObtainPairSerializer): @classmethod - def get_token(cls, user) -> dict: + def get_token(cls, user) -> Token: """ Get token method to generate tokens for the user. - This method overrides the get_token method of TokenObtainPairSerializer to generate tokens for the user. + This method overrides the get_token method of TokenObtainPairSerializer to generate tokens for the user. It generates the tokens for the user and returns the tokens. Args: user: The user object. - + Returns: dict: The dictionary containing the tokens. """ token = super().get_token(user) token["custom_claims"] = {"username": user.username, "password": user.password} - return token \ No newline at end of file + return token +