관리-도구
편집 파일: cursor_cext.cpython-38.pyc
U )�e�d � @ s( d Z ddlmZ ddlZddlZddlmZmZ ddlm Z m Z ddl m Z ddl mZ dd lmZmZmZmZmZmZ dd lmZ G d d � d e�ZG d d� de�ZG dd� de�ZG dd� de�Z G dd� de�Z G dd� de�Z G dd� de�Z G dd� de�Z G dd � d e�Z!G d d � d e�Z"dS ) z%Cursor classes using the C Extension � )� namedtupleN� )�MySQLConnectionAbstract�MySQLCursorAbstract)�PY2� isunicode)�errors)�CR_NO_RESULT_SET)� RE_PY_PARAM�RE_SQL_INSERT_STMT�RE_SQL_ON_DUPLICATE�RE_SQL_COMMENT�RE_SQL_INSERT_VALUES�RE_SQL_SPLIT_STMTS)�MySQLInterfaceErrorc @ s, e Zd ZdZdd� Zdd� Zedd� �ZdS ) �_ParamSubstitutorz4 Substitutes parameters into SQL statement. c C s || _ d| _d S )Nr )�params�index)�selfr � r �p/opt/hc_python/lib64/python3.8/site-packages/../../../lib/python3.8/site-packages/mysql/connector/cursor_cext.py�__init__2 s z_ParamSubstitutor.__init__c C sD | j }| j d7 _ z | j| W S tk r> t�d��Y nX d S )Nr z+Not enough parameters for the SQL statement)r r � IndexErrorr �ProgrammingError)r Zmatchobjr r r r �__call__6 s �z_ParamSubstitutor.__call__c C s t | j�| j S )z8Returns number of parameters remaining to be substituted)�lenr r �r r r r � remaining? s z_ParamSubstitutor.remainingN)�__name__� __module__� __qualname__�__doc__r r �propertyr r r r r r , s r c s( e Zd ZdZdZdZdZdd� Zd>� fdd� Zdd � Z d d � Z d d � Z dd� Z dd� Z dd� Zd?dd�Zdd� Zdd� Zedd � �Zed d � �Zed d � �Zd!d"� Zd@d#d$�Zd%d&� Zd'd(� ZdAd*d+�Zd,d-� Zd.d/� Z d0d1� Z e r�d2d3� Z d4d5� Z ed6d7� �Z!ed8d9� �Z"ed:d;� �Z#d<d=� Z$� Z%S )B� CMySQLCursorz;Default cursor for interacting with MySQL using C ExtensionFc C sf t �| � d| _d| _d| _d| _d| _d| _d| _g | _ g | _ t |t �sVt jdd��t�|�| _dS )� Initializer N���i ��errno)r r � _insert_id�_warning_count� _warnings�_affected_rows� _rowcount�_nextrow� _executed�_executed_list�_stored_results� isinstancer r �InterfaceError�weakref�proxy�_cnx�r � connectionr r r r M s zCMySQLCursor.__init__Tc sh d| _ d| _d| _d| _d| _d| _d| _d| _d| _d| _g | _|rV| j rV| j � � t t | �� � dS )zXReset the cursor When free is True (default) the result will be freed. r% Nr )r, r- r+ r( r) r* � _descriptionr. r/ r5 � free_result�superr# �reset�r �free�� __class__r r r; _ s zCMySQLCursor.resetc C s� g }z.| j �� | j �d�}| j �� }| j �� W nl tk rj } ztj|j|j|j d��W 5 d}~X Y n6 t k r� } zt� d� t |����W 5 d}~X Y nX |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 errors.Error (or subclass) on errors. Returns list of tuples or None. z SHOW WARNINGS��msgr' �sqlstateNz Failed getting warnings; {0})r5 Zconsume_results� cmd_query�get_rowsr r �get_mysql_exceptionrA r' rB � Exceptionr2 �format�str)r �warnings�_�exc�errr r r �_fetch_warningss s" � �z CMySQLCursor._fetch_warningsc C s | j jdkr | jr | �� | _dS )z7Handle possible warnings after all results are consumedTN)r5 Z get_warningsr) rM r* r r r r �_handle_warnings� s z CMySQLCursor._handle_warningsc C s| d|kr"|d | _ d| _| �� nV|d | _|d | _|d | _d| _| �� | jjdkrx| j rxt j | j d dd � � �d S ) z,Handles the result after statement execution�columnsr Z insert_id� warning_countZ affected_rowsr% Tr � N) r8 r, �_handle_resultsetr( r) r+ rN r5 �raise_on_warningsr* r rE )r �resultr r r �_handle_result� s zCMySQLCursor._handle_resultc C s dS )�Handle a result setNr r r r r rR � s z CMySQLCursor._handle_resultsetc C sR | j j| _| �� | j jdkr<| jr<tj| jd dd� � �| j jsN| j � � dS )zTHandle end of reading the result Raises an errors.Error on errors. Tr r rQ N) r5 rP r) rN rS r* r rE � more_resultsr9 r r r r � _handle_eof� s zCMySQLCursor._handle_eofc c s� t �| j�}d}|| | _| V z| �� s.t�W n0 tjk r` } z|jtkrP� W 5 d}~X Y nX |d7 }|| � � | _| V q dS )a� Generator returns MySQLCursor objects for multiple statements Deprecated: use nextset() method directly. This method is only used when multiple statements are executed by the execute() method. It uses zip() to make an iterator from the given query_iter (result of MySQLConnection.cmd_query_iter()) and the list of statements that were executed. r Nr ) r �splitr. �nextset� StopIterationr r2 r' r �strip)r Z executed_list�irK r r r � _execute_iter� s zCMySQLCursor._execute_iterr c C s� |sdS | j st�d��| j �� d}| �� z t|�rH|�| j j�}n|}W n4 tt fk r� } zt�t |���W 5 d}~X Y nX |�r | j � |�}t |t �r�|�� D ]6\}}tr�|�d�|�|�}q�|�d�|��� |�}q�n:t |ttf��r t|�} t�| |�}| jdk�r t�d��z | j j|| j| j| jd�} W n: t k �rt } ztj | j | j | j d��W 5 d} ~ X Y nX || _!| �"| � |�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� �%({0})sr �1Not all parameters were used in the SQL statement)�raw�buffered� raw_as_stringr@ )$r5 r r �handle_unread_resultr; r �encodeZpython_charset�UnicodeDecodeError�UnicodeEncodeErrorrH �prepare_for_mysqlr1 �dict�itemsr �replacerG �list�tupler r �subr rC �_raw� _buffered�_raw_as_stringr rE rA r' rB r. rU r^ ) r � operationr �multi�stmtrL �prepared�key�value�psubrT rK r r r �execute� sV � � � � zCMySQLCursor.executec C s� dd� }t �tdt �t||��}t �t|�}|s:t�d��|�d�� | j j �}g }z�|� | j j �}|D ]�} |}| j � | �} t | t�r�| �� D ] \} } |�d�| �� � | �}q�n6t | ttf�r�t| �} t�| |�}| jdkr�t�d��|�|� qf||k�r|�|d �|�d�}|| _|W S W d S W nf tt fk �rX } zt�t |���W 5 d }~X Y n2 t k �r� } zt�d | ��W 5 d }~X Y nX d S ) zImplements multi row insertc S s | � d�rdS | � d�S dS )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` � N)�group)�matchr r r �remove_comments s z3CMySQLCursor._batch_insert.<locals>.remove_commentsr` zBFailed rewriting statement for multi-row INSERT. Check SQL syntax.r ra r rb � ,Nz"Failed executing the operation; %s) �rerp r r �searchr r r2 r} rg r5 �charsetrj r1 rk rl rm rG rn ro r r r r �append�joinr. rh ri rH rF )r rt � seq_paramsr �tmp�matches�fmt�valuesrv r rw rx ry rz rL r r r � _batch_insert sL � � � �zCMySQLCursor._batch_insertc C s" |r|s dS | j s t�d��| j �� t|ttf�s>t�d��t�t |�rv|sXd| _ dS | � ||�}|dk rv| � |�S d}zd|D ]Z}| � ||� z4| j r�|t| j �� �7 }n || j7 }| �� s�q�q�W q� tk r� Y q�X q�W n8 ttfk �r } zt�d�|���W 5 d}~X Y nX || _ dS )z*Execute the given operation multiple timesNr_ z+Parameters for query must be list or tuple.r z#Failed executing the operation; {0})r5 r r rf r1 rn ro r� r~ r r, r� r{ r8 r rD r+ rZ r[ � ValueError� TypeErrorrG )r rt r� rv Zrowcntr rL r r r � executemanyB sB � �zCMySQLCursor.executemanyc C s | j S )z*Returns description of columns in a result)r8 r r r r � descriptionn s zCMySQLCursor.descriptionc C s | j dkr| jS | j S dS )z/Returns the number of rows produced or affectedr% N)r, r+ r r r r �rowcounts s zCMySQLCursor.rowcountc C s | j S )z8Returns the value generated for an AUTO_INCREMENT column)r( r r r r � lastrowid{ s zCMySQLCursor.lastrowidc C s$ | j s dS | j �� d| _d| _ dS )z<Close the cursor The result will be freed. FNT)r5 rf r* r r r r �close� s zCMySQLCursor.closec C s� |rt |t�std��t |ttf�s,td��d}g | _g }�zpg }g }|r�t|�D ]|\}}|j||d d�} |�| � t |t�r�|�d�| |d �� | � d�| �|d f� qR|�| � | � d�| �|f� qRd �|d � |��} | j j | | j | jd �} g }| j j�rb| j �� } | j �r(t| j �� �} nt| j �� �} d �| �| _| �| � |�| � | j �� q�|| _| �� |�r�| �� d �d � |��} | � | � | �� W S t� W S W nL tjk �r� � Y n4 t k �r� } zt� d�|���W 5 d}~X Y nX dS )z1Calls a stored procedure with the given argumentszprocname must be a stringzargs must be a sequencez@_{name}_arg{index}r )�namer z CAST({0} AS {1})z SET {0}=%sr z CALL {0}({1})�,)rc re z(a result of {0})z SELECT {0}z"Failed calling stored routine; {0}N) r1 rH r� ro rn r0 � enumeraterG r� r{ r� r5 rC rq rs �result_set_available�fetch_eof_columns�CMySQLCursorBufferedRawZ _get_self�CMySQLCursorBufferedr. rU � next_resultrX r; �fetchoner �ErrorrF r2 )r Zprocname�argsZargfmt�results�argnamesZargtypes�idx�arg�argname�callrT �cur�selectrL r r r �callproc� sb � � �zCMySQLCursor.callprocc C sb | j �� s| jdd� dS | jdd� | j jsN| j �� }| �|� tjtd��| �| j � � � dS )z%Skip to the next available result setT�r= NFr& ) r5 r� r; r� Zfetch_eof_statusrU r r2 r r� )r �eofr r r rZ � s zCMySQLCursor.nextsetc C s^ | j jst�d��| j �� }| jr0|�d| j� |s@| �� g S | jt |�7 _| �� |S )zRReturns all rows of a query result set Returns a list of tuples. � No result set to fetch from.r ) r5 � unread_resultr r2 rD r- �insertrX r, r )r �rowsr r r �fetchall� s zCMySQLCursor.fetchallr c C s� | j r| j g}|d8 }ng }|r:| jjr:|�| j�|�� |rb| j�� | _ | j sb| jjsb| j�� |sr| �� g S | j t |�7 _ |S )z,Returns the next set of rows of a result setr ) r- r5 r� �extendrD �get_rowrW r9 rX r, r )r �sizer� r r r � fetchmany� s zCMySQLCursor.fetchmanyc C sd | j }|s | jjr | j�� }|rF| j�� | _ | j sR| jjsR| j�� n | �� dS | jd7 _|S )z&Returns next row of a query result setNr )r- r5 r� r� rW r9 rX r, �r �rowr r r r� s zCMySQLCursor.fetchonec 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 r r r �__iter__ s zCMySQLCursor.__iter__c c s* t t| j��D ]}| j| 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)�ranger r0 )r r] r r r �stored_results s zCMySQLCursor.stored_resultsc C s | � � S )z'Used for iterating over the result set.)�__next__r r r r �next, s zCMySQLCursor.nextc C s4 z | � � }W n tjk r& t�Y nX |s0t�|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. )r� r r2 r[ r� r r r r� 0 s zCMySQLCursor.__next__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. r c S s g | ] }|d �qS )r r )�.0�dr r r � <listcomp>I s z-CMySQLCursor.column_names.<locals>.<listcomp>)r� ro r r r r � column_names? s zCMySQLCursor.column_namesc C s6 z| j �� �d�W S tk r0 | j �� Y S X 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)r. r\ �decode�AttributeErrorr r r r � statementK 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 r r r � with_rowsX s zCMySQLCursor.with_rowsc C sj d}| j rTz| j �d�}W n tk r4 | j }Y nX t|�dkrX|d d� d }nd}|j| jj|d�S )Nz{class_name}: {stmt}zutf-8�( z..z(Nothing executed yet))� class_namerv )r. r� r� r rG r? r )r r� Zexecutedr r r �__str__e s zCMySQLCursor.__str__)T)r F)r )r )&r r r r! rq rr rs r r; rM rN rU rR rX r^ r{ r� r� r" r� r� r� r� r� rZ r� r� r� r� r� r r� r� r� r� r� r� � __classcell__r r r> r r# E sN 87, ? r# c sX e Zd ZdZ� fdd�Zdd� Zd� fdd� Zd d � Zd d � Zddd�Z dd� Z � Z S )r� z*Cursor using C Extension buffering resultsc s t t| ��|� d| _d| _dS )r$ Nr )r: r� r �_rows� _next_rowr6 r> r r r x s z CMySQLCursorBuffered.__init__c C s* | j �� | _d| _t| j�| _| �� dS )rV r N)r5 rD r� r� r r, rX r r r r rR s z&CMySQLCursorBuffered._handle_resultsetTc s" d| _ d| _tt| �j|d� dS )zReset the cursor to defaultNr r� )r� r� r: r� r; r< r> r r r; � s zCMySQLCursorBuffered.resetc C s> d}z| j | j }W n tk r* Y dS X | jd7 _|S )zQReturns the next row in the result set Returns a tuple or None. Nr )r� r� r r� r r r � _fetch_row� s z CMySQLCursorBuffered._fetch_rowc C s4 | j d krt�d��| j | jd � }t| j �| _|S )Nr� )r� r r2 r� r �r �resr r r r� � s z CMySQLCursorBuffered.fetchallr c C s> g }|p | j }|dkr:|d8 }| �� }|r:|�|� qq:q|S )Nr r )Z arraysizer� r� )r r� r� Zcntr� r r r r� � s z CMySQLCursorBuffered.fetchmanyc C s | � � S �N)r� r r r r r� � s z CMySQLCursorBuffered.fetchone)T)r ) r r r r! r rR r; r� r� r� r� r� r r r> r r� t s r� c @ s e Zd ZdZdZdS )�CMySQLCursorRawz+Cursor using C Extension return raw resultsTN�r r r r! rq r r r r r� � s r� c @ s e Zd ZdZdZdS )r� z.Cursor using C Extension buffering raw resultsTNr� r r r r r� � s r� c s> e Zd ZdZdZ� fdd�Zd � fdd� Z� fdd �Z� ZS ) �CMySQLCursorDictz7Cursor using C Extension returning rows as dictionariesFc s* t t| ��� }|r"tt| j|��S dS dS �z/Returns all rows of a query result set N)r: r� r� rk �zipr� r� r> r r r� � s zCMySQLCursorDict.fetchoner c s$ t t� �j|d�}� fdd�|D �S )z0Returns next set of rows as list of dictionaries�r� c s g | ]}t t� j|���qS r �rk r� r� �r� r� r r r r� � s z.CMySQLCursorDict.fetchmany.<locals>.<listcomp>)r: r� r� �r r� r� r> r r r� � s zCMySQLCursorDict.fetchmanyc s t t� ��� }� fdd�|D �S )z>Returns all rows of a query result set as list of dictionariesc s g | ]}t t� j|���qS r r� r� r r r r� � s z-CMySQLCursorDict.fetchall.<locals>.<listcomp>)r: r� r� r� r> r r r� � s zCMySQLCursorDict.fetchall)r ) r r r r! rq r� r� r� r� r r r> r r� � s r� c s0 e Zd ZdZdZ� fdd�Z� fdd�Z� ZS )�CMySQLCursorBufferedDictzECursor using C Extension buffering and returning rows as dictionariesFc s* t t| ��� }|r"tt| j|��S d S d S r� )r: r� r� rk r� r� r� r> r r r� � s z#CMySQLCursorBufferedDict._fetch_rowc s t t� ��� }� fdd�|D �S )Nc s g | ]}t t� j|���qS r r� r� r r r r� � s z5CMySQLCursorBufferedDict.fetchall.<locals>.<listcomp>)r: r� r� r� r> r r r� � s z!CMySQLCursorBufferedDict.fetchall)r r r r! rq r� r� r� r r r> r r� � s r� c sF e Zd ZdZ� fdd�Z� fdd�Zd � fdd� Z� fd d �Z� ZS ) �CMySQLCursorNamedTuplez7Cursor using C Extension returning rows as named tuplesc s t t| ��� td| j�| _dS )rV �RowN)r: r� rR r r� � named_tupler r> r r rR � s z(CMySQLCursorNamedTuple._handle_resultsetc s$ t t| ��� }|r | j|� S dS dS r� )r: r� r� r� r� r> r r r� � s z CMySQLCursorNamedTuple.fetchoner c s$ t t� �j|d�}� fdd�|D �S )z0Returns next set of rows as list of named tuplesr� c s g | ]}� j |� �qS r �r� r� r r r r� s z4CMySQLCursorNamedTuple.fetchmany.<locals>.<listcomp>)r: r� r� r� r> r r r� s z CMySQLCursorNamedTuple.fetchmanyc s t t� ��� }� fdd�|D �S )z>Returns all rows of a query result set as list of named tuplesc s g | ]}� j |� �qS r r� r� r r r r� s z3CMySQLCursorNamedTuple.fetchall.<locals>.<listcomp>)r: r� r� r� r> r r r� s z CMySQLCursorNamedTuple.fetchall)r ) r r r r! rR r� r� r� r� r r r> r r� � s r� c s8 e Zd ZdZ� fdd�Z� fdd�Z� fdd�Z� ZS )� CMySQLCursorBufferedNamedTuplezECursor using C Extension buffering and returning rows as named tuplesc s t t| ��� td| j�| _d S )Nr� )r: r� rR r r� r� r r> r r rR s z0CMySQLCursorBufferedNamedTuple._handle_resultsetc s$ t t| ��� }|r | j|� S d S d S r� )r: r� r� r� r� r> r r r� s z)CMySQLCursorBufferedNamedTuple._fetch_rowc s t t� ��� }� fdd�|D �S )Nc s g | ]}� j |� �qS r r� r� r r r r� s z;CMySQLCursorBufferedNamedTuple.fetchall.<locals>.<listcomp>)r: r� r� r� r> r r r� s z'CMySQLCursorBufferedNamedTuple.fetchall)r r r r! rR r� r� r� r r r> r r� s r� c s e Zd ZdZ� fdd�Z� ZS )�CMySQLCursorPreparedz#Cursor using Prepare Statement c s t t| ��|� td��d S )Nz/Alternative: Use connection.MySQLCursorPrepared)r: r� r �NotImplementedErrorr6 r> r r r & s �z CMySQLCursorPrepared.__init__)r r r r! r r� r r r> r r� ! s r� )#r! � collectionsr r� r3 Z abstractsr r Zcatch23r r r` r � errorcoder �cursorr r r r r r Z_mysql_connectorr �objectr r# r� r� r� r� r� r� r� r� r r r r �<module> s. 3>