Question

我正在为具有公司和部门的应用程序设计REST API。一些用户可以是公司的成员。这导致以下API结构：

/companies/ - 可以GET，POST。

/companies/<pk>/ - 可以GET，POST，PUTCH，PATCH，DELETE。

/companies/<pk>/membership/ - 可以GET（给所有公司成员的用户）POST。

/companies/<pk>/membership/<pk>/ - 可以删除。

我设法实现了前3个端点，但是在实现最后一个端点时遇到了问题 - 如何在URL中实现具有多个<pk>值的端点？这就是我到目前为止所拥有的：

目前在api应用中有一个urls.py文件，如下所示：

...
    url(r'^company', include(company_urls.company_router.urls,
    namespace="company")),
...

urls.py应用程序中的

company。

from rest_framework import routers

from .views import CompanyViewSet


company_router = routers.DefaultRouter()

company_router.register(r'^', CompanyViewSet)

serializers.py文件：

from rest_framework import serializers

from .models import Company, CompanyMembership
from My_App.users.models import Profile


class CompanySerializer(serializers.ModelSerializer):

    class Meta:
        model = Company
        fields = ('pk', 'name', 'departments', 'members')
        read_only_fields = ('pk', 'departments', 'members')


class CompanyMembershipSerializer(serializers.Serializer):
    user = serializers.PrimaryKeyRelatedField(queryset=Profile.objects.all())

    def create(self, validated_data):
        pass

    def delete(self, instance, validated_data):
        pass

views.py文件：

from .models import Company, CompanyMembership
from .serializers import CompanySerializer, CompanyMembershipSerializer

from My_Appc.users.models import Profile


class CompanyViewSet(viewsets.ModelViewSet):
    queryset = Company.objects.all()
    serializer_class = CompanySerializer

    @decorators.detail_route(methods=['get', 'post', 'delete'])
    def membership(self, request, pk):
        company = self.get_object()
        if request.method == 'GET':
            serializer = CompanyMembershipSerializer(company)
        elif request.method == 'POST':

            serializer = CompanyMembershipSerializer(data=request.data)
            if serializer.is_valid():
                try:
                    user = Profile.objects.get(pk=request.data.get('user'))
                    user_company_membership = CompanyMembership(user=user,
                                                                          company=company)
                    user_company_membership.save()
                    return Response({'status': 'User added to Company.'}, status=status.HTTP_201_CREATED)
                except IntegrityError:
                    result = {
                        'status': 'Failed to add user to Company.',
                        'reason': 'User already part of Company.'
                    }
                    return Response(result, status=settings.ADDITIONAL_HTTP_STATUS_CODES['422_UNPROCESSABLE_ENTITY'])
            else:
                return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)

Answer 1

在您的网址中为参数使用不同的名称：

/companies/<company_pk>/membership/<membership_pk>/

在ViewSet中添加lookup_field和lookup_url_kwarg以指向公司pk字段/参数：

class CompanyViewSet(viewsets.ModelViewSet):
    lookup_field = 'pk'
    lookup_url_kwarg = 'company_pk'

get_object方法使用这两个查找来过滤查询集，这样您就可以根据网址中的第一个pk获得公司。

在您的会员资格方法和管理会员资格对象的自定义逻辑中，您可以通过以下方式访问会员资格pk：

membership_pk = self.kwargs.get('membership_pk', None)

Django REST框架 - 在URL中添加2个PK

1 个答案: