o DDiB^@s UdZddlmZddlZddlZddlmZmZmZm Z m Z ddl m Z m Z ddlmZddlmZddlmZdd lmZd d lmZd d lmZmZmZd d lmZmZmZd dl m!Z!m"Z"d dl#m$Z$d dl%m&Z&d dl'm(Z(d dl)m*Z*d dl+m,Z,m-Z-m.Z.d dl/m0Z0d dl1m2Z2d dl3m4Z4d dl5m6Z6d dl7m8Z8erddl9m:Z:d dl;mej?j@Z@ej?jAZAejBjCZCejBjDZDejEjFZFejEjGZGejEjHZHejIjJZJejIjKZKe4LZMeNdZOGddde ZPdeP_QeejRgdfZSdeTd <eePgdfZUdeTd!<Gd"d#d#ee&ZVdS)$z psycopg connection objects ) annotationsN) TYPE_CHECKINGAnyGeneric NamedTuple TypeAlias) ReferenceTyperef)warn)partial)deque)Callable)errors) generatorspostgrespq)PQGen PQGenConnQueryNoTemplate)SQL Composable)Xid)Row) AdaptersMap)IsolationLevel) LiteralStringSelfTypeVar)connection_summary)PrepareManager) capabilities) BasePipeline)ConnectionInfo)BasePool)PGconnPGresult CursorRowpsycopgc@s.eZdZUdZded< ded< ded<dS)Notifyz8An asynchronous notification received from the database.strchannelpayloadintpidN)__name__ __module__ __qualname____doc____annotations__r4r4Q/var/www/Datamplify/venv/lib/python3.10/site-packages/psycopg/_connection_base.pyr)<s r)r NoticeHandler NotifyHandlerc@seZdZdZejZejZejZejZej Z ej Z ej Z ej Z ej Z ejZdddZefdd d Zdd dZedddZedddZedddZejdddZdddZdddZedddZejdd dZdd!d"Zdd#d$Zedd&d'Zejdd(d'Zdd)d*Zdd+d,Zedd-d.Z e jdd/d.Z dd0d1Z!dd2d3Z"dd5d6Z#edd8d9Z$edd;d<Z%edd>d?Z&ddAdBZ'ddCdDZ(ddEdFZ)ddJdKZ*ddNdOZ+ddPdQZ,e-ddVdWZ.ddYdZZ/dd[d\Z0e-dd_d`Z1eddbdcZ2e2jddddcZ2eddedfZ3e3jddgdfZ3e4 hddidjddmdnZ5e6fddtduZ7ddxdyZ8ddzd{Z9dd|d}Z:dddZ;dddZdddZ?dddZ@dddZAdddZBdS)BaseConnectionz Base class for different types of connections. Share common functionalities such as access to the wrapped PGconn, but allow different interfaces (sync/async). pgconnr%cCs||_d|_d|_g|_g|_d|_d|_t|_d|_ t |_ t |}t tj||_t tj||_|d|_||d|_d|_d|_d|_dS)NFr)r9 _autocommit _adapters_notice_handlers_notify_handlers_num_transactions_closedr _prepared_tpcr _notifies_backlogr r r8_notice_handlernotice_handler_notify_handlernotify_handler _pipeline_isolation_level _read_only _deferrable_begin_statement)selfr9wselfr4r4r5__init__cs* zBaseConnection.__init___BaseConnection__warnrreturnNonecCs@t|dsdS|jr dSt|drdS|t|dtdS)Nr9_poolz_ was deleted while still open. Please use 'with' or '.close()' to close the connection properly)hasattrclosedobject__repr__ResourceWarning)rMrPr4r4r5__del__s  zBaseConnection.__del__r*cCs>|jjd|jj}t|j}d|d|dt|ddS)N.< z at 0xx>) __class__r0r1rr9id)rMclsinfor4r4r5rWs zBaseConnection.__repr__boolcCs |jjtkS)z$`!True` if the connection is closed.)r9statusBADrMr4r4r5rUs zBaseConnection.closedcCs|jjtko |j S)z `!True` if the connection was interrupted. A broken connection is always `closed`, but wasn't closed in a clean way, such as using `close()` or a `!with` block. )r9rdrer@rfr4r4r5brokenszBaseConnection.brokencC|jS)z'The autocommit state of the connection.)r;rfr4r4r5 autocommitszBaseConnection.autocommitvaluecC||dSN)_set_autocommitrMrjr4r4r5ricCtrlNotImplementedErrorrnr4r4r5rmzBaseConnection._set_autocommit PQGen[None]ccs |dEdHt||_dS)Nri)_check_intrans_genrcr;rnr4r4r5_set_autocommit_gensz"BaseConnection._set_autocommit_genIsolationLevel | NonecCrh)zX The isolation level of the new transactions started on the connection. )rIrfr4r4r5isolation_levelzBaseConnection.isolation_levelcCrkrl)_set_isolation_levelrnr4r4r5rxrocCrprlrqrnr4r4r5rzrsz#BaseConnection._set_isolation_levelcc2|dEdH|durt|nd|_d|_dS)Nrxr:)rurrIrLrnr4r4r5_set_isolation_level_gen z'BaseConnection._set_isolation_level_gen bool | NonecCrh)zX The read-only state of the new transactions started on the connection. )rJrfr4r4r5 read_onlyryzBaseConnection.read_onlycCrkrl)_set_read_onlyrnr4r4r5rrocCrprlrqrnr4r4r5rrszBaseConnection._set_read_onlyccr{)Nrr:)rurcrJrLrnr4r4r5_set_read_only_genr}z!BaseConnection._set_read_only_gencCrh)zY The deferrable state of the new transactions started on the connection. )rKrfr4r4r5 deferrableryzBaseConnection.deferrablecCrkrl)_set_deferrablernr4r4r5rrocCrprlrqrnr4r4r5rrszBaseConnection._set_deferrableccr{)Nrr:)rurcrKrLrnr4r4r5_set_deferrable_genr}z"BaseConnection._set_deferrable_gen attributeccsr|jj}tkr|jr|jEdH|jj}|tkr7|jr(td|dtd|dt |j dS)Nz can't change z2 now: connection.transaction() context in progressz' now: connection in transaction status ) r9transaction_statusIDLErH _sync_genr?eProgrammingErrorrTransactionStatusname)rMrrdr4r4r5rus  z!BaseConnection._check_intrans_genr#cCs t|jS)z>A `ConnectionInfo` attribute to inspect connection properties.)r#r9rfr4r4r5rbs zBaseConnection.inforcCs|js ttj|_|jSrl)r<rradaptersrfr4r4r5rs zBaseConnection.adaptersBaseConnection[Row]cCs|Srlr4rfr4r4r5 connectionzBaseConnection.connectionr-cC|jjS)zReturn the file descriptor of the connection. This function allows to use the connection as file-like object in functions waiting for readiness, such as the ones defined in the `selectors` module. )r9socketrfr4r4r5filenoszBaseConnection.filenocCs"|r|j}|dSdS)z/Cancel the current operation on the connection.N)_should_cancelr9 get_cancelcancel)rMcr4r4r5r(s  zBaseConnection.cancelcCs(|jrdS|jr|jdrtddS)zhCheck whether the current command should actually be cancelled when invoking cancel*(). Frz=cancel() cannot be used with a prepared two-phase transactionT)rUrBrrrfr4r4r5r.szBaseConnection._should_canceltimeoutfloatPQGenConn[None]ccs,|j}|tj||dEdHdS)Nr)r9 cancel_connstartrr)rMrrr4r4r5 _cancel_gen=s zBaseConnection._cancel_gencallbackr6cC|j|dS)z Register a callable to be invoked when a notice message is received. :param callback: the callback to call upon message received. :type callback: Callable[[~psycopg.errors.Diagnostic], None] N)r=appendrMrr4r4r5add_notice_handlerBz!BaseConnection.add_notice_handlercCr)z Unregister a notice message callable previously registered. :param callback: the callback to remove. :type callback: Callable[[~psycopg.errors.Diagnostic], None] N)r=removerr4r4r5remove_notice_handlerKrz$BaseConnection.remove_notice_handlerrN"ReferenceType[BaseConnection[Row]]resr&c Csr|}|r|js dSt||jj}|jD]!}z||Wqty6}z td||WYd}~qd}~wwdS)Nz)error processing notice callback '%s': %s)r=r Diagnosticr9 _encoding Exceptionlogger exception)rNrrMdiagcbexr4r4r5rDTs   zBaseConnection._notice_handlerr7cCr)z Register a callable to be invoked whenever a notification is received. :param callback: the callback to call upon notification received. :type callback: Callable[[~psycopg.Notify], None] N)r>rrr4r4r5add_notify_handlercrz!BaseConnection.add_notify_handlercCr)z Unregister a notification callable previously registered. :param callback: the callback to remove. :type callback: Callable[[~psycopg.Notify], None] N)r>rrr4r4r5remove_notify_handlerlrz$BaseConnection.remove_notify_handlerpgn pq.PGnotifycCsr|}sdS|jj}t|j||j||j}|jr)|jD]}||q dS|j}dur7| |dSdSrl) r9rr)relnamedecodeextrabe_pidr>rCr)rNrrMencnrdr4r4r5rFus   zBaseConnection._notify_handler int | NonecCr)a& Number of times a query is executed before it is prepared. - If it is set to 0, every query is prepared the first time it is executed. - If it is set to `!None`, prepared statements are disabled on the connection. Default value: 5 rAprepare_thresholdrfr4r4r5rs z BaseConnection.prepare_thresholdcCs ||j_dSrlrrnr4r4r5rs cCs|jj}|tjkr |SdS)z Maximum number of prepared statements on the connection. `!None` means no max number of prepared statements. The default value is 100. N)rA prepared_maxsysmaxsize)rMrvr4r4r5rszBaseConnection.prepared_maxcCs|durtj}||j_dSrl)rrrArrnr4r4r5rs grconninfoPQGenConn[Self]ccs"tj||dEdH}||}|S)z?Generator to connect to the database and create a new instance.rN)rconnect)rarrr9connr4r4r5 _connect_genszBaseConnection._connect_gencommandr result_format pq.FormatPQGen[PGresult | None]ccs|t|tr||jj}n t|tr||}|jr9t |jj |d|d}|jj ||jj ddS|tkrD|j|n |jj |d|dt|jEdHd}|jtkr|jtkr|jtkrptj||jjdtdt|jjd||S)z Generator to send a command and receive the result to the backend. Only used to implement internal commands such as "commit", with eventual arguments bound client-side. The cursor can do more complex stuff. N)rencodingunexpected result z from command )_check_connection_ok isinstancer*encoder9rras_bytesrHr send_query_params command_queuer result_queueTEXT send_queryrexecuterd COMMAND_OK TUPLES_OK FATAL_ERRORrerror_from_resultInterfaceErrorr ExecStatusrr)rMrrcmdresultr4r4r5 _exec_commands:    zBaseConnection._exec_commandr bytes | Noneccs|dusts|durd|nd}||EdHdS||jr9t|jj|}|jj||jj ddS|j|t |jEdHd}|j t krk|j tkr]tj||jjdtdt|j jddS)ae Deallocate one, or all, prepared statement in the session. ``name == None`` stands for DEALLOCATE ALL. If possible, use protocol-level commands; otherwise use SQL statements. Note that PgBouncer doesn't support DEALLOCATE name, but it supports protocol-level Close from 1.21 and DEALLOCATE ALL from 1.22. Ns DEALLOCATE sDEALLOCATE ALLrrrz0 from sending closing prepared statement message)_HAS_SEND_CLOSErrrHr r9send_close_preparedrrrrrrdrrrrrrrrr)rMrstmtrrr4r4r5 _deallocates(    zBaseConnection._deallocatecCs:|jjtkrdS|jjtkrtdtd|jj)Nzthe connection is closedz7cannot execute operations: the connection is in status )r9rdOKrerOperationalErrorrrfr4r4r5rs   z#BaseConnection._check_connection_okccsN|jrdS|jjtkrdS||EdH|jr%|jEdHdSdS)z.Generator to start a transaction if necessary.N)r;r9rrr_get_tx_start_commandrHrrfr4r4r5 _start_querys zBaseConnection._start_querybytescCs|jr|jSdg}|jdur$t|j}|d||jdd|jdur3||jr0dnd|jdurB||jr?dndd ||_|jS) NsBEGINsISOLATION LEVEL_r\s READ ONLYs READ WRITEs DEFERRABLEsNOT DEFERRABLE ) rLrxrrrreplacerrrjoin)rMpartsvalr4r4r5r&s      z$BaseConnection._get_tx_start_commandccs`|jr td|jrtd|jjtkrdS|dEdH|jr.|j EdHdSdS)z-Generator implementing `Connection.commit()`.zExplicit commit() forbidden within a Transaction context. (Transaction will be automatically committed on successful exit from context.)z6commit() cannot be used during a two-phase transactionNsCOMMIT) r?rrrBr9rrrrHrrfr4r4r5 _commit_gen:s zBaseConnection._commit_genccs|jr td|jrtd|jr|jEdH|jjtkr$dS| dEdH|j |j |EdH|jrG|jEdHdSdS)z/Generator implementing `Connection.rollback()`.zExplicit rollback() forbidden within a Transaction context. (Either raise Rollback() or allow an exception to propagate out of the context.)z8rollback() cannot be used during a two-phase transactionNsROLLBACK) r?rrrBrHrr9rrrrAclear maintain_genrfr4r4r5 _rollback_genNs&  zBaseConnection._rollback_gen format_idgtridbqualrcCs|t|||S)aL Returns a `Xid` to pass to the `!tpc_*()` methods of this connection. The argument types and constraints are explained in :ref:`two-phase-commit`. The values passed to the method will be available on the returned object as the members `~Xid.format_id`, `~Xid.gtrid`, `~Xid.bqual`. ) _check_tpcr from_parts)rMrrrr4r4r5xidis zBaseConnection.xidr Xid | strccsx|t|tst|}|jjtkr#tdt |jjj |j r+td|df|_ ||EdHdS)Nz8can't start two-phase transaction: connection in status z3can't use two-phase transactions in autocommit modeF)rrr from_stringr9rrrrrrrr;rBrrrMrr4r4r5_tpc_begin_genvs     zBaseConnection._tpc_begin_genccsv|js td|jdrtd|jd}|df|_|tdt|EdH|jr9|jEdHdSdS)Nz='tpc_prepare()' must be called inside a two-phase transactionrzF'tpc_prepare()' cannot be used during a prepared two-phase transactionrTzPREPARE TRANSACTION {}) rBrrrrformatr*rHrrr4r4r5_tpc_prepare_gens   zBaseConnection._tpc_prepare_genactionrXid | str | Noneccsd|d}|dur|jst|d|jd}n|jr)t|dt|ts3t|}|jrQ|jdsQt|d|d}d|_|EdHdS|t d  t |t |EdHd|_dS) Ntpc_z()z? without xid must must be called inside a two-phase transactionrz= with xid must must be called outside a two-phase transactionrr_genz{} PREPARED {}) lowerrBrrrrrgetattrrrrr*)rMrrfnamemethr4r4r5_tpc_finish_gens,    zBaseConnection._tpc_finish_gencCsdS)z0Raise NotSupportedError if TPC is not supported.Nr4rfr4r4r5rrzBaseConnection._check_tpcN)r9r%)rPrrQrR)rQr*)rQrc)rjrcrQrR)rjrcrQrt)rQrw)rjrwrQrR)rjrwrQrt)rQr~)rjr~rQrR)rjr~rQrt)rr*rQrt)rQr#)rQr)rQr)rQr-)rQrR)rrrQr)rr6rQrR)rNrrr&rQrR)rr7rQrR)rNrrrrQrR)rQr)rjrrQrR)r)rr*rrrQr)rrrrrQr)rrrQrt)rQrt)rQr)rr-rr*rr*rQr)rrrQrt)rrrrrQrt)Cr/r0r1r2rWarningErrorr DatabaseError DataErrorrIntegrityError InternalErrorrNotSupportedErrorrOr rYrWpropertyrUrgrisetterrmrvrxrzr|rrrrrrrurbrrrrrrrr staticmethodrDrrrFrr classmethodrrrrrrrrrrrrr rr4r4r4r5r8Os )                                     . $     r8)Wr2 __future__rrloggingtypingrrrrrweakrefrr warningsr functoolsr collectionsr collections.abcr rrrrrrabcrrrsqlrrrBrrowsradaptr_enumsr_compatrrrpq.miscr _preparingr _capabilitiesr!_pipeline_baser"_connection_infor#psycopg_pool.baser$pq.abcr%r&r'FormatrBINARY ConnStatusrrerrrrrrINTRANShas_send_close_preparedr getLoggerrr)r0rr6r3r7r8r4r4r4r5sX