Django ORM中多个到多个字段的复杂计数

时间:2011-06-22 22:53:31

标签: sql django django-models django-aggregation

所以我有一组可以出现在许多类别中的任务:

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. 所有TaskGroup的列表应该是 返回。
  2. 任务数量的计数 在特定的组内。即 家具(3),床上用品(2),地板 灯(6)
  3. 如果没有任务 具体的TaskGroup,组说 应该有0
  4. 每组内的任务是 仅限当前用户使用。
  5. 到目前为止,我提出的最好的是这样的:

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

    但它在计算之前过滤了所有内容,所以我没有看到任务组没有任务。

    也许我正在思考但是我已经尝试过这么多年了,而我现在正试图循环并自己做计算。

    我真的希望你能帮助我拯救我的理智!

1 个答案:

答案 0 :(得分:0)

我有一次完全相同的问题。一种解决方案是使用额外的子查询:

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)

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