Django ORM 中多对多字段的复杂计数

Question

因此，我有一组可以出现在多个类别中的任务：

class TaskGroup(models.Model):
    name = models.CharField(max_length=200)
    slug = models.SlugField(max_length=200)
    icon = models.CharField(max_length=200, blank=True, null=True)

    def __unicode__(self):
        return unicode(self.name)


class Task(models.Model):
    start_date = models.DateField()
    end_date = models.DateField()
    is_date_fuzzy = models.BooleanField()
    name = models.CharField(max_length=200)
    assignee = models.ForeignKey(User, verbose_name="users who is assigned the task", blank=True, null=True)
    task_groups = models.ManyToManyField(TaskGroup)

如您所见，每个任务可以出现在多个任务组中。

我希望我的查询满足以下条件：

1. 所有任务组的列表应该是 回来了。
2. 任务数量的计数 在特定群体内。 IE 家具 (3)、床上用品 (2)、地板 灯 (6)
3. 如果没有任务 具体TaskGroup，表示组 应该有 0 个
4. 每组内的任务是 仅限于当前用户。

到目前为止我想出的最好的办法是这样的：

TaskGroup.objects.filter(
    task__assignee=current_usr
).annotate(
    task_count=Count('task__id')
).order_by('name')

但它在进行计数之前过滤所有内容，所以我看不到任务为零的任务组。

也许我想得太多了，但多年来我一直在尝试这样做，而且在这个阶段我只是想自己循环计数。

我真的希望你能帮助挽救我仅存的一点理智！

Answer 1

我曾经遇到过完全相同的问题。一种解决方案是使用额外的子查询：

TaskGroup.objects.extra(
    select={"task_count": "SELECT COUNT(*) from app_task where app_task.id=app_taskgroup.task_id AND app_task.assignee = '?'"},
    select_params = [current_usr]
).order_by('name')

或类似的东西。可能是连接密钥错误。

但这太丑了。并打破了很多 DRY 和 DB 独立原则。

更好的选择可能是使用 2 个查询。第二个针对没有分配给该用户任何任务的特定组：

empty_groups = TaskGroup.objects.exclude(task__assignee=current_usr)

然后只需迭代两组而不是一组。