与经理相关的Django重复查询

Question

我正在优化我的 ORM 查询。我有两个应用程序，“app1”和“app2”。一类“app2”具有指向一类 app1 的外键，如下所示：

#app1/models.py
class C1App1(WithDateAndOwner):
  def get_c2_app2(self):
    res = self.c2app2_set.all()
    if res.count() > 0:
      return res[0]
    else:
      return None

#app2/models.py
class C2App2(WithDateAndOwner):
  c1app1 = models.ForeignKey("app1.C1App1")
  is_ok = models.BooleanField(default=False)

现在我在管理页面中显示 C1App1 的所有实例的 C2App2：

#app1/admin.py
@admin.register(C1App1)
class C1App1Admin(admin.MyAdmin):
  list_display = ("get_c2_app2")
  list_select_related = ()
  list_prefetch_related = ("c2app2_set",)
  list_per_page = 10

prefetch_related 减少这个查询：

SELECT ••• FROM `app2_c2app2` WHERE `app2_c2app2`.`c1app1_id` = 711
  Duplicated 19 times.

到：

SELECT ••• FROM `app2_c2app2` WHERE `app2_c2app2`.`c1app1_id` IN (704, 705, 706, 707, 708, 709, 710, 711, 702, 703) ORDER BY `app2_c2app2`.`id` DESC

没关系。现在，如果我想过滤 C2App2 的属性“is_ok”上的查询：

#app1/models.py
class C1App1(WithDateAndOwner):
  def get_c2_app2(self):
    res = self.c2app2_set.filter(is_ok=False)
    if res.count() > 0:
      return res[0]
    else:
      return None

我仍然有这个预取的查询：

SELECT ••• FROM `c2app2_set` WHERE `app2_c2app2`.`c1app1_id` IN (704, 705, 706, 707, 708, 709, 710, 711, 702, 703) ORDER BY `app2_c2app2`.`id` DESC

但是对于 C1App1 (10) 的每个显示实例都复制了一个。：

SELECT ••• FROM `app2_c2app2` WHERE (`app2_c2app2`.`c1app1_id` = 711 AND `app2_c2app2`.`is_ok` = 1)
  Duplicated 13 times.

实际上，对于显示的 10 个 id 中的 3 个，该查询也再次重复，这导致了这 13 个重复的查询。我该怎么做才能不重复这些查询？似乎 prefetch_related 在这里不再有帮助。

Answer 1

prefetch_related 仅在您使用.all(). 如果您应用任何其他转换，例如.filter()，将进行新的数据库查询。这是因为prefetch_related简单地将所有相关实例缓存在列表中，因此 Django 无法filter()在列表上执行。要解决您的问题，您应该使用Prefetch对象。您可以将queryset参数传递给它。因此，不要使用 list_prefetch_related，而是get_queryset在您的管理类中覆盖方法。

def get_queryset(*args, **kwargs):
     qs = super().get_queryset(*args, **kwargs)
     qs = qs.prefetch_related(Prefetch('c2app2_set', queryset=C2App2.objects.filter(is_ok=False)))
     return qs

和

class C1App1(WithDateAndOwner):
   def get_c2_app2(self):
      res = self.c2app2_set.all()
      if res.count() > 0:
         return res[0]
      else:
         return None