darkor

login serializer

Aug 3rd, 2023
993
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.73 KB | None | 0 0
  1. class CustomLoginSerializer(JWTSerializer):
  2.     is_first_login = serializers.SerializerMethodField()
  3.  
  4.     def get_is_first_login(self, obj: dict) -> bool:
  5.         if obj["user"].country.code is None:
  6.             return True
  7.         return False
  8.  
  9.     def to_representation(self, instance: dict) -> dict:
  10.         ret = super().to_representation(instance)
  11.         del ret["user"]
  12.         return ret
  13.  
  14.  
  15. class CustomLoginSerializerV2(serializers.Serializer):
  16.     access = serializers.CharField()
  17.     refresh = serializers.CharField()
  18.     is_first_login = serializers.SerializerMethodField()
  19.  
  20.     def get_is_first_login(self, obj: dict) -> bool:
  21.         if obj["user"].country.code is None:
  22.             return True
  23.         return False
Advertisement
Add Comment
Please, Sign In to add comment