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

Unirsi con la funzione di restituzione del set (SRF) e accedere alle colonne in SQLAlchemy

Si scopre che questo non è supportato direttamente da SA, ma il comportamento corretto può essere ottenuto con un ColumnClause e un FunctionElement . Prima importa questa ricetta come descritto da zzzeek in questo problema di SA . Quindi crea uno speciale unnest funzione che include il WITH ORDINALITY modificatore:

class unnest_func(ColumnFunction):
    name = 'unnest'
    column_names = ['unnest', 'ordinality']

@compiles(unnest_func)
def _compile_unnest_func(element, compiler, **kw):
    return compiler.visit_function(element, **kw) + " WITH ORDINALITY"

Puoi quindi usarlo in join, ordini, ecc. in questo modo:

act_ref = unnest_func(Activity.ob_refs)
query = (query
    .add_columns(act_ref.c.unnest, act_ref.c.ordinality)
    .outerjoin(act_ref, sa.true())
    .outerjoin(Subscription, Subscription.ob_ref == act_ref.c.unnest)
    .order_by(act_ref.c.ordinality.desc()))