SQLAlchemy 核心连接上下文管理器

Question

我可以为 SQLAlchemy Core 连接使用上下文管理器吗？换句话说，是

conn = engine.connect()
conn.execute('SELECT * FROM FOO')
conn.close()

和

一样

with engine.connect() as conn:
    conn.execute('SELECT * FROM FOO')

我不需要在上下文管理器退出时对提交、回滚等进行任何检查。

Answer 1

您可以将连接用作上下文管理器。

请参阅连接的__enter__ 和__exit__ 方法以了解其实现方式。 __enter__ 只返回 self 和 __exit__ 调用关闭。

或者你可以试试：

from sqlalchemy import create_engine
engine = create_engine("sqlite:///:memory:")
with engine.connect() as conn:
    print(conn.closed)
print(conn.closed)

输出：

错误的 真的