Django QuerySet 使用子查询进行注释

Question

给出以下 model

class Pizza(Model):
  pass

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

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

所以现在我有一个披萨，我有前三名的披萨（就配料数量而言排名第一）

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

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

<- 这是我要避免的，我不想在循环中进行查询，我想获取当前的比萨饼及其浇头数量，并获取 top3 比萨饼及其浇头数量，一站式查询。

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

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

这是在 class function 中（在模型上）

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

    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.

EDIT#2：上述错误仅发生在 SQLite

Answer 1

您可以制作一个.union(…) [Django-doc]来构造两个查询集的 uion。 这是这样的：

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上运行它，那么这将出错。 可能对于其他数据库，这将运行良好。

Answer 2

如果这些是你的披萨：

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>
]>

这是你的 model：

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>]>

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