SqlAlchemy 中的引用完整性和事务

Referential Integrity and Transactions in SqlAlchemy

我正在尝试将数据插入到两个表中,后者引用了前者。类似于:

CREATE TABLE foo (foo_id bigserial primary key, name varchar)
CREATE TABLE bar (bar_id bigserial primary key, name varchar,
                  foo_id int constraint foo_fkey references foo)

我想在一个事务中添加两行,这样如果有任何失败,就不会向数据库写入任何内容。如果我只是将这两个对象添加到我的会话中,我不会得到自动递增的 foo_id,所以我添加 foo 对象,执行刷新(填充 id),然后设置 bar 对象,然后再次刷新(我稍后还有更多工作要做,但没有被这种关系引用。在第二次刷新时,我收到引用完整性错误,因为 foo 中的行实际上没有被写入。

我可以在 SQL 中轻松完成此操作:

BEGIN TRANSACTION;
INSERT INTO foo (name) values('foo');
INSERT INTO bar (name, foo_id) VALUES('bar', currval(pg_get_serial_sequence('foo', 'foo_id')));
COMMIT TRANSACTION;

有没有办法让我暂时停止RI检查?或者,有没有更好的方法让我获得新的 foo_id 而不是冲洗?

这是我正在执行的那种 sqla 代码:

session.begin()
f = Foo()
f.name = 'Foo'
session.add(f)
session.flush()
b = Bar()
b.name = 'Bar'
b.foo_id = f.foo_id
session.add(b)
session.flush()
...
session.commit()
session.close()

我后面用的是bar_id值,这就是为什么我需要第二次刷新的原因。错误发生在第二次冲洗时。

Postgres 允许在 CTE 中插入语句。这使您可以在一个 SQL 中完成您想要的。 (参见 demo

with new_foo as 
     ( insert into foo(name) 
        values('foo') 
       returning foo_id
     ) 
insert into bar(name, foo_id) 
     select 'bar', foo_id 
       from new_foo;

即提供您的模糊化层 (SalAlchemy) 也允许它。

我无法在第二次刷新时重现完整性错误。假设模型设置具有如下一对一关系:

class Foo(Base):
    __tablename__ = 'foo'

    id = sa.Column(sa.Integer, primary_key=True)
    bar = orm.relationship('Bar', back_populates='foo', uselist=False)


class Bar(Base):
    __tablename__ = 'bar'

    id = sa.Column(sa.Integer, primary_key=True)
    foo_id = sa.Column(sa.Integer, sa.ForeignKey('foo.id'))
    foo = orm.relationship('Foo', back_populates='bar')

这些操作的输出:

# No explicit id, no flush, commit
foo1 = Foo()
bar1 = Bar(foo=foo1)
session.add(bar1)
session.commit()
print(f'foo {foo1.id}, bar {bar1.id}')

# No explicit id, flush
bar2 = Bar()
foo2 = Foo(bar=bar2)
session.add(foo2)
session.flush()
print(f'foo {foo2.id}, bar {bar2.id}')
session.commit()


# Explicit id, two flushes
foo3 = Foo()
session.add(foo3)
session.flush()
bar3 = Bar()
bar3.foo_id = foo3.id
session.add(bar3)
session.flush()
print(f'foo {foo3.id}, bar {bar3.id}')
session.commit()

是我所期望的:

foo 1, bar 1
foo 2, bar 2
foo 3, bar 3