Django-rest-framework drf-extensions：嵌套路由器/url 未过滤

Question

我无法让我的路由器根据“parents_query_lookup”过滤我的请求。

这是我的代码：

网址.py：

from rest_framework_extensions.routers import ExtendedSimpleRouter
from .views import OrganizationViewSet, GroupViewSet, BootGroupViewSet


router = ExtendedSimpleRouter()
(router.register(r'organizations', OrganizationViewSet,
                 base_name='organization')
    .register(r'groups', GroupViewSet,  base_name='organizations-group',
              parents_query_lookups=['resource__organization'])
    .register(r'boot_groups', BootGroupViewSet,
              base_name='organizations-groups-boot_group',
              parents_query_lookups=['group__resource__organization', 'group']))

urlpatterns = router.urls

视图.py：

from rest_framework.viewsets import ModelViewSet
from rest_framework_extensions.mixins import NestedViewSetMixin
from .models import Organization, OrganizationSerializer, \
    Group, GroupSerializer, BootGroup, BootGroupSerializer


class OrganizationViewSet(NestedViewSetMixin, ModelViewSet):
    queryset = Organization.objects.all()
    serializer_class = OrganizationSerializer


class GroupViewSet(NestedViewSetMixin, ModelViewSet):
    queryset = Group.objects.all()
    serializer_class = GroupSerializer


class BootGroupViewSet(NestedViewSetMixin, ModelViewSet):
    queryset = BootGroup.objects.all()
    serializer_class = BootGroupSerializer

枚举.py：

class ResourceTypeEnum:

    RESOURCE_TYPE_GROUP = 'group'
    RESOURCE_TYPE_VM = 'vm'

    RESOURCE_TYPE_CHOICES = (
        (RESOURCE_TYPE_GROUP, RESOURCE_TYPE_GROUP),
        (RESOURCE_TYPE_VM, RESOURCE_TYPE_VM)
    )

模型.py：

from django.db.models import Model
from rest_framework.serializers import ModelSerializer

from .enums import ResourceTypeEnum


class Organization(Model):
    organization_id = models.AutoField(primary_key=True)
    name = models.CharField(max_length=255)
    parent = models.ForeignKey('self', blank=True, null=True)

    class Meta:
        unique_together = (("name", "parent"))
        verbose_name = "Organization"
        verbose_name_plural = "Organizations"
        app_label = 'api_manager'
        db_table = 'organization'

    def __unicode__(self):
        return self.name


class OrganizationSerializer(ModelSerializer):
    class Meta:
        model = Organization
        fields = ('name', 'parent')
        depth = 2


class Resource(Model):
    resource_id = models.AutoField(primary_key=True)
    name = models.CharField(max_length=55)
    type = models.CharField(
        max_length=5, choices=ResourceTypeEnum.RESOURCE_TYPE_CHOICES)
    organization = models.ForeignKey(Organization)

    class Meta:
        verbose_name = "Resource"
        verbose_name_plural = "Resources"
        app_label = 'api_manager'
        db_table = 'resource'

    def __unicode__(self):
        return self.name


class ResourceSerializer(ModelSerializer):
    class Meta:
        model = Resource
        fields = ('name', 'type', 'organization')
        depth = 2


class Group(Model):
    resource = models.OneToOneField(Resource, primary_key=True)
    is_consistent = models.BooleanField()
    parent = models.ForeignKey('self', blank=True, null=True)

    class Meta:
        verbose_name = "Group"
        verbose_name_plural = "Groups"
        app_label = 'api_manager'
        db_table = 'group'

    def __unicode__(self):
        return "%s: %s" % (self.resource.organization, self.resource)


class GroupSerializer(ModelSerializer):
    class Meta:
        model = Group
        fields = ('resource', 'is_consistent', 'parent')
        depth = 2


class BootGroup(Model):
    boot_group_id = models.AutoField(primary_key=True)
    name = models.CharField(max_length=100)
    boot_order = models.PositiveSmallIntegerField(default=1)
    group = models.ForeignKey(Group)

    class Meta:
        unique_together = (("boot_order", "group"))
        verbose_name = "BootGroup"
        verbose_name_plural = "BootGroups"
        app_label = 'api_manager'
        db_table = 'boot_group'

    def __unicode__(self):
        return "%s: %s" % (self.group.resource, self.name)


class BootGroupSerializer(ModelSerializer):
    class Meta:
        model = BootGroup
        fields = ('name', 'boot_order', 'group')
        depth = 2


class Vm(Model):
    resource = models.OneToOneField(Resource, primary_key=True)
    hostname = models.CharField(max_length=200, blank=True, null=True)
    group = models.ForeignKey(Group, blank=True, null=True)
    boot_group = models.ForeignKey(BootGroup, blank=True, null=True)

    class Meta:
        verbose_name = "Vm"
        verbose_name_plural = "Vms"
        app_label = 'api_manager'
        db_table = 'vm'

    def __unicode__(self):
        return "%s: %s" % (self.resource.organization, self.resource)


class VmSerializer(ModelSerializer):
    class Meta:
        model = Vm
        fields = ('resource', 'hostname', 'group',  'boot_group')
        depth = 2

无论我尝试什么，“organizations/1/groups”之类的东西都会返回所有Group模型，无论Organization如何。我错过了什么吗？先谢谢您的帮助。

Answer 1

根据您的路由器定义，您需要在中定义一个查询集GroupViewSet

from rest_framework_extensions.utils import compose_parent_pk_kwarg_name

 def get_queryset(self):
     resource_id =  self.kwargs.get(compose_parent_pk_kwarg_name('resource_organization')
     return Group.objects.filter(resource=resource_id)

过滤器值可以在 kwargs as 中找到parent_lookup_<parents_query_lookups>，或者您可以使用以下命令检索父查询字典self.get_parents_query_dict()