Sqlalchemy session as contextmanager

with_session.py

@contextmanager
def session(*args: Any, **kwargs: Any) -> Iterator[sa.orm.Session]:
    with closing(session_factory(*args, **kwargs)) as _session:
        try:
            yield _session
        except Exception:
            _session.rollback()
            raise
        else:
            _session.commit()