如何在Django中复制QuerySet注解并实现关联字段排序去重？

问题背景 #

在Django中使用外键属性对QuerySet排序时，会出现重复条目；若在同一QuerySet上同时指定排序和去重操作，会触发SELECT DISTINCT ON expressions must match initial ORDER BY expressions错误。

我尝试过两种思路解决：一是创建新QuerySet，通过已有注解QuerySet的ID筛选后再排序；二是复制queryset.query.annotations字典复用注解，但均未成功。

模型定义 #

class Home(models.Model):
    street = models.CharField(...)
    city = models.CharField(...)
  
class Person(models.Model):
    name = models.CharField(...)
    home = models.ForeignKey(Home, ...)

需求目标 #

• 获取按street与city拼接字符串排序的Person QuerySet，且无重复条目。

已尝试的方案 #

尝试1：同一QuerySet注解+排序+去重 #

queryset = Person.objects.annotate(addr=Concat('home__street', Value(' '), 'home__city'))

queryset.filter(...).order_by('addr').distinct('pk')
# 执行失败，触发`SELECT DISTINCT ON expressions must match initial ORDER BY expressions`错误

尝试2：通过主键筛选新QuerySet #

Person.objects.filter(
    pk__in=queryset.filter(...).values('pk').order_by('pk').distinct('pk')
).order_by('addr')

# 执行失败，新QuerySet未添加`addr`注解，无法按该字段排序
# 虽可自定义BaseModel复用注解，但涉及大量复杂关联模型，成本过高

尝试3：复制QuerySet注解 #

annotations = queryset.query.annotations

Person.objects.annotate(**annotations)\
  .filter(
       pk__in=queryset.order_by('pk').distinct('pk').values('pk')
  ).order_by('addr')

# 执行失败，注解与原QuerySet绑定紧密，无法直接复用

可行解决方案 #

方案1：子查询去重+主查询注解排序 #

先通过子查询获取去重后的Person主键集合，再在主查询中重新添加注解、筛选主键并排序，避免注解复用的问题：

# 子查询：获取去重后的Person主键
distinct_pks = Person.objects.filter(...).values('pk').distinct()

# 主查询：添加地址注解、筛选主键、按地址排序
result = Person.objects.annotate(
    addr=Concat('home__street', Value(' '), 'home__city')
).filter(pk__in=distinct_pks).order_by('addr')

方案2：PostgreSQL专属：调整排序与去重字段顺序 #

若使用PostgreSQL数据库，DISTINCT ON要求排序的起始字段与去重字段匹配。可以先按pk去重，再按addr排序：

queryset = Person.objects.annotate(
    addr=Concat('home__street', Value(' '), 'home__city')
).filter(...).order_by('pk', 'addr').distinct('pk')

注：这种方式下QuerySet会先按pk排序，再按addr排序，若需要纯按addr排序，建议使用方案1。

方案3：直接去重后再添加注解排序 #

如果重复条目是由跨表查询的笛卡尔积导致，可先对Person QuerySet去重，再添加注解并排序：

result = Person.objects.filter(...).distinct().annotate(
    addr=Concat('home__street', Value(' '), 'home__city')
).order_by('addr')

内容的提问来源于stack exchange，提问作者thetva