sql >> Base de Datos >  >> RDS >> PostgreSQL

Unirse con la función de devolución de conjuntos (SRF) y columnas de acceso en SQLAlchemy

Resulta que esto no es compatible directamente con SA, pero el comportamiento correcto se puede lograr con un ColumnClause y un FunctionElement . Primero importa esta receta como lo describe zzzeek en este problema de SA . Luego crea un unnest especial función que incluye el WITH ORDINALITY modificador:

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"

Luego puede usarlo en uniones, pedidos, etc. como este:

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()))