在聚合中使用模型方法

Question

我正在尝试在 django 聚合查询中使用模型方法。我不确定这是否可能，而且我可能会以错误的方式进行。

这是我要查询的型号。

class ClassNumbers(models.Model):
   """
   The class year level and number inline model for a booking
   """
   booking = models.ForeignKey('Booking')
   yearLevel = models.CharField(max_length=10, choices=YEAR_CHOICES, verbose_name='Year Level')
   numberOfStudents = models.IntegerField(verbose_name='Number of Students') 

class Booking(models.Model):
   # A shorter version of the model
   date = models.DateField()
   institution = models.ForeignKey(Institution)

   def getStudentTotal(self):
      # Total class numbers
      classes = ClassNumbers.objects.filter(booking=self.id)
      classTotal = 0
      if ( classes ):
          for c in classes:
              classTotal += c.numberOfStudents
      return classTotal

   def getHDSV(self):
      HDSVunits = {
                'Full-Day': 2.0,
                'Half-Day AM': 1.0,
                'Half-Day PM': 1.0,
                'Three-Quarter Day': 1.5,
                '1 Hour': 0.5,
                'Custom': 1.0,
                }
      numStudents = self.getStudentTotal()
      result = numStudents * HDSVunits[self.price.name]
      return result

getHDSV 方法返回应用所在内部使用的报告指标。我希望将指标聚合为日期期间内一个月的总计。

我不是聚合/注释大师。到目前为止我的尝试还没有达到我想要的结果。

最终，我查询了指定日期之间的 Bookings，然后循环遍历结果，并通过每次迭代调用 getHDSV 方法将报告单元记录到字典中。当然，生成的字典没有按照我想要的方式排序。 所以我现在寻求一些帮助。

考虑到指标的生成方式，我可以在查询中聚合数据时调用模型方法吗？或者我应该在创建聚合时使用HDSVunits字典？或者有更好的方法吗？

谢谢。

Answer 1

您的设置相当困难，在 Price 模型上映射 HDSVunits 可能会更容易，以便更容易在查询中访问。

我能想到的最好的办法是这样的：

Booking.objects.aggregate(
    hdsv=(
        Sum('classnumbers__numberofstudents') * 
        Case(
            When(price__name='Full-Day', then=2.0),
            When(price__name='Half-Day AM', then=1.0),
            When(price__name='Full-Day PM', then=1.0),
            When(price__name='Three-Quarter Day', then=1.5),
            When(price__name='1 Hour', then=0.5),
            When(price__name='Custom', then=1.0),
            output_field=FloatField(),
        )
    )
)

如果 HDSV 值存储为 Price 模型上的字段，您可以简单地执行以下操作：

Booking.objects.aggregate(
    hdsv=Sum('classnumbers__numberofstudents') * F('price__hdsv'))

顺便说一句，您确实应该考虑遵循 < a href="https://www.python.org/dev/peps/pep-0008/#naming-conventions" rel="nofollow">Python 命名约定 这将使其他 Python 开发人员更容易帮助你。

Answer 2

如果 getHDSV 返回的数据不是来自数据库，则 aggregate 和 annotate 将无法用于收集其统计信息。