使用 Django AUTH_USER 模型通过 JWT Token 传递数据（例如，first_name，last_name）

Question

我正在使用 ModelViewSet 开发 Django RF，我将在其中使用 JWT 令牌身份验证。我想使用默认的 django auth_user 模型而不是创建新的自定义用户模型。

#serializer.py
class MyTokenObtainPairSerializer(TokenObtainPairSerializer):
    class Meta:
    model = User
    fields = ('username','first_name')
    def validate(self, attrs):
        data = super().validate(attrs)
        refresh = self.get_token(self.user)
        data['refresh'] = str(refresh)
        data.pop('refresh', None) # remove refresh from the payload
        data['access'] = str(refresh.access_token)

        # Add extra responses here
        data['username'] = self.user.username
        data['first_name'] = self.user.first_name
        data['last_name'] = self.user.last_name
        return data

views.py
class TokenObtainPairView(TokenObtainPairView):
# Replace the serializer with your custom
serializer_class = MyTokenObtainPairSerializer
# token_obtain_pair = TokenObtainPairView.as_view()



class MyTokenObtainPairSerializer(TokenObtainPairSerializer):
class Meta:
    model = User
    fields = ('username','first_name')
@classmethod
def get_token(cls, user):
    token = super().get_token(user)
    token['username'] = user.username
    token['first_name'] = user.first_name
    return token

Answer 1

如果您没有在 settings.py 中设置AUTH_USER_MODEL = 'myapp.MyUser'，则默认使用User 而不是MyUser。

settings.py

如果您使用simplejwt，则必须分配身份验证类。

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

---

CustomTokenObtainPairView 不自定义令牌

好的，明白了。非常感谢@Klim Bim。你可以帮我一个忙吗。你能告诉我如何使用 simple-jwt 令牌添加其他字段。我曾尝试使用自定义 TokenObtaibPairView'，但我得到的只是 user_id。

您可以自定义token claims，也可以将数据添加到响应中。

serializer.py

class CustomTokenObtainPairSerializer(TokenObtainPairSerializer):
    @classmethod
    def get_token(cls, user):
        token = super().get_token(user)
        token['username'] = user.username
        token['first_name'] = user.first_name

        return token

views.py

class CustomTokenObtainPairView(TokenObtainPairView):

   serializer_class = CustomTokenObtainPairSerializer

   def post(self, request, *args, **kwargs):

      response = super().post(request, *args, **kwargs)
      response.data['custom_key'] =  'my_custom_data'
      
      return response

urls.py

urlpatterns = [
    path('token/', CustomTokenObtainPairView.as_view(), name='token_obtain_pair'),
]

邮递员