sqlalchemy：关闭声明性多态连接？

Question

sqlalchemy 有没有办法在单个查询中关闭声明式的多态连接加载？大多数时候这很好，但我有：

class A(Base) : 
   discriminator = Column('type', mysql.INTEGER(1), index=True, nullable=False)
   __mapper_args__ = { 'polymorphic_on' : discriminator }
   id = Column(Integer, primary_key=True)
   p = Column(Integer)

class B(A) : 
   __mapper_args__ = { 'polymorphic_identity' : 0 }
   id = Column(Integer, primary_key=True)
   x = Column(Integer)

class C(A) : 
   __mapper_args__ = { 'polymorphic_identity' : 1 }
   id = Column(Integer, primary_key=True)
   y = Column(String)

我想进行一个查询，以便获得 Bx > 的所有 A.ids。 10，如果 A 实际上是 B，或者 Cy == 'blah'，如果 A 实际上是 C，则全部按 p 排序。

为了迭代地做到这一点，我从第一部分开始 - “如果 A 实际上是 B，则获取 Bx > 10 的所有 A.id”。所以我想我应该从外部连接开始：

session.query(A.id).outerjoin((B, B.id == A.id)).filter(B.x > 10)

...除了似乎没有办法避免让outerjoin((B, B.id == A.id))子句生成A中所有内容的完整连接子选择中 B 中的所有内容。如果 B 不继承自 A，那么这种情况就不会发生，所以我认为是多态声明性代码生成做到了这一点。有办法把它关掉吗？或者强制外连接做我想做的事？

我想要的是这样的：

select a.id from A a left outer join B b on b.id == a.id where b.x > 10

但我得到的是这样的：

select a.id from A a left outer join (select B.id, B.x, A.id from B inner join A on B.id == A.id)

...顺便说一句，如果不可能，那么后者的效率是否比前者低？ SQL 引擎实际上会执行该内连接，还是会忽略它？

Answer 1

您可以尝试单独为每个子类构建查询，然后将它们联合在一起。查询 B.id 时，SQLAlchemy 隐式加入超类并返回 A.id，因此采用 B.id 的选择并集C.id 仅返回单个列。

>>> b_query = session.query(B.id).filter(B.x > 10)
>>> c_query = session.query(C.id).filter(C.y == 'foo')
>>> print b_query.union(c_query)
SELECT anon_1."A_id" AS "anon_1_A_id" 
FROM (SELECT "A".id AS "A_id" 
FROM "A" JOIN "B" ON "A".id = "B".id 
WHERE "B".x > ? UNION SELECT "A".id AS "A_id" 
FROM "A" JOIN "C" ON "A".id = "C".id 
WHERE "C".y = ?) AS anon_1

您仍然会得到一个子选择，但只有一个连接“层” - 外部选择只是重命名列。

Answer 2

您应该使用 with_polymorphic()而不是outerjoin()，它似乎返回了预期的结果：

session.query(A).with_polymorphic(B).filter(B.x > 10).all()
# BEGIN
# SELECT "A".type AS "A_type", "A".id AS "A_id", "A".p AS "A_p", "B".id AS "B_id", "B".x AS "B_x" 
# FROM "A" LEFT OUTER JOIN "B" ON "A".id = "B".id 
# WHERE "B".x > ?
# (10,)
# Col ('A_type', 'A_id', 'A_p', 'B_id', 'B_x')

相比：

session.query(A.id).outerjoin((B, B.id == A.id)).filter(B.x > 10)
# BEGIN
# SELECT "A".id AS "A_id" 
# FROM "A" LEFT OUTER JOIN (SELECT "A".type AS "A_type", "A".id AS "A_id", "A".p AS "A_p", "B".id AS "B_id", "B".x AS "B_x" 
# FROM "A" JOIN "B" ON "A".id = "B".id) AS anon_1 ON anon_1."A_id" = "A".id 
# WHERE anon_1."B_x" > ?
# (10,)
# Col ('A_id',)

我用来测试这个的代码，以防有人想测试这个简洁的SQLAlchemy：

#!/usr/bin/env python
import logging
from sqlalchemy import *
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

Base = declarative_base()

class A(Base) :
   __mapper_args__ = { 'polymorphic_on' : discriminator }
   __tablename__ = 'A'

   id = Column(Integer, primary_key=True)
   discriminator = Column('type', Integer, index=True, nullable=False)
   p = Column(Integer)

class B(A) :
   __mapper_args__ = { 'polymorphic_identity' : 0 }
   __tablename__ = 'B'

   id = Column(Integer, ForeignKey('A.id'), primary_key=True)
   x = Column(Integer)

class C(A) :
   __mapper_args__ = { 'polymorphic_identity' : 1 }
   __tablename__ = 'C'

   id = Column(Integer, ForeignKey('A.id'), primary_key=True)
   y = Column(String)

meta = Base.metadata
meta.bind = create_engine('sqlite://')
meta.create_all()

Session = sessionmaker()
Session.configure(bind=meta.bind)
session = Session()

log = logging.getLogger('sqlalchemy')
log.addHandler(logging.StreamHandler())
log.setLevel(logging.DEBUG)

session.query(A.id).outerjoin((B, B.id == A.id)).filter(B.x > 10).all()
session.query(A).with_polymorphic(B).filter(B.x > 10).all()

我在Python 2.7上用SQLAlchemy 0.6.4运行了这个。