帶有 Djoser 令牌身份驗證令牌響應的 Django 休息框架

Question

因為我正在使用名為 djoser 的第三方包來處理令牌身份驗證。 我想自定義響應。 但是在嘗試改變它之后，結果不是我想要的。 我只是想獲取令牌的令牌值，而不是在前面有“auth_token:”。

這是 djoser 的鏈接： https : //github.com/sunscrapers/djoser

這是我的代碼：

序列化程序.py

class TokenCreateSerializer(serializers.Serializer):
    password = serializers.CharField(
        required=False, style={'input_type': 'password'}
    )

    default_error_messages = {
        'invalid_credentials': constants.INVALID_CREDENTIALS_ERROR,
        'inactive_account': constants.INACTIVE_ACCOUNT_ERROR,
    }

    def __init__(self, *args, **kwargs):
        super(TokenCreateSerializer, self).__init__(*args, **kwargs)
        self.user = None
        self.fields[User.USERNAME_FIELD] = serializers.CharField(
            required=False
        )

    def validate(self, attrs):
        self.user = authenticate(
            username=attrs.get(User.USERNAME_FIELD),
            password=attrs.get('password')
        )

        self._validate_user_exists(self.user)
        self._validate_user_is_active(self.user)
        return attrs

    def _validate_user_exists(self, user):
        if not user:
            self.fail('invalid_credentials')

    def _validate_user_is_active(self, user):
        if not user.is_active:
            self.fail('inactive_account')

視圖.py

    class TokenCreateView(utils.ActionViewMixin, generics.GenericAPIView):
    """
    Use this endpoint to obtain user authentication token.
    """
    serializer_class = settings.SERIALIZERS.token_create
    permission_classes = [permissions.AllowAny]

    def _action(self, serializer):
        token = utils.login_user(self.request, serializer.user)
        token_serializer_class = settings.SERIALIZERS.token
        return Response(
            data=token_serializer_class(token).data,
            status=status.HTTP_200_OK,
        )

我的自定義views.py

    class CustomTokenCreateView(utils.ActionViewMixin, generics.GenericAPIView):
    """
    Use this endpoint to obtain user authentication token.
    """
    serializer_class = TokenCreateSerializer
    permission_classes = [permissions.AllowAny]

    def _action(self, serializer):
        token = utils.login_user(self.request, serializer.user)
        token_serializer_class = settings.SERIALIZERS.token
        content = {
            'Token': token_serializer_class(token).data,
            'promptmsg': 'You have successfully login',
            'status': '200'
        }
        return Response(
            data=content,
            status=status.HTTP_200_OK,
        )

djoser 令牌認證的結果：

成功 ：

{
"auth_token": "da57cd11c34cb4332eaa6cc2cac797d0ee95cafb"
}

我編輯的 djoser 身份驗證的結果

成功：

{
    "Token": {
        "auth_token": "da57cd11c34cb4332eaa6cc2cac797d0ee95cafb"
    },
    "promptmsg": "You have successfully login",
    "status": "200"
}

我想要的結果

成功：

{
    "Token": "da57cd11c34cb4332eaa6cc2cac797d0ee95cafb",
    "promptmsg": "You have successfully login",
    "status": "200"
}

有沒有辦法刪除 auth_token 標簽？ 我不介意它的 auth_token 但只要格式是預期的

Answer 1

試試這個來訪問值：

content = {
        'Token': token_serializer_class(token).data["auth_token"],
        'promptmsg': 'You have successfully login',
        'status': '200'
    }

這樣你就不會分配整個對象。