File: //var/opt/nydus/ops/mysql/connector/__pycache__/cursor_cext.cpython-39.pyc
a
�,�h=� � @ s� d Z ddlmZ ddlZddlZddlZddlmZ ddlm Z m
Z
m Z m Z m
Z
mZmZmZmZmZmZ ddlmZ ddlmZmZmZmZmZm Z m Z m Z m Z dd l m!Z!m"Z"m#Z#m$Z$ dd
l%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/ dd l0m1Z1 dd l2m3Z3m4Z4m5Z5m6Z6m7Z7 e �r"dd
l8m9Z9 dZ:G dd� d�Z;G dd� de$�Z<G dd� de<�Z=G dd� de<�Z>G dd� de=�Z?G dd� de<�Z@G dd � d e=�ZAG d d � d e<�ZBG d d � d e=�ZCG d!d"� d"e<�ZDG d#d$� d$e@eD�ZEG d%d&� d&eBeD�ZFG d'd(� d(eD�ZGdS ))z%Cursor classes using the C Extension.� )� annotationsN)�
namedtuple) �
TYPE_CHECKING�Any�Dict� Generator�Iterator�List�NoReturn�Optional�Sequence�Tuple�Union)�MySQLInterfaceError� ) �CextEofPacketType�CextResultType�DescriptionType�ParamsSequenceOrDictType�ParamsSequenceType� RowItemType�RowType�
StrOrBytes� WarningType)�NAMED_TUPLE_CACHE�CMySQLPrepStmt�MySQLConnectionAbstract�MySQLCursorAbstract)
� 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)�CR_NO_RESULT_SET)�Error�InterfaceError�NotSupportedError�ProgrammingError�get_mysql_exception)�CMySQLConnectionzNo result set to fetch fromc @ sB e Zd ZdZddd�dd�Zddd �d
d �Zed d
�dd��ZdS )�_ParamSubstitutorz4
Substitutes parameters into SQL statement.
zSequence[bytes]�None)�params�returnc C s || _ d| _d S )Nr )r1 �index)�selfr1 � r5 �N/opt/nydus/tmp/pip-target-wkfpz8uv/lib64/python/mysql/connector/cursor_cext.py�__init__m s z_ParamSubstitutor.__init__�object�bytes)�matchobjr2 c C sB | j }| j d7 _ z | j| W S ty< td�d �Y n0 d S )Nr z+Not enough parameters for the SQL statement)r3 r1 �
IndexErrorr, )r4 r: r3 r5 r5 r6 �__call__q s ��z_ParamSubstitutor.__call__�int�r2 c C s t | j�| j S )z8Returns number of parameters remaining to be substituted)�lenr1 r3 �r4 r5 r5 r6 � remaining{ s z_ParamSubstitutor.remainingN)�__name__�
__module__� __qualname__�__doc__r7 r< �propertyrA r5 r5 r5 r6 r/ h s
r/ c s� e Zd ZU dZdZded< dZded< dZded< ddd �d
d �Zd]ddd
�� fdd�
Z dd�dd�Z
dd�dd�Z dd�dd�Z ddd�dd�Z
dd�d d �Zdd�d d �Zd d�d!d"�Zd^d$d%dd&d'�d(d)�Zd*d+d,d-�d.d/�Zd*d+d&d-�d0d1�Zed2d�d3d4��Zed5d�d6d7��Zdd�d8d9�Zd_d*d:d;d<�d=d>�Zd?d�d@dA�ZdBd�dCdD�Zd`d5dBdF�dGdH�ZdId�dJdK�Z dLd�dMdN�Z d d�dOdP�Z dQd�dRdS�Z edTd�dUdV��Z ed*d�dWdX��Z!edd�dYdZ��Z"d*d�d[d\�Z#� Z$S )a� CMySQLCursorz;Default cursor for interacting with MySQL using C ExtensionF�bool�_raw� _buffered�_raw_as_stringr. r0 )�
connectionr2 c C s@ t �| � d| _d| _d| _t|t�s0tdd��t� |�| _
dS )�
Initialize����NNi ��errnoN) r r7 �_affected_rows� _rowcount�_nextrow�
isinstancer r* �weakref�proxy�_cnx�r4 rL r5 r5 r6 r7 � s
zCMySQLCursor.__init__T��freer2 c s^ d| _ d| _d| _d| _d| _d| _d| _d| _d| _g | _|rP| jrP| j� � t
� � � dS )zXReset the cursor
When free is True (default) the result will be freed.
rN Nr ) rS rT rR �_last_insert_id�_warning_count� _warnings� _descriptionZ_executed_listrX � free_result�super�reset�r4 r[ �� __class__r5 r6 rb � s
zCMySQLCursor.resetr> c C s | j du rtt��dS )zsCheck if the statement has been executed.
Raises an error if the statement has not been executed.
N)� _executedr* �ERR_NO_RESULT_TO_FETCHr@ r5 r5 r6 �_check_executed� s
z CMySQLCursor._check_executedzOptional[List[WarningType]]c
C s� g }z2| j �� | j �d�}| j �� d }| j �� W nl typ } z"t|j|j|jd�|�W Y d}~n<d}~0 t y� } z t
d|� ��d�W Y d}~n
d}~0 0 |r�|S dS )a Fetch warnings
Fetch warnings doing a SHOW WARNINGS. Can be called after getting
the result.
Returns a result set or None when there were no warnings.
Raises Error (or subclass) on errors.
Returns list of tuples or None.
z
SHOW WARNINGSr ��msgrQ �sqlstateNzFailed getting warnings; ) rX Zconsume_results� cmd_query�get_rowsr r- rj rQ rk � Exceptionr* )r4 Zwarns�_�errr5 r5 r6 �_fetch_warnings� s"
��&z CMySQLCursor._fetch_warningsc C sf | j jr| jr| �� | _| js"dS t| jd dd� d| j j i�}| j jrP|�tjt |�dd� dS )z�Handle possible warnings after all results are consumed.
Raises:
Error: Also raises exceptions if raise_on_warnings is set.
Nr r � �warning� )�
stacklevel)
rX Z get_warningsr] rq r^ r- Zraise_on_warnings�warnings�warn�str)r4 rp r5 r5 r6 �_handle_warnings� s
��z CMySQLCursor._handle_warningsz(Union[CextEofPacketType, CextResultType])�resultr2 c C sR d|v r"|d | _ d| _| �� n,|d | _|d | _|d | _d| _| �� dS )z,Handles the result after statement execution�columnsr Z insert_id�
warning_countZ
affected_rowsrN N)r_ rS �_handle_resultsetr\ r] rR ry �r4 rz r5 r5 r6 �_handle_result� s
zCMySQLCursor._handle_resultc C s dS )�Handle a result setNr5 r@ r5 r5 r6 r} � s z CMySQLCursor._handle_resultsetc C s( | j j| _| �� | j js$| j �� dS )zMHandle end of reading the result
Raises an Error on errors.
N)rX r| r] ry � more_resultsr` r@ r5 r5 r6 � _handle_eof� s
zCMySQLCursor._handle_eofz#Generator[CMySQLCursor, None, None]c
c s� t | jd�}d| _d}d}zV|�� �d�r.|s>|r:|�� nd}| jj}t|�rRW q|�� | _| V | � � snt
�W q t y� } z|j t
kr�� W Y d}~qd}~0 t
y� Y dS 0 qdS )aY 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.
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_stmtN� Fs CALLs stmt_overflow!)r' rf �upper�
startswith�popleftrX �result_set_availabler& �rstrip�nextset�
StopIterationr* rQ r( )r4 Z
executed_list�stmtrz rp r5 r5 r6 �
_execute_iter� s( * �
zCMySQLCursor._execute_iterr5 r r z-Optional[Generator[CMySQLCursor, None, None]]�� operationr1 �multir2 c
C s� |sdS z| j r| j �� r t�W n4 ttfyT } ztdd�|�W Y d}~n
d}~0 0 | j �� d}| �� z"t|t�r�|�| j j �}n|}W n6 t
t fy� } ztt|��|�W Y d}~n
d}~0 0 |�rH| j � |�}t|t
��r|�� D ] \}}|�d|� d��� |�}q�n8t|ttf��rHt|�} t�| |�}| jdk�rHtd��z | j j|| j| j| jd �}
W n< t�y� } z"t|j |j |j d
�|�W Y d}~n
d}~0 0 || _ | � |
� |�r�| �!� S dS ) z�Execute given statement using given parameters
Deprecated: The multi argument is not needed and nextset() should
be used to handle multiple result sets.
N�Cursor is not connected� r� �%(�)sr �1Not all parameters were used in the SQL statement)�rawZbuffered�
raw_as_stringri )"rX � is_closedr, �ReferenceError�handle_unread_resultrb rU rx �encode�python_charset�UnicodeDecodeError�UnicodeEncodeError�prepare_for_mysql�dict�items�replace�list�tupler/ r �subrA rl rI rJ rK r r- rj rQ rk rf r r� ) r4 r� r1 r� rp r� �prepared�key�value�psubrz r5 r5 r6 �executeF s\ "
$ ��
��
zCMySQLCursor.executerx z"Sequence[ParamsSequenceOrDictType]zOptional[bytes]�r� �
seq_paramsr2 c
C s� ddd�dd�}t �tdt �t||��}t �t|�}|s@td��|�d��| j j
�}g }z�|�| j j
�}|D ]�} |}| j � | �}
t |
t
�r�|
�� D ] \} } |�d | � d
��� | �}q�n4t |
ttf�r�t|
�}
t�|
|�}|
jd kr�td ��|�|� ql||v �r |�|d
�|�d�}|| _|W S W dS ttf�y\ } ztt|��|�W Y d}~n>d}~0 t �y� } z td|� ��d�W Y d}~n
d}~0 0 dS )zImplements multi row insertzre.Matchrx )�matchr2 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 � � )�group)r� r5 r5 r6 �remove_comments� s
z3CMySQLCursor._batch_insert.<locals>.remove_commentsr� zAFailed rewriting statement for multi-row INSERT. Check SQL syntaxr r� r� r r� � ,N� Failed executing the operation; ) �rer� r# r �searchr" r* r� r� rX r� r� rU r� r� r� r� r� r/ r rA r, �append�joinrf r� r� rx rn )r4 r� r� r� �tmp�matches�fmt�valuesr� r1 r� r� r� r� rp r5 r5 r6 �
_batch_insert� sP � �
�
�
"zCMySQLCursor._batch_insertc
C s< |r|s dS z| j st�W n2 ttfyL } ztd�|�W Y d}~n
d}~0 0 | j �� t|ttf�sntd��t�t |�r�|s�d| _
dS | � ||�}|dur�|| _ | �
|�S d}zF|D ]<}| �
||� | jr�| j jr�| �� || jr�| j
n| j7 }q�W n: ttf�y0 } z td|� ��d�W Y d}~n
d}~0 0 || _
dS )a� Execute the given operation multiple times
The executemany() method will execute the operation iterating
over the list of parameters in seq_params.
Example: Inserting 3 new employees and their phone number
data = [
('Jane','555-001'),
('Joe', '555-001'),
('John', '555-003')
]
stmt = "INSERT INTO employees (name, phone) VALUES ('%s','%s)"
cursor.executemany(stmt, data)
INSERT statements are optimized by batching the data, that is
using the MySQL multiple rows syntax.
Results are discarded! If they are needed, consider looping over
data using the execute() method.
Nr� z+Parameters for query must be list or tuple.r r� )rX r, r� r� rU r� r� r� r� r! rS r� rf r� � with_rows�
unread_result�fetchall� descriptionrR �
ValueError� TypeErrorr* )r4 r� r� rp r� �rowcntr1 r5 r5 r6 � executemany� s:
&zCMySQLCursor.executemanyz Optional[List[DescriptionType]]c C s | j S )z*Returns description of columns in a result)r_ r@ r5 r5 r6 r� s zCMySQLCursor.descriptionr= c C s | j dkr| jS | j S )z/Returns the number of rows produced or affectedrN )rS rR r@ r5 r5 r6 �rowcount s
zCMySQLCursor.rowcountc C s$ | j s
dS | j �� d| _d| _ dS )z<Close the cursor
The result will be freed.
FNT)rX r� r^ r@ r5 r5 r6 �close s
zCMySQLCursor.closer z0Optional[Union[Dict[str, RowItemType], RowType]]��procname�argsr2 c
C sl |rt |t�std��t |ttf�s,td��d}g | _�z�g }g }|�d�d }|r�g }t|�D ]n\}} |j||d d�}
|� |
� t | t�r�|� d|
� d | d � d
�� |� | d � q`|� |
� |� | � q`d �
d
d� |D ��} | � d| � �|� d|� dd �
|�� d
�} | j j
| | j| jd�}
g }| j j�r�| j �� }
t | ttf��rRt}n(t | ttf��rht}n| j�rvt}nt}|| j �� �}d| � d
�|_|�|
� |� |� | j �� �q(|| _| � � |�r| � � dd� t |dd� |D ��D �}dd �
|�� �}| � |� | � � W S t� W S t �y2 � Y n6 t!�yf } z t"d|� ��d�W Y d}~n
d}~0 0 dS )z1Calls a stored procedure with the given argumentszprocname must be a stringzargs must be a sequencez@_{name}_arg{index}�.rN r )�namer3 z CAST(� AS �)r �,c s s | ]}|� d �V qdS )z=%sNr5 ��.0�argr5 r5 r6 � <genexpr>C r� z(CMySQLCursor.callproc.<locals>.<genexpr>zSET zCALL �()r� r� z
(a result of c S s g | ]\}}|� d |� ��qS )r� r5 )r� r� �aliasr5 r5 r6 �
<listcomp>g s �z)CMySQLCursor.callproc.<locals>.<listcomp>c S s g | ]}|� d ��qS )z@_)�lstripr� r5 r5 r6 r� j r� zSELECT z Failed calling stored routine; N)#rU rx r� r� r� �_stored_results�split� enumerate�formatr� r� r� rX rl rI rK r� �fetch_eof_columns�CMySQLCursorDict�CMySQLCursorBufferedDict�CMySQLCursorNamedTuple� CMySQLCursorBufferedNamedTuple�CMySQLCursorBufferedRaw�CMySQLCursorBufferedZget_selfrf r � next_resultr� rb �zip�fetchoner) rn r* )r4 r� r� Zargfmt�argnames�argtypesZ procname_absZ argvalues�idxr� �argnameZ placeholders�callrz �resultsZ cursor_class�cur�selectrp r5 r5 r6 �callproc sz
�
�
��
zCMySQLCursor.callproczOptional[bool]c C s` | j �� s| jdd� dS | jdd� | j jsL| j �� }| �|� ttd��| �| j �� � dS )z%Skip to the next available result setT�r[ NFrP ) rX r� rb r� Zfetch_eof_statusr r* r( r� )r4 �eofr5 r5 r6 r� x s
zCMySQLCursor.nextset�
List[RowType]c C s~ | � � | jjsg S | j�� }| jrD| jd rD|d �d| jd � |d sX| �� g S | jt|d �7 _| �� |d S )��Return all rows of a query result set.
Returns:
list: A list of tuples with all rows of a query result set.
r ) rh rX r� rm rT �insertr� rS r? �r4 �rowsr5 r5 r6 r� � s
zCMySQLCursor.fetchallr ��sizer2 c C s� | � � | jr.| jd r.| jd g}|d8 }ng }|rT| jjrT|�| j�|�d � |r�| jjr�| j�� | _| jr�| jd s�| jjs�| j�� nd| _|s�| � � g S | j
t |�7 _
|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 rO ) rh rT rX r� �extendrm �get_rowr� r` r� rS r? )r4 r� r� r5 r5 r6 � fetchmany� s.
��� zCMySQLCursor.fetchmany�Optional[RowType]c C s| | � � | j}|s$| jjr$| j�� }|rZ|d rZ| j�� | _| jd sf| jjsf| j�� n | �� dS | jd7 _|d S )�yReturn next row of a query result set.
Returns:
tuple or None: A row from query result set.
r Nr ) rh rT rX r� r� r� r` r� rS �r4 �rowr5 r5 r6 r� � s
zCMySQLCursor.fetchonezIterator[RowType]c C s t | jd�S )z�Iteration over the result set
Iteration over the result set which calls self.fetchone()
and returns the next row.
N)�iterr� r@ r5 r5 r6 �__iter__� s zCMySQLCursor.__iter__c c s | j D ]
}|V qg | _ dS )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.
Returns a iterator.
N)r� r~ r5 r5 r6 �stored_results� s
zCMySQLCursor.stored_resultsr c C s4 z | � � }W n ty$ td�Y n0 |s0td�|S )z�Iteration over the result set
Used for iterating over the result set. Calls self.fetchone()
to get the next row.
Raises StopIteration when no more rows are available.
N)r� r* r� r� r5 r5 r6 �__next__� s zCMySQLCursor.__next__zTuple[str, ...]c C s | j s
dS tdd� | j D ��S )ztReturns column names
This property returns the columns names as a tuple.
Returns a tuple.
r5 c s s | ]}|d V qdS )r Nr5 )r� �dr5 r5 r6 r� r� z,CMySQLCursor.column_names.<locals>.<genexpr>)r� r� r@ r5 r5 r6 � column_names� s zCMySQLCursor.column_namesc C s4 z| j �� �d�W S ty. | j �� Y S 0 dS )z�Returns the executed statement
This property returns the executed statement. When multiple
statements were executed, the current statement in the iterator
will be returned.
�utf8N)rf �strip�decode�AttributeErrorr@ r5 r5 r6 � statement s zCMySQLCursor.statementc C s | j r
dS dS )z�Returns whether the cursor could have rows returned
This property returns True when column descriptions are available
and possibly also rows, which will need to be fetched.
Returns True or False.
TF)r� r@ r5 r5 r6 r� s zCMySQLCursor.with_rowsc C sh d}| j rRz| j �d�}W n ty2 | j }Y n0 t|�dkrV|d d� d }nd}|j| jj|d�S )Nz{class_name}: {stmt}zutf-8�( z..z(Nothing executed yet))�
class_namer� )rf r r r? r� re rB )r4 r� Zexecutedr5 r5 r6 �__str__% s zCMySQLCursor.__str__)T)r5 F)r5 )r )%rB rC rD rE rI �__annotations__rJ rK r7 rb rh rq ry r r} r� r� r� r� r� rF r� r� r� r� r� r� r� r� r� r r r r r� r �
__classcell__r5 r5 rd r6 rG � sN
J �A=C �X' rG c s� e Zd ZdZdd�� fdd� Zdd�dd �Zd d dd �� fd
d�
Zdd�dd�Zdd�dd�Zd ddd�dd�Z dd�dd�Z
e d d�d d ��Z � Z
S ) r� z*Cursor using C Extension buffering resultsr. �rL c s t � �|� d| _d| _dS )rM Nr )ra r7 �_rows� _next_rowrY rd r5 r6 r7 7 s z CMySQLCursorBuffered.__init__r0 r> c C s. | j �� d | _d| _t| j�| _| �� dS )r� r N)rX rm r r r? rS r� r@ r5 r5 r6 r} > s z&CMySQLCursorBuffered._handle_resultsetTrH rZ c s d| _ d| _t� j|d� dS )zReset the cursor to defaultNr r� )r r ra rb rc rd r5 r6 rb E s zCMySQLCursorBuffered.resetr� c C s< d}z| j | j }W n ty( Y dS 0 | jd7 _|S )�QReturns the next row in the result set
Returns a tuple or None.
Nr )r r r; r� r5 r5 r6 �
_fetch_rowK s z CMySQLCursorBuffered._fetch_rowr� c C s( | � � | j| jd� }t| j�| _|S )r� N)rh r r r? �r4 �resr5 r5 r6 r� X s z CMySQLCursorBuffered.fetchallr r= r� c C sF | � � g }|p| j}|dkrB|d8 }| �� }|rB|�|� qqBq|S �r� r r )rh � arraysizer r� �r4 r� r Zcntr� r5 r5 r6 r� c s
z CMySQLCursorBuffered.fetchmanyc C s | � � | �� S )r� �rh r r@ r5 r5 r6 r� y s z CMySQLCursorBuffered.fetchonec C s
| j duS )z�Returns whether the cursor could have rows returned
This property returns True when rows are available,
which will need to be fetched.
Returns True or False.
N)r r@ r5 r5 r6 r� � s z CMySQLCursorBuffered.with_rows)T)r )rB rC rD rE r7 r} rb r r� r� r� rF r� r
r5 r5 rd r6 r� 4 s
r� c @ s e Zd ZU dZdZded< dS )�CMySQLCursorRawz+Cursor using C Extension return raw resultsTrH rI N�rB rC rD rE rI r r5 r5 r5 r6 r � s
r c @ s e Zd ZU dZdZded< dS )r� z.Cursor using C Extension buffering raw resultsTrH rI Nr r5 r5 r5 r6 r� � s
r� c s\ e Zd ZU dZdZded< dd�� fdd� Zdd
d d �� fd
d�
Zd d�� fdd� Z� Z S )r� z7Cursor using C Extension returning rows as dictionariesFrH rI � Optional[Dict[str, RowItemType]]r> c s" t � �� }|r tt| j|��S dS )zyReturn next row of a query result set.
Returns:
dict or None: A dict from query result set.
N)ra r� r� r� r r� rd r5 r6 r� � s
zCMySQLCursorDict.fetchoner r= � List[Dict[str, RowItemType]]r� c s t � j|d�}� fdd�|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.
�r� c s g | ]}t t� j|���qS r5 �r� r� r �r� r� r@ r5 r6 r� � r� z.CMySQLCursorDict.fetchmany.<locals>.<listcomp>�ra r� �r4 r� r rd r@ r6 r� � s zCMySQLCursorDict.fetchmanyc s t � �� }� fdd�|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 | ]}t t� j|���qS r5 r r r@ r5 r6 r� � r� z-CMySQLCursorDict.fetchall.<locals>.<listcomp>�ra r� r rd r@ r6 r� � s
zCMySQLCursorDict.fetchall)r )
rB rC rD rE rI r r� r� r� r
r5 r5 rd r6 r� � s
r� c s< e Zd ZdZdZdd�� fdd� Zdd�� fdd � Z� ZS )
r� zECursor using C Extension buffering and returning rows as dictionariesFr r> c s" t � �� }|r tt| j|��S d S �N)ra r r� r� r r� rd r5 r6 r � s
z#CMySQLCursorBufferedDict._fetch_rowr c s t � �� }� fdd�|D �S )r� c s g | ]}t t� j|���qS r5 r r r@ r5 r6 r� � r� z5CMySQLCursorBufferedDict.fetchall.<locals>.<listcomp>r" r rd r@ r6 r� � s
z!CMySQLCursorBufferedDict.fetchall)rB rC rD rE rI r r� r
r5 r5 rd r6 r� � s r� c sn e Zd ZU dZdZded< dd�� fdd� Zd d�� fd
d � Zdd
dd�� fdd�
Zdd�� fdd� Z � Z
S )r� z7Cursor using C Extension returning rows as named tuplesNr � named_tupler0 r> c sP t � �� t| j�}zt| | _W n( tyJ td|�| _| jt|< Y n0 dS )r� �RowN)ra r} r� r r r$ �KeyErrorr )r4 r{ rd r5 r6 r} � s
z(CMySQLCursorNamedTuple._handle_resultsetr� c s t � �� }|r| j|� S dS �r� N)ra r� r$ r� rd r5 r6 r� � s
z CMySQLCursorNamedTuple.fetchoner r= r� r� c s( t � j|d�}|sg S � fdd�|D �S )r� r c s g | ]}� j |� �qS r5 �r$ r r@ r5 r6 r� � r� z4CMySQLCursorNamedTuple.fetchmany.<locals>.<listcomp>r r! rd r@ r6 r� � s
z CMySQLCursorNamedTuple.fetchmanyc s t � �� }� fdd�|D �S )r� c s g | ]}� j |� �qS r5 r( r r@ r5 r6 r� r� z3CMySQLCursorNamedTuple.fetchall.<locals>.<listcomp>r" r rd r@ r6 r� � s
z CMySQLCursorNamedTuple.fetchall)r ) rB rC rD rE r$ r r} r� r� r� r
r5 r5 rd r6 r� � s
r� c sX e Zd ZU dZdZded< dd�� fdd� Zd d�� fd
d � Zd d�� fd
d� Z� Z S )r� zECursor using C Extension buffering and returning rows as named tuplesNr r$ r0 r> c s t � �� td| j�| _d S )Nr% )ra r} r r r$ r@ rd r5 r6 r} s
z0CMySQLCursorBufferedNamedTuple._handle_resultsetr� c s t � �� }|r| j|� S d S r# )ra r r$ r� rd r5 r6 r s
z)CMySQLCursorBufferedNamedTuple._fetch_rowr� c s t � �� }� fdd�|D �S )r� c s g | ]}� j |� �qS r5 r( r r@ r5 r6 r� r� z;CMySQLCursorBufferedNamedTuple.fetchall.<locals>.<listcomp>r" r rd r@ r6 r� s
z'CMySQLCursorBufferedNamedTuple.fetchall)
rB rC rD rE r$ r r} r r� r
r5 r5 rd r6 r� s
r� c s� e Zd ZdZdd�� fdd� Zdd�dd �Zd/d d d
�dd�Zd0dddd�dd�Zdd�� fdd� Zd1d dd�� fdd�
Z d2d d d dd �d d �Z
d!d"dd#�d$d%�Z d d�d&d'�Z d3d(d)d*�d+d,�Z
d)d�d-d.�Z� ZS )4�CMySQLCursorPreparedz&Cursor using MySQL Prepared Statementsr. r c s. t � �|� d | _d| _d| _d| _d | _d S )Nr T)ra r7 r rS r �_binary�_stmtrY rd r5 r6 r7 % s z CMySQLCursorPrepared.__init__r0 r> c C s d| _ | �� dS )zHandle EOF packetrO N)rT ry r@ r5 r5 r6 r� - s z CMySQLCursorPrepared._handle_eofFrH r� )r� r2 c C s� | j r| j jsdS d}| jdkr@| jj| j| j|| j d�\}}n
| j\}}|r�| jj| j| j|| j d�| _| jd }|dur�|d | _| �� | j dkr�d| _ n| j d7 _ |r�|d | _| �� |S )r NrO )�binaryr{ r� � prep_stmtr r| rN )
r+ �have_result_setrT rX r� r* r� r] r� rS )r4 r� r� r� r5 r5 r6 r 2 s:
�
�
z CMySQLCursorPrepared._fetch_rowNr r
r� c C s
t � �dS )zRCalls a stored procedue
Not supported with CMySQLCursorPrepared.
N)r+ )r4 r� r� r5 r5 r6 r� Z s z CMySQLCursorPrepared.callprocc s0 | j r"| �� | j�| j � d| _ t� �� dS )zzClose the cursor
This method will try to deallocate the prepared statement and close
the cursor.
N)r+ rb rX �cmd_stmt_closera r� r@ rd r5 r6 r� a s
zCMySQLCursorPrepared.closeTrZ c s& | j r| j�| j � t� j|d� dS )z Resets the prepared statement.r� N)r+ rX �cmd_stmt_resetra rb rc rd r5 r6 rb m s zCMySQLCursorPrepared.resetr z"Optional[ParamsSequenceOrDictType]r� c
s� |sdS z| j r| j �� r t�W n4 ttfyT } ztdd�|�W Y d}~n
d}~0 0 | j jdd� | j j}|dkrxd}t|t�s�z|�|�}W n2 t y� } ztt|��|�W Y d}~n
d}~0 0 t� t
��r6t � t
|�}zt� fdd �|D ��� W n0 t�y& } ztd
�|�W Y d}~n
d}~0 0 t �td |�}|| ju�r�| j�rX| j �| j� || _z|�|�}W n4 t�y� } ztt|��|�W Y d}~n
d}~0 0 d |v �r�t �td
|�}z| j �|�| _W n" t�y� d| _d| _� Y n0 | j �| j� | jjdk�r� �sdS � �rjt� tt f��sLtddt � �j � d� � d�d��| jjt � �k�rjtddd��� du �rxd� | j j | jg� �R � }|�r�| �!|� dS )a) Prepare and execute a MySQL Prepared Statement
This method will prepare the given operation and execute it using
the given parameters.
If the cursor instance already had a prepared statement, it is
first closed.
Note: argument "multi" is unused.
Nr� r� T)r� Zutf8mb4r c 3 s | ]}� | V qd S r# r5 )r� r� �r1 r5 r6 r� � r� z/CMySQLCursorPrepared.execute.<locals>.<genexpr>z6Not all placeholders were found in the parameters dict�?s %s� ?r i� z Incorrect type of argument: r� zP), it must be of type tuple or list the argument given to the prepared statement)rQ rj z:Incorrect number of arguments executing prepared statementr5 )"rX r� r, r� r� �charsetrU rx r r� r� r� �findallr$ r� r& r� r% rf r+ r/ r� r� r Zcmd_stmt_preparer) r0 Z param_countr� �typerB r? Zcmd_stmt_executer )r4 r� r1 r� rp r4 Zreplacement_keysr r5 r1 r6 r� s s| "
$ �� $
��
z CMySQLCursorPrepared.executerx z Sequence[ParamsSequenceType]r� c
C sz d}z2|D ](}| � ||� | jr(| �� || j7 }q
W n8 ttfyn } z td|� ��|�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.
r r� N)r� r� r� rS r� r� r* )r4 r� r� r� r1 rp r5 r5 r6 r� � s &z CMySQLCursorPrepared.executemanyc C s | � � | �� pdS r' r r@ r5 r5 r6 r� � s z CMySQLCursorPrepared.fetchonez
Optional[int]r� r� c C sJ | � � g }|p| j}|dkrF| jjrF|d8 }| �� }|r|�|� q|S r )rh r r+ r. r r� r r5 r5 r6 r� � s
z CMySQLCursorPrepared.fetchmanyc C s� | � � | jjsg S | jj| jd�}| jrJ| jd rJ|d �d| jd � |d s^| �� g S | jt |d �7 _| �� |d S )r� )r- r )
rh r+ r. rX rm rT r� r� rS r? r� r5 r5 r6 r� s z CMySQLCursorPrepared.fetchall)F)N)T)NF)N)rB rC rD rE r7 r� r r� r� rb r� r� r� r� r� r
r5 r5 rd r6 r) " s ( �^ r) c @ s e Zd ZdZdS )�CMySQLCursorPreparedDicta� This class is a blend of features from CMySQLCursorDict and CMySQLCursorPrepared
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. CMySQLCursorPreparedDict (current class)
2. CMySQLCursorDict (left parent class)
3. CMySQLCursorPrepared (right parent class)
4. CMySQLCursor (base class)
N�rB rC rD rE r5 r5 r5 r6 r7 s r7 c @ s e Zd ZdZdS )� CMySQLCursorPreparedNamedTuplezVThis class is a blend of features from CMySQLCursorNamedTuple and CMySQLCursorPreparedNr8 r5 r5 r5 r6 r9 - s r9 c @ s e Zd ZU dZdZded< dS )�CMySQLCursorPreparedRawzOThis class is a blend of features from CMySQLCursorRaw and CMySQLCursorPreparedTrH rI Nr r5 r5 r5 r6 r: 1 s
r: )HrE �
__future__r r� rv rV � collectionsr �typingr r r r r r r
r r r
r Z_mysql_connectorr �typesr r r r r r r r r Z abstractsr r r r �cursorr r r r! r" r# r$ r% r& r' � errorcoder( �errorsr) r* r+ r, r- Zconnection_cextr. rg r/ rG r� r r� r� r� r� r� r) r7 r9 r: r5 r5 r5 r6 �<module> sD 4 ,0 8Z'3 z