Django - 从子查询中注释多个字段

Question

我正在处理一个 Django 项目，在该项目上我有一个“A”对象（A.objects.all()）的查询集，我需要从“B”对象的子查询中注释多个字段。问题是 annotate 方法只能处理每个参数的一种字段类型（DecimalField、CharField 等），因此，为了注释多个字段，我必须使用类似的方法：

A.objects.all().annotate(b_id          =Subquery(B_queryset.values('id')[:1],
                         b_name        =Subquery(B_queryset.values('name')[:1],
                         b_other_field =Subquery(B_queryset.values('other_field')[:1],
                         ... )

这是非常低效的，因为它为我要注释的每个字段在最终 SQL 上创建了一个新的子查询/子选择。我想在它的 values() 参数上使用具有多个字段的相同子选择，并将它们全部注释在 A 的查询集上。我想使用这样的东西：

b_subquery = Subquery(B_queryset.values('id', 'name', 'other_field', ...)[:1])
A.objects.all().annotate(b=b_subquery)

但是当我尝试这样做（并访问第一个元素A.objects.all().annotate(b=b_subquery)[0]）时，它引发了一个异常：

{FieldError}Expression contains mixed types. You must set output_field.

如果我设置了Subquery(B_quer...[:1], output_field=ForeignKey(B, models.DO_NOTHING))，我会收到一个数据库异常：

{ProgrammingError}subquery must return only one column

简而言之，整个问题是我有多个“属于”A 的 B，所以我需要使用子查询来为每个 AA.objects.all()选择一个特定的 B 并将其附加到该 A 上，使用 OuterRefs 和一些过滤器（我只想要 B 的几个字段），这对我来说是一个微不足道的问题。

提前感谢您的任何帮助！

Answer 1

我在这种情况下所做的是使用与预取相关的

a_qs = A.objects.all().prefetch_related(
    models.Prefetch('b_set',
        # NOTE: no need to filter with OuterRef (it wont work anyway)
        # Django automatically filter and matches B objects to A
        queryset=B_queryset,
        to_attr='b_records'
    )
)

现在a.b_records将是一个包含a's相关b对象的列表。根据您的过滤方式，B_queryset此列表可能仅限于 1 个对象。