Django подзапрос и аннотации с OuterRef

У меня проблемы с использованием annotate() с OuterRef в подзапросах Django 1.11. Пример модели:

class A(models.Model):
    name = models.CharField(max_length=50)


class B(models.Model):
    a = models.ForeignKey(A)

Теперь запрос с подзапросом (который на самом деле не имеет никакого смысла, но иллюстрирует мою проблему):

A.objects.all().annotate(
    s=Subquery(
        B.objects.all().annotate(
            n=OuterRef('name')
        ).values('n')[:1],
        output_field=CharField()
    )
)

Это дает следующую ошибку:

Traceback (most recent call last):
  File "<console>", line 1, in <module>
  File "myapp/models.py", line 25, in a
    n=OuterRef('name')
  File ".virtualenv/lib/python2.7/site-packages/django/db/models/query.py", line 948, in annotate
    if alias in annotations and annotation.contains_aggregate:
AttributeError: 'ResolvedOuterRef' object has no attribute 'contains_aggregate'

Разве невозможно аннотировать подзапрос, основанный на OuterRef?


Обновление № 1

Найден обходной путь для этого, который позволит мне двигаться вперед, но это нехорошо.

class RawCol(Expression):

    def __init__(self, model, field_name, output_field=None):
        field = model._meta.get_field(field_name)
        self.table = model._meta.db_table
        self.column = field.column
        super().__init__(output_field=output_field)

    def as_sql(self, compiler, connection):
        sql = f'"{self.table}"."{self.column}"'
        return sql, []

изменения OuterRef использовать пользовательское выражение

A.objects.all().annotate(
    s=Subquery(
        B.objects.all().annotate(
            n=RawCol(A, 'name')
        ).values('n')[:1],
        output_field=CharField()
    )
)

Урожайность

SELECT "myapp_a"."id",
       "myapp_a"."name",

  (SELECT "myapp_a"."name" AS "n"
   FROM "myapp_b" U0 LIMIT 1) AS "s"
FROM "myapp_a"

0 ответов

Это известная ошибка в Django, исправленная в версии 3.0.

См. https://code.djangoproject.com/ticket/28621 для обсуждения.

Если вам, как и мне, нужно аннотировать поле, чтобы вы могли использовать его в следующем подзапросе, помните, что вы можете складывать OuterRef нравиться:


id__in=SubQuery(
    MyModel.objects.filter(
        field=OuterRef(OuterRef(some_outer_field))
    )
)

Одно поле одной связанной строки B может быть аннотировано таким образом для каждой строки A.

subq = Subquery(B.objects.filter(a=OuterRef('pk')).order_by().values('any_field_of_b')[:1])
qs = A.objects.all().annotate(b_field=subq)

(Это было более читабельно, чтобы написать его в виде двух команд с временной переменной. Это стиль, похожий на документы.)

Он скомпилирован в один запрос SQL:

>>> print(str(qs.suery))
SELECT a.id, a.name,
  (SELECT U0.any_field_of_b FROM b U0 WHERE U0.a_id = (a.id)  LIMIT 1) AS b_field
FROM a

(упрощено без "appname_")

Другие вопросы по тегам