Sqlalcheny sugar for sessions

session.py

from contextlib import ContextDecorator
from functools import wraps
from typing import Any, Callable, Optional, Type


class SessionContext(ContextDecorator):
    """
    class Session(SessionContext):
        scoped_session = ...
    
    with Session():
        session.add(...)
    
    Examples:
    with Session() as s:
        s.add(...)

    @Session()
    def func():
        session.add()

    with Session(commit=False):
        session.add(...)
        session.commit()
    """
    scoped_session: _scoped_session
    
    def __init__(self, commit: bool = True) -> None:
        self._commit = commit

    def __enter__(self) -> _scoped_session:
        return self.scoped_session

    def __exit__(self, error: Optional[Type], value: Any, traceback: Any) -> None:
        try:
            if not error and self._commit:
                self.scoped_session.commit()
        finally:
            self.scoped_session.remove()  # with rollback