python - 如何创建默认等于主键的 sqlalchemy 列？

Question

我有一个看起来像这样的类:

class Foo(Base):
    pk = Column(Integer, Sequence('foo'), primary_key=True)

我想创建另一个字段 ref ，对于新创建的对象，它默认等于 pk 。不幸的是，我使用的是 Postgres，它有一个 non-standard nextval() function ，所以我不能简单地将两列都默认为 Sequence 对象。

这是我目前的解决方案:

class Foo(Base):
    pk = Column(Integer, Sequence('foo'), primary_key=True)
    ref = Column(Integer, index=True, nullable=False, default=func.currval('foo'))

然而，这取决于两件事:SQLAlchemy 在 pk 之前生成带有 ref 的 INSERT 语句，和 Postgres 从左到右评估表达式(因此 nextval() 生成的 pk 调用总是发生在 currval 调用生成之前通过 ref )。

对于如何更好地完成这项工作的任何建议，我将不胜感激!

编辑:还有设置 default=-1 或其他东西并强制调用者更新 ref 字段的更多解决方案。这对 SQLA/Postgres 语义的更改更安全、更健壮，但对于类的客户端来说非常不方便。

Answer 1

可以利用 SQLAlchemy 的 ORM after_insert映射器事件:

from sqlalchemy import event

# Model from OP
class Foo(Base):
    pk = Column(Integer, Sequence('foo'), primary_key=True)
    ref = Column(Integer, index=True, nullable=False, default=-1)

    @staticmethod
    def fill_ref(mapper, connection, target):
        # Note: You cannot use ORM in the event callbacks
        connection.execute(mapper.mapped_table.update()
                                 .values(ref=target.pk))
                                 .where(table.c.pk==target.pk))

event.listen(Foo, 'after_insert', Foo.fill_ref)