Django-treebeard MP节点预取后代

时间:2016-04-12 09:08:38

标签: django django-rest-framework django-treebeard

我正在django-rest-framework使用django-treebeard开发一个具有分层数据结构的应用程序。我的(简化)主模型看起来像这样

class Task(MP_Node):
    name = models.CharField(_('name'), max_length=64)
    started = models.BooleanField(default=True)

我目前正在尝试实现的是所有根节点的列表视图,其中显示了额外的字段(例如是否所有子节点都已启动)。为此,我指定了一个视图:

class TaskViewSet(viewsets.ViewSet):

    def retrieve(self, request, pk=None):
        queryset = Task.get_tree().filter(depth=1, job__isnull=True)
        operation = get_object_or_404(queryset, pk=pk)
        serializer = TaskSerializer(operation)
        return Response(serializer.data)

和序列化程序

class TaskSerializer(serializers.ModelSerializer):
    are_children_started = serializers.SerializerMethodField()

    def get_are_children_started(self, obj):
        return all(task.started for task in Task.get_tree(obj))

这一切都有效,我得到了预期的结果。但是,我遇到了一个N​​ + 1查询问题,对于每个根任务,我需要分别获取所有子项。通常这可以使用prefetch_related解决,但是当我使用django-treebeard中的物化路径结构时,任务模型之间没有Django关系,因此prefetch_related不知道该怎么做盒子。我试图使用自定义Prefetch对象,但因为这仍然需要一个Django关系路径,我无法让它工作。

我目前的想法是使用指向其根节点的外键扩展Task模型,如下所示:

root_node = models.ForeignKey('self', null=True,
                              related_name='descendant_tasks',
                              verbose_name=_('root task')
                              )

为了使MP关系显式化,可以查询它。然而,这确实感觉像是一种非干燥的方法,所以我想知道是否有人有另外的建议如何解决它。

1 个答案:

答案 0 :(得分:3)

最后,我最终为每个指向其根节点的任务添加了一个外键,如下所示:

root_node = models.ForeignKey('self', null=True,
                          related_name='descendant_tasks',
                          verbose_name=_('root task')
                          )

我在我的Task模型上更新了我的save方法,以确保我总是指向正确的根节点

def save(self, force_insert=False, force_update=False, using=None, update_fields=None):
    try:
        self.root_task = self.get_root()
    except ObjectDoesNotExist:
        self.root_task = None

    return super(Task, self).save(force_insert=False, force_update=False, using=None,
                                  update_fields=None
                                  )

这使我可以使用prefetch_related('descendants')简单地预取所有后代。

每当我需要以嵌套的方式使用后代时,我使用以下函数再次嵌套扁平的后代列表

def build_nested(tasks):

    def get_basepath(path, depth):
        return path[0:depth * Task.steplen]

    container, link = [], {}
    for task in sorted(tasks, key=attrgetter('depth')):
        depth = int(len(task.path) / Task.steplen)
        try:
            parent_path = get_basepath(task.path, depth - 1)
            parent_obj = link[parent_path]
            if not hasattr(parent_obj, 'sub_tasks'):
                parent_obj.sub_tasks = []
            parent_obj.sub_tasks.append(task)
        except KeyError:  # Append it as root task if no parent exists
            container.append(task)

        link[task.path] = task

    return container