Django遍历QuerySet为每条记录添加关联查询值的实现问题

annotate 是 Django 提供的 QuerySet 级别的 API，作用是在数据库查询阶段为整个结果集追加注解字段，不能作用于单个 ORM 模型实例，这是你最初报错的核心原因。

你修改后直接给实例动态追加 discount_rate 属性的方案确实可以正常运行，但存在严重的 N+1 性能问题：每遍历一条 Invoice 记录就会触发一次数据库查询获取 FunderProgramMember，数据量稍大时接口响应速度会大幅下降，不建议在线上环境使用。

推荐方案1：数据库层面Subquery注解（性能最优） #

直接通过 Django 的 Subquery 能力在 SQL 阶段完成关联字段的注入，全程只执行2次数据库查询，完全避免循环查库，代码如下：

from django.db.models import Subquery, OuterRef

class InvoiceViewSet(ModelViewSet):
    queryset = Invoice.objects.all()
    serializer_class = InvoiceSerializer

    def get_queryset(self):
        user = self.request.user
        customer_id = Customer.objects.only('ref').get(user_id=user.id)
        
        # 构造子查询，匹配每个Invoice对应的discount_rate
        discount_subquery = FunderProgramMember.objects.filter(
            supplier=OuterRef('supplier_id'),
            buyer=OuterRef('buyer')
        ).values('discount_rate')[:1]

        # 给整个QuerySet追加注解字段，数据库层面完成关联
        queryset = Invoice.objects.filter(
            supplier_id=customer_id.ref
        ).annotate(
            discount_rate=Subquery(discount_subquery)
        )
        return queryset

该方案注入的 discount_rate 字段会和原生表字段一样挂载到每个 Invoice 实例上，无需额外处理即可正常序列化返回。

推荐方案2：批量查询映射（适合需要额外逻辑处理的场景） #

如果你确实需要在 Python 层面对每条记录做自定义处理，可以先批量查询所有关联的 FunderProgramMember 做成映射字典，再遍历赋值，避免 N+1 查询：

class InvoiceViewSet(ModelViewSet):
    queryset = Invoice.objects.all()
    serializer_class = InvoiceSerializer

    def get_queryset(self):
        user = self.request.user
        customer_id = Customer.objects.only('ref').get(user_id=user.id)
        queryset = Invoice.objects.filter(supplier_id=customer_id.ref)

        # 批量提取所有需要匹配的关联键
        relation_keys = {(inv.supplier_id, inv.buyer_id) for inv in queryset}
        # 批量查询所有关联的折扣配置
        programs = FunderProgramMember.objects.filter(
            supplier__in=[k[0] for k in relation_keys],
            buyer__in=[k[1] for k in relation_keys]
        )
        # 构造映射字典，O(1)读取
        discount_map = {(p.supplier_id, p.buyer_id): p.discount_rate for p in programs}

        # 遍历赋值，无额外数据库查询
        for invoice in queryset:
            invoice.discount_rate = discount_map.get((invoice.supplier_id, invoice.buyer_id), 0)
        
        return queryset

序列化注意事项 #

两种方案都需要在 InvoiceSerializer 中显式声明 discount_rate 字段才能正常返回：

class InvoiceSerializer(serializers.ModelSerializer):
    discount_rate = serializers.FloatField(read_only=True)
    # 其余原有字段配置
    class Meta:
        model = Invoice
        fields = '__all__' # 或者把discount_rate加入fields列表

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