是否可以在没有“ look_up”字段的情况下使用“ RetrieveUpdateAPIView”？

Question

RetrieveUpdateAPIView的url与特定字段一起使用。（例如。/profile/update/<pk>）

我正在从HTTP标头中的令牌获取用户帐户，因此很难将用户的pk包含到URL中。

该视图是我的RetrieveUpdateAPIView。

class ProfileRetrieveAndUpdateProfile(generics.RetrieveUpdateAPIView):
    serializer_class = ProfileRetrieveAndUpdateSerializer
    # lookup_field = 'pk'

    def get_queryset(self):
        qs = Profile.objects.all()
        logged_in_user_profile = qs.filter(user=self.request.user)
        return logged_in_user_profile

我应该只使用views.APIView来实现此功能吗？

Answer 1

我通过重写get_object解决了这个问题。

def get_object(self):
  queryset = self.get_queryset()
  obj = get_object_or_404(queryset, user=self.request.user)
  return obj

干杯