Question

我正在使用Django REST Framework设计API端点，并希望它在以下端点接受GET和POST方法：

companies/<company_pk>/membership

以及以下端点的DELETE方法：

companies/<company_pk>/membership/<membership_pk>

有两种相关模型：Company和CompanyMembership。每个用户都可以是一个或多个公司的成员，由CompanyMembership的实例表示，该实例通过FK链接到Company。

我已经创建了相应的GET和POST请求，但是我遇到了DELETE请求的问题 - 我的视图中的destroy方法似乎缺少membership_pk参数，无论是否我在我的网址中定义它。如何让我的视图接受删除CompanyMembership实例的删除请求，这些请求由网址中提供的membership_pk标识？

urls.py

...
    url(r'^companies/(?P<company_pk>[0-9]+)/membership', include(company_urls.company_membership_router.urls,
        namespace='company_memberships')),
...

views.py


class CompanyMembershipViewSet(viewsets.ModelViewSet):
    """
    A ViewSet for creating, retrieving, and deleting CompanyMemberships.
    """
    queryset = CompanyMembership.objects.all()
    serializer_class = CompanyMembershipSerializer
    lookup_url_kwarg = 'company_pk'

    def get_queryset(self):
        uid = self.kwargs.get(self.lookup_url_kwarg)
        memberships = CompanyMembership.objects.filter(company_id=uid)
        return memberships

    def create(self, request, company_pk):
        """
        Custom create method since company_pk does not have to be sent
        explicitly within request.data

        Returns: instance of rest_framework Response.
        """
        EXISTING_USER_ERROR = {'non_field_errors':
            ['The fields user, company must make a unique set.']
        }
        data = {
            'user': request.data['user'],
            'company': company_pk
        }
        serializer = CompanyMembershipSerializer(data=data)
        if serializer.is_valid():
            serializer.save()
            return Response(serializer.data, status=status.HTTP_201_CREATED)
        if serializer.errors == EXISTING_USER_ERROR:
            return Response(serializer.errors, status=settings.ADDITIONAL_HTTP_STATUS_CODES['422_UNPROCESSABLE_ENTITY'])
        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)

    def destroy(self, request, **kwargs):
        company_membership = self.get_object(pk=membership_pk)
        company_membership.delete()
        return Response(status=status.HTTP_204_NO_CONTENT)


routers.py
    from rest_framework import routers

    from .views import CompanyViewSet, CompanyMembershipViewSet


    company_router = routers.DefaultRouter()
    company_membership_router = routers.DefaultRouter()

    company_router.register(r'^', CompanyViewSet)
    company_membership_router.register(r'^', CompanyMembershipViewSet)

Answer 1

公司会员路由器生成以下网址：

companies/<company_pk>/membership/ # for handling list requests
companies/<company_pk>/membership/<pk>/ # for handling detail requests

您可以使用membership_pk网址kwarg访问pk 。

此外，您在视图集中错误地定义了lookup_url_kwarg字段。 DRF使用此kwarg来使用get_object()方法检索对象。您没有获得对象，因为它尝试使用pk作为company_pk url kwarg值而不是pk值执行查找。

Django REST框架 - destroy方法不会采用URL参数吗？

1 个答案: