jL'dZddlmZddlZddlmZddlmZddlm Z m Z m Z ddl m Z dd lmZdd lmZmZdd lmZmZdd lmZdd lmZe r ddlmZddlmZejjZej j!Z!ej"e#Z$Gdde%Z&Gddej'Z(Gdde eZ)Gdde)dZ*Gdde)dZ+dS)zC Transaction context managers returned by Connection.transaction() ) annotationsN)Enum) TracebackType) TYPE_CHECKINGAnyGeneric)Iterator)errors)pqsql)ConnectionTypePQGen)Self)connection_summary) Connection)AsyncConnectionc(eZdZdZdZd d dZd d ZdS) 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%Transaction | AsyncTransaction | Nonec||_dSN)r)selfrs ^/DATA/AppData/hermes/projects/honcho/.venv/lib/python3.11/site-packages/psycopg/transaction.py__init__zRollback.__init__+s&returnstrc0|jjd|jdS)N()) __class__ __qualname__rrs r__repr__zRollback.__repr__.s!.-EE0@EEEErr)rrrr )__name__ __module__r%__doc__rr'rrrr sXJ'''''FFFFFFrrceZdZdZdS)OutOfOrderTransactionNestingz)Out-of-order transaction nesting detectedN)r)r*r%r+r,rrr.r.2s3333rr.ceZdZGddeeZ d%d&d Zed'd Zd(dZ d)dZ d*dZ d)dZ d+dZ d,dZd,dZd,dZd-d!Zd.d$ZdS)/BaseTransactionc*eZdZdZdZdZdZdZdZdZ dS) BaseTransaction.Statusz The current status of a transaction. Exposed as the `~psycopg.Transaction.status` attribute of `~psycopg.Transaction` and `~psycopg.AsyncTransaction` objects. not_startedactive committedfailedrolled_back_explicitlyrolled_back_with_errorN) r)r*r%r+ NOT_STARTEDACTIVE COMMITTEDFAILEDROLLED_BACK_EXPLICITLYROLLED_BACK_WITH_ERRORr,rrStatusr27s<  $  !9!9rr?NF connectionrsavepoint_name str | Noneforce_rollbackboolc||_|jj|_|pd|_||_|jj|_d|_d|_dS)NF) _connpgconn_savepoint_namerCr?r9status_outer_transaction _stack_index)rr@rArCs rrzBaseTransaction.__init__FsN   j' -3,k- "'rrc|jS)zV The name of the savepoint; `!None` if handling the main transaction. )rJr&s rrAzBaseTransaction.savepoint_nameTs ##rr c |jjd|jj}t|j}|jr |jdnd}d|d|d|jjd|dt|dd S) N. rF) r$r*r%rrIrArKvalueid)rclsinfosps rr'zBaseTransaction.__repr__]s*JJT^-HJJ!$+..,0,? G# ( ( ( (RL3LLLLdk/LL4LLr$xxLLLLLr PQGen[None]c# K|j|jjkrtd|jj|_||D]"}|j|Ed{V#dS)Nz(transaction blocks can be used only once) rKr?r9 TypeErrorr:_push_savepoint_get_enter_commandsrH _exec_command)rcommands r _enter_genzBaseTransaction._enter_gencs ;$+1 1 1FGG Gk(  //11 9 9Gz//88 8 8 8 8 8 8 8 8 9 9rexc_typetype[BaseException] | Noneexc_valBaseException | Noneexc_tbTracebackType | None PQGen[bool]c#K|s#|js|Ed{VdS ||Ed{VS#t$rt$r'}t d||Yd}~dSd}~wwxYw)NFz#error ignored in rollback of %s: %s)rC _commit_gen _rollback_genr. Exceptionloggerwarning)rrbrdrfexc2s r _exit_genzBaseTransaction._exit_genls  t2 '')) ) ) ) ) ) ) )5 #'#5#5g#>#>>>>>>>?/       DdDQQQuuuuu sAA?A::A?c#K|d}|jj|_|r||D]"}|j|Ed{V#dS)Ncommit)_pop_savepointr?r;rK_get_commit_commandsrHr_)rexr`s rrjzBaseTransaction._commit_gens   * *k+  H0022 9 9Gz//88 8 8 8 8 8 8 8 8 9 9rc#xKt|tr$t|jdd|d}t|ts|jr|jj|_ n|jj |_ |r|| D]"}|j |Ed{V#|jj |jj |jEd{Vt|tr|jr |j|urdSdS)Nz: Explicit rollback from: T)exc_inforollbackF) isinstancerrmdebugrHrsrCr?r=rKr>_get_rollback_commandsr_ _preparedclear maintain_genr)rrdrur`s rrkzBaseTransaction._rollback_gensP gx ( ( S LLDJBBBTL R R R   , , gx ( ( =D,? =+>??$*%%     &&#.)=>>??$*%%     " z3 3 33       rNonec|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)rItransaction_statusIDLErLrHrrJrMr&s rr]zBaseTransaction._push_savepoints #'+"@D"H  " Rz3 3 33 3' R'Qtz/Ka/O'Q'Q$ J8 $$)$$$$ractionException | Nonec|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: )rHrrMr.)rrs rrszBaseTransaction._pop_savepointsW $$)$$ : '4+< < <4+ F6 F F F F   rNF)r@rrArBrCrD)rrBr()rrZ)rbrcrdrerfrgrrh)rdrerrh)rr)rr)rr rr)r)r*r%r rr?rpropertyrAr'rarprjrkr^rtr{r]rsr,rrr0r06sM : : : : :d : : :$&*$     $$$X$MMMM 9999299996        "****&       rr0c>eZdZdZdZeddZddZddZdS) TransactionzO Returned by `Connection.transaction()` to handle a transaction block. rrConnection[Any]c|jS)zLThe connection the object is managing. :type: `Connection` rHr&s rr@zTransaction.connection zrrc|jj5|j|dddn #1swxYwY|SrrHlockwaitrar&s r __enter__zTransaction.__enter__s Z_ / / JOODOO-- . . . / / / / / / / / / / / / / / / s-AA  A rbrcrdrerfrgrDc|jjtkrU|jj5|j||||cdddS#1swxYwYdS|jj|_dSr rIrKOKrHrrrpr?r<rrbrdrfs 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 R+,DK5s/AA"%A"N)rrrrrbrcrdrerfrgrrD)r)r*r%r+rr@rrr,rrrrkJ X      rrrc>eZdZdZdZeddZddZddZdS)AsyncTransactionzT Returned by `AsyncConnection.transaction()` to handle a transaction block. rrAsyncConnection[Any]c|jS)zQThe connection the object is managing. :type: `AsyncConnection` rr&s rr@zAsyncTransaction.connectionrrrcK|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$rbrcrdrerfrgrDc0K|jjtkrm|jj4d{V|j||||d{Vcdddd{VS#1d{VswxYwYdS|jj|_dSrrrs r __aexit__zAsyncTransaction.__aexit__)sL ;  # #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 X+,DK5s5A22 A<?A<N)rrrr)r)r*r%r+rr@rrr,rrrrrrrr),r+ __future__rloggingenumrtypesrtypingrrrcollections.abcr rFr er r abcrr_compatrpq.miscrr@rconnection_asyncrTransactionStatusr ConnStatusr getLoggerr)rmrlrProgrammingErrorr.r0rrr,rrrs; #"""""..........$$$$$$&&&&&&&&''''''2&&&&&&111111  ]  8 $ $FFFFFyFFF$444441#5444z z z z z gn-z z z z/"34D'=>r