Django REST框架自定义令牌（带有令牌访问和求章的用户对象）

Question

我正在尝试创建一个自定义令牌，以将用户对象与访问和刷新一起发送。

注意：如果还有另一种方法，请告诉我。从头开始启动令牌代码是一个选择。

token_serializer.py

from rest_framework_simplejwt.serializers import (
    TokenObtainSerializer,
    RefreshToken,
    api_settings,
    update_last_login,
)
from django.core import serializers
from app.models.user import AppUser


class TokenSerializer(TokenObtainSerializer):
    @classmethod
    def get_token(cls, user):
        return RefreshToken.for_user(user)

    def validate(self, attrs):
        data = super().validate(attrs)

        refresh = self.get_token(self.user)

        data["refresh"] = str(refresh)
        data["access"] = str(refresh.access_token)

        @@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@
        serialized_app_user = serializers.serialize("json", AppUser)
        app_user = serialized_app_user .objects.filter(pk=self.user.id).first()
        @@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@

        
        if app_user:
            data["user"] = app_user
        else:
            data["user"] = None

        if api_settings.UPDATE_LAST_LOGIN:
            update_last_login(None, self.user)

        return data

token_view.py

from rest_framework_simplejwt.views import TokenViewBase
from app.api.serializers import TokenSerializer


class TokenView(TokenViewBase):
    serializer_class = TokenSerializer

此代码引起此错误：'modelbase'对象是不可能的

导致此错误的行，

serialized_app_user = serializers.serialize("json", AppUser)

但我添加了，因为没有以下错误发生：object object object类型Appuser不是JSON序列化

Answer 1

如果您确实需要令牌中的用户信息获得端点，则需要序列化appuser对象：

...
from rest_framework import serializers
from app.models.user import AppUser


class UserSerializer(serializers.ModelSerializer):
    class Meta:
        model = AppUser
        fields = '__all__'  # or any specific fields


class TokenSerializer(TokenObtainSerializer):
    def validate(self, attrs):
        ...
        app_user = serialized_app_user .objects.filter(pk=self.user.id).first()
        if app_user:
            data["user"] = UserSerializer(app_user).data
        else:
            data["user"] = None