*Jf$dZddlZddlmZddlmZmZmZmZm Z m Z ddl m Z ddl m Z ddl mZdd lmZmZdd lmZdd lmZe rdd lmZdd lmZddlmZe jjZe jjZej e!Z"Gdde#Z$Gddej%Z&GddeeZ'Gdde'dZ(Gdde'dZ)dS)zC Transaction context managers returned by Connection.transaction() N) TracebackType)GenericIteratorOptionalTypeUnion TYPE_CHECKING)pq)sql)errors)ConnectionTypePQGen)Self)connection_summary)Any) Connection)AsyncConnectionc>eZdZdZdZ d dedfdZdefdZdS) Rollbacka Exit the current `Transaction` context immediately and rollback any changes made within this context. If a transaction context is specified in the constructor, rollback enclosing transactions contexts up to and including the one specified. psycopgN transaction) TransactionAsyncTransactionNc||_dSN)r)selfrs V/home/alex/cs2snipeproduction/venv/lib/python3.11/site-packages/psycopg/transaction.py__init__zRollback.__init__*s'returnc0|jjd|jdS)N()) __class__ __qualname__rrs r__repr__zRollback.__repr__0s!.-EE0@EEEEr r)__name__ __module__r&__doc__rrstrr(r rrrsuJGK''BC'''' F#FFFFFFr rceZdZdZdS)OutOfOrderTransactionNestingz)Out-of-order transaction nesting detectedN)r)r*r&r+r-r rr/r/4s3333r r/c eZdZ ddedeedefdZedeefdZ defd Z de dfd Z d ee ed eed eede efdZde dfdZd eede efdZdeefdZdeefdZdeefdZddZdedeefdZdS)BaseTransactionNF connectionsavepoint_nameforce_rollbackc||_|jj|_|pd|_||_dx|_|_d|_d|_dS)NF)_connpgconn_savepoint_namer4_entered_exited_outer_transaction _stack_index)rr2r3r4s rrzBaseTransaction.__init__9sP   j' -3,',,  "'r r!c|jS)zV The name of the savepoint; `!None` if handling the main transaction. )r:r's rr3zBaseTransaction.savepoint_nameGs ##r c |jjd|jj}t|j}|jsd}n |jsd}nd}|jr |jdnd}d|d|d|d |d t|d d S) N.inactiveactive terminated r6) r%r*r&rr9r;r<r3id)rclsinfostatussps rr(zBaseTransaction.__repr__Ps*JJT^-HJJ!$+..} "FF "FF!F,0,? G# ( ( ( (RA3AAAAfAAAABtHHAAAAAr c#K|jrtdd|_||D]"}|j|Ed{V#dS)Nz(transaction blocks can be used only onceT)r; TypeError_push_savepoint_get_enter_commandsr8 _exec_command)rcommands r _enter_genzBaseTransaction._enter_gen]s = HFGG G  //11 9 9Gz//88 8 8 8 8 8 8 8 8 9 9r exc_typeexc_valexc_tbc#K|s#|js|Ed{VdS ||Ed{VS#t$rt$r'}t d||Yd}~dSd}~wwxYw)NFz#error ignored in rollback of %s: %s)r4 _commit_gen _rollback_genr/ Exceptionloggerwarning)rrUrVrWexc2s r _exit_genzBaseTransaction._exit_genfs  t2 '')) ) ) ) ) ) ) )5 #'#5#5g#>#>>>>>>>?/       DdDQQQuuuuu sAA?A::A?c#K|d}d|_|r||D]"}|j|Ed{V#dS)NcommitT)_pop_savepointr<_get_commit_commandsr8rR)rexrSs rrYzBaseTransaction._commit_gens}   * *  H0022 9 9Gz//88 8 8 8 8 8 8 8 8 9 9r c#xKt|tr$t|jdd|d}d|_|r||D]"}|j|Ed{V#t|tr|j r |j |urdSdS)Nz: Explicit rollback from: T)exc_inforollbackF) isinstancerr\debugr8rbr<_get_rollback_commandsrRr)rrVrdrSs rrZzBaseTransaction._rollback_gens gx ( ( S LLDJBBBTL R R R   , ,  H2244 9 9Gz//88 8 8 8 8 8 8 8 8 gx ( ( & '*=*E*Etur c#K|jr|jV|jrZt jdt j|j|jVdSdS)Nz SAVEPOINT {}) r=r8_get_tx_start_commandr:r SQLformat Identifieras_bytesr's rrQz#BaseTransaction._get_enter_commandss  " 5*2244 4 4 4   ''t';<<==$*%%       r c# K|jr_|jsXtjdtj|j|jV|jr|jjrJdVdSdS)Nz RELEASE {}sCOMMIT) r:r=r rmrnrorpr8_num_transactionsr's rrcz$BaseTransaction._get_commit_commandss   (?  %%t';<<==$*%%     " z3 3 3 3OOOOO  r c#DK|jr|jstjdtj|j|jVtjdtj|j|jV|jr|jjrJdV|jj r&|jj Ed{VdSdS)NzROLLBACK TO {n})nz RELEASE {n}sROLLBACK) r:r=r rmrnrorpr8rr _preparedclearget_maintenance_commandsr's rrjz&BaseTransaction._get_rollback_commandss-   (? )**#.)=>>??$*%%     &&#.)=>>??$*%%     " z3 3 3 3    :  % % ' ' Gz+DDFF F F F F F F F F F G Gr c|jjtk|_|jr|jjrJn|jsd|jjdz|_|jj|_|jxjdz c_dS)z Push the transaction on the connection transactions stack. Also set the internal state of the object and verify consistency. _pg3_r N)r9transaction_statusIDLEr=r8rrr:r>r's rrPzBaseTransaction._push_savepoints #'+"@D"H  " Rz3 3 3 3 3' R'Qtz/Ka/O'Q'Q$ J8 $$)$$$$r actionc|jxjdzc_|jj|jkrdStd|d|S)zy Pop the transaction from the connection transactions stack. Also verify the state consistency. r Nz transaction z at the wrong nesting level: )r8rrr>r/)rr|s rrbzBaseTransaction._pop_savepointsW $$)$$ : '4+< < <4+ F6 F F F F   r NF)r!N)r)r*r&rrr,boolrpropertyr3r(rrTr BaseExceptionrr_rYrZrbytesrQrcrjrPr[rbr-r rr1r18s)-$  " !      $ $$$X$ B# B B B B9E$K99994 ./-('  t 29U4[9999Xm%<t$ Xe_     huo    GGGGG*****&  S  Xi-@       r r1ceZdZdZdZed dZdefdZdee e dee d ee de fd Z d S) rzO Returned by `Connection.transaction()` to handle a transaction block. rr!Connection[Any]c|jS)z&The connection the object is managing.r8r's rr2zTransaction.connections zr c|jj5|j|dddn #1swxYwY|Srr8lockwaitrTr's r __enter__zTransaction.__enter__s Z_ / / JOODOO-- . . . / / / / / / / / / / / / / / / s-AA  A rUrVrWc|jjtkrU|jj5|j||||cdddS#1swxYwYdSdSr~r9rLOKr8rrr_rrUrVrWs r__exit__zTransaction.__exit__s ;  # # R Rzt~~h'P'PQQ R R R R R R R R R R R R R R R R R R5s/AA"%A"N)r!r)r)r*r&r+rr2rrrrrrrrr-r rrrsJ X4 4 ./ -( '        r rrceZdZdZdZed dZdefdZdee e dee d ee de fd Z d S) rzT Returned by `AsyncConnection.transaction()` to handle a transaction block. rr!AsyncConnection[Any]c|jSrrr's rr2zAsyncTransaction.connection s zr cK|jj4d{V|j|d{Vdddd{Vn#1d{VswxYwY|Srrr's r __aenter__zAsyncTransaction.__aenter__s:? 5 5 5 5 5 5 5 5*//$//"3"344 4 4 4 4 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 s3A A$'A$rUrVrWcK|jjtkrm|jj4d{V|j||||d{Vcdddd{VS#1d{VswxYwYdSdSr~rrs r __aexit__zAsyncTransaction.__aexit__s@ ;  # #z X X X X X X X X!Z__T^^Hgv-V-VWWWWWWWW X X X X X X X X X X X X X X X X X X X X X X X X X X X X X X5s5A22 A<?A<N)r!r)r)r*r&r+rr2rrrrrrrrr-r rrrsJ X$ 4 ./ -( '        r rr)*r+loggingtypesrtypingrrrrrr r6r r r eabcrr_compatrpq.miscrrr2rconnection_asyncrTransactionStatusr{ ConnStatusr getLoggerr)r\r[rProgrammingErrorr/r1rrr-r rrs4 JJJJJJJJJJJJJJJJ&&&&&&&&''''''2&&&&&&111111  ]  8 $ $FFFFFyFFF*444441#5444m m m m m gn-m m m `/"34<'=>r