a hd @sddlmZmZddlmZddlmZddl Z ddl m Z m Z 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/m0Z1m2Z2m3Z4m5Z6ddl7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKddlLmMZMmNZNmOZOezCursor.c Csf|j}|durdS|j}|dur$dSt|dkr4dSg}|D]$}||d|ddddddfq<|S)NrnameZtype_oid)r_columnslenappend)rbrrrow_descrxcolrUrUrVrts "zCursor._getDescriptionrUc Cslz|jjs|jjs|jdt|dkrB|durB|j||_n*t|j||\}}|jj|||j |d|_|jj dur|gn|jj }t ||_ d|_ Wnt y}z:|jdurtdn|jjdurtdn|WYd}~nd}~0ty`}z`|jd}t|trD|d} | d kr&t} n| d kr6t} nt} | |nt|WYd}~n d}~00g|_|S) 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)valsoidsstreamrU Cursor closedconnection is closedC2800023505)r[_in_transaction autocommitZexecute_simpleryr_rAr^execute_unnamedrarowsiterr`AttributeErrorrD_sockrBargs isinstancedictr8r<Z input_types) rb operationrr statementr~remsg response_codeclsrUrUrVexecutes>             zCursor.executecCsng}|j}|D]$}||_|||||jjqt|dkrLtd|_nd|vr^d|j_n t||j_|S)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. rNrp)rarrzr_rqryr/sum)rbrZ param_setsZ rowcountsZ input_oids parametersrUrUrV executemany"s     zCursor.executemanycCsTz t|WStyYdSty6tdYntyNtdYn0dS)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)next StopIteration TypeErrorr<rrfrUrUrVfetchone@s      zCursor.fetchonecCs>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)tuplerr\rr<)rbZnumrUrUrV fetchmanySs 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)rrr<rfrUrUrVfetchallis   zCursor.fetchallcCs d|_dS)zCloses the cursor. This method is part of the `DBAPI 2.0 specification `_. N)r[rfrUrUrVrhysz Cursor.closecCs|S)zoA cursor object is iterable to retrieve the rows from a query. This is a DBAPI 2.0 extension. rUrfrUrUrV__iter__szCursor.__iter__c GsVg}|D]B}t|tr|}n$z t|}Wnty>t}Yn0||q||_dS)z3This method is part of the `DBAPI 2.0 specificationN)rintrKeyErrorr&rzra)rbZsizesrsizeZoidrUrUrV setinputsizess     zCursor.setinputsizescCsdS)zThis method is part of the `DBAPI 2.0 specification `_, however, it is not implemented by pg8000. NrU)rbrZcolumnrUrUrV setoutputsizeszCursor.setoutputsizec Csz t|jWSty4|jdur.tdnYnVty}z>|jdurVtdnt|jjdkrptdn|WYd}~n d}~00dS)NzA query hasn't been issued.rz no result set)rr`rr_r<rryrx)rbrrUrUrV__next__s       zCursor.__next__)N)rUN)N)N)__name__ __module__ __qualname__rdrgrlpropertyrcrs descriptionrtrrrrrrhrrrrrUrUrUrVrZs&     E  rZcseZdZeddZeddZeddZeddZeddZeddZ eddZ ed dZ ed dZ fd d Z d dZddZeddZeddZddZddZd*ddZddZddZd d!Zd"d#Zd+d$d%Zd,d&d'Zd(d)ZZS)-rScCs |tSre) _getErrorr@rfrUrUrVrurvzConnection.cCs |tSre)rrCrfrUrUrVrurvcCs |tSre)rrDrfrUrUrVrurvcCs |tSre)rrBrfrUrUrVrurvcCs |tSre)rr;rfrUrUrVrurvcCs |tSre)rr8rfrUrUrVrurvcCs |tSre)rr9rfrUrUrVrurvcCs |tSre)rr<rfrUrUrVrurvcCs |tSre)rr:rfrUrUrVrurvc sztj|i|Wnrty}zZ|jd}t|trl|d}|dkrPt}n|dkr^t}nt}||nt|WYd}~n d}~00t |dd|_ d|_ dS)NrrrrZnamed)r^F) superrdrBrrrrDr8r<rZ _run_cursorr)rbrkwargsrrrr __class__rUrVrds   zConnection.__init__cCstd|jdd|S)Nz#DB-API extension connection.%s usedrmrn)rr)rberrorrUrUrVrszConnection._getErrorcCst|S)zCreates a :class:`Cursor` object bound to this connection. This function is part of the `DBAPI 2.0 specification `_. )rZrfrUrUrVcursorszConnection.cursorcCs |jSre)rrtrfrUrUrVrszConnection.descriptioncCs|jttfvSre)Z_transaction_statusr2r1rfrUrUrVrszConnection._in_transactioncCs|ddS)zCommits the current database transaction. This function is part of the `DBAPI 2.0 specification `_. commitN)rrfrUrUrVrszConnection.commitcCs|js dS|ddS)zRolls back the current database transaction. This function is part of the `DBAPI 2.0 specification `_. Nrollback)rrrfrUrUrVrszConnection.rollbackNcKs8|jj|||d|jjjdur&tSt|jjjSdS)N)r)rrr_rr)rbZsqlrparamsrUrUrVrunszConnection.runcCs t||Sre)PreparedStatement)rbrrUrUrVprepareszConnection.preparecCs |||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)rU)rbZ format_idZglobal_transaction_idZbranch_qualifierrUrUrVxidszConnection.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 `_. r}N)_xidrr)rbrrUrUrV tpc_begin s zConnection.tpc_begincCsd|jdf}||dS)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';rXN)rr)rbqrUrUrV tpc_prepares zConnection.tpc_preparecCsr|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';rX)rr<r tpc_recoverrrrbrprevious_autocommit_moderUrUrV tpc_commit)s  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';rX)rr<rrrrrrUrUrV tpc_rollbackMs  zConnection.tpc_rollbackcsFz8j}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)r)r).0rowrfrUrV |rvz*Connection.tpc_recover..N)rrr)rbrZcursrUrfrVrps zConnection.tpc_recover)N)N)N)rrrrr@rCrDrBr;r8r9r<r:rdrrrrrrrrrrrrrr __classcell__rUrUrrVrSs4               $ #rSc snd}d}d}d}d}d}d}gg}|} d} t|D]\} } | dt|kr\|| d} nd} | |kr| dkr|| | d kr|} q|} nj| d kr|| |} nR| d kr|| | d kr|} n2| d kr| d vr| d kr|} dn || nR| |kr>| dkr0|rd}n| dkr,d}n|} || n| |krb| d krV|} || n| |kr| dkr| dkr|} || n| |kr,d| 7<| dus| sN| dkrN|} z2ddd}|dt|dd=Wn*ty(|dttYn0n"| |krN|| | dkrN|} | } q4fdd}d||fS)NrrXrmF'E"-:z:=rT\rp_$ cstfddDS)Nc3s|]}|VqdSrerU)rprrUrV rvz2to_statement..make_vals..)rrZ placeholdersrrV make_valsszto_statement..make_vals) enumerateryrzisalnumindexstr ValueErrorjoin)ZqueryZOUTSIDEZ INSIDE_SQZ INSIDE_QIZ INSIDE_ESZ INSIDE_PNZ INSIDE_COZin_quote_escapeZ output_querystateZprev_cicZnext_cZpidxrrUrrV to_statements                  rc@s$eZdZddZddZddZdS)rcCs6||_||_t|\}|_||d\|_|_|_dS)NrU)conrr make_argsZprepare_statementname_binr{ input_funcs)rbrrrrUrUrVrds zPreparedStatement.__init__c Kst|jj||}z>|jjs2|jjs2|jd|j|j||j |j |j |_ WnRt y}z:|jdurvtdn|jjdurtdn|WYd}~n d}~00|j jdurtSt|j jS)Nr}rr)r,rZpy_typesrrrrZ execute_namedrr{rrr_rrDrrr)rbr~rrrUrUrVrs      zPreparedStatement.runcCs|j|jd|_dSre)rZclose_prepared_statementrrfrUrUrVrhszPreparedStatement.closeN)rrrrdrrhrUrUrUrVrsr)@ BIGINTEGERr4rrr r5r r rSrZr DATETIMEDECIMAL DECIMAL_ARRAYr6rBDater7rCr rrrrrrr8rDr9rrrrrrNUMBERr:rr;rr<ROWIDrr r!r"r# TIMEDELTAr$r%Timer=r>r?r&r'r(r)r@r*rW pginterval_inpginterval_out timedelta_in) rENrFNNNNNTNNN)bdatetimerrrr itertoolsrwarningsrr]Zpg8000.convertersrrrr r r r r rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+rr,r-rr.rZ pg8000.corer/r0r1r2r3Z pg8000.dbapir4r5r6r7r8r9r:r;r<r=r>r?r@rAZpg8000.exceptionsrBrCrD __version__ __author__rrrrrrrrWapilevel threadsafetyr^rZrSrr__all__rUrUrUrVsN  +@   SU!