python - 如何在 sqlalchemy 混合表达式中运行查询？

Question

有一个与类标签具有“多”关系的类 Run。我需要能够根据标签的函数(tag_collection)过滤运行。

下面是我的代码片段。在本例中，我使用的是自动映射器，但我认为这并不重要。

Base = automap_base()

class Run(Base):
    __tablename__ = 'runs'
    macro_id = Column(ForeignKey('macros.id'))  

    @hybrid_property
    def tag_count(self):
        return(len(self.tag_collection))

    @hybrid_property
    def default_region(self):
        return(reduce(lambda  memo, t: memo or t.name == "region" and t.value == "default", self.tag_collection, False))

    @default_region.expression
    def default_region(cls):
        # how do I do the query here?

class Tag(Base):
    __tablename__ = 'tags'
    run_id = Column(ForeignKey('runs.id'))    

如果我获得 Run 的实例，我可以使用 default_region 属性，但是当我运行查询时，并尝试使用 default_region 属性进行过滤，我要么在 @hybird_property 装饰的 default_region 方法中传入 Run 类(作为 self)，要么最终进入 @default_region.expression 修饰的 default_region 方法，它也接收 Run 类。

示例查询:

session.query(Run).filter(Run.default_region == True).all()

我认为我需要在 @default_region.expression 修饰的 default_region 方法中执行查询，但我不知道如何获取 session 句柄。我知道 Session.object_session(someobject)，但我认为 someobject 需要是 orm 类的实例而不是类本身。

我认为我的过滤方法要么完全错误，要么我需要以某种方式获取 @default_region.expression 装饰的 default_region 中的 session 对象的句柄> 方法。

有没有更好的方法来完成这种过滤？

如何在@default_region.expression修饰的default_region方法中获取 session ？

Answer 1

在我看来，这是一个 Correlated Subquery Relationship Hybrid 。当 Run.default_region 添加为过滤器时，您想要更新要添加的查询

WHERE EXISTS (
    SELECT 1 FROM tags
    WHERE run_id = runs.id
      AND name = "region"
      AND value = "default")

将其作为表达式返回，并带有 exists() function :

from sqlalchemy.sql import exists, and_

@default_region.expression
def default_region(cls):
    return exists([1]).where(and_(
        Tag.run_id == cls.id,
        Tag.name == 'region',
        Tag.value == 'default'))

请注意，您要避免在这里使用== True或== False，没有必要!直接使用该属性即可:

session.query(Run).filter(Run.default_region).all()

或使用~来否定:

session.query(Run).filter(~Run.default_region).all()

通过定义.expression method对于属性，您已告诉 SQLAlchemy 在访问类上的属性时使用该方法，因此上下文始终是那里的类。如果不定义 .exists (或 .comparator )，Run.default_region 将调用 default_region getter (用 `@hybrid_property 修饰的方法)作为类方法，因此只有那时该方法才必须与传入的实例或类一起使用。对于某些混合属性实现，这恰好取决于它们使用属性的方式关于上下文。

至于 default_region instance 属性的实现，我不会在那里使用 reduce() ；使用any() function :

@hybrid_property
def default_region(self):
    return any(
        t.name == "region" and t.value == "default"
        for t in self.tag_collection)

any() 在找到匹配项时停止迭代提供的生成器表达式。