From 70665dffcc0ce1e656d5eddb4d557f4b68d66424 Mon Sep 17 00:00:00 2001 From: Luca Deluigi Date: Fri, 30 Aug 2024 09:58:41 +0200 Subject: [PATCH] Add schema info to custom token pair serializer --- backend/backend/login/jwt.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/backend/backend/login/jwt.py b/backend/backend/login/jwt.py index fd45f9d7..3bc15d1a 100644 --- a/backend/backend/login/jwt.py +++ b/backend/backend/login/jwt.py @@ -23,6 +23,8 @@ def refresh_token(self) -> RefreshToken: class TokenObtainPairSerializer(BaseTokenObtainPairSerializer): + refresh = CharField(read_only=True) + access = CharField(read_only=True) code_param = 'code' code = False @@ -33,6 +35,10 @@ def __init__(self, *args, **kwargs) -> None: self.fields.clear() self.fields[self.code_param] = CharField(write_only=True) self.code = True + else: + self.fields[self.code_param] = CharField(write_only=True, required=False) + for field in self.fields.values(): + field.required = False def validate(self, attrs: dict[str, Any]) -> dict[str, str]: if not self.code: