PostgreSQL
 sql >> Base de données >  >> RDS >> PostgreSQL

SQLAlchemy, PostgreSQL et array_agg :comment sélectionner des éléments dans array_agg ?

Vous faites probablement tout correctement, mais obtenez d'abord des tableaux vides. Dans votre requête précédente, vous avez utilisé le filtrage en python (len(x[1]) > 1 ). Vous pouvez imprimer Query expression avant de l'exécuter pour vous en assurer.

Vous devriez probablement ajouter un having clause à votre requête de 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()
)

Ensuite, vous n'aurez pas non plus besoin d'un filtrage en python.