python - django 使用 Count() 一次性注释相关字段

Question

我的Profile 模型在一个模型中有两个 m2m 字段 - region 和 country。如您所知，country 有自己的带有外键的 region。

我想尝试计算每个区域的个人资料 - 不仅包括 region，还包括 country__region。

即)如果一些国家只有非洲地区，而其他国家只有刚果国家(地区为非洲)，我想将它们过滤为一个。

我尝试使用注释来解决它。我可以单独找到区域的数量，如下所示

    profiles = Profile.objects.all()
    region_count = profiles.values('region').annotate(region_count=Count('region'))
    country_count = profiles.values('region').annotate(region_count=Count('country__region'))

但是如何计算特定区域的查询集，同时使用 region 和 region__country 进行过滤？有什么可行的方法吗？

这是我的个人资料/国家/地区模型。区域模型只有名称字段。

class Profile(models.Model):
    region = models.ManyToManyField(
        Region,
        verbose_name="Region(s) of interest",
        blank=True,
    )
    country = models.ManyToManyField(
        Country,
        related_name="country",
        verbose_name="Countries of interest",
        blank=True,
    )
    ...

class Country(models.Model):
    region = models.ForeignKey(
        Region,
        null=True,
        blank=True,
    )
    ...

感谢您的帮助。

摘要

我想使用annotate同时对region和country__region的查询集进行计数。

Answer 1

您可以尝试使用conditional-expressions计数前:

from django.db.models import Case, When, F, Count

Profile.objects.annotate(
    reg=Case(
        When(region__isnull=True, then=F('country__region')),
        default=F('region'))
    ).values('reg').annotate(region_count=Count('reg'))