Сводное аннотированное поле из подзапроса в Django

Question

Я пытаюсь реализовать подзапрос с Django ORM, но не могу найти рабочего решения. Запрос SQL, который мне понадобится для реинжиниринга:

select t1.location, sum(t1.value_relative::numeric) as total
from (
    select
        administrative_division_id as "location",
        jsonb_array_elements("values"->'loss'->'values')->>'dim_value' as dim_value,
        jsonb_array_elements("values"->'loss'->'values')->>'value_relative' as value_relative
    from assessments_entrybaserisk
    where "values"->'scenario'->>'value' = 'Total'
) t1
where t1.dim_value::numeric = 2
group by t1.location

Я пытался использовать Django Подзапрос, как показано ниже:

subq = EntryBaseRisk.objects \
    .filter(id=OuterRef('pk'), values__scenario__value='Total') \
    .annotate(
    _dim_value=Cast(
        KeyTextTransform('dim_value',
                         JsonbArrayElements(KeyTransform('values', KeyTransform('loss', 'values')))),
        IntegerField()),
    _value_relative=Cast(
        KeyTextTransform('value_relative',
                         JsonbArrayElements(KeyTransform('values', KeyTransform('loss', 'values')))),
        FloatField())) \
    .values('assessment', 'administrative_division', '_dim_value', '_value_relative')

data = EntryBaseRisk.objects \
    .annotate(
    dim_value=Subquery(subq.values('_dim_value'), output_field=IntegerField()),
    total_relative_value=Sum(Subquery(subq.values('_value_relative'),
                                      output_field=FloatField()))) \
    .filter(**filters) \
    .values('administrative_division', 'total_relative_value') \
    .order_by('administrative_division')

, но я получаю сообщение об ошибке ProgrammingError: more than one row returned by a subquery used as an expression. Проблема в том, что SQL, сгенерированный Django, отличается от моего SQL выше, потому что он не выбирает из выбора и работает только в том случае, если подзапрос возвращает только одну строку. Я не могу выполнить агрегирование Sum в подзапросе, потому что вызовы агрегатных функций не могут содержать вызовов функций, возвращающих набор. Есть ли другой способ заставить Django генерировать SQL в соответствии с моими потребностями?

Answer 1

Вы можете попробовать что-то вроде этого:

relative_subquery = subq.values('_value_relative').annotate(total=Sum("*")).values('total')
dim_subquery = subq.values('_dim_value')

data = EntryBaseRisk.objects.annotate(
    dim_value=Subquery(dim_subquery[:1], output_field=IntegerField()),
    total_relative_value=Subquery(relative_subquery, output_field=FloatField())) \
    .filter(**filters) \
    .values('administrative_division', 'total_relative_value') \
    .order_by('administrative_division')

Здесь я суммирую значение _value_relative из subq и использую первое значение _dim_value для аннотации EntryBaseRisk набор запросов.