Django注释另一个模型的字段值

时间:2020-10-16 14:51:14

标签: python django django-rest-framework

我想用另一个历史记录模型中的值注释MyModel查询集。我的模型关系如下:

class Stage(models.Model):
    name = models.CharField()


class History(models.Model):
    mymodel = models.ForeignKey(
        MyModel,
        on_delete=models.CASCADE,
    )
    stage = models.ForeignKey(
        Stage,
        on_delete=models.DO_NOTHING
     )
    created_at = models.DateTimeField(
        auto_now_add=True
    )

class MyModel(models.Model):
    id = models.AutoField(
        primary_key=True,
    )
    ...

我想在ViewSet中执行的近似操作(order_by,我需要获取最新的阶段):

class MyModelViewSet(viewsets.ModelViewSet):
    ...
    def get_queryset(self):
        qs = super(MyModelViewSet, self).get_queryset()
        qs = qs.annotate(last_stage_name=Value(
            History.objects.filter(mymodel__id=F('id')).order_by('-created_at')[0].stage.name,
            output_field=CharField()))

但是我从第一个对象得到的所有对象处于同一阶段,而不是从每个对象的最新阶段得到,因此操作逻辑是错误的。我想我应该摆脱History.objects.filter但找不到合适的解决方案。

1 个答案:

答案 0 :(得分:3)

您可以使用Subquery expression [Django-doc]

from django.db.models import OuterRef, Subquery

class MyModelViewSet(viewsets.ModelViewSet):
    
    # …
    
    def get_queryset(self):
        return super().get_queryset().annotate(
            last_stage_name=Subquery(
                History.objects.filter(
                    mymodel_id=OuterRef('pk')
                ).order_by('-created_at').values('stage__name')[:1]
            )
        )