使用带有psycopg2的try / except或“with closing”？

Question

我在Python中使用Psycopg2来访问PostgreSQL数据库。我很好奇使用with closing()模式来创建和使用游标是否安全，或者我是否应该使用围绕查询的显式try/except。我的问题是关于插入或更新，以及交易。

据我所知，所有Psycopg2查询都发生在一个事务中，并且由调用代码来提交或回滚事务。如果在with closing(...块内发生错误，是否发出回滚？在旧版本的Psycopg2中，在close()上明确发布了回滚，但现在不再是这种情况了（参见http://initd.org/psycopg/docs/connection.html#connection.close）。

通过一个例子，我的问题可能更有意义。这是使用with closing(...的示例

with closing(db.cursor()) as cursor:
     cursor.execute("""UPDATE users                    
             SET password = %s, salt = %s
             WHERE user_id = %s""",
             (pw_tuple[0], pw_tuple[1], user_id))
     module.rase_unexpected_error()
     cursor.commit()

当module.raise_unexpected_error（）引发错误时会发生什么？交易是否回滚？据我了解事务，我要么提交它们要么回滚它们。那么在这种情况下，会发生什么？

或者我可以这样写我的查询：

cursor = None
try:
    cursor = db.cursor()
    cursor.execute("""UPDATE users                    
            SET password = %s, salt = %s
            WHERE user_id = %s""",
            (pw_tuple[0], pw_tuple[1], user_id))
    module.rase_unexpected_error()
    cursor.commit()
except BaseException:
    if cursor is not None:
        cursor.rollback()
finally:
    if cursor is not None:
        cursor.close()

另外我应该提一下，我不知道Psycopg2的连接类cursor()方法是否会引发错误（文档没有说）所以更安全而不是抱歉，不是吗？

我应该使用哪种方法发布查询和管理事务？

Answer 1

Your link to the Psycopg2 docs有点解释它，不是吗？

...请注意，关闭连接而不先提交更改将导致任何挂起的更改被丢弃，就像执行了ROLLBACK一样（除非选择了不同的隔离级别：请参阅set_isolation_level（））。

版本2.2中已更改：之前在close（）上由Psycopg发出了显式的ROLLBACK。该命令可能已在不适当的时间发送到后端，因此Psycopg当前依赖后端隐式丢弃未提交的更改。已知一些中间件在交易期间（当状态为STATUS_IN_TRANSACTION时）关闭连接时表现不正确，例如， PgBouncer报告一个不干净的服务器并丢弃该连接。要避免此问题，您可以确保在关闭之前使用commit（）/ rollback（）终止事务。

因此，除非您使用不同的隔离级别或使用PgBouncer，否则您的第一个示例应该可以正常工作。但是，如果您希望对事务期间发生的事情进行更精细的控制，那么try / except方法可能是最好的，因为它与数据库事务状态本身并行。