o "4g\@sUdZddlmZddlZddlZddlmZmZmZm Z ddl m Z m Z ddl mZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZmZmZddlmZmZddlmZddl m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'm(Z(m)Z)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4e rddl5m6Z6m7Z7ddl8m9Z9e*dZ:ej;jj?Z?ej>j@Z@ejAjBZBejAjCZCejAjDZDejEjFZFejEjGZGe2HZIeJdZKGdddeZLdeL_MeejNgdfZOdePd <eeLgdfZQdePd!<Gd"d#d#ee!ZRdS)$z psycopg connection objects ) annotationsN)CallableGeneric NamedTuple TYPE_CHECKING)ref ReferenceType)warn)partial)pq)errors)postgres) generators)PQGen PQGenConnQuery) ComposableSQL)Xid)Row) AdaptersMap)IsolationLevel) LiteralStringSelf TypeAliasTypeVar)connection_summary) BasePipeline)PrepareManager) capabilities)ConnectionInfo)PGconnPGresult)BasePool CursorRowpsycopgc@s.eZdZUdZded< ded< ded<dS)Notifyz8An asynchronous notification received from the database.strchannelpayloadintpidN)__name__ __module__ __qualname____doc____annotations__r2r2a/var/www/html/authentication-server/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 ejZdddZddd Zdd d ZedddZedddZedddZejdddZdddZdddZedddZejdddZddd Zdd!d"Zedd$d%Zejdd&d%Zdd'd(Zdd)d*Zedd+d,Zejdd-d,Zdd.d/Z dd0d1Z!dd3d4Z"edd6d7Z#edd9d:Z$edddddZ?dddZ@dddZAdS)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 j ||_t t j||_|d|_|d|_d|_d|_d|_dS)NFr)r7 _autocommit _adapters_notice_handlers_notify_handlers_num_transactions_closedr _prepared_tpcrr r6_notice_handlernotice_handler_notify_handlernotify_handler _pipeline_isolation_level _read_only _deferrable_begin_statement)selfr7wselfr2r2r3__init__bs& zBaseConnection.__init__returnNonecCs<t|dsdS|jr dSt|drdStd|dtdS)Nr7_poolz connection zV was deleted while still open. Please use 'with' or '.close()' to close the connection)hasattrclosedr ResourceWarningrJr2r2r3__del__s   zBaseConnection.__del__r(cCs>|jjd|jj}t|j}d|d|dt|ddS)N.< z at 0xx>) __class__r.r/rr7id)rJclsinfor2r2r3__repr__s zBaseConnection.__repr__boolcCs |jjtkS)z$`!True` if the connection is closed.)r7statusBADrSr2r2r3rQs 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. )r7r`rar>rSr2r2r3brokenszBaseConnection.brokencC|jS)z'The autocommit state of the connection.)r9rSr2r2r3 autocommitszBaseConnection.autocommitvaluecC||dSN)_set_autocommitrJrer2r2r3rdcCtrgNotImplementedErrorrir2r2r3rhzBaseConnection._set_autocommit PQGen[None]ccs |dEdHt||_dS)Nrd)_check_intrans_genr_r9rir2r2r3_set_autocommit_gensz"BaseConnection._set_autocommit_genIsolationLevel | NonecCrc)zX The isolation level of the new transactions started on the connection. )rFrSr2r2r3isolation_levelzBaseConnection.isolation_levelcCrfrg)_set_isolation_levelrir2r2r3rsrjcCrkrgrlrir2r2r3rurnz#BaseConnection._set_isolation_levelcc2|dEdH|durt|nd|_d|_dS)Nrsr8)rprrFrIrir2r2r3_set_isolation_level_gen z'BaseConnection._set_isolation_level_gen bool | NonecCrc)zX The read-only state of the new transactions started on the connection. )rGrSr2r2r3 read_onlyrtzBaseConnection.read_onlycCrfrg)_set_read_onlyrir2r2r3rzrjcCrkrgrlrir2r2r3r{rnzBaseConnection._set_read_onlyccrv)Nrzr8)rpr_rGrIrir2r2r3_set_read_only_genrxz!BaseConnection._set_read_only_gencCrc)zY The deferrable state of the new transactions started on the connection. )rHrSr2r2r3 deferrablertzBaseConnection.deferrablecCrfrg)_set_deferrablerir2r2r3r}rjcCrkrgrlrir2r2r3r~rnzBaseConnection._set_deferrableccrv)Nr}r8)rpr_rHrIrir2r2r3_set_deferrable_genrxz"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 ) r7transaction_statusIDLErE _sync_genr=eProgrammingErrorr TransactionStatusname)rJrr`r2r2r3rps   z!BaseConnection._check_intrans_genr!cCs t|jS)z>A `ConnectionInfo` attribute to inspect connection properties.)r!r7rSr2r2r3r]s zBaseConnection.inforcCs|js ttj|_|jSrg)r:rradaptersrSr2r2r3r s zBaseConnection.adaptersBaseConnection[Row]cCs|Srgr2rSr2r2r3 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. )r7socketrSr2r2r3filenoszBaseConnection.filenocCs"|r|j}|dSdS)z/Cancel the current operation on the connection.N)_should_cancelr7 get_cancelcancel)rJcr2r2r3r"s  zBaseConnection.cancelcCs(|jrdS|jr|jdrtddS)zhCheck whether the current command should actually be cancelled when invoking cancel*(). Fr z=cancel() cannot be used with a prepared two-phase transactionT)rQr@rrrSr2r2r3r(szBaseConnection._should_canceltimeoutfloatPQGenConn[None]ccs,|j}|tj||dEdHdS)Nr)r7 cancel_connstartrr)rJrrr2r2r3 _cancel_gen7s zBaseConnection._cancel_gencallbackr4cC|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;appendrJrr2r2r3add_notice_handler<z!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;removerr2r2r3remove_notice_handlerErz$BaseConnection.remove_notice_handlerrK"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 Diagnosticr7 _encoding Exceptionlogger exception)rKrrJdiagcbexr2r2r3rANs   zBaseConnection._notice_handlerr5cCr)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<rrr2r2r3add_notify_handler]rz!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<rrr2r2r3remove_notify_handlerfrz$BaseConnection.remove_notify_handlerpgn pq.PGnotifycCsR|}|r|js dS|jj}t|j||j||j}|jD]}||q dSrg)r<r7rr'relnamedecodeextrabe_pid)rKrrJencnrr2r2r3rCos   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 r?prepare_thresholdrSr2r2r3r|s z BaseConnection.prepare_thresholdcCs ||j_dSrgrrir2r2r3rs 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)r? prepared_maxsysmaxsize)rJrvr2r2r3rszBaseConnection.prepared_maxcCs|durtj}||j_dSrg)rrr?rrir2r2r3rs grconninfoPQGenConn[Self]ccs"tj||dEdH}||}|S)z?Generator to connect to the database and create a new instance.rN)rconnect)r\rrr7connr2r2r3 _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(encoder7rras_bytesrEr send_query_params command_queuer result_queueTEXT send_queryrexecuter` COMMAND_OK TUPLES_OK FATAL_ERRORrerror_from_resultInterfaceErrorr ExecStatusrr)rJrrcmdresultr2r2r3 _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_CLOSErrrEr r7send_close_preparedrrrrrr`rrrrrrr rr)rJrstmtrrr2r2r3 _deallocates.    zBaseConnection._deallocatecCs:|jjtkrdS|jjtkrtdtd|jj)Nzthe connection is closedz7cannot execute operations: the connection is in status )r7r`OKrarOperationalErrorrrSr2r2r3rs   z#BaseConnection._check_connection_okccsN|jrdS|jjtkrdS||EdH|jr%|jEdHdSdS)z.Generator to start a transaction if necessary.N)r9r7rrr_get_tx_start_commandrErrSr2r2r3 _start_querys zBaseConnection._start_querybytescCs|jr|jSdg}|jdur$t|j}|d||jdd|jdur3||jr0dnd|jdurB||jr?dndd ||_|jS) NsBEGINsISOLATION LEVEL_rWs READ ONLYs READ WRITEs DEFERRABLEsNOT DEFERRABLE ) rIrsrrrreplacerrzr}join)rJpartsvalr2r2r3rs      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=rrr@r7rrrrErrSr2r2r3 _commit_gen2s 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=rrr@rErr7rrrr?clear maintain_genrSr2r2r3 _rollback_genFs&  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)rJrrrr2r2r3xidas 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_stringr7rrrrr rrr9r@rrrJrr2r2r3_tpc_begin_genns     zBaseConnection._tpc_begin_genccsv|js td|jdrtd|jd}|df|_|tdt|EdH|jr9|jEdHdSdS)Nz='tpc_prepare()' must be called inside a two-phase transactionr zF'tpc_prepare()' cannot be used during a prepared two-phase transactionrTzPREPARE TRANSACTION {}) r@rrrrformatr(rErrr2r2r3_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 transactionr r_genz{} PREPARED {}) lowerr@rrrrrgetattrrrrr()rJrrfnamemethr2r2r3_tpc_finish_gens,    zBaseConnection._tpc_finish_gencCsdS)z0Raise NotSupportedError if TPC is not supported.Nr2rSr2r2r3rrzBaseConnection._check_tpcN)r7r")rMrN)rMr()rMr_)rer_rMrN)rer_rMro)rMrr)rerrrMrN)rerrrMro)rMry)reryrMrN)reryrMro)rr(rMro)rMr!)rMr)rMr)rMr+)rrrMr)rr4rMrN)rKrrr#rMrN)rr5rMrN)rKrrrrMrN)rMr)rerrMrN)r)rr(rrrMr)rrrrrMr)rrrMro)rMro)rMr)rr+rr(rr(rMr)rrrMro)rrrrrMro)Br-r.r/r0rWarningErrorr DatabaseError DataErrorrIntegrityError InternalErrorrNotSupportedErrorrLrTr^propertyrQrbrdsetterrhrqrsrurwrzr{r|r}r~rrpr]rrrrrrrr staticmethodrArrrCrr classmethodrrrrrrrrrrrrrrr2r2r2r3r6Ns  #                                     . '     r6)Sr0 __future__rrloggingtypingrrrrweakrefrrwarningsr functoolsr rr r rrrabcrrrsqlrrr@rrowsradaptr_enumsr_compatrrrrpq.miscrrEr _preparingr _capabilitiesr _connection_infor!pq.abcr"r#psycopg_pool.baser$r%FormatrBINARY ConnStatusrrarrrrrrINTRANShas_send_close_preparedr getLoggerrr'r.rr4r1r5r6r2r2r2r3sX