从烧瓶中的 PostgreSQL 函数提交事务

Posted 2023-02-23

【中文标题】从烧瓶中的 PostgreSQL 函数提交事务

【英文标题】：Commiting a transaction from a PostgreSQL function in flask

【发布时间】：2014-02-23 13:06:56

【问题描述】：

我是 Flask 和 SQLAlchemy 的新手（过去 3 年一直在使用 Django）。我需要调用一个 existing PostgreSQL 函数来写入数据库中的 3 个不同的表。这是我无法控制的（我只需要让它工作）。该函数返回一条记录（自定义 Postgres 类型），其中包含有关结果的信息。代码如下：

from flask.ext.sqlalchemy import SQLAlchemy

db = SQLAlchemy()

...

retval = db.engine.execute(
    'SELECT * FROM add_purchase(%s, %s, %s, %s, %s, %s, %s);',
    clean_data.account_id, amount, tax, deposit, clean_data.pos_id, 
    g.token_id, clean_data.external_reference_id).first()

app.logger.info(retval) # for debugging

db.session.commit()

上面的代码运行没有错误。从日志消息中，我可以看到从数据库返回了正确的数据。但是，如果我去 psql，我看不到新插入的数据。事务似乎从未真正提交过。

我能找到的大部分文档和示例都是基于使用 SQLAlchemy 的 ORM。我在这里做错了什么？

堆栈信息：

Flask==0.10.1
Flask-SQLAlchemy==1.0
psycopg2==2.5.2
Python 2.7.3
Postgresql 9.3

更新

我已经找到了一种使它起作用的方法，这里有一个对我真正有用的示例：

from sqlalchemy import exc

...

connection = db.engine.connect()
trans = connection.begin()
try:
    retval = connection.execute(
        'SELECT * FROM add_purchase(%s, %s, %s, %s, %s, %s, %s);',
        clean_data.account_id, amount, tax, deposit,
        clean_data.pos_id, g.token_id,
        clean_data.external_reference_id).first()
except exc.SQLAlchemyError:
    trans.rollback()
    raise
else:
    trans.commit()

我已经查看了 SQLAlchemy 的连接方法中的代码。似乎调用“connect()”只是从池中获取一个新连接（默认情况下似乎设置为 size=5 - 至少我没有将其设置为 5）。所以，我不确定它是否有很大的危害。但是，似乎应该通过使原始问题中的示例正常工作来使此代码更清晰。我仍然希望有人能告诉我该怎么做。

【参考方案1】：

如果您在 SQLAlchemy 中使用原始 SQL，那么我认为您必须自己发出 BEGIN 和 COMMIT 语句。尝试添加如下内容：

...
begin_retval = db.engine.execute('BEGIN;')
retval = db.engine.execute('SELECT * FROM add_purchase(%s, %s, %s, %s, %s, %s, %s);',
clean_data.account_id, amount, tax, deposit, clean_data.pos_id, 
g.token_id, clean_data.external_reference_id).first()
commit_retval = db.engine.execute('COMMIT;')
...

【讨论】：

非常感谢您的回复。我已经测试了那个确切的代码，它没有提交事务。它也不会引发错误。

嗯。那么 add_purchase 是如何工作的呢？你有那个代码吗？这可能会有所帮助...

是的；请参阅原始问题。 “上面的代码运行没有错误。从日志消息中，我可以看到从数据库返回正确的数据。”