o ޭh5@sddlmZddlmZddlmZddlmZmZm Z ddl m Z ddl m Z er2ddlmZmZedZed efZed ed ZdddZddddZddddZdS)) annotations)Callable)wraps) TYPE_CHECKINGTypeVarcast) connections) ParamsError)BaseDBAsyncClientTransactionContextT.F)boundconnection_name str | Nonereturnr cCsT|r t|}|Sttjdkr tttj}t|}|Stdttj)NzPYou are running with multiple databases, so you should specify connection_name: ) rgetlen db_confignextiterkeysr listr connectionrZ/var/www/html/stock_analysis/be/venv/lib/python3.10/site-packages/tortoise/transactions.py_get_connections   rNr cCst|}|S)aG Transaction context manager. You can run your code inside ``async with in_transaction():`` statement to run it into one transaction. If error occurs transaction will rollback. :param connection_name: name of connection to run with, optional if you have only one db connection )r_in_transactionrrrrin_transaction s r Callable[[F], F]csdfdd }|S)a( Transaction decorator. You can wrap your function with this decorator to run it into one transaction. If error occurs transaction will rollback. :param connection_name: name of connection to run with, optional if you have only one db connection funcr rcs"tdfdd }tt|S)Nrr c sVt4IdH|i|IdHWdIdHS1IdHs$wYdSN)r )argskwargs)rr"rrwrapped:s0z(atomic..wrapper..wrapped)rr )rrr )r"r&r)r"rwrapper9s zatomic..wrapperN)r"r rr r)rr(rr'ratomic.s r))rrrr r#)rrrr )rrrr!) __future__rcollections.abcr functoolsrtypingrrrtortoisertortoise.exceptionsr tortoise.backends.base.clientr r r FuncTyper rr r)rrrrs