Add schema info to custom token pair serializer

SpaceCowMedia · Aug 30, 2024 · 70665df · 70665df
1 parent 1b7b77c
commit 70665df
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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: