python - SQLAlchemy:按多态子类过滤现有查询(相当于 `.of_type()` 但针对 `session.query()` )

Question

问题

如果我有一个实体的现有查询，我如何限制它只返回该实体的多态子类的结果？

详细信息

使用Mapping Class Inheritance Hierarchies中的员工/工程师/经理设置sqlalchemy 文档的部分:

假设我有一个从某处获得的复杂查询，并且我不想更改哪个原始定义:

def get_complex_employee_query():
    """ Super complex query """
    query = session.query(Employee).filter(Employee.name.like('John %'))

    [... imagine a bunch of other `.filter()`, `.join()` and/or `.options()` here ...]

    return query

query = get_complex_query()

我知道我可以通过执行以下操作来过滤查询以仅列出工程师

query = query.filter(Employee.type='engineer')

但假设:

• 有 Engineer 的多态子类，
• 或者我不知道/关心Employee是type列的polymorphic_on，
• 或者我不知道/不关心type列的正确值，仅过滤给工程师

有没有办法应用 .filter()、.options() 或其他 query 方法来限制在不了解配置多态继承的特定字段的情况下向工程师(和子类)查询？

我也可以使用 .join() ，只要我不必知道/关心属于其一部分的类/表之间的主/外键关系多态层次结构。

简而言之

我想要类似Model.relationship.of_type()的东西方法，但用于查询而不是关系。

有这样的事吗？

Answer 1

简短回答

from sqlalchemy import inspect

    [...]
    eng_mapper = inspect(Engineer)
    query.filter(
        eng_mapper.polymorphic_on.in_(
            m.polymorphic_identity
            for m in eng_mapper.polymorphic_iterator()
        ),
    )

我更喜欢稍微不那么冗长的咒语，但这很有效，并且不需要了解多态层次结构的具体配置。

详细信息

何时 inspect()在 ORM 映射类上调用，它返回 Mapper对于那堂课。这与 Model.__mapper__ 类属性相同。

Mapper 包含内省(introspection)多态层次结构所需的所有信息。特别是:

• .polymorphic_on是模型中位于层次结构顶部的字段(列)，其中包含记录的多态标识值(例如，对于 Engineer 来说，这将是 Employee.type 字段)。
• .polymorphic_identity是映射模型的每个实例在 .polymorphic_on 字段中的值(例如，Engineer 将为 "engineer")。
• .polymorphic_iterator()迭代模型 Mapper 集合，其中包括 Model.__mapper__ 和 Model 所有子类的 .__mapper__递归地(例如，对于 Engineer 来说，这将是一个仅包含 Engineer.__mapper__ 的迭代器)。

为了使其更具可读性，可以轻松地将上述过滤表达式转换为函数:

from sqlalchemy import inspect

def filter_instances_of(cls):
    mapper = inspect(cls)
    return mapper.polymorphic_on.in_(
        m.polymorphic_identity for m in mapper.polymorphic_iterator()
    )

并像这样使用它:

query = query.filter(
    filter_instances_of(Engineer),
    [... other filter criteria ...]
)