53

我在 Django 中有两个模型。第一个具有哪些工作职能（职位）向哪些其他职位报告的层次结构，第二个是人员及其担任的工作职能。

class PositionHierarchy(model.Model):
    pcn = models.CharField(max_length=50)
    title = models.CharField(max_length=100)
    level = models.CharField(max_length=25)
    report_to = models.ForeignKey('PositionHierachy', null=True)


class Person(model.Model):
    first_name = models.CharField(max_length=50)
    last_name = models.CharField(max_length=50)
    ...
    position = models.ForeignKey(PositionHierarchy)

当我有一个人记录并且我想找到这个人的经理时，我必须这样做

manager = person.position.report_to.person_set.all()[0]
# Can't use .first() because we haven't upgraded to 1.6 yet

如果我让人们使用QuerySet，我可以使用 position 和 report_to 加入（并避免第二次访问数据库）Person.objects.select_related('position', 'position__reports_to').filter(...)，但是有没有办法避免再次访问数据库来获取 person_set？我尝试添加'position__reports_to__person_set'或只是添加position__reports_to__person到select_related，但这似乎并没有改变查询。这prefetch_related是为了什么？

我想制作一个自定义管理器，这样当我进行查询以获取人员记录时，我还可以获得他们的 PositionHeirarchy 和他们经理的人员记录，而无需更多地往返数据库。这是我到目前为止所拥有的：

class PersonWithManagerManager(models.Manager):
    def get_query_set(self):
        qs = super(PersonWithManagerManager, self).get_query_set()
        return qs.select_related(
            'position',
            'position__reports_to',
        ).prefetch_related(
        )
4

1 回答 1

41

是的，这就是prefetch_related()目的。它将需要一个额外的查询，但想法是它会一次获取所有相关信息，而不是一次Person。

在你的情况下：

qs.select_related('position__report_to')
  .prefetch_related('position__report_to__person_set')

Persons无论原始查询集中的数量如何，都应该需要两个查询。

比较文档中的这个例子：

>>> Restaurant.objects.select_related('best_pizza')
                      .prefetch_related('best_pizza__toppings')
于 2014-04-16T23:22:56.727 回答