summaryrefslogtreecommitdiff
path: root/django/contrib/postgres/expressions.py
blob: 469f4e9fb6813ccd23d1eabf8b25ec8b602801f1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
from django.contrib.postgres.fields import ArrayField
from django.db.models import Subquery
from django.utils.functional import cached_property


class ArraySubquery(Subquery):
    template = "ARRAY(%(subquery)s)"

    def __init__(self, queryset, **kwargs):
        super().__init__(queryset, **kwargs)

    @cached_property
    def output_field(self):
        return ArrayField(self.query.output_field)