如何在 O(1) 查询中过滤父母和孩子？

Question

我有一对具有基本父/子关系的模型。孩子有一个字段foo。我想做一个 SQLAlchemy 查询，该查询将返回所有具有 foo 匹配条件的子对象的 Parent 对象。与这些父母一起，我想返回孩子，但前提是他们符合过滤条件。

这些是我的模型：

class Parent(Model):
    __tablename__ = 'parent'
    id = Column(Integer, primary_key=True)
    children = relationship('Child', backref='parent')

class Child(Model):
    __tablename__ = 'child'
    id = Column(Integer, primary_key=True)
    foo = Column(String(128))
    parent_id = Column(Integer, ForeignKey('parent.id'), nullable=False)

这是一个例子。假设我有父母A 和B。我有孩子a1、a2、b1 和b2。子 a1 具有 foo = 1、a2.foo = 2 等等。

我想进行类似“child.foo == 1”的查询，并取回这些数据：

[
    {
        id: 'A',
        children: ['a1'],
    },
    {
        id: 'B',
        children: ['b1'],
    }
]

请注意，a2 和 b2 不包含在结果中。

到目前为止我已经考虑过两个想法，但还没有实现：

1. 选择子对象，然后构建父对象。

   执行此操作时，我无法查询父项上的字段，这也是必需的。

2. 选择父母，然后遍历匹配的父母并进行更多查询以收集匹配的孩子。

   这（似乎）需要 O(n) 次查询，我不想这样做。

Answer 1

要从单个查询中获得所需的一切，您需要使用 join() 和 contains_eager()：

q = session.query(Parent).\
    join(Parent.children).\
    options(contains_eager(Parent.children)).\
    filter(Child.foo == 1)