o "4gI=@sdZddlmZddlZddlmZddlmZddlm Z m Z m Z m Z m Z mZddlmZdd lmZdd lmZdd lmZdd lmZmZmZmZmZmZmZdd lmZddl m!Z!m"Z"m#Z#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,m-Z-ddl+m.Z.m/Z/ddl0m1Z1ddl2m3Z3ddl4m5Z5ddl6m7Z7ddl8m9Z9ddl:m;Z;ddlZ>m?Z?ddl@mAZAerddlBmCZCdZDejEjFZFejEjGZGejHjIZIejHjJZJejHjKZKeLZMeNdZOGdd d e=e!ZPdS)!z* Psycopg connection object (sync version) ) annotationsN) monotonic) TracebackType)Any GeneratorIteratorcastoverload TYPE_CHECKING)contextmanager)pq)errors)waiting) AdaptContextConnDict ConnParamParamsPQGenQueryRV)Xid)Row RowFactory tuple_rowargs_row) AdaptersMap)IsolationLevel)Self) make_conninfoconninfo_to_dict)conninfo_attemptstimeout_from_conninfo)Pipeline)notifies) Transaction)Cursor) capabilities) ServerCursor)BaseConnection CursorRowNotify)Lock)PGconng?psycopgcs$eZdZUdZdZded<ded<ded<d ed <eeeefdfd d Z e ddddddddd"d#Z dd$d%Z dd-d.Z e dd1d2Zdd3d4Zedd5dd8d9Zedd5ddconnectOsD     zConnection.connectcCs|Sr6r=r:r=r=r> __enter__zConnection.__enter__exc_typetype[BaseException] | Noneexc_valBaseException | Noneexc_tbTracebackType | NoneNonec Csr|jrdS|r'z|Wnty&}z td||WYd}~n d}~ww|t|dds7|dSdS)Nz#error ignored in rollback on %s: %s_pool)closedrollback Exceptionr\warningcommitgetattrclose)r:rorqrsexc2r=r=r>__exit__s   zConnection.__exit__rrcKst|fi|S)z3Manipulate connection parameters before connecting.)r )rcrDrKr=r=r>rUsz!Connection._get_connection_paramscCs|jrdSd|_|jdS)zClose the database connection.NT)rw_closedr5finishrlr=r=r>r}szConnection.close)binaryr Cursor[Row]cCdSr6r=)r:rr=r=r>cursorrnzConnection.cursorRowFactory[CursorRow]Cursor[CursorRow]cCrr6r=)r:rr3r=r=r>rs)r scrollablewithholdnamer bool | NonerServerCursor[Row]cCrr6r=)r:rrrrr=r=r>rsServerCursor[CursorRow]cCrr6r=)r:rrr3rrr=r=r>rs )rr3rrRowFactory[Any] | NoneCursor[Any] | ServerCursor[Any]cCsH||s |j}|r|j|||||d}n|j||d}|r"t|_|S)zW Return a new `Cursor` to send commands and queries to the connection. )rr3rrr3)_check_connection_okr3r1r0BINARYformat)r:rrr3rrcurr=r=r>rs )preparerqueryrrd Params | Nonerc CsJz|}|r t|_|j|||dWStjy$}z|dd}~ww)z8Execute a query and return a cursor to read its results.)rN)rrrexecuterYrZr_)r:rrdrrrrir=r=r>rs  zConnection.executecC:|j||WddS1swYdS)z/Commit any pending transaction to the database.N)r9wait _commit_genrlr=r=r>r{"zConnection.commitcCr)z2Roll back to the start of any pending transaction.N)r9r _rollback_genrlr=r=r>rxrzConnection.rollbackg>@rMrNfloatcCs:|sdStrtj|j|dtddS|dS)aCancel the current operation on the connection. :param timeout: raise a `~errors.CancellationTimeout` if the cancellation request does not succeed within `timeout` seconds. Note that a successful cancel attempt on the client is not a guarantee that the server will successfully manage to cancel the operation. This is a non-blocking version of `~Connection.cancel()` which leverages a more secure and improved cancellation feature of the libpq, which is only available from version 17. If the underlying libpq is older than version 17, the method will fall back to using the same implementation of `!cancel()`. NrMrO)_should_cancelr'has_cancel_saferrW _cancel_genrXcancel)r:rNr=r=r> cancel_safes  zConnection.cancel_safe@c CsFz |j|dWdSty"}z td|WYd}~dSd}~ww)NrMzquery cancellation failed: %s)rryr\rz)r:rNrir=r=r> _try_cancel!s zConnection._try_cancelsavepoint_name str | Noneforce_rollbackIterator[Transaction]c cst|||}|jrR|:|| |VWdn1s$wYWdn1s3wYWddSWddS1sKwYdS| |VWddS1scwYdS)a Start a context block with a new transaction or nested transaction. :param savepoint_name: Name of the savepoint used to manage a nested transaction. If `!None`, one will be chosen automatically. :param force_rollback: Roll back the transaction at the end of the block even if there were no error (e.g. to try a no-op process). :rtype: Transaction N)r%r4pipeline)r:rrtxr=r=r> transaction's P"zConnection.transaction)rN stop_after float | NonerGenerator[Notify, None, None]c cs|durt|}t|t}nd}t}d}|ji|jj} z |jt|j|d}Wntj y=}z| dd}~ww|D]} t | j || j || j} | V|d7}q@|durb||krbn|rqtt|t}|dkrqn qWddSWddS1swYdS)a Yield `Notify` objects as soon as they are received from the database. :param timeout: maximum amount of time to wait for notifications. `!None` means no timeout. :param stop_after: stop after receiving this number of notifications. You might actually receive more than this number if more than one notifications arrives in the same packet. NrTrOr g)rminrXr9r5 _encodingrr$rYrZr_r+relnamedecodeextrabe_pid) r:rNrdeadlinerP nreceivedencnsripgnnr=r=r>r$<sB    "zConnection.notifiesIterator[Pipeline]ccs|j||j}|durt|}|_Wdn1s!wYz=| |VWdn1s7wYW|jdkrb|j||jusMJd|_WddS1s[wYdSdS|jdkr|j||justJd|_Wdw1swYw)zrns. "zConnection.pipelinerh PQGen[RV]rPrc Csnz tj||jj|dWSty6|jjtkr5|jddz tj||jj|dWtj y4Yww)z Consume a generator operating on the connection. The function must be used on generators that don't change connection fd (i.e. not on connect and reset). rOrrM) rrr5socket _INTERRUPTEDtransaction_statusACTIVErrY QueryCanceled)r:rhrPr=r=r>rs   zConnection.waitvaluecC||dSr6)set_autocommitr:rr=r=r>_set_autocommitzConnection._set_autocommitcC<|j|||WddS1swYdS)z6Method version of the `~Connection.autocommit` setter.N)r9r_set_autocommit_genrr=r=r>r"zConnection.set_autocommitIsolationLevel | NonecCrr6)set_isolation_levelrr=r=r>_set_isolation_levelrzConnection._set_isolation_levelcCr)z;Method version of the `~Connection.isolation_level` setter.N)r9r_set_isolation_level_genrr=r=r>rrzConnection.set_isolation_levelcCrr6) set_read_onlyrr=r=r>_set_read_onlyrzConnection._set_read_onlycCr)z5Method version of the `~Connection.read_only` setter.N)r9r_set_read_only_genrr=r=r>rrzConnection.set_read_onlycCrr6)set_deferrablerr=r=r>_set_deferrablerzConnection._set_deferrablecCr)z6Method version of the `~Connection.deferrable` setter.N)r9r_set_deferrable_genrr=r=r>rrzConnection.set_deferrablexid Xid | strcCr)zO Begin a TPC transaction with the given transaction ID `!xid`. N)r9r_tpc_begin_genr:rr=r=r> tpc_begins"zConnection.tpc_beginc Csjz|j||WdWdS1swYWdStjy4}ztt|dd}~ww)zV Perform the first phase of a transaction started with `tpc_begin()`. N)r9r_tpc_prepare_genrYObjectNotInPrerequisiteStateNotSupportedErrorrE)r:rir=r=r> tpc_prepares&zConnection.tpc_prepareXid | str | NonecC>|j||d|WddS1swYdS)z: Commit a prepared two-phase transaction. COMMITNr9r_tpc_finish_genrr=r=r> tpc_commit"zConnection.tpc_commitcCr)z= Roll back a prepared two-phase transaction. ROLLBACKNrrr=r=r> tpc_rollbackrzConnection.tpc_rollback list[Xid]cCsx||jj}|jttjd}|t| }Wdn1s'wY|t kr:|jjt kr:| |S)Nr) _check_tpcinforrrr _from_recordr_get_recover_queryfetchallIDLEINTRANSrx)r:statusrresr=r=r> tpc_recovers zConnection.tpc_recover)r5r-r3r2rT)rDrErArFrBrGrCrHr3rIr0rJrKrrLr)rLr)rorprqrrrsrtrLru)rDrErKrrLr)rLru)rrFrLr)rrFr3rrLr) rrErrFrrrrFrLr) rrErrFr3rrrrrFrLr) rrErrFr3rrrrrFrLrr6) rrrdrrrrrFrLr)rNrrLru)NF)rrrrFrLr)rNrrrGrLr)rLr)rhrrPrrLr)rrFrLru)rrrLru)rrrLru)rrrLru)rrrLru)rLr)+__name__ __module__ __qualname____doc____annotations__rrrrr8 classmethodrkrmrrUr}r rrr{rxrrr rr$rrXrrrrrrrrrrrrrr __classcell__r=r=r;r>r/8s    3      %  2            r/)Qr __future__rloggingtimertypesrtypingrrrrr r contextlibr r?r rrYrabcrrrrrrr_tpcrrowsrrrradaptr_enumsr_compatrrDrr r!r"r4r# generatorsr$rr%rr& _capabilitiesr' server_cursorr(_connection_baser)r*r+ threadingr,pq.abcr-rXFormatTEXTrTransactionStatusrrrKeyboardInterruptr getLoggerr\r/r=r=r=r>sJ        $