用Django Rest框架返回当前用户

Question

我目前正在使用Django开发一个API。

但是，我想创建一个返回具有以下endpoint的当前用户的视图：/用户/当前/。

为此，我创建了一个列表视图，并过滤了发出请求的用户的查询集。这行得通，但结果是一个列表，而不是一个单一的对象。结合分页，与其他endpoint相比，结果看起来太复杂和不一致了。

我也试图创建一个详细视图并过滤查询集，但是DRF抱怨我没有提供pk或slug。

Answer 1

您可以使用mixins，而不是使用< code>ModelViewSet的全部功能。有一个< code>RetrieveModelMixin用于检索单个对象，就像这里提到的那样-http://www . django-rest-framework . org/API-guide/view sets/# example _ 3

class UserViewSet(mixins.RetrieveModelMixin, viewsets.GenericViewSet):
    permission_classes = (permissions.IsAuthenticated,)
    serializer_class = UserSerializer

    def get_object(self):
        return self.request.user

如果您还需要更新模型，只需添加 UpdateModelMixin。

Answer 2

最好的方法是使用view set. Model ViewSet的力量，如下所示：

class UserViewSet(viewsets.ModelViewSet):
    queryset = User.objects.all()
    serializer_class = UserSerializer

    def get_object(self):
        pk = self.kwargs.get('pk')

        if pk == "current":
            return self.request.user

        return super().get_object()

视图集。Model ViewSet是混音的组合。CreateModel Mi新混音。RetrieveModel Mi新混音。UpdateModel Mi新混音。ListModel Mi新视图集。通用视图集。如果您需要列出所有或获得特定用户，包括当前经过身份验证的用户，您只需像这样替换它

class UserViewSet(mixins.RetrieveModelMixin, mixins.ListModelMixin, viewsets.GenericViewSet):
    # ...

Answer 3

有了这样的东西，你可能最好打破常规视图，自己编写视图。

@api_view(['GET'])
def current_user(request):
    serializer = UserSerializer(request.user)
    return Response(serializer.data)

您也可以使用基于类的视图做同样的事情，如下所示...

class CurrentUserView(APIView):
    def get(self, request):
        serializer = UserSerializer(request.user)
        return Response(serializer.data)

当然，也不要求您使用序列化程序，您同样可以从用户实例中提取您需要的字段。

@api_view(['GET'])
def current_user(request):
    user = request.user
    return Response({
        'username': user.username,
        'email': user.email,
        ...
    })

希望有帮助。