sqlalchemy: отключить декларативное полиморфное соединение? - PullRequest
Новая
       6

sqlalchemy: отключить декларативное полиморфное соединение?

4 голосов
/ 02 сентября 2010

Есть ли в 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)

Я хочу сделать запрос таким, чтобы я получил все A.ids, для которых Bx> 10, если это A на самом деле B, или гдеCy == 'бла', если это A на самом деле C, все упорядочено по p.

Чтобы сделать это итеративно, я начинаю только с первой части - "получить все A.id, для которых Bx> 10, если это A на самом деле B ".Поэтому я подумал, что начну с внешнего соединения: 

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

... за исключением того, что, похоже, нет способа избежать использования этого предложения externaljoin ((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 engine на самом деле выполнять это внутреннее соединение, или он его исключит?

Ответы [ 2 ]

1 голос
/ 03 сентября 2010

Вы можете попробовать построить запросы для каждого подкласса индивидуально, а затем объединить их вместе. При запросе 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

Вы по-прежнему получаете подвыбор, но только один «слой» объединений - внешний выбор просто переименовывает столбец.

1 голос
/ 02 сентября 2010

Вы должны использовать with_polymorphic () вместо externaljoin (), что, похоже, возвращает ожидаемые результаты: 

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.

...