sql - Django:来自两个模型的多个计数

Question

我正在尝试创建一个个人资料页面，显示分配给每个相应职业的矮人数量。我有 4 个职业，每个职业中有 2 个工作，当然还有许多矮人，每个人都有一份工作。我怎样才能计算出每个职业中矮人的数量？我的解决方案是硬核 HTML 中的职业名称，并对每个职业进行查询，但这似乎查询过多。

这是我“想要”看到的:

Unassigned: 3
Construction: 2
Farming: 0
Gathering: 1

这是我的模型。我通过不将 Careers 直接连接到我的 Dwarves 模型(他们通过他们的工作连接)来增加一些复杂性。

from django.contrib.auth.models import User
from django.db import models

class Career(models.Model):
    name = models.CharField(max_length = 64)

    def __unicode__(self):
        return self.name

class Job(models.Model):
    career = models.ForeignKey(Career)
    name = models.CharField(max_length = 64)
    career_increment = models.DecimalField(max_digits = 4, decimal_places = 2)
    job_increment = models.DecimalField(max_digits = 4, decimal_places = 2)

    def __unicode__(self):
        return self.name

class Dwarf(models.Model):
    job = models.ForeignKey(Job)
    user = models.ForeignKey(User)
    created = models.DateTimeField(auto_now_add = True)
    modified = models.DateTimeField(auto_now = True)
    name = models.CharField(max_length = 64)

    class Meta:
        verbose_name_plural = 'dwarves'

    def __unicode__(self):
        return self.name

编辑 1 我的观点看起来像:

def fortress(request):
    careers = Career.objects.annotate(Count('dwarf_set'))
    return render_to_response('ragna_base/fortress.html', {'careers': careers})

和模板:

{% for career in careers %}
    <li>{{ career.dwarf_set__count }}</li>
{% endfor %}

错误是:

Cannot resolve keyword 'dwarf_set' into field. Choices are: id, job, name

解决方案

查看:

def fortress(request):
    careers = Career.objects.all().annotate(dwarfs_in_career = Count('job__dwarf'))
    return render_to_response('ragna_base/fortress.html', {'careers': careers})

模板:

{% for career in careers reversed %}
    <li>{{ career.name }}: {{ career.dwarves_in_career }}</li>
{% endfor %}

更好的解决方案

careers = Career.objects.filter(Q(job__dwarf__user = 1) | Q(job__dwarf__user__isnull = True)) \
    .annotate(dwarves_in_career = Count('job__dwarf'))

不要忘记from django.db.models import Count, Q

我喜欢上述解决方案的一点是，它不仅会返回有矮人工作的职业，甚至还会返回没有矮人工作的职业，这是我遇到的下一个问题。以下是我对完整性的看法:

<ul>
{% for career in careers %}
    <li>{{ career.name }}: {{ career.dwarves_in_career }}</li>
{% endfor %}
</ul>

Answer 1

Django 的 ORM 不会让这个 super 简单。简单的方法是做类似的事情:

Career.objects.all() 中的职业: career.dwarf_set.all().count()

这将为每个作业执行 1 个查询(O(n) 复杂度)。

您可以尝试使用 Django's Aggregation feature 来加快速度，但我不完全确定它是否会满足您的需要。你得看看。

第三个选项是使用自定义 SQL，这绝对可以完成工作。您只需要编写它，并随着您的应用程序的增长和变化而维护它...