SQLAlchemy ORM从子查询中选择多个实体

Question

我需要查询多个实体,例如session.query(Entity1, Entity2),只从子查询而不是直接从表中查询.文档有关于从子查询中选择一个实体的内容,但我无法在文档中或通过实验找到如何选择多个实体.

我的用例是我需要通过窗口函数过滤映射类底层的表,这在PostgreSQL中只能在子查询或CTE中完成.

编辑:子查询跨越两个表的JOIN,所以我不能这样做aliased(Entity1, subquery).

Answer 1

from sqlalchemy import *
from sqlalchemy.orm import *
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class A(Base):
    __tablename__ = "a"

    id = Column(Integer, primary_key=True)
    bs = relationship("B")

class B(Base):
    __tablename__ = "b"

    id = Column(Integer, primary_key=True)

    a_id = Column(Integer, ForeignKey('a.id'))

e = create_engine("sqlite://", echo=True)
Base.metadata.create_all(e)
s = Session(e)
s.add_all([A(bs=[B(), B()]), A(bs=[B()])])
s.commit()

# with_labels() here is to disambiguate A.id and B.id.
# without it, you'd see a warning
# "Column 'id' on table being replaced by another column with the same key."
subq = s.query(A, B).join(A.bs).with_labels().subquery()


# method 1 - select_from()
print s.query(A, B).select_from(subq).all()

# method 2 - alias them both.  "subq" renders
# once because FROM objects render based on object
# identity.
a_alias = aliased(A, subq)
b_alias = aliased(B, subq)
print s.query(a_alias, b_alias).all()