try-finally с SqlAlchemy: это хорошая привычка?

Question

Я привык делать это:

from sqlalchemy.orm import sessionmaker 
from sqlalchemy.engine import create_engine 

Session = sessionmaker() 
engine = create_engine("some connection db string", echo=False) 
Session.configure(bind=engine) 

db_con = Session() 

try: 
    # DB MANIPULATION 
finally: 
    db_con.close() 

Это хорошая привычка? Если да, то почему sqlalchemy не позволяет вам просто:

with Session() as db_con: 
    # DB MANIPULATION 

?

Answer 1

Нет, это не очень хорошая практика. Это легко забыть и сделает код более запутанным.

Вместо этого вы можете использовать контекстный менеджер contextlib.closing и сделать это единственным способом получить сеанс.

# Wrapped in a custom context manager for better readability 
@contextlib.contextmanager 
def get_session(): 
    with contextlib.closing(Session()) as session: 
     yield session 

with get_session() as session: 
    session.add(...) 

Answer 2

Во-первых, если вы закончили работу с сеансом, вы должны закрыть сеанс. session.close вернет соединение обратно в пул двигателей, и если вы выходите из программы, вы должны распоряжаться пулом двигателя engine.dispose.

Теперь на ваш вопрос. В большинстве случаев сеансы будут использоваться для приложений с большим сроком выполнения, таких как веб-сервер. Там, где имеет смысл централизовать управление сеансом. Например, в flask-sqlalchemy сеанс создается с запуском каждого веб-запроса и закрывается, когда запрос завершен.