具有跨不同模型注释的复杂查询集

Question

我正在使用 Django 1.6 并有以下模型（为了便于阅读，稍微简化了一些）：

class Person(models.Model):
    name = models.CharField(max_length=128)

class Room(models.Model):
    max_persons = models.SmallIntegerField()
    name = models.CharField(max_length=30, unique=True)

class Stay(models.Model):
    person = models.ForeignKey(Person)
    STAY_TYPE = (
        (1, 'Type 1'),
        (2, 'Type 2'),
        (3, 'Type 3'),
    )
    stay_type = models.CharField(max_length=1, choices=STAY_TYPE)

class Hospitalization(models.Model):
    stay = models.ForeignKey(Stay)
    date_in = models.DateField()
    date_out = models.DateField(blank=True, null=True)
    room = models.ForeignKey(Room)

使用queryset（在经理中），我想获取给定日期的所有可用房间，即不满足max_persons 的位置。

到目前为止，我尝试过使用Q 和Room.objects.filter(hospitalization__date_out__lt="2014-04-25") 之类的东西，但我无法弄清楚max_persons 的比较。

你有什么想法吗？

编辑

与每个房间相关的 Person 对象的数量代表了该房间中的人数。

因此，根据 Karl 的以下建议，我尝试了以下方法：

• Room.objects.filter(hospitalization__date_out__lte="2014-04-25").annotate(num_persons=hospitalization_set__stay__person.Count()).exclude(num_persons__lte=max_persons)
  • 产量NameError: name 'hospitalization_set__stay__person' is not defined
• Room.objects.filter(hospitalization__date_out__lte="2014-04-25").annotate(num_persons=Count('hospitalization_set__stay__person')).exclude(num_persons__lte=max_persons)
  • 产量FieldError: Cannot resolve keyword 'hospitalization_set' into field.
• Room.objects.filter(hospitalization__date_out__lte="2014-04-25").annotate(num_persons=Count('hospitalization__stay__person')).exclude(num_persons__lte=F('max_persons'))
  • 没有任何结果 ([])，而根据我当前的数据库，我预计至少有三个结果。
  • 删除.exclude() 时，我仍然没有得到任何结果。看起来我正在使用的 .annotate() 做错了。

Answer 1

我相信类似于Room.objects.filter(hospitalization__date_out="2014-04-25", max_persons__lt=x) 的东西应该可以解决问题。请记住，您可以在过滤器操作中有多个声明，并且您还可以链接排除和过滤器。这不会增加数据库活动； queryset 过滤器等仅在评估 queryset 本身时执行。见The Django Docs on making queries。

编辑： 这实际上比我最初想象的要复杂一些，但我相信我已经解决了（在导入你的模型声明并进行 shell 会话之后！）

rooms = Room.objects.filter(DATE FILTERING HERE).annotate(num_persons=Count('hospitalization__room__id')).filter(num_persons__lt=F('max_persons')).count()

将为您提供num_persons（我们上面的注释变量）小于max_persons的房间数量，并预先进行日期过滤。

请记住包含所需的导入： from django.db.models import Count from django.db.models import F