U ¯Dx`W_ã<@sddlmZddlmZddlZddlmZmZmZm Z m Z mZmZm 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/m0Z0m1Z1m2Z2m3Z3ddl4m5Z5ddl6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>ddl?m@Z@mAZAmBZBdd lCmDZDeDƒd ZE[DdZFe"ZGddd„ZHdZIdZJdZKGdd„dƒZLGdd„de5ƒZMdd„ZNGdd„dƒZOe=e7e@eHeBe)Ú_cÚ arraysizeÚpg8000Ú paramstyleÚ_contextÚ _row_iterÚ_input_oids)ÚselfÚ connectionrXrPrPrQÚ__init__\s zCursor.__init__cCs|S©NrP©r\rPrPrQÚ __enter__iszCursor.__enter__cCs| ¡dSr_)Úclose)r\Úexc_typeÚ exc_valueÚ tracebackrPrPrQÚ__exit__lszCursor.__exit__cCstddd|jS)Nz'DB-API extension cursor.connection usedé©Ú stacklevel)rrUr`rPrPrQr]oszCursor.connectioncCs|j}|dkrdS|jS)Néÿÿÿÿ)rYÚ row_count)r\ÚcontextrPrPrQÚrowcounttszCursor.rowcountcCs| ¡Sr_)Ú_getDescriptionr`rPrPrQÚ|ózCursor.c Csf|j}|dkrdS|j}|dkr$dSt|ƒdkr4dSg}|D]$}| |d|ddddddf¡q<|S)NrÚnameZtype_oid)rYÚcolumnsÚlenÚappend)r\rlÚrow_descrrÚcolrPrPrQrn~sÿzCursor._getDescriptionrPc Cs,z^|jjs|jjs|j d¡t|j||ƒ\}}|jj|||j|d|_t|jj ƒ|_ d|_WnÂtk r®}z2|jdkr„tdƒ‚n|jj dkrštdƒ‚n|‚W5d}~XYnttk r }zT|jd}t|tƒr|d}|dkrìt} n|d krút} nt} | |ƒ‚nt|ƒ‚W5d}~XYnXg|_|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 transaction)ÚvalsZ input_oidsÚstreamNú Cursor closedúconnection is closedrÚCÚ28000Ú23505)rUÚin_transactionÚ autocommitÚexecute_unnamedr:rXr[rYÚiterÚrowsrZÚAttributeErrorr=Ú_sockr;ÚargsÚ isinstanceÚdictr4r8Zinput_types) r\Ú operationr†ryÚ statementrxÚeÚmsgÚ response_codeÚclsrPrPrQÚexecute’sFÿþ zCursor.executecCsDg}|D]}| ||¡| |jj¡qd|kr4dnt|ƒ|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. rj)rrtrYrkÚsum)r\r‰Z param_setsZ rowcountsÚ parametersrPrPrQÚexecutemanyÔszCursor.executemanycCsZz t|ƒWStk r YdStk r:tdƒ‚Yntk rTtdƒ‚YnXdS)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Ú TypeErrorr8r„r`rPrPrQÚfetchoneês zCursor.fetchonecCs@ztt||dkr|jn|ƒƒWStk r:tdƒ‚YnXdS)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“)ÚtuplerrVr–r8)r\ÚnumrPrPrQÚ fetchmanyýsÿzCursor.fetchmanycCs,z t|ƒWStk r&tdƒ‚YnXdS)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. r“N)r˜r–r8r`rPrPrQÚfetchalls zCursor.fetchallcCs d|_dS)zŽCloses the cursor. This method is part of the `DBAPI 2.0 specification `_. N)rUr`rPrPrQrb$szCursor.closecCs|S)zoA cursor object is iterable to retrieve the rows from a query. This is a DBAPI 2.0 extension. rPr`rPrPrQÚ__iter__,szCursor.__iter__c Gsdg}|D]P}t|tƒr|}n2z|jj|\}}Wntk rLtjj}YnX| |¡q||_ dS)z`_, however, it is not implemented by pg8000. NrP)r\r¢ÚcolumnrPrPrQÚ setoutputsizeCszCursor.setoutputsizec CsŽzt|jƒWStk r6|jdkr0tdƒ‚n‚YnTtk rˆ}z6|jdkrZtdƒ‚nt|jjƒdkrttdƒ‚n|‚W5d}~XYnXdS)NzA query hasn't been issued.rz no result set)r”rZr„rYr8r•rsrr)r\r‹rPrPrQÚ__next__Js zCursor.__next__)N)rPN)N)N)Ú__name__Ú __module__Ú__qualname__r^rarfÚpropertyr]rmÚdescriptionrnrr’r—ršr›rbrœr¥r§r¨rPrPrPrQrT[s& B rTcsúeZdZedd„ƒZedd„ƒZedd„ƒZedd„ƒZedd„ƒZedd„ƒZ edd„ƒZ ed d„ƒZed d„ƒZ‡fdd„Z d d„Zdd„Zedd„ƒZdd„Zdd„Zd(dd„Zdd„Zdd„Zdd„Zd d!„Zd)d"d#„Zd*d$d%„Zd&d'„Z‡ZS)+rNcCs | t¡Sr_)Ú _getErrorr9r`rPrPrQro^rpzConnection.cCs | t¡Sr_)r®r<r`rPrPrQro_rpcCs | t¡Sr_)r®r=r`rPrPrQro`rpcCs | t¡Sr_)r®r;r`rPrPrQroarpcCs | t¡Sr_)r®r7r`rPrPrQrobrpcCs | t¡Sr_)r®r4r`rPrPrQrocrpcCs | t¡Sr_)r®r5r`rPrPrQrodrpcCs | t¡Sr_)r®r8r`rPrPrQroerpcCs | t¡Sr_)r®r6r`rPrPrQrogrpc s–ztƒj||ŽWnptk r‚}zR|jd}t|tƒrj|d}|dkrNt}n|dkr\t}nt}||ƒ‚nt|ƒ‚W5d}~XYnXt |dd|_ dS)Nrr|r}r~Únamed)rX)Úsuperr^r;r†r‡rˆr=r4r8rTÚ_run_cursor)r\r†Úkwargsr‹rŒrrŽ©Ú __class__rPrQr^is zConnection.__init__cCstd|jdd|S)Nz#DB-API extension connection.%s usedrgrh)rr©)r\ÚerrorrPrPrQr®~sÿþzConnection._getErrorcCst|ƒS)zÀCreates a :class:`Cursor` object bound to this connection. This function is part of the `DBAPI 2.0 specification `_. )rTr`rPrPrQÚcursor„szConnection.cursorcCs |j ¡Sr_)r±rnr`rPrPrQrszConnection.descriptioncCs| d¡dS)z§Commits the current database transaction. This function is part of the `DBAPI 2.0 specification `_. ÚcommitN)rr`rPrPrQr·‘szConnection.commitcCs|js dS| d¡dS)zªRolls back the current database transaction. This function is part of the `DBAPI 2.0 specification `_. NÚrollback)rrr`rPrPrQr¸™szConnection.rollbackNcKs |jj|||dt|jjjƒS)N)ry)r±rr˜rYrƒ)r\ZsqlryÚparamsrPrPrQÚrun£szConnection.runcCs t||ƒSr_)ÚPreparedStatement)r\r‰rPrPrQÚprepare§szConnection.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)rP)r\Z format_idZglobal_transaction_idZbranch_qualifierrPrPrQÚxidªszConnection.xidcCs||_|jr| d¡dS)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 `_. rwN)Ú_xidr€r)r\r½rPrPrQÚ tpc_begin²s zConnection.tpc_begincCsd|jdf}| |¡dS)a¯Performs 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';r>N)r¾r)r\ÚqrPrPrQÚtpc_prepareÃszConnection.tpc_preparecCsl|dkr|j}|dkrtdƒ‚z:|j}d|_|| ¡krN| d|df¡n| ¡W5||_Xd|_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';r>)r¾r8r€Útpc_recoverrr·©r\r½Úprevious_autocommit_moderPrPrQÚ tpc_commitÑsÿzConnection.tpc_commitcCsl|dkr|j}|dkrtdƒ‚z:|j}d|_|| ¡krN| d|df¡n| ¡W5||_Xd|_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';r>)r¾r8r€rÂrr¸rÃrPrPrQÚtpc_rollbackösÿzConnection.tpc_rollbackcsBz4ˆj}dˆ_ˆ ¡}| d¡‡fdd„|DƒW¢S|ˆ_XdS)zíReturns 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|dd¡‘qS)rÚ)r½)Ú.0Úrowr`rPrQÚ $sz*Connection.tpc_recover..N)r€r¶r)r\rÄZcursrPr`rQrÂs zConnection.tpc_recover)N)N)N)r©rªr«r¬r9r<r=r;r7r4r5r8r6r^r®r¶rr·r¸rºr¼r½r¿rÁrÅrÆrÂÚ __classcell__rPrPr³rQrN[s4ÿ % "rNc spd}d}d}d}d}d}d}g‰g}|} d} t|ƒD]\}}|dt|ƒkr\||d} nd} | |krü|dkrŽ| |¡| d krˆ|} qø|} nj|d kr¦| |¡|} nR|dkrÆ| |¡| dkrø|} n2|dkrî| d krî| dkrî|} ˆ d¡n | |¡nT| |kr>|dkr0|rd}n| dkr,d}n|} | |¡n| |krb|d krV|} | |¡nî| |kr|dkr„| dkr„|} | |¡nÀ| |kr.ˆd|7<| dksÈ| ¡sP| dkrP|} z2ˆ ˆddd¡}| dt|dƒ¡ˆd=Wn,tk r*| dttˆƒƒ¡YnXn"| |krP| |¡|dkrP|} |} q4‡fdd„}d |¡|fS)Nrr>érgééFú'ÚEú"ú-ú:z:=rÇTú\rjr¤ú$Ú cst‡fdd„ˆDƒƒS)Nc3s|]}ˆ|VqdSr_rP)rÈÚp©r†rPrQÚ ysz2to_statement..make_vals..)r˜rØ©ZplaceholdersrØrQÚ make_valsxszto_statement..make_vals)Ú enumeratersrtÚisalnumÚindexÚstrÚ ValueErrorÚjoin)ÚqueryZOUTSIDEZ INSIDE_SQZ INSIDE_QIZ INSIDE_ESZ INSIDE_PNZ INSIDE_COZin_quote_escapeZoutput_queryÚstateZprev_cÚiÚcZnext_cZpidxrÛrPrÚrQÚto_statement)s‚ ræc@s$eZdZdd„Zdd„Zdd„ZdS)r»cCs&||_||_t|ƒ\|_|_i|_dSr_)Úconr‰rærŠÚ make_argsÚname_map)r\rçr‰rPrPrQr^szPreparedStatement.__init__c Ksò|j | |¡¡\}}z|j|\}}}Wn4tk r^|j |j|¡\}}}|j|<YnXz4|jjs~|jjs~|j d¡|j ||||¡|_WnPtk rä}z2|jdkrºt dƒ‚n|jjdkrÐt dƒ‚n|‚W5d}~XYnXt|jjƒS)Nrwrzr{)rçZmake_paramsrèréržZprepare_statementrŠrr€rZ execute_namedrYr„r=r…r˜rƒ)r\rxr¡r¹Zname_binruZinput_funcsr‹rPrPrQrº…s,ÿÿ zPreparedStatement.runcCs,|j ¡D]\}}}|j |¡q d|_dSr_)réÚvaluesrçZclose_prepared_statement)r\Zstatement_name_binr¤rPrPrQrbžszPreparedStatement.closeN)r©rªr«r^rºrbrPrPrPrQr»~sr»)rANrBNNNNNTNN)QÚ itertoolsrÚwarningsrrWZpg8000.convertersrrrrrr r rrr rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1Zpg8000.corer2Zpg8000.dbapir3r4r5r6r7r8r9r:Zpg8000.exceptionsr;r<r=Ú_versionr?Ú__version__Ú __author__ZROWIDrRZapilevelZthreadsafetyrXrTrNrær»Ú__all__rPrPrPrQÚs¼À( ý OU'÷