lavkach3

1
from uuid import uuid4
2

3
from .session import session, set_session_context, reset_session_context
4

5

6
def standalone_session(func):
7
    async def _standalone_session(*args, **kwargs):
8
        session_id = str(uuid4())
9
        context = set_session_context(session_id=session_id)
10

11
        try:
12
            await func(*args, **kwargs)
13
        except Exception as e:
14
            await session.rollback()
15
            raise e
16
        finally:
17
            await session.remove()
18
            reset_session_context(context=context)
19

20
    return _standalone_session
21