从flask中的PostgreSQL函数提交事务

Question

我是Flask和SQLAlchemy的新手（在过去3年中一直与Django合作）。我需要调用一个现有的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连接方法中的代码。似乎调用“连接（）”只是从池中获得一个新的连接（默认情况下似乎设置为size=5——至少我没有设置为5）。所以，我不确定它是否有很大的危害。然而，通过使原始问题中的示例工作，似乎应该有办法使代码更干净。我仍然希望有人能告诉我怎么做。

Answer 1

如果您将原始SQL与SQLAlchemy一起使用，那么我认为您必须自己发出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;')
...