Фильтр Sqlalchemy родительский и дочерний стол для родителей

Question

У меня есть несколько моделей, которые я пытаюсь найти, поэтому я хочу показать, как возвращаются результаты всех родительских объектов, где либо имя Родителя - "foo", либо имя ребенка - "foo".

У меня есть запрос:

parents = Session.query(Parent).\
            join(Child_s3).\
            filter(Parent.name.ilike("%foo%")).\
            filter(Child_s3.name.ilike("%foo%")).\
            order_by(asc(Product.name))

И модели:

class Parent(BaseSO):
    __tablename__ = 'parents'
    id = Column(Integer, primary_key=True)
    name = Column(Unicode(100), nullable=False, unique=True)
    colours = relationship('Child_s3', secondary=Parent_images, backref='Parentc')

class Child_s3(BaseSO):
    __tablename__ = 'children'
    id = Column(Integer, primary_key=True)
    name = Column(Unicode)

Parent_images = Table(
    'Parent_images', BaseSO.metadata,
    Column('parent_id', Integer, ForeignKey('parents.id')),
    Column('child_id', Integer, ForeignKey('children.id'))
)

В моем запросе отображаются родители с именем «foo», но не отображаются родительские объекты, у которого также есть дочерние элементы с именем "foo", может ли кто-нибудь помочь построить этот запрос для поиска в обеих таблицах соответствующих родительских объектов?

Answer 1

Этот код показывает, как получить результат, используя явное соединение или подзапрос:

import sqlalchemy as sa
from sqlalchemy import orm
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()


Parent_images = sa.Table(
    'Parent_images', Base.metadata,
    sa.Column('parent_id', sa.Integer, sa.ForeignKey('parents.id')),
    sa.Column('child_id', sa.Integer, sa.ForeignKey('children.id'))
)


class Parent(Base):
    __tablename__ = 'parents'
    id = sa.Column(sa.Integer, primary_key=True)
    name = sa.Column(sa.Unicode(100), nullable=False, unique=True)
    colours = orm.relationship('Child_s3', secondary=Parent_images, backref='parents')

    def __repr__(self):
        return 'Parent(name=%s)' % self.name

    __str__ = __repr__


class Child_s3(Base):
    __tablename__ = 'children'
    id = sa.Column(sa.Integer, primary_key=True)
    name = sa.Column(sa.Unicode)

    def __repr__(self):
        return 'Child_s3(name=%s)' % self.name

    __str__ = __repr__


if __name__ == '__main__':
    engine = sa.create_engine('sqlite:///')
    Base.metadata.drop_all(engine)
    Base.metadata.create_all(engine)
    Session = orm.sessionmaker(bind=engine)

    session = Session()
    for parent, child in [('boofoo', 'spam'), ('baz', 'foobar'), ('bar', 'quux')]:
        p1 = Parent(name=parent)
        session.add(p1)
        p1.colours.append(Child_s3(name=child))
    session.commit()

    print('Join')
    session = Session()
    q = (session.query(Parent)
                .join(Child_s3, Parent.colours)
                .filter(sa.or_(Parent.name.ilike('%foo%'),
                               Child_s3.name.ilike('%foo%'))))
    for p in q.all():
        print(p, p.colours)
    session.commit()
    print()

    print('Subquery')
    session = Session()
    q = (session.query(Parent)
                .filter(sa.or_(Parent.name.ilike('%foo%'),
                               Parent.colours.any(Child_s3.name.ilike('%foo%')))))
    for p in q.all():
        print(p, p.colours)
    session.commit()
    print()

Запрос на соединение

q = (session.query(Parent)
            .join(Child_s3, Parent.colours)
            .filter(sa.or_(Parent.name.ilike('%foo%'),
                           Child_s3.name.ilike('%foo%'))))

генерирует это SQL

SELECT parents.id AS parents_id, parents.name AS parents_name 
FROM parents JOIN "Parent_images" AS "Parent_images_1" ON parents.id = "Parent_images_1".parent_id JOIN children ON children.id = "Parent_images_1".child_id 
WHERE lower(parents.name) LIKE lower(?) OR lower(children.name) LIKE lower(?)

Подзапрос

q = (session.query(Parent)
            .filter(sa.or_(Parent.name.ilike('%foo%'),
                            Parent.colours.any(Child_s3.name.ilike('%foo%')))))

генерирует это SQL:

SELECT parents.id AS parents_id, parents.name AS parents_name            
FROM parents                                                                                                                        
WHERE lower(parents.name) LIKE lower(?) OR (EXISTS (SELECT 1                                                                        
FROM "Parent_images", children                                                                                                      
WHERE parents.id = "Parent_images".parent_id AND children.id = "Parent_images".child_id AND lower(children.name) LIKE lower(?)))

Сценарий производит этот вывод из данных примера:

Join
Parent(name=baz) [Child_s3(name=foobar)]
Parent(name=boofoo) [Child_s3(name=spam)]

Subquery
Parent(name=boofoo) [Child_s3(name=spam)]
Parent(name=baz) [Child_s3(name=foobar)]