SQLAlchemy Core Connection Context Manager

Question

Can I use a context manager for SQLAlchemy Core connections? In other words, is

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

the same as

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

I don't need any checks for commits, rollbacks, etc at context manager exit.

Answer 1

You can use a connection as context manager.

See the connections's __enter__ and __exit__ methods for how this is implemented. __enter__ just returns self and __exit__ calls close.

Or you could just try it:

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

output:

False
True