python - 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})

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

获得我正在做的其他顶级披萨:

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

这是在类函数中(在模型上)

我要的是上面的查询，包括当前的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，我的查询没有外部引用(它都是相同的模型)我只想在一个中获得 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。