时间戳

首页 > 解决方案 > 如何使用 ManyToManyField 中的公共 ID 数量注释查询集

python - 如何使用 ManyToManyField 中的公共 ID 数量注释查询集

问题描述

我有以下型号：

class Ingredient(models.Model):
    id = models.CharField(primary_key=True, max_length=24)
    name = models.CharField(max_length=100)

class Recipe(models.Model):
    id = models.CharField(primary_key=True, max_length=24)
    name = models.CharField(max_length=100)
    ingredients = models.ManyToManyField(to=Ingredient)

以及具有有效成分 ID 的列表，例如：

ing_id_list = ['5d481cf3abe2d800150de7b6', '5c8c05bee3f3391eda4320b2', ...]

我一直在构建一个带有注释字段的查询集，该字段使用常见相关成分 ID 的数量和列表中的 ID 来丰富食谱：

annotated_recipe_queryset = Recipe.objects.annotate(no_of_common_ings=Count(##PLEASE HELP##))

任何人都可以帮忙吗？

标签： pythondjangodjango-queryset

解决方案

如果您使用的是 Django>=1.8，则可以使用条件聚合。对于您的示例，查询可能是：

from django.db.models import Count, Case, When, IntegerField
annotated_recipe_queryset = Recipe.objects.annotate(
        no_of_common_ings=Count(Case(When(
            ingredients__id__in=ing_id_list, then=1), output_field=IntegerField(), )))

推荐阅读