python 为Django restframework生成JWT（JSON WEB TOKEN）

Posted 2021-05-09

from rest_framework_jwt.settings import api_settings


def jwt_encode(user):
    """
    See https://github.com/Tivix/django-rest-auth/blob/master/rest_auth/utils.py#L19
    and http://getblimp.github.io/django-rest-framework-jwt/#creating-a-new-token-manually
    """
    jwt_payload_handler = api_settings.JWT_PAYLOAD_HANDLER
    jwt_encode_handler = api_settings.JWT_ENCODE_HANDLER

    payload = jwt_payload_handler(user)
    return jwt_encode_handler(payload)

class LoginView(GenericAPIView):
    '''Login endpoint, returns the auth token'''
    permission_classes = (permissions.AllowAny,)
    serializer_class = LoginSerializer

    def post(self, request):
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        return Response({
            'token': jwt_encode(serializer.user)
        }, status=status.HTTP_200_OK)