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
我正在尝试将数据插入到两个表中,后者引用了前者。类似于:
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