Django，如何根据查询集设置过滤器？

Question

我有一个由三个不同模型给出的查询集：

class Sottocategoria(models.Model):
    name = models.CharField('Nome del sottoprodotto', max_length=30)

class A(models.Model):
    codice_commessa=models.ForeignKey()
    prodotto=models.ForeignKey()
    sottocategoria=models.ForeignKey(Sottocategoria)

class B(models.Model):
    quantity=models.ForeignKey()
    price=models.DecimalField()
    sottocategoria=models.ForeignKey(Sottocategoria)

现在我设置了以下 for 循环：

for sottocategoria_id, totale in 
 (B.objects.values_list('sottocategoria__id').annotate(totale=(Sum(F('quantity') * F('price')))):
....

我需要过滤模型 B 中的 sottocategoria__id，它们存在于模型 A 中。

广告示例如果我在 model A sottocategoria 中等于 {'abc','abcd','abcdf'} 并且模型 B sottocategoria 等于 {'abc','abcd','abcdf', '1234'}，则在我的 for 循环中我只想过滤 {'abc','abcd','abcdf'}。

Answer 1

您可以使用__in lookup [Django-doc] 进行过滤：

B.objects.filter(
    sottocategoria__name__in={'abc','abcd','abcdf'}
).values_list(
    'sottocategoria_id'
).annotate(
    totale=Sum(F('quantity') * F('price'))
)

您还可能想要.order_by('sottocategoria_id')，这样如果您下标，您将在sottocategoria_id 上下标，而不是在B 对象的主键上：

B.objects.filter(
    sottocategoria__name__in={'abc','abcd','abcdf'}
).values_list(
    'sottocategoria_id'
).annotate(
    totale=Sum(F('quantity') * F('price'))
).order_by('sottocategoria_id')

例如，如果您查找被A 引用的sottocategorias，您可以使用：

B.objects.filter(
    sottocategoria__in=Sottocategoria.objects.filter(a__isnull=False).distinct()
).values_list(
    'sottocategoria_id'
).annotate(
    totale=Sum(F('quantity') * F('price'))
).order_by('sottocategoria_id')

对于某些数据库，例如 MySQL，最好先实现 id：

sottocategoria_ids = list(Sottocategoria.objects.filter(a__isnull=False).values_list('pk', flat=True).distinct())

B.objects.filter(
    sottocategoria__in=sottocategoria_ids
).values_list(
    'sottocategoria_id'
).annotate(
    totale=Sum(F('quantity') * F('price'))
).order_by('sottocategoria_id')

我们也可以从A模型中查询：

sottocategoria_ids = list(A.objects.values_list('sottocategoria_id', flat=True).distinct())

B.objects.filter(
    sottocategoria__in=sottocategoria_ids
).values_list(
    'sottocategoria_id'
).annotate(
    totale=Sum(F('quantity') * F('price'))
).order_by('sottocategoria_id')