PostgreSQL
 sql >> Database >  >> RDS >> PostgreSQL

SQLAlchemy, PostgreSQL e array_agg:come selezionare elementi da array_agg?

Probabilmente stai facendo tutto bene, ma prima ottieni array vuoti. Nella tua query precedente hai utilizzato il filtro in Python (len(x[1]) > 1 ). Puoi stampare Query espressione prima di eseguirla per essere sicuro.

Probabilmente dovresti aggiungere un having clausola alla tua query di base:

from sqlalchemy import Integer
from sqlalchemy.dialects.postgresql import ARRAY

cats_agg = func.array_agg(ProductCategory.id, type_=ARRAY(Integer)).label('cats')
prods = (
    session.query(
        Product.id.label('id'),
        cats_agg,
    .outerjoin(
        ProductCategory,
        ProductCategory.product_id == Product.id)
    .group_by(Product.id)
    .having(func.array_length(cats_agg, 1) > 1)
    .subquery()
)

Quindi non avrai nemmeno bisogno del filtro in Python.