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