使用 annotate() 包含一个 django 模型对象属性

Question

我有一个包含 @property 的模型，我希望在执行以下操作时包含该模型 vals = MyModel.objects.values() Another S/O post 建议这样做的方法是重写管理器类的 get_queryset 方法，并通过调用 annotate(xtra_field=...) 来增强。

问题是我找不到如何使用模型属性执行此操作的示例。我尝试过类似的东西 super(SuperClass, self).get_queryset().annotate(xtra_fld=self.model.xtra_prop) 及其众多变体，但都不起作用。

问题，我认为注释参数的 RHS 应该是什么？

（而且，是的，我知道还有其他更笨重的方法可以做到这一点，比如迭代 values() 返回的内容并在我去的时候添加属性。对我来说，这不是一个非常优雅（甚至是 django）的解决方案.

Answer 1

除非您在数据库中生成属性值，否则您无法在不覆盖 django 的内部方法的情况下执行此操作 - 我建议不要这样做。

一个更优雅的解决方案可能是向a custom queryset 添加一个额外的方法，它会根据请求生成额外的字段：

from django.db import models

class AreaQuerySet(models.QuerySet):
    def values_with_area(self, *fields):
        # width & height are always required to calculate the area
        fields = set(fields)
        fields.update(['width', 'height'])

        for row in self.values(*fields):
            row['area'] = Widget.calculate_area(row['width'], row['height'])
            yield row

class Widget(models.Model):
    # ...
    objects = AreaQuerySet.as_manager()

    @classmethod
    def calculate_area(cls, width, height):
        return width * height

    @property
    def area(self):
        return Widget.calculate_area(self.width, self.height)