1、修改serializer.py，增加自定义类
# 自定义用户登录token等返回信息
class MyTokenObtainPair(TokenObtainPairView):
def post(self, request, *args, **kwargs):
serializer = self.get_serializer(data=request.data)
try:
serializer.is_valid(raise_exception=True)
except TokenError as e:
raise InvalidToken(e.args[0])
# 获取所有值
result = serializer.validated_data
result["email"] = serializer.user.email
result["username"] = serializer.user.username
result["id"] = serializer.user.id
# 重写token名字
result["token"] = result.pop("access")
return Response(result, status=status.HTTP_200_OK)
完整代码：
from django.shortcuts import render, HttpResponse
from rest_framework.views import APIViewfrom .models import Linkfrom rest_framework import serializers, status
from rest_framework.response import Response
from rest_framework_simplejwt.exceptions import TokenError, InvalidToken
from rest_framework_simplejwt.views import TokenObtainPairView# 和模型直接耦合，确定不灵活，上面的是自己写
class LinkSerializer(serializers.ModelSerializer):# 自定义接口字段名称source="name"是真实的，mingzi是解耦的名称mingzi = serializers.CharField(max_length=300, source="name")class Meta:# 指定关联哪个模型集model = Link# fields = "__all__"# 排除去掉哪些字段exclude = ["gid", "name"]# 自定义用户登录token等返回信息
class MyTokenObtainPair(TokenObtainPairView):def post(self, request, *args, **kwargs):serializer = self.get_serializer(data=request.data)try:serializer.is_valid(raise_exception=True)except TokenError as e:raise InvalidToken(e.args[0])# 获取所有值result = serializer.validated_dataresult["email"] = serializer.user.emailresult["username"] = serializer.user.usernameresult["id"] = serializer.user.id# 重写token名字result["token"] = result.pop("access")return Response(result, status=status.HTTP_200_OK)
2、修改api模块中的urls.py
# 登录
# path("token/", TokenObtainPairView.as_view(), name="token_obtain_pair"),
path("token/", serializer.MyTokenObtainPair.as_view(), name="token_obtain_pair"),
完整代码
from django.urls import path, re_path
from . import viewsfrom . import serializerfrom rest_framework import routers
from rest_framework_simplejwt.views import (TokenObtainPairView,TokenRefreshView,TokenVerifyView,TokenBlacklistView,
)routers = routers.DefaultRouter()
routers.register("linkapi", views.Linkapi)app_name = "api"
urlpatterns = [# 登录# path("token/", TokenObtainPairView.as_view(), name="token_obtain_pair"),path("token/", serializer.MyTokenObtainPair.as_view(), name="token_obtain_pair"),# 刷新path("token/refresh/", TokenRefreshView.as_view(), name="token_refresh"),# 校验path("token/verify/", TokenVerifyView.as_view(), name="token_verify"),# 黑名单path("token/blacklist/", TokenBlacklistView.as_view(), name="token_blacklist"),
]urlpatterns += routers.urls
3、基于上一节的代码，其他代码不变
4、测试
