U ڲgI=@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?psycopgc s eZdZUdZdZded<ded<ded<d ed <eeeefd dd fd d Z e ddddddddddddddddddZ ddd d!Z d"d#d$d%d&d'd(Z e dd)d*d+d,d-Zd%dd.d/Zedd0dd1d2d3d4Zedd0dd5d6d7d8d4Zedddd9ddd:dd;d<d=d4Zedddd9ddd5d:dd>d?d@d4ZddddddAdddBd:ddCd?dDd4ZddddEdFdGd:dd1dHdIdJZd%ddKdLZd%ddMdNZdOdPdQd%dRdSdTZdUdPdQd%dRdVdWZeddXddYdZd[d\Zddd]d^dd_d`dadbZedcddddeZefdfd^dgdhdidjZdd%dkdldmZdd%dkdndoZdpd%dkdqdrZdpd%dkdsdtZ d:d%dkdudvZ!d:d%dkdwdxZ"d:d%dkdydzZ#d:d%dkd{d|Z$d}d%d~ddZ%d%dddZ&ddd%d~ddZ'ddd%d~ddZ(ddddZ)Z*S) Connectionz3 Wrapper for a connection to the database. r.ztype[Cursor[Row]]cursor_factoryztype[ServerCursor[Row]]server_cursor_factoryzRowFactory[Row] row_factoryzPipeline | None _pipeliner-)pgconnr2cs*t|||_t|_t|_t|_dSN) super__init__r2r,lockr&r0r(r1)selfr4r2 __class__6/tmp/pip-unpacked-wheel-b_ea6rx_/psycopg/connection.pyr7Ds  zConnection.__init__FN) autocommitprepare_thresholdcontextr2r0strboolz int | NonezAdaptContext | NonezRowFactory[Row] | Noneztype[Cursor[Row]] | Nonerr)conninfor@rArBr2r0kwargsreturnc Ks|j|f|}t|} d} t|} | D]} z*td | }|j|| d} tj| td} Wn\tj k r}ztimeoutintervalr z=connection attempt failed: host: %r port: %r, hostaddr %r: %shostportZhostaddr)r>)_get_connection_paramsr"r!rZ _connect_genr wait_conn_WAIT_INTERVALe _NO_TRACEBACKlenloggerdebuggetrCAssertionErrorwith_tracebackrDZ _autocommitr2r0radaptersZ _adaptersrA)clsrEr@rArBr2r0rFparamsrIrvattemptsattemptgenexZlast_exr<r<r=connectOs@     zConnection.connect)rGcCs|Sr5r<r9r<r<r= __enter__szConnection.__enter__ztype[BaseException] | NonezBaseException | NonezTracebackType | NoneNone)exc_typeexc_valexc_tbrGc Csn|jr dS|rNz |WqVtk rJ}ztd||W5d}~XYqVXn|t|ddsj|dS)Nz#error ignored in rollback on %s: %s_pool)closedrollback ExceptionrTwarningcommitgetattrclose)r9rerfrgZexc2r<r<r=__exit__s " zConnection.__exit__rr)rErFrGcKs t|f|S)z3Manipulate connection parameters before connecting.)r )rZrErFr<r<r=rNsz!Connection._get_connection_paramscCs|jr dSd|_|jdS)zClose the database connection.NT)ri_closedr4finishrbr<r<r=roszConnection.close)binaryz Cursor[Row])rsrGcCsdSr5r<)r9rsr<r<r=cursorszConnection.cursorzRowFactory[CursorRow]zCursor[CursorRow])rsr2rGcCsdSr5r<)r9rsr2r<r<r=rts)rs scrollablewithholdz bool | NonezServerCursor[Row])namersrurvrGcCsdSr5r<)r9rwrsrurvr<r<r=rtszServerCursor[CursorRow])rwrsr2rurvrGcCsdSr5r<)r9rwrsr2rurvr<r<r=rts )rsr2rurvzRowFactory[Any] | NonezCursor[Any] | ServerCursor[Any]cCsH||s|j}|r,|j|||||d}n|j||d}|rDt|_|S)zW Return a new `Cursor` to send commands and queries to the connection. )rwr2rurvr2)_check_connection_okr2r1r0BINARYformat)r9rwrsr2rurvcurr<r<r=rts )preparersrz Params | None)queryr[r}rsrGc CsXz$|}|rt|_|j|||dWStjk rR}z|dW5d}~XYnXdS)z8Execute a query and return a cursor to read its results.)r}N)rtrzr{executerQrRrX)r9r~r[r}rsr|r`r<r<r=rs zConnection.executec Cs$|j||W5QRXdS)z/Commit any pending transaction to the database.N)r8waitZ _commit_genrbr<r<r=rmszConnection.commitc Cs$|j||W5QRXdS)z2Roll back to the start of any pending transaction.N)r8rZ _rollback_genrbr<r<r=rjszConnection.rollbackg>@rHfloat)rIrGcCs8|s dStr,tj|j|dtdn|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()`. NrHrJ)Z_should_cancelr'Zhas_cancel_saferrOZ _cancel_genrPcancel)r9rIr<r<r= cancel_safes zConnection.cancel_safe@c CsDz|j|dWn.tk r>}ztd|W5d}~XYnXdS)NrHzquery cancellation failed: %s)rrkrTrl)r9rIr`r<r<r= _try_cancel!szConnection._try_cancelz str | NonezIterator[Transaction])savepoint_nameforce_rollbackrGc cslt|||}|jrR|0| | |VW5QRXW5QRXW5QRXn| |VW5QRXdS)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%r3pipeline)r9rrZtxr<r<r= transaction's &zConnection.transaction)rI stop_afterz float | NonezGenerator[Notify, None, None])rIrrGc cs|dk rt|}t|t}nd}t}d}|j|jj}z|jt|j|d}Wn.tj k r}z| dW5d}~XYnX|D]0} t | j || j || j} | V|d7}q|dk r||krq|r:tt|t}|dkr:qq:W5QRXdS)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. NrrJr g)rminrPr8r4 _encodingrr$rQrRrXr+ZrelnamedecodeextraZbe_pid) r9rIrdeadlinerKZ nreceivedencnsr`Zpgnnr<r<r=r$<s4    zConnection.notifieszIterator[Pipeline]ccs|j*||j}|dkr,t|}|_W5QRXz| |VW5QRXW5|jdkr|j||jksrtd|_W5QRXXdS)z)r>)N)NF)N)N)+__name__ __module__ __qualname____doc____annotations__rrrrr7 classmethodrarcrprNror rtrrmrjrrr rr$rrPrrrrrrrrrrrrrr __classcell__r<r<r:r=r/8s   "3   %2 r/)Qr __future__rloggingtimertypesrtypingrrrrr r contextlibr r>r rrQrabcrrrrrrrZ_tpcrZrowsrrrrZadaptrZ_enumsr_compatrrErr r!r"r3r# generatorsr$rr%rtr&Z _capabilitiesr'Z server_cursorr(Z_connection_baser)r*r+ threadingr,Zpq.abcr-rPZFormatZTEXTrzZTransactionStatusrrrKeyboardInterruptr getLoggerrTr/r<r<r<r=sH        $