a hx @s,ddlmZmZmZddlmZmZddlm Z ddl m Z ddl m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0ddl1m2Z2m3Z3m4Z4m5Z5m6Z6ddl7m8Z8m9Z9m:Z:ddl;me#Z?d Z@d ZAd ZBeCZDd dZEddZFddZGddZHddZIddZJddZKd8dd ZLd Z@d ZAd ZBd!d"ZMGd#d$d$ZNGd%d&d&e3ZOGd'd(d(ePZQGd)d*d*e8ZRGd+d,d,e8ZSGd-d.d.e8ZTGd/d0d0e8ZUGd1d2d2e8ZVGd3d4d4e8ZWGd5d6d6eWZXgd7ZYdS)9)datedatetimetime)countislice) localtime)warn)$BIGINTBOOLEAN BOOLEAN_ARRAYBYTESCHAR CHAR_ARRAYDATEFLOAT FLOAT_ARRAYINET INT2VECTORINTEGER INTEGER_ARRAYINTERVALJSONJSONBMACADDRNAME NAME_ARRAYNULLTYPENUMERIC NUMERIC_ARRAYOID PGIntervalPY_PGSTRINGTEXT TEXT_ARRAYTIME TIMESTAMP TIMESTAMPTZUNKNOWN UUID_TYPEVARCHAR VARCHAR_ARRAYXID)ContextCoreConnectionIN_FAILED_TRANSACTIONIN_TRANSACTIONver) DatabaseErrorErrorInterfaceError)RangezMathieu Fenniakz2.0formatcCs t|||S)zConstruct an object holding a date value. This function is part of the `DBAPI 2.0 specification `_. :rtype: :class:`datetime.date` )Date)yearmonthdayr<G/var/www/html/efn/efnconsultor/selenium/selenium-4.34.2/pg8000/dbapi.pyPgDateusr>cCs t|||S)zConstruct an object holding a time value. This function is part of the `DBAPI 2.0 specification `_. :rtype: :class:`datetime.time` )Time)hourminutesecondr<r<r=PgTimesrCcCst||||||S)zConstruct an object holding a timestamp value. This function is part of the `DBAPI 2.0 specification `_. :rtype: :class:`datetime.datetime` )Datetime)r9r:r;r@rArBr<r<r= TimestampsrEcCstt|ddS)aConstruct an object holding a date value from the given ticks value (number of seconds since the epoch). This function is part of the `DBAPI 2.0 specification `_. :rtype: :class:`datetime.date` N)r8rZticksr<r<r= DateFromTickss rHcCstt|ddS)aConstruct an object holding a time value from the given ticks value (number of seconds since the epoch). This function is part of the `DBAPI 2.0 specification `_. :rtype: :class:`datetime.time` rF)r?rrGr<r<r= TimeFromTickss rJcCstt|ddS)a Construct an object holding a timestamp value from the given ticks value (number of seconds since the epoch). This function is part of the `DBAPI 2.0 specification `_. :rtype: :class:`datetime.datetime` NrI)rErrGr<r<r=TimestampFromTickss rKcCs|S)zConstruct an object holding binary data. This function is part of the `DBAPI 2.0 specification `_. r<)valuer<r<r=BinarysrM localhostN8TcCs$t|||||||||| | | | | dS)N) hostdatabaseportpasswordsource_address unix_sock ssl_contexttimeout tcp_keepaliveapplication_name replicationstartup_paramssock) Connection)userrPrQrRrSrTrUrVrWrXrYrZr[r\r<r<r=connects r_c sd}d}d}d}d}d}d} d} d} g} g} td d td}|}d}t|D]4\}}|dt|krt||dnd}||kr@|d kr| ||d kr|}n|}q|d kr| ||}q|dkr| ||dkr|}q|dkr | ||dkr<| }q|dkr2|dkr2| t|q|dkrf|dkrf|dvrf|dkrf| dn|dkr|dkr|dvr|dkr|}| dn|dkr|dkr|dkr|}| dnp|dvr2|dkr2d}| rd} | |n6|dkr d} n&|dkr(|}| t|ntdn | |n@||kr|d krt| r`d} n|d krpd} n|}| |n||kr|d kr|}| |n||kr|d kr|dkr|}| |n|| kr|dkr|dkr|}| |nx||kr^|dkr| d |7<|dusJ|s\|d!kr\|}z2| | d dd }| dt |d| d =Wn*t y| dt t| Yn0n|dkrN|d"kr0|dkr0|}z2| | d dd }| dt |d| d =Wn*t y,| dt t| Yn0n|d#vr z$convert_paramstyle..'E"-rcqmark?numeric:z:=namedpyformat%()r7rrr7Tsz*Only %s and %% are supported in the query.\_)z() )rnrlr7c3s|]}|VqdSNr<).0pargsr<r= rgz%convert_paramstyle..) mapr enumeratelenappendnextr4isalnumindexrd ValueErrortuplejoin)ZstylequeryrOUTSIDE INSIDE_SQ INSIDE_QI INSIDE_ES INSIDE_PN INSIDE_COZ INSIDE_DQin_quote_escapeZin_param_escape placeholders output_queryZ param_idxstateprev_cicnext_cpidxvalsr<r~r=convert_paramstyles         (                       rc@seZdZddZeddZeddZeddZd!d d Zd dZ d"ddZ ddZ ddZ ddZ d#ddZddZddZddZd$dd Zd S)%CursorcCs"||_d|_d|_d|_d|_dS)Nr6r<)_c arraysize_context _row_iter _input_oids)self connectionr<r<r=__init__s zCursor.__init__cCstddd|jS)Nz'DB-API extension cursor.connection usedrF stacklevel)rrrr<r<r=rs zCursor.connectioncCs|j}|durdS|jS)Nrw)r row_count)rcontextr<r<r=rowcountszCursor.rowcountc Csf|j}|durdS|j}|dur$dSt|dkr4dSg}|D]$}||d|ddddddfq<|S)Nrnametype_oid)rcolumnsrr)rrrow_descrcolr<r<r= descriptions "zCursor.descriptionr<Nc Csz|jjs|jjs|jdt|dkrB|durB|j||_n(tt||\}}|jj|||j |d|_|jj dur~d|_ nt |jj |_ d|_ WnRt y}z:|jdurtdn|jjdurtdn|WYd}~n d}~00g|_dS)aeExecutes a database operation. Parameters may be provided as a sequence, or as a mapping, depending upon the value of :data:`pg8000.paramstyle`. This method is part of the `DBAPI 2.0 specification `_. :param operation: The SQL statement to execute. :param args: If :data:`paramstyle` is ``qmark``, ``numeric``, or ``format``, this argument should be an array of parameters to bind into the statement. If :data:`paramstyle` is ``named``, the argument should be a dict mapping of parameters. If the :data:`paramstyle` is ``pyformat``, the argument value may be either an array or a mapping. :param stream: This is a pg8000 extension for use with the PostgreSQL `COPY `_ command. For a COPY FROM the parameter must be a readable file-like object, and for COPY TO it must be writable. .. versionadded:: 1.9.11 begin transactionrN)roidsstreamr< Cursor closedconnection is closed)r_in_transaction autocommitexecute_simplerrr paramstyleexecute_unnamedrrowsriterAttributeErrorr4_sock input_types)r operationrr statementrer<r<r=executes(        zCursor.executecCsng}|j}|D]$}||_|||||jjqt|dkrLtd|_nd|vr^d|j_n t||j_dS)a1Prepare a database operation, and then execute it against all parameter sequences or mappings provided. This method is part of the `DBAPI 2.0 specification `_. :param operation: The SQL statement to execute :param parameter_sets: A sequence of parameters to execute the statement with. The values in the sequence should be sequences or mappings of parameters, the same as the args argument of the :meth:`execute` method. rNrw)rrrrrrr-sum)rr param_sets rowcounts input_oids parametersr<r<r= executemanys    zCursor.executemanyc Cs|dur gn|}d|dddd|Dd}zHtd||\}}|jj||d|_|jjdurld|_nt|jj|_WnRty}z:|jdurt d n|jj durt d n|WYd}~n d}~00dS) NzCALL rtz, cSsg|]}dqS)z%sr<)r|rxr<r<r= rgz#Cursor.callproc..ryr7)rrr) rrrrrrrrrr4r)rZprocnamerrrrrrr<r<r=callprocs$     zCursor.callproccCs<z t|WStyYdSty6tdYn0dS)a Fetch the next row of a query result set. This method is part of the `DBAPI 2.0 specification `_. :returns: A row as a sequence of field values, or ``None`` if no more rows are available. N#attempting to use unexecuted cursor)r StopIteration TypeErrorProgrammingErrorrr<r<r=fetchones    zCursor.fetchonecCs|S)zoA cursor object is iterable to retrieve the rows from a query. This is a DBAPI 2.0 extension. r<rr<r<r=__iter__/szCursor.__iter__c Csz t|jWSty4|jdur.tdnYnVty}z>|jdurVtdnt|jjdkrptdn|WYd}~n d}~00dS)NzA query hasn't been issued.rz no result set)rrrrrrrr)rrr<r<r=__next__6s       zCursor.__next__cCs>ztt||dur|jn|WSty8tdYn0dS)aFetches the next set of rows of a query result. This method is part of the `DBAPI 2.0 specification `_. :param size: The number of rows to fetch when called. If not provided, the :attr:`arraysize` attribute value is used instead. :returns: A sequence, each entry of which is a sequence of field values making up a row. If no more rows are available, an empty sequence will be returned. Nr)rrrrr)rnumr<r<r= fetchmanyFs zCursor.fetchmanycCs*z t|WSty$tdYn0dS)a$Fetches all remaining rows of a query result. This method is part of the `DBAPI 2.0 specification `_. :returns: A sequence, each entry of which is a sequence of field values making up a row. rN)rrrrr<r<r=fetchall\s   zCursor.fetchallcCs d|_dS)zCloses the cursor. This method is part of the `DBAPI 2.0 specification `_. N)rrr<r<r=closelsz Cursor.closec GsVg}|D]B}t|tr|}n$z t|}Wnty>t}Yn0||q||_dS)z3This method is part of the `DBAPI 2.0 specificationN) isinstanceintr!KeyErrorr(rr)rsizesrsizeoidr<r<r= setinputsizests     zCursor.setinputsizescCsdS)zThis method is part of the `DBAPI 2.0 specification `_, however, it is not implemented by pg8000. Nr<)rrcolumnr<r<r= setoutputsizeszCursor.setoutputsize)r<N)N)N)N)__name__ __module__ __qualname__rpropertyrrrrrrrrrrrrrrr<r<r<r=rs$     6  rcseZdZfddZeddZeddZeddZeddZeddZ ed dZ ed dZ ed dZ ed dZ d dZeddZddZddZddZddZddZddZd$ddZd%d d!Zd"d#ZZS)&r]cstj|i|d|_dS)NF)superrr)rrkwargs __class__r<r=rszConnection.__init__cCs |tSr{) _getErrorWarningrr<r<r=rfrgzConnection.cCs |tSr{)rr3rr<r<r=rfrgcCs |tSr{)rr4rr<r<r=rfrgcCs |tSr{)rr2rr<r<r=rfrgcCs |tSr{)rOperationalErrorrr<r<r=rfrgcCs |tSr{)rIntegrityErrorrr<r<r=rfrgcCs |tSr{)r InternalErrorrr<r<r=rfrgcCs |tSr{)rrrr<r<r=rfrgcCs |tSr{)rNotSupportedErrorrr<r<r=rfrgcCstd|jddd|S)NzDB-API extension connection.z usedrFr)rr)rerrorr<r<r=rszConnection._getErrorcCs|jttfvSr{)_transaction_statusr0r/rr<r<r=rszConnection._in_transactioncCst|S)zCreates a :class:`Cursor` object bound to this connection. This function is part of the `DBAPI 2.0 specification `_. )rrr<r<r=cursorszConnection.cursorcCs|ddS)zCommits the current database transaction. This function is part of the `DBAPI 2.0 specification `_. commitN)rrr<r<r=rszConnection.commitcCs|js dS|ddS)zRolls back the current database transaction. This function is part of the `DBAPI 2.0 specification `_. Nrollback)rrrr<r<r=rszConnection.rollbackcCs |||fS)a)Create a Transaction IDs (only global_transaction_id is used in pg) format_id and branch_qualifier are not used in postgres global_transaction_id may be any string identifier supported by postgres returns a tuple (format_id, global_transaction_id, branch_qualifier)r<)r format_idglobal_transaction_idbranch_qualifierr<r<r=xidszConnection.xidcCs||_|jr|ddS)a.Begins a TPC transaction with the given transaction ID xid. This method should be called outside of a transaction (i.e. nothing may have executed since the last .commit() or .rollback()). Furthermore, it is an error to call .commit() or .rollback() within the TPC transaction. A ProgrammingError is raised, if the application calls .commit() or .rollback() during an active TPC transaction. This function is part of the `DBAPI 2.0 specification `_. rN)_xidrr)rrr<r<r= tpc_begins zConnection.tpc_begincCs|d|jdfdS)aPerforms the first phase of a transaction started with .tpc_begin(). A ProgrammingError is be raised if this method is called outside of a TPC transaction. After calling .tpc_prepare(), no statements can be executed until .tpc_commit() or .tpc_rollback() have been called. This function is part of the `DBAPI 2.0 specification `_. zPREPARE TRANSACTION '%s';r6N)rrrr<r<r= tpc_prepares zConnection.tpc_prepareNcCsr|dur|j}|durtdz@|j}d|_||vrN|d|dfn|W||_n||_0d|_dS)a%When called with no arguments, .tpc_commit() commits a TPC transaction previously prepared with .tpc_prepare(). If .tpc_commit() is called prior to .tpc_prepare(), a single phase commit is performed. A transaction manager may choose to do this if only a single resource is participating in the global transaction. When called with a transaction ID xid, the database commits the given transaction. If an invalid transaction ID is provided, a ProgrammingError will be raised. This form should be called outside of a transaction, and is intended for use in recovery. On return, the TPC transaction is ended. This function is part of the `DBAPI 2.0 specification `_. Nz.Cannot tpc_commit() without a TPC transaction!TzCOMMIT PREPARED '%s';r6)rrr tpc_recoverrrrrprevious_autocommit_moder<r<r= tpc_commits  zConnection.tpc_commitcCsr|dur|j}|durtdz@|j}d|_||vrN|d|dfn|W||_n||_0d|_dS)aDWhen called with no arguments, .tpc_rollback() rolls back a TPC transaction. It may be called before or after .tpc_prepare(). When called with a transaction ID xid, it rolls back the given transaction. If an invalid transaction ID is provided, a ProgrammingError is raised. This form should be called outside of a transaction, and is intended for use in recovery. On return, the TPC transaction is ended. This function is part of the `DBAPI 2.0 specification `_. Nz9Cannot tpc_rollback() without a TPC prepared transaction!TzROLLBACK PREPARED '%s';r6)rrrrrrrr<r<r= tpc_rollbacks  zConnection.tpc_rollbackcsJz<j}d_}|dfdd|DW|_S|_0dS)zReturns a list of pending transaction IDs suitable for use with .tpc_commit(xid) or .tpc_rollback(xid). This function is part of the `DBAPI 2.0 specification `_. Tz!select gid FROM pg_prepared_xactscsg|]}d|ddqS)rrq)r)r|rowrr<r=r7rgz*Connection.tpc_recover..N)rrrr)rrcursr<rr=r+s zConnection.tpc_recover)N)N)rrrrrrr3r4r2rrrrrrrrrrrrrrrr __classcell__r<r<rr=r]s,              $ #r]c@seZdZdZdS)rzGeneric exception raised for important database warnings like data truncations. This exception is not currently used by pg8000. This exception is part of the `DBAPI 2.0 specification `_. Nrrr__doc__r<r<r<r=r<src@seZdZdZdS) DataErrorzGeneric exception raised for errors that are due to problems with the processed data. This exception is not currently raised by pg8000. This exception is part of the `DBAPI 2.0 specification `_. Nrr<r<r<r=rGsrc@seZdZdZdS)ra= Generic exception raised for errors that are related to the database's operation and not necessarily under the control of the programmer. This exception is currently never raised by pg8000. This exception is part of the `DBAPI 2.0 specification `_. Nrr<r<r<r=rRs rc@seZdZdZdS)ra Generic exception raised when the relational integrity of the database is affected. This exception is not currently raised by pg8000. This exception is part of the `DBAPI 2.0 specification `_. Nrr<r<r<r=r_src@seZdZdZdS)raNGeneric exception raised when the database encounters an internal error. This is currently only raised when unexpected state occurs in the pg8000 interface itself, and is typically the result of a interface bug. This exception is part of the `DBAPI 2.0 specification `_. Nrr<r<r<r=rksrc@seZdZdZdS)ra&Generic exception raised for programming errors. For example, this exception is raised if more parameter fields are in a query string than there are available parameters. This exception is part of the `DBAPI 2.0 specification `_. Nrr<r<r<r=rwsrc@seZdZdZdS)rzGeneric exception raised in case a method or database API was used which is not supported by the database. This exception is part of the `DBAPI 2.0 specification `_. Nrr<r<r<r=rsrc@seZdZdZdS)ArrayContentNotSupportedErrorz Raised when attempting to transmit an array where the base type is not supported for binary data transfer by the interface. Nrr<r<r<r=rsr):r BINARYr r r rMr rr]rrrr2r8rHr3rrrrrrrrr4rrrrrrrrrrrrr rROWIDr5r"r#r$r%r&r'r?rJrErKr(r)r*r+rr,r_) rNNrONNNNNTNNNN)Zrrr8rDrr? itertoolsrrrwarningsrpg8000.convertersr r r r r rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r, pg8000.corer-r.r/r0r1pg8000.exceptionsr2r3r4 pg8000.typesr5 __version__ __author__rapilevel threadsafetyrbytesrr>rCrErHrJrKrMr_rrr] Exceptionrrrrrrrr__all__r<r<r<r=sl  &          " 2