Django QuerySet 使用子查询进行注释

Question

给定以下模型

class Pizza(Model):
  pass

class Topping(Model):
  on = ManyToMany(Pizza, related_name='on_pizza')

我正在尝试获取我的比萨饼和浇头数量，以及前 3 个其他比萨饼（就浇头数量而言排名第一） 最终结果应该是： 一个列表，其中包含当前的比萨饼、前 3 个比萨饼以及所有配料的数量。

所以现在我有一个披萨， 我有前 3 名的比萨饼（就浇头数量而言排名第一）

但我必须遍历顶级比萨来创建字典

for pizza in top_pizzas:
  data.append({'pizza':pizza, 'num':pizza.tcount})

得到我正在做的其他顶级比萨：

top = Pizza.objects.all().exclude(pk=self.pk).annotate(tcount=Count('on_pizza')).order_by('-on_pizza')

这是在一个类函数中（在模型上）

我想要的是上面的查询，包括当前的披萨， 这是我尝试过的：

    def compared_with_top(self):
      top = Pizza.objects.all().exclude(pk=self.pk).annotate(tcount=Count('on_pizza')).order_by('-tcount')
      me_with_top = Pizza.objects\
          .filter(pk=self.pk)\
          .annotate(tcount=Count('on_pizza'))\
          .annotate(other_top_3=Subquery(top[:3]))
      return me_with_top

这给了我一个错误：FieldError: Cannot resolve expression type, unknown output_field

我注意到子查询的所有示例都涉及 2 个单独的模型，并且使用 OuterRef，我的查询没有外部 ref（它的所有模型都是相同的）我只想将 2 个查询合二为一，如果这样的话.

上述错误指向“output_field”，但我找不到任何关于它应该是什么的信息。

编辑：有人建议我合并 2 个查询，这听起来不错，但随后我收到错误 DatabaseError: ORDER BY not allowed in subqueries of compound statements.

编辑#2： 上述错误只发生在 SQLite 上

Answer 1

如果这些是你的披萨：

Pizza.objects.values()
Out[15]: <QuerySet [
    {'id': 1, 'name': 'Hawaiian'},
    {'id': 2, 'name': 'Cheese'},
    {'id': 3, 'name': 'Veggie'},
    {'id': 4, 'name': 'Meat Lovers'},
    {'id': 5, 'name': 'Pineapple ONLY'}
]>

这些是你的配料：

Topping.objects.all()
Out[17]: <QuerySet [
    <Topping: Canadian Bacon>, <Topping: Pineapple>, <Topping: Cheese>,
    <Topping: Green peppers>, <Topping: Olives>, <Topping: Mushrooms>,
    <Topping: Onions>, <Topping: Tomatoes>, <Topping: Sausage>,
    <Topping: Pepperoni>, <Topping: Beef>
]>

这是你的模型：

from django.db import models
from django.db.models import Q, Count

class Pizza(models.Model):
    name = models.CharField(max_length=50)
    toppings = models.ManyToManyField('Topping', related_name='on_pizza')

    def __str__(self):
        return self.name

    def compared_with_top(self):
        top = Pizza.objects.annotate(Count('toppings')).order_by(
            '-toppings__count').exclude(id=self.id)[:3]
        return Pizza.objects.filter(Q(id=self.id) | Q(id__in=top.values('id')))
        

class Topping(models.Model):
    name = models.CharField(max_length=50)

    def __str__(self):
        return self.name

每个披萨的浇头数量是多少：

In [32]: Pizza.objects.annotate(Count('toppings')).values()
Out[32]: <QuerySet [
{'id': 1, 'name': 'Hawaiian', 'toppings__count': 2},
{'id': 2, 'name': 'Cheese', 'toppings__count': 1},
{'id': 3, 'name': 'Veggie', 'toppings__count': 5},
{'id': 4, 'name': 'Meat Lovers', 'toppings__count': 6},
{'id': 5, 'name': 'Pineapple ONLY', 'toppings__count': 1}
]>

我们做一些比萨饼：

hawaiian = Pizza.objects.get(name='Hawaiian')
cheese = Pizza.objects.get(name='Cheese')
veggie = Pizza.objects.get(name='Veggie')
meat = Pizza.objects.get(name='Meat Lovers')
pineapple = Pizza.objects.get(name='Pineapple ONLY')

Pizza.compared_with_top 基于计数结果：

In [26]: hawaiian.compared_with_top()
Out[26]: <QuerySet [<Pizza: Hawaiian>, <Pizza: Cheese>, <Pizza: Veggie>, <Pizza: Meat Lovers>]>

In [27]: cheese.compared_with_top()
Out[27]: <QuerySet [<Pizza: Cheese>, <Pizza: Hawaiian>, <Pizza: Veggie>, <Pizza: Meat Lovers>]>

In [28]: veggie.compared_with_top()
Out[28]: <QuerySet [<Pizza: Veggie>, <Pizza: Hawaiian>, <Pizza: Cheese>, <Pizza: Meat Lovers>]>

In [29]: meat.compared_with_top()
Out[29]: <QuerySet [<Pizza: Meat Lovers>, <Pizza: Hawaiian>, <Pizza: Cheese>, <Pizza: Veggie>]>

In [30]: pineapple.compared_with_top()
Out[30]: <QuerySet [<Pizza: Pineapple ONLY>, <Pizza: Hawaiian>, <Pizza: Veggie>, <Pizza: Meat Lovers>]>

这将为您提供正确的输出，其中查询集中的第一个对象是当前的披萨。不过，后面的三个不是从高到低排序的。它是倒退的。我不确定为什么它们被排序为最小到最大，但也许其他人确实知道。我怀疑是因为使用了Q。

Answer 2

您可以创建一个.union(…) [Django-doc] 来构造两个查询集的联合。因此是这样的：

def compared_with_top(self):
    top = Pizza.objects.exclude(pk=self.pk).annotate(
        tcount=Count('on_pizza')
    ).order_by('-tcount')
    pizzas = Pizza.objects.filter(pk=self.pk).annotate(
        tcount=Count('on_pizza')
    ).union(top[:3])

@ReedJones：对此进行了测试。如果您在sqlite 上运行此程序，则会出错。可能对于其他数据库，这将运行良好。