HEX

File: //lib/python3.6/site-packages/pymysql/__pycache__/cursors.cpython-36.opt-1.pyc
3

(�Y_z?�@s�ddlmZmZddlmZddlZddlmZmZm	Z	ddl
mZejdej
ejB�ZGd
d�de�ZGdd
�d
e�ZGdd�dee�ZGdd�de�ZGdd�dee�ZdS)�)�print_function�absolute_import)�partialN�)�
range_type�	text_type�PY2)�errz'\s*((?:INSERT|REPLACE)\b.+\bVALUES?\s*)z6(\(\s*(?:%s|%\(.+\)s)\s*(?:,\s*(?:%s|%\(.+\)s)\s*)*\))z(\s*(?:ON DUPLICATE.*)?);?\s*\Zc@s0eZdZdZdZdd�Zdd�Zdd�Zd	d
�Zdd�Z	d
d�Z
dd�Zdd�Zdd�Z
d:dd�Zdd�Zd;dd�Zdd�Zd<dd �Zd=d!d"�Zd#d$�Zd%d&�Zffd'd(�Zd)d*�Zd>d+d,�Zd-d.�Zd?d0d1�Zd2d3�Zd4d5�Zd6d7�Zd8d9�Zej Z ej!Z!ej"Z"ej#Z#ej$Z$ej%Z%ej&Z&ej'Z'ej(Z(ej)Z)dS)@�Cursora

    This is the object you use to interact with the database.

    Do not create an instance of a Cursor yourself. Call
    connections.Connection.cursor().

    See `Cursor <https://www.python.org/dev/peps/pep-0249/#cursor-objects>`_ in
    the specification.
    i�cCs4||_d|_d|_d|_d|_d|_d|_d|_dS)Nrr���)�
connection�description�	rownumber�rowcount�	arraysize�	_executed�_result�_rows)�selfr�r�/usr/lib/python3.6/cursors.py�__init__%szCursor.__init__cCs2|j}|dkrdSzx|j�r qWWdd|_XdS)zD
        Closing a cursor just exhausts all remaining data.
        N)r�nextset)r�connrrr�close/s
zCursor.closecCs|S)Nr)rrrr�	__enter__<szCursor.__enter__cGs~|j�dS)N)r)r�exc_inforrr�__exit__?szCursor.__exit__cCs|jstjd��|jS)Nz
Cursor closed)rr	�ProgrammingError)rrrr�_get_dbCs
zCursor._get_dbcCs|jstjd��dS)Nzexecute() first)rr	r)rrrr�_check_executedHszCursor._check_executedcCs|S)Nr)r�rowrrr�	_conv_rowLszCursor._conv_rowcGsdS)z!Does nothing, required by DB API.Nr)r�argsrrr�
setinputsizesOszCursor.setinputsizescGsdS)z!Does nothing, required by DB API.Nr)rr#rrr�setoutputsizesRszCursor.setoutputsizesFcCsT|j�}|j}|dks ||jk	r$dS|js.dSd|_|j�|j|d�|j�dS)zGet the next query setN)�
unbufferedT)rrZhas_next�
_clear_resultZnext_result�_do_get_result)rr&rZcurrent_resultrrr�_nextsetUszCursor._nextsetcCs
|jd�S)NF)r))rrrrrcszCursor.nextsetNcsDt|t�r|j��}n*t|ttf�r@t|���fdd�|D��}|S)Nc3s|]}�j|�d�VqdS))�encodingN)�
_ensure_bytes)�.0�v)r*rrr�	<genexpr>jsz'Cursor._ensure_bytes.<locals>.<genexpr>)�
isinstancer�encode�tuple�list�type)r�xr*r)r*rrr+fs

zCursor._ensure_bytescs�t|j�jd��t|ttf�rFtr0tt�|��}t�fdd�|D��St|t�r�trj�fdd�|j	�D�}�fdd�|j	�D�Str��|�}�j
|�SdS)N)r*c3s|]}�j|�VqdS)N)�literal)r,�arg)rrrr.ssz&Cursor._escape_args.<locals>.<genexpr>csi|]\}}�|��|��qSrr)r,�key�val)�ensure_bytesrr�
<dictcomp>vsz'Cursor._escape_args.<locals>.<dictcomp>csi|]\}}�j|�|�qSr)r5)r,r7r8)rrrr:xs)rr+r*r/r1r2r�map�dict�items�escape)rr#rr)rr9r�_escape_argsms

zCursor._escape_argscCs8|j�}tr|j||jd�}|dk	r4||j||�}|S)z�
        Returns the exact string that is sent to the database by calling the
        execute() method.

        This method follows the extension to the DB API 2.0 followed by Psycopg.
        )r*N)rrr+r*r?)r�queryr#rrrr�mogrify�szCursor.mogrifycCs.x|j�rqW|j||�}|j|�}||_|S)a�Execute a query

        :param str query: Query to execute.

        :param args: parameters used with query. (optional)
        :type args: tuple, list or dict

        :return: Number of affected rows
        :rtype: int

        If args is a list or tuple, %s can be used as a placeholder in the query.
        If args is a dict, %(name)s can be used as a placeholder in the query.
        )rrA�_queryr)rr@r#�resultrrr�execute�s

zCursor.executecs||sdStj��}|r\|jd�f}|jd�j�}|jd�p>d}�j||||�j�j�j�St��fdd�|D���_	�j	S)a�Run several data against one query

        :param query: query to execute on server
        :param args:  Sequence of sequences or mappings.  It is used as parameter.
        :return: Number of rows affected, if any.

        This method improves performance on multiple-row INSERT and
        REPLACE. Otherwise it is equivalent to looping over args with
        execute().
        Nr���c3s|]}�j�|�VqdS)N)rD)r,r6)r@rrrr.�sz%Cursor.executemany.<locals>.<genexpr>)
�RE_INSERT_VALUES�match�group�rstrip�_do_execute_many�max_stmt_lengthrr*�sumr)rr@r#�mZq_prefixZq_valuesZ	q_postfixr)r@rr�executemany�s
zCursor.executemanyc
CsN|j�}|j}t|t�r"|j|�}tr:t|t�r:|j|�}t|t�rN|j|�}t|�}	t|�}||t|�|�}
t|
t�r�tr�|
j|�}
n|
j|d�}
|	|
7}	d}x�|D]�}||||�}
t|
t�r�tr�|
j|�}
n|
j|d�}
t	|	�t	|
�t	|�d|k�r||j
|	|�7}t|�}	n|	d7}	|	|
7}	q�W||j
|	|�7}||_|S)N�surrogateescaperr�,)rr?r/rr0r�	bytearray�iter�next�lenrDr)
r�prefix�valuesZpostfixr#rMr*rr>Zsqlr-�rowsr6rrrrL�s>







"
zCursor._do_execute_manycs�|j��|rFdj���|jddj��fdd�t|�D���|j�d�dj�fdd�tt|��D��f}|j|�||_|S)	aExecute stored procedure procname with args

        procname -- string, name of procedure to execute on server

        args -- Sequence of parameters to use with procedure

        Returns the original args.

        Compatibility warning: PEP-249 specifies that any modified
        parameters must be returned. This is currently impossible
        as they are only available by storing them in a server
        variable and then retrieved by a query. Since stored
        procedures return zero or more result sets, there is no
        reliable way to get at OUT or INOUT parameters via callproc.
        The server variables are named @_procname_n, where procname
        is the parameter above and n is the position of the parameter
        (from zero). Once all result sets generated by the procedure
        have been fetched, you can issue a SELECT @_procname_0, ...
        query using .execute() to get any OUT or INOUT values.

        Compatibility warning: The act of calling a stored procedure
        itself creates an empty result set. This appears after any
        result sets generated by the procedure. This is non-standard
        behavior with respect to the DB-API. Be sure to use nextset()
        to advance through all result sets; otherwise you may get
        disconnected.
        z@_{0}_%d=%szSET %s�,c3s$|]\}}�|�j|�fVqdS)N)r>)r,�indexr6)r�fmtrrr.sz"Cursor.callproc.<locals>.<genexpr>zCALL %s(%s)csg|]}d�|f�qS)z@_%s_%dr)r,�i)�procnamerr�
<listcomp>sz#Cursor.callproc.<locals>.<listcomp>)	r�formatrB�join�	enumeraterrrVr)rr^r#�qr)rr\r^r�callproc�s

zCursor.callproccCsD|j�|jdks"|jt|j�kr&dS|j|j}|jd7_|S)zFetch the next rowNr)r rrrV)rrCrrr�fetchoneszCursor.fetchonecCsL|j�|jdkrfS|j|p"|j}|j|j|�}t|t|j��|_|S)zFetch several rowsN)r rrr�minrV)r�size�endrCrrr�	fetchmanys
zCursor.fetchmanycCsD|j�|jdkrfS|jr.|j|jd�}n|j}t|j�|_|S)zFetch all the rowsN)r rrrV)rrCrrr�fetchall$s
zCursor.fetchall�relativecCsh|j�|dkr|j|}n|dkr*|}ntjd|��d|koPt|j�kns^td��||_dS)Nrk�absolutezunknown scroll mode %srzout of range)r rr	rrVr�
IndexError)r�value�mode�rrrr�scroll0sz
Cursor.scrollcCs.|j�}||_|j�|j|�|j�|jS)N)r�_last_executedr'r@r(r)rrcrrrrrB=s
z
Cursor._querycCs(d|_d|_d|_d|_d|_d|_dS)Nr)rrrr
�	lastrowidr)rrrrr'EszCursor._clear_resultcCs8|j�}|j|_}|j|_|j|_|j|_|j|_dS)N)	rrZ
affected_rowsrr
Z	insert_idrsrYr)rrrCrrrr(NszCursor._do_get_resultcCst|jd�S)N)rTre)rrrr�__iter__XszCursor.__iter__)F)N)N)N)N)rk)*�__name__�
__module__�__qualname__�__doc__rMrrrrrr r"r$r%r)rr+r?rArDrPrLrdrerirjrqrBr'r(rtr	�Warning�ErrorZInterfaceErrorZ
DatabaseErrorZ	DataErrorZOperationalErrorZIntegrityErrorZ
InternalErrorr�NotSupportedErrorrrrrr
sL	





$*	



	
r
cs(eZdZeZ�fdd�Zdd�Z�ZS)�DictCursorMixincsztt��j�g}�jrVx6�jjD]*}|j}||krB|jd|}|j|�q"W|�_	|rv�j
rv�fdd��j
D��_
dS)N�.csg|]}�j|��qSr)r")r,rp)rrrr_wsz2DictCursorMixin._do_get_result.<locals>.<listcomp>)�superr|r(r
r�fields�nameZ
table_name�append�_fieldsr)rr�fr�)�	__class__)rrr(ks
zDictCursorMixin._do_get_resultcCs|dkrdS|jt|j|��S)N)�	dict_type�zipr�)rr!rrrr"yszDictCursorMixin._conv_row)rurvrwr<r�r(r"�
__classcell__rr)r�rr|gsr|c@seZdZdZdS)�
DictCursorz.A cursor which returns results as a dictionaryN)rurvrwrxrrrrr�sr�c@speZdZdZdd�Zdd�ZeZdd�Zdd	�Zd
d�Z	dd
�Z
dd�Zdd�Zdd�Z
ddd�Zddd�ZdS)�SSCursora�
    Unbuffered Cursor, mainly useful for queries that return a lot of data,
    or for connections to remote servers over a slow network.

    Instead of copying every row of data into a buffer, this will fetch
    rows as needed. The upside of this is the client uses much less memory,
    and rows are returned much faster when traveling over a slow network
    or if the result set is very big.

    There are limitations, though. The MySQL protocol doesn't support
    returning the total number of rows, so the only way to tell how many rows
    there are is to iterate over every row returned. Also, it currently isn't
    possible to scroll backwards, as only the current row is held in memory.
    cCs|S)Nr)rr!rrrr"�szSSCursor._conv_rowcCsR|j}|dkrdS|jdk	r2|j|jkr2|jj�zx|j�r@q6WWdd|_XdS)N)rrZ_finish_unbuffered_queryr)rrrrrr�s

zSSCursor.closecCs2|j�}||_|j�|j|dd�|j�|jS)NT)r&)rrrr'r@r(r)rrcrrrrrB�szSSCursor._querycCs|jdd�S)NT)r&)r))rrrrr�szSSCursor.nextsetcCs|j|jj��S)z
Read next row)r"rZ_read_rowdata_packet_unbuffered)rrrr�	read_next�szSSCursor.read_nextcCs.|j�|j�}|dkrdS|jd7_|S)zFetch next rowNr)r r�r)rr!rrrre�szSSCursor.fetchonecCst|j��S)z�
        Fetch all, as per MySQLdb. Pretty useless for large queries, as
        it is buffered. See fetchall_unbuffered(), if you want an unbuffered
        generator version of this method.
        )r2�fetchall_unbuffered)rrrrrj�szSSCursor.fetchallcCst|jd�S)z�
        Fetch all, implemented as a generator, which isn't to standard,
        however, it doesn't make sense to return everything in a list, as that
        would use ridiculous memory for large result sets.
        N)rTre)rrrrr��szSSCursor.fetchall_unbufferedcCs|j�S)N)r�)rrrrrt�szSSCursor.__iter__NcCsZ|j�|dkr|j}g}x:t|�D].}|j�}|dkr:P|j|�|jd7_q$W|S)z
Fetch manyNr)r rrr�r�r)rrgrYr]r!rrrri�s
zSSCursor.fetchmanyrkcCs�|j�|dkrL|dkr"tjd��xt|�D]}|j�q,W|j|7_nV|dkr�||jkrhtjd��||j}xt|�D]}|j�q|W||_ntjd|��dS)Nrkrz0Backwards scrolling not supported by this cursorrlzunknown scroll mode %s)r r	r{rr�rr)rrnro�_rhrrrrq�s"

zSSCursor.scroll)N)rk)rurvrwrxr"r�__del__rBrr�rerjr�rtrirqrrrrr��s	
r�c@seZdZdZdS)�SSDictCursorz;An unbuffered cursor, which returns results as a dictionaryN)rurvrwrxrrrrr��sr�z]\s*((?:INSERT|REPLACE)\b.+\bVALUES?\s*)(\(\s*(?:%s|%\(.+\)s)\s*(?:,\s*(?:%s|%\(.+\)s)\s*)*\))z|\s*((?:INSERT|REPLACE)\b.+\bVALUES?\s*)(\(\s*(?:%s|%\(.+\)s)\s*(?:,\s*(?:%s|%\(.+\)s)\s*)*\))(\s*(?:ON DUPLICATE.*)?);?\s*\Z)Z
__future__rr�	functoolsr�reZ_compatrrrrGr	�compile�
IGNORECASE�DOTALLrH�objectr
r|r�r�r�rrrr�<module>sUu