Django 反向查询最后创建的对象

Question

我有两个模型：

class SomeActivity(models.Model):
    name = models.ChartField(max_length=100)

class SomeStatus(models.Model):
    name = models.CharField(max_length=100)
    status = models.IntegerField(choises=STATUS_CHOISES)
    some_activity = models.ForeignKey(SomeActivity, related_name='statuses')

活动的最后创建状态是当前状态。为了得到它，我使用以下代码：

try:
    last_status = some_activity.statuses.latest('id')
except:
    last_status = None

但问题是当我想进行查询时，返回所有 Activities 与 last_status 匹配 status__in=[1, 2]。

Answer 1

这个解决方案有点扭曲，但我认为它会起作用：

from django.db.models import Max

max_status_ids = SomeActivity.objects.filter(statuses__isnull=False).annotate(
              last_status_id=Max('statuses__id')
              ).values_list('last_status_id', flat=True)
status_satisfied_ids = SomeStatus.objects.filter(id__in=list(max_status_ids),
              status__in=[1, 2]).values_list('id', flat=True)
activities = SomeActivity.objects.filter(statuses__id__in=list(
              status_satisfied_ids))

希望有更好的解决方案。

---

更新

试试

max_status_ids = SomeActivity.objects.annotate(last_status_id=Max('statuses')
                                    ).values('last_status_id')
activities = SomeActivity.objects.filter(statuses__in=max_status_ids,
                                         statuses__status__in=(1,2))

1. 在__in查找后使用qs、qs.values()或qs.values_list()时，Django会自动生成子查询。因此，无需使用list() 包装查询集（这也引入了不必要的评估，因此也无需使用中间SQL）或在qs.values_list() 中写入flat=True
2. statuses__id__in 在activities 中查找已经引入了表连接，因此最好将status__in 从status_satisfied_ids 移动到activities 以利用连接。否则，status_satisfied_ids 会引入额外的选择。

或者您可以使用Window functions of PostgreSQL 直接按排名进行过滤。