自定义 rest_framework_simplejwt 返回值

小毛驴SVIP

已于 2022-12-22 15:45:14 修改

阅读量449

点赞数

分类专栏： django 文章标签： django

于 2022-12-22 15:34:47 首次发布

本文链接：https://blog.csdn.net/qq_31080741/article/details/128409334

版权

django 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

# # settings.py

from datetime import timedelta

# # 认证
REST_FRAMEWORK = {
    'DEFAULT_AUTHENTICATION_CLASSES': (
        'rest_framework_simplejwt.authentication.JWTAuthentication',
    ),
}


SIMPLE_JWT = {
    'REFRESH_TOKEN_LIFETIME': timedelta(days=1),
    'ACCESS_TOKEN_LIFETIME': timedelta(days=1),
    'ROTATE_REFRESH_TOKENS': True,
    # 'AUTH_HEADER_TYPES': ("Bearer",),
    'AUTH_HEADER_NAME': 'HTTP_TOKEN',
    'AUTH_TOKEN_CLASSES': ('rest_framework_simplejwt.tokens.AccessToken',
                           'rest_framework_simplejwt.tokens.SlidingToken'),
}

# # custom_token_serializer.py

from rest_framework_simplejwt.serializers import TokenObtainPairSerializer, TokenObtainSlidingSerializer


class MyTokenObtainPairSerializer(TokenObtainSlidingSerializer):

    def validate(self, attrs):

        data = super().validate(attrs)
        username = self.user.username
        data["username"] = username

    @classmethod
    def get_token(cls, user):
        token = super().get_token(user)

        token['username'] = user.username

        return token