Use dynamically generated schema, and use specified username field

Author: demiurg

ninja_jwt/schema.py

@@ -4,11 +4,11 @@
 from django.contrib.auth import authenticate, get_user_model
 from django.contrib.auth.models import AbstractUser, update_last_login
 from django.utils.translation import gettext_lazy as _
+from ninja.orm import create_schema
+from ninja_jwt.utils import token_error
 from ninja_schema import ModelSchema, Schema
 from pydantic import root_validator
 
-from ninja_jwt.utils import token_error
-
 from . import exceptions
 from .settings import api_settings
 from .tokens import RefreshToken, SlidingToken, Token, UntypedToken
@@ -19,6 +19,9 @@
 user_name_field = get_user_model().USERNAME_FIELD  # type: ignore
 
 
+AuthUserSchema = create_schema(get_user_model(), fields=[user_name_field])
+
+
 class TokenObtainSerializer(ModelSchema):
     class Config:
         model = get_user_model()
@@ -69,10 +72,9 @@ def get_token(cls, user: Type[AbstractUser]) -> Type[Token]:
         )
 
 
-class TokenObtainPairOutput(Schema):
+class TokenObtainPairOutput(AuthUserSchema):
     refresh: str
     access: str
-    username: str
 
 
 class TokenObtainPairSerializer(TokenObtainSerializer):
@@ -97,9 +99,8 @@ def output_schema(self):
         return TokenObtainPairOutput(**self.dict(exclude={"password"}))
 
 
-class TokenObtainSlidingOutput(Schema):
+class TokenObtainSlidingOutput(AuthUserSchema):
     token: str
-    username: str
 
 
 class TokenObtainSlidingSerializer(TokenObtainSerializer):