File: //var/opt/nydus/ops/mysql/connector/aio/__pycache__/cursor.cpython-39.pyc
a
�,�h � � @ s� d Z dgZddlZddlZddlmZ ddlmZ ddlm Z m
Z
m Z m Z m
Z
mZmZmZmZmZ ddlmZ dd lmZmZmZmZmZmZm Z m Z m Z m Z m Z m!Z! dd
l"m#Z#m$Z$m%Z%m&Z&m'Z' dd l(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2 d d
l3m4Z4m5Z5m6Z6 dZ7G dd� d�Z8e9e e9e9f e9d�dd�Z:G dd� de6�Z;G dd� de;�Z<G dd� de;�Z=G dd� de<�Z>G dd � d e;�Z?G d d � d e;�Z@G d d � d e?e<�ZAG d!d"� d"e@e<�ZBG d#d$� d$e;�ZCG d%d&� d&e?eC�ZDG d'd(� d(e@eC�ZEG d)d*� d*eC�ZFdS )+z0Implementation of cursor classes in pure Python.� MySQLCursor� N)�
namedtuple)�Decimal)
�Any�AsyncGenerator�Dict�Iterator�List�NoReturn�Optional�Sequence�Tuple�Union� )�
ServerFlag) � MAX_RESULTS�RE_PY_MAPPING_PARAM� RE_PY_PARAM�RE_SQL_COMMENT�RE_SQL_FIND_PARAM�RE_SQL_INSERT_STMT�RE_SQL_INSERT_VALUES�RE_SQL_ON_DUPLICATE� RE_SQL_PYTHON_CAPTURE_PARAM_NAME�RE_SQL_PYTHON_REPLACE_PARAM�is_eol_comment�parse_multi_statement_query)�Error�InterfaceError�NotSupportedError�ProgrammingError�get_mysql_exception)
�DescriptionType�
EofPacketType�ParamsDictType�ParamsSequenceOrDictType�ParamsSequenceType�
ResultType� RowItemType�RowType�
StrOrBytes� WarningType� )�NAMED_TUPLE_CACHE�MySQLConnectionAbstract�MySQLCursorAbstractzNo result set to fetch fromc @ sH e Zd ZdZee dd�dd�Zejed�dd�Z e
e d �d
d ��Z dS ) �_ParamSubstitutorz+Substitute parameters into a SQL statement.N��params�returnc C s || _ d| _d S �Nr )r2 �index)�selfr2 � r7 �M/opt/nydus/tmp/pip-target-wkfpz8uv/lib64/python/mysql/connector/aio/cursor.py�__init___ s z_ParamSubstitutor.__init__��matchobjr3 c C sF | j }| j d7 _ zt| j| �W S ty@ td�d �Y n0 d S )Nr, z+Not enough parameters for the SQL statement)r5 �bytesr2 �
IndexErrorr )r6 r; r5 r7 r7 r8 �__call__c s ��z_ParamSubstitutor.__call__�r3 c C s t | j�| j S )z<Return the number of parameters remaining to be substituted.)�lenr2 r5 �r6 r7 r7 r8 � remainingm s z_ParamSubstitutor.remaining)
�__name__�
__module__� __qualname__�__doc__r r< r9 �re�Matchr> �property�intrB r7 r7 r7 r8 r0 \ s
r0 )�bytestr�
value_dictr3 c s& t jtd�� fdd� }t�|| �}|S )ah
>>> _bytestr_format_dict(b'%(a)s', {b'a': b'foobar'})
b'foobar
>>> _bytestr_format_dict(b'%%(a)s', {b'a': b'foobar'})
b'%%(a)s'
>>> _bytestr_format_dict(b'%%%(a)s', {b'a': b'foobar'})
b'%%foobar'
>>> _bytestr_format_dict(b'%(x)s %(y)s',
... {b'x': b'x=%(y)s', b'y': b'y=%(x)s'})
b'x=%(y)s y=%(x)s'
r: c sV d}| � � }|d dkr d}|d dkr8|d }� | }|du rRtd|d � ���|S )zReplace pattern.NZconversion_type� %� sZ mapping_keyz Unsupported conversion_type: )� groupdict�
ValueError)r; �value�groups�key�rL r7 r8 �replace� s �z%_bytestr_format_dict.<locals>.replace)rG rH r< r �sub)rK rL rU �stmtr7 rT r8 �_bytestr_format_dicts s
rX c @ s\ e Zd ZdZed�dd�Zed�dd�Zee e
e e
e f f d�dd �Z
eee e
e f d
f d�d d �Zeed
f eed
f d�dd�ZdDeee d�dd�Zed
d�dd�Zd
d�dd�Zed
d�dd �Zd
d�d d �Zed
d �d d!�Z d
d�d"d#�Z ed�d$d%�Z d
d�d&d'�Z dEe e e!e" e e#e"f f e
d)�d*d+�Z$ee%e& d�d,d-�Z'e#e!e( ee
d.�d/d0�Z)e*e d�d1d2�Z+dFe#e!e" ee e e#e,f ef d3�d4d5�Z-dGe e e!e" e e#e"f f ed
d6�d7d8�Z.dHe e e!e" e e#e"f f eed
f d)�d9d:�Z/e#e!e d
d.�d;d<�Z0ee d�d=d>�Z1e%e d�d?d@�Z2dIee3 e%e dA�dBdC�Z4d
S )Jr a� Default cursor for interacting with MySQL.
This cursor will execute statements and handle the result. It will not automatically
fetch all rows.
MySQLCursor should be inherited whenever other functionallity is required.
An example would to change the fetch* member functions to return dictionaries instead
of lists of values.
Implements the Python Database API Specification v2.0 (PEP-249).
r? c � s" | � � I d H }|d ur|S t�d S �N)�fetchone�StopAsyncIteration�r6 �resr7 r7 r8 � __anext__� s zMySQLCursor.__anext__c � s> | j s
dS | j �| � | j �� I d H | �� I d H d | _ dS )NFT)� _connectionZ
remove_cursor�handle_unread_result�
_reset_resultrA r7 r7 r8 �close� s zMySQLCursor.closer1 c
C s� i }zf| j jj}| j jj}| j jj}|�� D ]:\}}|}||�}||�}t|t�sZ||�}|||�� < q,W n4 t y� } z t
d| � ��| �W Y d} ~ n
d} ~ 0 0 |S )z-Process query parameters given as dictionary.z'Failed processing pyformat-parameters; N) r_ � converter�to_mysql�escape�quote�items�
isinstancer �encode� Exceptionr )
r6 r2 r] rd re rf rS rQ �conv�errr7 r7 r8 �_process_params_dict� s&
��z MySQLCursor._process_params_dict.c
s� �dd� }z^| j jj�| j jj� | j jj��fdd�|D �}� fdd�|D �}��fdd�t|�D �}W n4 ty� } z td|� ��|�W Y d}~n
d}~0 0 t|�S )zProcess query parameters.Nc s g | ] }� |��qS r7 r7 ��.0rQ )rd r7 r8 �
<listcomp>� � z/MySQLCursor._process_params.<locals>.<listcomp>c s g | ] }� |��qS r7 r7 rn )re r7 r8 rp � rq c s* g | ]"\}}t � | t�s"�|�n|�qS r7 )rh r )ro �irQ )r2 rf r7 r8 rp � s �z%Failed processing format-parameters; ) r_ rc rd re rf � enumeraterj r �tuple)r6 r2 �resultrl r7 )re r2 rf rd r8 �_process_params� s"
�
��zMySQLCursor._process_paramsN)�
query_iterr3 c C s� t | jd�}d| _|r |�� nd}|2 zj3 dH W }| �� I dH | �|�I dH t|�rZq&|�� | _| V |�� �d�r�d|vr&|r�|�� nd}q&6 dS )a� Generator returns MySQLCursor objects for multiple statements
This method is only used when multiple statements are executed
by the `cursor.execute(multi_stmt_query, multi=True)` method.
It matches the given `query_iter` (result of `MySQLConnection.cmd_query_iter()`)
and the list of statements that were executed.
How does this method work? To properly map each statement (stmt) to a result,
the following facts must be considered:
1. Read operations such as `SELECT` produce a non-empty result
(calling `next(query_iter)` gets a result that includes at least one column).
2. Write operatios such as `INSERT` produce an empty result
(calling `next(query_iter)` gets a result with no columns - aka empty).
3. End-of-line (EOL) comments do not produce a result, unless is the last stmt
in which case produces an empty result.
4. Calling procedures such as `CALL my_proc` produce a sequence `(1)*0` which
means it may produce zero or more non-empty results followed by just one
empty result. In other words, a callproc stmt always terminates with an
empty result. E.g., `my_proc` includes an update + select + select + update,
then the result sequence will be `110` - note how the write ops results get
annulated, just the read ops results are produced. Other examples:
* insert + insert -> 0
* select + select + insert + select -> 1110
* select -> 10
Observe how 0 indicates the end of the result sequence. This property is
vital to know what result corresponds to what callproc stmt.
In this regard, the implementation is composed of:
1. Parsing: the multi-statement is broken down into single statements, and then
for each of these, leading white spaces are removed (including
jumping line, vertical line, tab, etc.). Also, EOL comments are removed from
the stream, except when the comment is the last statement of the
multi-statement string.
2. Mapping: the facts described above as used as "game rules" to properly match
statements and results. In case, if we run out of statements before running out
of results we use a sentinel named "stmt_overflow!" to indicate that the mapping
went wrong.
Acronyms
1: a non-empty result
2: an empty result
)Z
multi_stmtNs stmt_overflow!s CALL�columns) r � _executed�popleftra �_handle_resultr �rstrip�upper�
startswith)r6 rw Z
executed_listrW ru r7 r7 r8 �
_execute_iter� s /
zMySQLCursor._execute_iterF��rawr3 c � s� | j js dS d}| jdkr<| j j| j| j|d�I dH \}}n
| j\}}|r�| j j| j| j|d�I dH | _| jd }|dur�| �|�I dH | jdkr�d| _n| jd7 _|r�| �|�I dH |S )�&Return the next row in the result set.N�NN)�binaryrx r� r, ���)r_ �
unread_result�_nextrowZget_row�_binary� description� _handle_eof� _rowcount)r6 r� �row�eofr7 r7 r8 �
_fetch_row s*
�
�
zMySQLCursor._fetch_row�ru r3 c � sj t |t�std��d|v r<|d | _d| j_| �� I dH n*d|v r^d| j_| �|�I dH ntd��dS )z�Handle the result after a command was send.
The result can be either an OK-packet or a dictionary containing column/eof
information.
Raises:
InterfaceError: When result is not a dict() or result is invalid.
zResult was not a dict()rx TN�
affected_rowsFzInvalid result)rh �dictr � _descriptionr_ r� �_handle_resultset�_handle_noresultset�r6 ru r7 r7 r8 r{ 6 s
zMySQLCursor._handle_resultc � s dS )��Handle the result set.
This method handles the result set and is called after reading and storing
column information in _handle_result(). For non-buffering cursors, this method
is usually doing nothing.
Nr7 rA r7 r7 r8 r� N s z MySQLCursor._handle_resultset�r] r3 c
� sn z"|d | _ |d | _|d | _W n8 ttfyZ } z td|� ��d�W Y d}~n
d}~0 0 | �� I dH dS )z�Handle result of execute() when there is no result set.
Raises:
ProgrammingError: When failing handling a non-resultset.
r� Z insert_id�
warning_countz Failed handling non-resultset; N)r� Z_last_insert_id�_warning_count�KeyError� TypeErrorr �_handle_warnings)r6 r] rl r7 r7 r8 r� V s
&z MySQLCursor._handle_noresultsetc � sn | j jr | jr | �� I dH | _| js(dS t| jd d | jd d | j j d�}| j jr\|�tj|dd� dS )z�Handle possible warnings after all results are consumed.
Raises:
Error: Also raises exceptions if raise_on_warnings is set.
Nr r, r )�warning� )�
stacklevel) r_ Z get_warningsr� �_fetch_warnings� _warningsr! Zraise_on_warnings�warnings�warn)r6 rl r7 r7 r8 r� e s �z MySQLCursor._handle_warnings�r� r3 c � s* d| j _d| _|d | _| �� I dH dS )zHandle EOF packet.Fr� r� N)r_ r� r� r� r� �r6 r� r7 r7 r8 r� | s
zMySQLCursor._handle_eofc � sB d| _ d| _d| _d| _g | _g | _d| _d| _| �� I dH dS )� Reset the cursor to default.Nr r� r� ) r� r� r� ry �_executed_list�_stored_resultsr� r� �resetrA r7 r7 r8 ra � s zMySQLCursor._reset_resultc C s$ z
| j jW S ty Y dS 0 dS )z(Check whether there is an unread result.FN)r_ r� �AttributeErrorrA r7 r7 r8 �_have_unread_result� s
z MySQLCursor._have_unread_resultc C s | j du rtt��dS )z�Check if the statement has been executed.
Raises:
InterfaceError: If the statement has not been executed.
N)ry r �ERR_NO_RESULT_TO_FETCHrA r7 r7 r8 �_check_executed� s
zMySQLCursor._check_executedr7 �� operationr2 r3 c
C s� z$t |ttf�r|n |�| jj�}W n6 ttfyZ } ztt |��|�W Y d}~n
d}~0 0 |r�t |t
�r|t || � |��}nXt |t
tf�r�t| �|��}t�||�}|jdkr�td��n tdt|�j� d|� d���|S )a' Prepare SQL statement for execution.
Converts the SQL statement to bytes and replaces the parameters in the
placeholders.
Raises:
ProgrammingError: On converting to bytes, missing parameters or invalid
parameters type.
Nr �1Not all parameters were used in the SQL statementz Could not process parameters: �(z)), it must be of type list, tuple or dict)rh r< � bytearrayri r_ �python_charset�UnicodeDecodeError�UnicodeEncodeErrorr �strr� rX rm �listrt r0 rv r rV rB �typerC )r6 r� r2 rW rl �psubr7 r7 r8 �_prepare_statement� s* � �$
��z MySQLCursor._prepare_statementc � sv g }| j jdd�I dH 4 I dH �4}|�d�I dH |�� I dH }W d �I dH qj1 I dH s`0 Y |rr|S dS )z%Fetch warnings doing a SHOW WARNINGS.F�r� Nz
SHOW WARNINGS)r_ �cursor�execute�fetchall)r6 ru �curr7 r7 r8 r� � s
8zMySQLCursor._fetch_warnings�r� �
seq_paramsr3 c
C s| t jtd�dd�}t �tdt �t||��}t �t|�}|sBtd��|� d��
| j j �}g }z�|�
| j j �}|D ]Z} |}t
| t�r�t|| �| ��}n,t| �| ��}
t�|
|�}|
jdkr�td��|�|� qn||v r�|�|d �|�d�}|| _|W S W d
S ttf�y. } ztt| ��| �W Y d
} ~ nRd
} ~ 0 t �yB � Y n6 t �yv } z td | � ��d
�W Y d
} ~ n
d
} ~ 0 0 d
S ) zImplements multi row insert)�matchr3 c S s | � d�rdS | � d�S )a Remove comments from INSERT statements.
This function is used while removing comments from INSERT
statements. If the matched string is a comment not enclosed
by quotes, it returns an empty string, else the string itself.
r, � r )�group)r� r7 r7 r8 �remove_comments� s
z2MySQLCursor._batch_insert.<locals>.remove_commentsr� zAFailed rewriting statement for multi-row INSERT. Check SQL syntaxr, r r� � ,N� Failed executing the operation; ) rG rH r� rV r r �searchr r r� ri r_ r� rh r� rX rm r0 rv r rB r �appendrU �joinry r� r� r rj ) r6 r� r� r� �tmp�matches�fmt�valuesrW r2 r� rl r7 r7 r8 �
_batch_insert� sJ � �
� "zMySQLCursor._batch_insertc C s
t | j�S )z�Returns an iterator for stored results.
This method returns an iterator over results which are stored when callproc()
is called. The iterator will provide MySQLCursorBuffered instances.
)�iterr� rA r7 r7 r8 �stored_results s zMySQLCursor.stored_results��procname�argsr3 c
� s� |rt |t�std��t |ttf�s,td��g | _g }�z g }g }|�d�d }|r�g }t|�D ]p\}} d|� d|d � �}
|�|
� t | t�r�|�d|
� d | d � d
�� |�| d � q`|�|
� |�| � q`d � d
d� |D ��} | �
d| � �|�I dH d|� dd � |�� d
�} | j j }
| j �
| �2 z�3 dH W }d| j _ t | ttf��rTt}n(t | ttf��rjt}n| j�rxt}nt}|| j �� �}d| � d
�|_|�|�I dH |jdu�r�|j| _d|v �r*|�|� �q*6 |
| j _ |�r8dd� t|dd� |D ��D �}dd � |�� �}| �
|�I dH || _| �� I dH W S || _t� W S t �yZ � Y n6 t �y� } z t d|� ��d�W Y d}~n
d}~0 0 dS )a� Calls a stored procedure with the given arguments
The arguments will be set during this session, meaning
they will be called like _<procname>__arg<nr> where
<nr> is an enumeration (+1) of the arguments.
Coding Example:
1) Defining the Stored Routine in MySQL:
CREATE PROCEDURE multiply(IN pFac1 INT, IN pFac2 INT, OUT pProd INT)
BEGIN
SET pProd := pFac1 * pFac2;
END
2) Executing in Python:
args = (5, 5, 0) # 0 is to hold pprod
await cursor.callproc('multiply', args)
print(await cursor.fetchone())
For OUT and INOUT parameters the user should provide the
type of the parameter as well. The argument should be a
tuple with first item as the value of the parameter to pass
and second argument the type of the argument.
In the above example, one can call callproc method like:
args = (5, 5, (0, 'INT'))
await cursor.callproc('multiply', args)
The type of the argument given in the tuple will be used by
the MySQL CAST function to convert the values in the corresponding
MySQL type (See CAST in MySQL Reference for more information)
Does not return a value, but a result set will be
available when the CALL-statement execute successfully.
Raises exceptions when something is wrong.
zprocname must be a stringzargs must be a sequence�.r� �@_Z_argr, z CAST(� AS �)r �,c s s | ]}|� d �V qdS )z=%sNr7 �ro �argr7 r7 r8 � <genexpr>U rq z'MySQLCursor.callproc.<locals>.<genexpr>zSET NzCALL r� Fz
(a result of rx c S s g | ]\}}|� d |� ��qS )r� r7 )ro �name�aliasr7 r7 r8 rp y s �z(MySQLCursor.callproc.<locals>.<listcomp>c S s g | ]}|� d ��qS )r� )�lstripr� r7 r7 r8 rp | rq zSELECT z Failed calling stored routine; ) rh r� rP rt r� r� �splitrs r� r� r� r_ �can_consume_results�cmd_query_iter�MySQLCursorDict�MySQLCursorBufferedDict�MySQLCursorNamedTuple� MySQLCursorBufferedNamedTuple�_raw�MySQLCursorBufferedRaw�MySQLCursorBufferedZget_selfry r{ r� r� �ziprZ r rj r )r6 r� r� �results�argnames�argtypesZ procname_absZ argvalues�idxr� �argnameZ placeholders�callr� ru Z cursor_classr� �selectrl r7 r7 r8 �callproc sx (
�
��zMySQLCursor.callproc�r� r2 �multir3 c
� s� | j std��|sdS |r"td��| j �� I dH | �� I dH | �||�}|| _z"| �| j �|�I dH �I dH W n8 ty� } z | j j r�td�|�� W Y d}~n
d}~0 0 dS )am Executes the given operation.
Executes the given operation substituting any markers with the given parameters.
For example, getting all rows where id is 5:
await cursor.execute("SELECT * FROM t1 WHERE id = %s", (5,))
If the `multi`` parameter is used a `ProgrammingError` is raised. The method for
executing multiple statements is `executemulti()`.
If warnings where generated, and connection.get_warnings is True, then
self._warnings will be a list containing these warnings.
Raises:
ProgramingError: If multi parameter is used.
�Cursor is not connectedNz�The `multi` parameter cannot be used in the Connector/Python Asyncio implementation. Please use `executemulti()` for executing multiple statementsz7Use `executemulti()` when executing multiple statements)
r_ r r` ra r� ry r{ Z cmd_queryr Zhave_next_result)r6 r� r2 r� rW rl r7 r7 r8 r� � s. � "��zMySQLCursor.executec C sx | j std��|st�| j �� I dH | �� I dH | �||�}|| _g | _| �| j � |��2 z3 dH W }|V q^6 dS )��Execute multiple statements.
Executes the given operation substituting any markers with the given
parameters.
r� N)
r_ r r[ r` ra r� ry r� r r� )r6 r� r2 rW ru r7 r7 r8 � executemulti� s zMySQLCursor.executemultic
� s, |r|s dS | j �� I dH z t|�}W n. tyV } ztd�|�W Y d}~n
d}~0 0 t�t|�r�|srd| _dS | � ||�}|dur�|| _
| � |�I dH S d}zF|D ]<}| � ||�I dH | j r�| �
� r�| �� I dH || j7 }q�W n: ttf�y } z td|� ��d�W Y d}~n
d}~0 0 || _dS )a Prepare and execute a MySQL Prepared Statement many times.
This method will prepare the given operation and execute with each tuple found
the list seq_params.
If the cursor instance already had a prepared statement, it is first closed.
N�(Parameters for query must be an Iterabler r� )r_ r` r� r� r rG r� r r� r� ry r� � with_rowsr� r� rP )r6 r� r� �_rl rW �rowcntr2 r7 r7 r8 � executemany� s2 &zMySQLCursor.executemanyc � s | j du rtt��| �� I dH S )z�Return next row of a query result set.
Raises:
InterfaceError: If there is no result to fetch.
Returns:
tuple or None: A row from query result set.
N)ry r r� r� rA r7 r7 r8 rZ s
zMySQLCursor.fetchonec � s� | j du rtt��| jjs g S | j�� I dH \}}| jd rN|�d| jd � | �|�I dH t |�}|dkr~| j
dkr~d| _
| j
|7 _
|S )��Return all rows of a query result set.
Returns:
list: A list of tuples with all rows of a query result set.
Nr r� ) ry r r� r_ r� �get_rowsr� �insertr� r@ r� �r6 �rowsr� Zrowcountr7 r7 r8 r� s
zMySQLCursor.fetchall��sizer3 c � sH | � � g }|p| j}|dkrD|d8 }| �� I dH }|r|�|� q|S )a< Return the next set of rows of a query result set.
When no more rows are available, it returns an empty list.
The number of rows returned can be specified using the size argument, which
defaults to one.
Returns:
list: The next set of rows of a query result set.
r r, N)r� � arraysizerZ r� �r6 r r] Zcntr� r7 r7 r8 � fetchmany' s
zMySQLCursor.fetchmany)F)r7 )r7 )r7 F)r7 )N)5rC rD rE rF r) r^ �boolrb r$ r r< r r rm r& r
rv r r' r/ r r r� r{ r� r� r� r# r� ra r� r� r* r r r� r� r r+ r� r% r� r r� r( r� r� r� r� rZ r� rJ r r7 r7 r7 r8 r � sv
� �
� ? �� ) � 8 �� ~ �� 6 �
� � *
c s� e Zd ZdZedd�� fdd� Zdd�dd�Zdeee d
�d d �Z
dedd�dd�Z ee d�dd�Z e
e d�dd�Zeed�dd��Z� ZS )r� z+Cursor which fetches rows within execute().N��
connectionr3 c s t � �|� d | _d| _d S r4 )�superr9 �_rows� _next_row�r6 r �� __class__r7 r8 r9 @ s z MySQLCursorBuffered.__init__r? c � sD | j �� I dH \| _}t| j�| _| �|�I dH d| _d| j _dS )r� Nr F)r_ r� r r@ r� r� r r� r� r7 r7 r8 r� E s
z%MySQLCursorBuffered._handle_resultsetFr� c � s@ d}z| j | j }W n ttfy, Y dS 0 | jd7 _|S )r� Nr, )r r r= r� )r6 r� r� r7 r7 r8 r� R s z MySQLCursorBuffered._fetch_rowT��freer3 c � s
d| _ dS )r� N�r �r6 r r7 r7 r8 r� \ s zMySQLCursorBuffered.resetc � s | � � | �� I dH S ��yReturn next row of a query result set.
Returns:
tuple or None: A row from query result set.
N�r� r� rA r7 r7 r8 rZ ` s z MySQLCursorBuffered.fetchonec � s@ | j du s| jdu r tt��g }| j| jd� }t| j�| _|S �r� N)ry r r r� r r@ r\ r7 r7 r8 r� i s z MySQLCursorBuffered.fetchallc C s
| j d uS rY r rA r7 r7 r8 r� v s z MySQLCursorBuffered.with_rows)F)T)rC rD rE rF r. r9 r� r r r) r� r� rZ r r� rI r� �
__classcell__r7 r7 r r8 r� = s
r� c sL e Zd ZdZedd�� fdd� Zee d�dd�Ze e d�d d
�Z
� Z S ) �MySQLCursorRawzHSkip conversion from MySQL datatypes to Python types when fetching rows.Nr c s t � �|� d| _d S )NT)r
r9 r� r
r r7 r8 r9 ~ s zMySQLCursorRaw.__init__r? c � s | � � | jdd�I dH S )r Tr� Nr rA r7 r7 r8 rZ � s zMySQLCursorRaw.fetchonec � s� | � � | �� sg S | jjdd�I dH \}}| jd rH|�d| jd � | �|�I dH t|�}|dkrx| jdkrxd| _| j|7 _|S )r� Tr� Nr r� ) r� r� r_ r� r� r� r� r@ r� r r7 r7 r8 r� � s
zMySQLCursorRaw.fetchall) rC rD rE rF r. r9 r r) rZ r r� r r7 r7 r r8 r { s r c @ s0 e Zd ZU dZdZeed< eed�dd��ZdS )r� z�
Cursor which skips conversion from MySQL datatypes to Python types when
fetching rows and fetches rows within execute().
Tr� r? c C s
| j d uS rY r rA r7 r7 r8 r� � s z MySQLCursorBufferedRaw.with_rowsN) rC rD rE rF r� r �__annotations__rI r� r7 r7 r7 r8 r� � s
r� c s| e Zd ZdZd eeee eee e
f d�dd�Z eee e
f d�� fdd� Z eeee e
f d�� fd d
� Z
� ZS ) r� z�
Cursor fetching rows as dictionaries.
The fetch methods of this class will return dictionaries instead of tuples.
Each row is a dictionary that looks like:
row = {
"col1": value1,
"col2": value2
}
N��rowdata�descr3 c C s |rt t| j|��S dS )zWConvert a MySQL text result row to Python types
Returns a dictionary.
N)r� r� � column_names�r6 r r r7 r7 r8 �_row_to_python� s z MySQLCursorDict._row_to_pythonr? c � s | � t� �� I dH | j�S )zyReturn next row of a query result set.
Returns:
dict or None: A dict from query result set.
N)r r
rZ r� rA r r7 r8 rZ � s zMySQLCursorDict.fetchonec � s � fdd�t � �� I dH D �S )z�Return all rows of a query result set.
Returns:
list: A list of dictionaries with all rows of a query
result set where column names are used as keys.
c s g | ]}|r� � |� j��qS r7 �r r� �ro r� rA r7 r8 rp � s �z,MySQLCursorDict.fetchall.<locals>.<listcomp>N�r
r� rA r rA r8 r� � s
�zMySQLCursorDict.fetchall)N)rC rD rE rF r) r r r" r r� r( r rZ r� r r7 r7 r r8 r� � s �
� r� c sd e Zd ZdZd eeee ee d�dd�Zee d�� fdd� Z eee d�� fd d
� Z
� Z S ) r� z�
Cursor fetching rows as named tuple.
The fetch methods of this class will return namedtuples instead of tuples.
Each row is returned as a namedtuple and the values can be accessed as:
row.col1, row.col2
Nr c C sP |}|rLt | j�}z t| }W n$ tyB td|�}|t|< Y n0 ||� S dS )zXConvert a MySQL text result row to Python types
Returns a named tuple.
ZRowN)rt r r- r� r )r6 r r r� rx Z named_tupler7 r7 r8 r � s
z$MySQLCursorNamedTuple._row_to_pythonr? c � s6 t � �� I dH }|sdS t| jd�r2| �|| j�S |S )r Nrc )r
rZ �hasattrr_ r r� �r6 r� r r7 r8 rZ � s
��z MySQLCursorNamedTuple.fetchonec � s � fdd�t � �� I dH D �S )r� c s g | ]}|r� � |� j��qS r7 r! r" rA r7 r8 rp s �z2MySQLCursorNamedTuple.fetchall.<locals>.<listcomp>Nr# rA r rA r8 r� s
�z MySQLCursorNamedTuple.fetchall)N) rC rD rE rF r) r r r" r rZ r� r r7 r7 r r8 r� � s �
� r� c @ sH e Zd ZdZeeeef d�dd�Ze eeeef d�dd�Z
dS )r� z8
Buffered Cursor fetching rows as dictionaries.
r? c � s, | � � | �� I dH }|r(| �|| j�S dS r �r� r� r r� r% r7 r7 r8 rZ s
z MySQLCursorBufferedDict.fetchonec � sZ | j du s| jdu r tt��g }| j| jd� D ]}|�| �|| j�� q0t| j�| _|S r � ry r r r� r r� r r� r@ �r6 r] r� r7 r7 r8 r� $ s z MySQLCursorBufferedDict.fetchallN) rC rD rE rF r r r� r( rZ r r� r7 r7 r7 r8 r� s r� c @ s8 e Zd ZdZee d�dd�Zeee d�dd�ZdS )r� z7
Buffered Cursor fetching rows as named tuple.
r? c � s, | � � | �� I dH }|r(| �|| j�S dS r r& r% r7 r7 r8 rZ 8 s
z&MySQLCursorBufferedNamedTuple.fetchonec � sZ | j du s| jdu r tt��g }| j| jd� D ]}|�| �|| j�� q0t| j�| _|S r r' r( r7 r7 r8 r� D s z&MySQLCursorBufferedNamedTuple.fetchallN) rC rD rE rF r r) rZ r r� r7 r7 r7 r8 r� 3 s r� c sX e Zd ZdZed�� fdd� Zd1edd�dd �Zedd
�� fd d � Z e
dd
�dd�Z e dd�� fdd� Z
d2eeed�dd�Zd3eeee eeef f eedf d�dd�Zdd�� fdd � Zd4eeed �d d �Zedd �d!d"�Zd5ee e edd$�d%d&�Z eee dd'�d(d)�Z e e! d�d*d+�Z"d6e e
e#e! d,�d-d.�Z$e#e! d�d/d0�Z%� Z&S )7�MySQLCursorPreparedz&Cursor using MySQL Prepared Statements)r c s: t � �|� d | _d| _d | _d| _d | _d| _d| _d S )Nr TF) r
r9 r r � _preparedr� � _have_result�_last_row_sent�_cursor_existsr
r r7 r8 r9 V s z MySQLCursorPrepared.__init__TNr c � sL | j r<z | j�| j d �I d H W n ty4 Y n0 d | _ d| _d| _d S )N� statement_idF)r* r_ �cmd_stmt_closer r, r- r r7 r7 r8 r� ` s zMySQLCursorPrepared.resetr� c � s0 | � |�d|�dd��� t� �|�I d H d S �N� status_flag�
server_statusr )�_handle_server_status�getr
r� r\ r r7 r8 r� k s �z'MySQLCursorPrepared._handle_noresultset)�flagsr3 c C s$ |t j@ dk| _|t j@ dk| _d S r4 )r ZSTATUS_CURSOR_EXISTSr- ZSTATUS_LAST_ROW_SENTr, )r6 r5 r7 r7 r8 r3 q s z)MySQLCursorPrepared._handle_server_statusr� c � s0 | � |�d|�dd��� t� �|�I d H d S r0 )r3 r4 r
r� r� r r7 r8 r� u s �z MySQLCursorPrepared._handle_eofr7 r� c � s
t � �dS )zQCalls a stored procedue
Not supported with MySQLCursorPrepared.
N�r )r6 r� r� r7 r7 r8 r� { s z MySQLCursorPrepared.callprocr� c � s
t � �dS )r� Nr6 )r6 r� r2 r7 r7 r8 r� � s
z MySQLCursorPrepared.executemultir? c � s" | � � I dH t� �� I dH dS )zzClose the cursor
This method will try to deallocate the prepared statement and close
the cursor.
N)r� r
rb rA r r7 r8 rb � s zMySQLCursorPrepared.closer c C s dS )z�Convert row data from MySQL to Python types
The conversion is done while reading binary data in the protocol module.
Nr7 r r7 r7 r8 r � s z"MySQLCursorPrepared._row_to_pythonr� c � s� t |t�r*d| j_d| _| �|�I dH nV|d | _d| j_d| _d|d v rb| �|d d � n d|d v r�| �|d d � dS )z Handle result after executionFNr, Tr1 r r2 )rh r� r_ r� r+ r� r� r3 r� r7 r7 r8 r{ � s
z"MySQLCursorPrepared._handle_resultFr� c
� sv | j std��|sdS | j j}|dkr*d}t|t�svz|�|�}W n2 tyt } ztt|��|�W Y d}~n
d}~0 0 t� t�r�t� t
|�}zt � fdd�|D ��� W n. t y� } ztd�|�W Y d}~n
d}~0 0 t�
td|�}|| ju�r�| j�r| j �| jd �I dH || _z|�|�}W n4 t�yX } ztt|��|�W Y d}~n
d}~0 0 d
|v �rrt�
td |�}z| j �|�I dH | _W n t�y� d| _� Y n0 | j �| jd �I dH | jd �rֈ �s�dS � �r0t� t tf��s td
dt� �j� d� � d�d��t| jd �t� �k�r0td
dd��� du �r>d� | j j | jd � | jd d�I dH }| � |�I dH dS )a4 Prepare and execute a MySQL Prepared Statement
This method will prepare the given operation and execute it using
the optionally given parameters.
If the cursor instance already had a prepared statement, it is
first closed.
Note: argument "multi" is unused.
r� NZutf8mb4�utf8c 3 s | ]}� | V qd S rY r7 )ro rS �r2 r7 r8 r� � rq z.MySQLCursorPrepared.execute.<locals>.<genexpr>z6Not all placeholders were found in the parameters dict�?r. s %s� ?�
parametersi� z Incorrect type of argument: r� zP), it must be of type tuple or list the argument given to the prepared statement)�errno�msgz:Incorrect number of arguments executing prepared statementr7 )�datar; ) r_ r �charsetrh r� �decoder� r� rG �findallr rt r� rV r ry r* r/ ri r� r Zcmd_stmt_preparer Zcmd_stmt_resetr� r� rC r@ Zcmd_stmt_executer{ )r6 r� r2 r� r? rl Zreplacement_keysr] r7 r8 r8 r� � s�
$
�� $
����
� zMySQLCursorPrepared.executer� c
� s� |r|s dS | j �� I dH z t|�}W n. tyV } ztd�|�W Y d}~n
d}~0 0 d}zF|D ]<}| �||�I dH | jr�| �� r�| �� I dH || j 7 }qbW n8 t
tfy� } z t d|� ��d�W Y d}~n
d}~0 0 || _ dS )aD Prepare and execute a MySQL Prepared Statement many times
This method will prepare the given operation and execute with each
tuple found the list seq_params.
If the cursor instance already had a prepared statement, it is
first closed.
executemany() simply calls execute().
Nr� r r� ) r_ r` r� r� r r� r� r� r� r� rP r )r6 r� r� r� rl r� r2 r7 r7 r8 r� s" &z MySQLCursorPrepared.executemanyc � s8 | � � | jr&| j�| jd �I dH | �� I dH p6dS )r r. N)r� r- r_ �cmd_stmt_fetchr* r� rA r7 r7 r8 rZ + s z MySQLCursorPrepared.fetchoner c � sP | � � g }|p| j}|dkrL| �� rL|d8 }| �� I dH }|r|�|� q|S )�< Return the next set of rows of a query result set.
When no more rows are available, it returns an empty list.
The number of rows returned can be specified using the size argument,
which defaults to one.
Returns:
list: The next set of rows of a query result set.
r r, N)r� r r� r� r� r r7 r7 r8 r 6 s
z MySQLCursorPrepared.fetchmanyc � s� | � � g }| jd r&|�| jd � | �� r�| jrN| j�| jd t�I dH | jj | j
| j d�I dH \}}|� |� | �
|�I dH q&t|�| _|S )r� r r. N)r� rx )r� r� r� r� r- r_ rB r* r r� r� r� �extendr� r@ r� �r6 r r� r� r7 r7 r8 r� J s
�
�
z MySQLCursorPrepared.fetchall)T)r7 )r7 )N)NF)N)'rC rD rE rF r. r9 r r� r' r� rJ r3 r# r� r r
r� r* r r r r� r r/ r� rb r r{ r r% r� r& r� r) rZ r r r� r r7 r7 r r8 r) S s>
�
� �� ^� # r) c s: e Zd ZdZdee eeee f d�� fdd�
Z
� Z S )�MySQLCursorPreparedDicta�
This class is a blend of features from MySQLCursorDict and MySQLCursorPrepared
Multiple inheritance in python is allowed but care must be taken
when assuming methods resolution. In the case of multiple
inheritance, a given attribute is first searched in the current
class if it's not found then it's searched in the parent classes.
The parent classes are searched in a left-right fashion and each
class is searched once.
Based on python's attribute resolution, in this case, attributes
are searched as follows:
1. MySQLCursorPreparedDict (current class)
2. MySQLCursorDict (left parent class)
3. MySQLCursorPrepared (right parent class)
4. MySQLCursor (base class)
Nr c � s" � fdd�t � j|d�I dH D �S )a� Return the next set of rows of a query result set.
When no more rows are available, it returns an empty list.
The number of rows returned can be specified using the size argument,
which defaults to one.
Returns:
list: The next set of rows of a query result set represented
as a list of dictionaries where column names are used as keys.
c s g | ]}|r� � |� j��qS r7 r! r" rA r7 r8 rp � s �z5MySQLCursorPreparedDict.fetchmany.<locals>.<listcomp>�r N�r
r �r6 r r rA r8 r t s
�z!MySQLCursorPreparedDict.fetchmany)N) rC rD rE rF r rJ r r r� r( r r r7 r7 r r8 rF b s ��rF c s2 e Zd ZdZdee ee d�� fdd�
Z� Z S )� MySQLCursorPreparedNamedTuplez^
This class is a blend of features from MySQLCursorNamedTuple and MySQLCursorPrepared
Nr c � s" � fdd�t � j|d�I dH D �S )a� Return the next set of rows of a query result set.
When no more rows are available, it returns an empty list.
The number of rows returned can be specified using the size argument,
which defaults to one.
Returns:
list: The next set of rows of a query result set represented
as a list of named tuples where column names are used as names.
c s g | ]}|r� � |� j��qS r7 r! r" rA r7 r8 rp � s �z;MySQLCursorPreparedNamedTuple.fetchmany.<locals>.<listcomp>rG NrH rI r rA r8 r � s
�z'MySQLCursorPreparedNamedTuple.fetchmany)N)
rC rD rE rF r rJ r r) r r r7 r7 r r8 rJ � s rJ c @ s\ e Zd ZU dZdZeed< ee d�dd�Z d
ee
e e d�d d
�Z e e d�d d �Z
dS )�MySQLCursorPreparedRawzW
This class is a blend of features from MySQLCursorRaw and MySQLCursorPrepared
Tr� r? c � s> | � � | jr&| j�| jd �I dH | j| jd�I dH p<dS )r r. Nr� )r� r- r_ rB r* r� r� rA r7 r7 r8 rZ � s z MySQLCursorPreparedRaw.fetchoneNr c � sV | � � g }|p| j}|dkrR| �� rR|d8 }| j| jd�I dH }|r|�|� q|S )rC r r, r� N)r� r r� r� r� r� r r7 r7 r8 r � s
z MySQLCursorPreparedRaw.fetchmanyc � s� | � � g }| jd r&|�| jd � | �� r�| jrN| j�| jd t�I dH | jj | j
| j | j d�I dH \}}|�
|� | �|�I dH q&t|�| _|S )r� r r. N)r� r� rx )r� r� r� r� r- r_ rB r* r r� r� r� r� rD r� r@ r� rE r7 r7 r8 r� � s
�
�
z MySQLCursorPreparedRaw.fetchall)N)rC rD rE rF r� r r r r) rZ rJ r r r� r7 r7 r7 r8 rK � s
rK )GrF �__all__rG r� � collectionsr �decimalr �typingr r r r r r
r r r
r � constantsr r� r r r r r r r r r r r r �errorsr r r r r! �typesr" r# r$ r% r& r' r( r) r* r+ Z abstractsr- r. r/ r� r0 r< rX r r� r r� r� r� r� r� r) rF rJ rK r7 r7 r7 r8 �<module> sB 0
8 0 />$
-: &