관리-도구
편집 파일: connection_cext.cpython-38.pyc
U )�e'Q � @ s� d Z ddlmZ ddlmZ ddlmZmZmZm Z m Z ddl m Z m Z ddlmZ dZzDdd lZdd lmZmZmZmZmZmZmZmZmZ dd lm Z W n4 e k r� Z ze d � e e ����W 5 d Z [ X Y nX d ZG dd� de �Z!d S )z'Connection class using the C Extension � )�errors)� INT_TYPES)� CharacterSet� FieldFlag� ServerFlag� ShutdownType� ClientFlag)�MySQLConnectionAbstract�MySQLCursorAbstract)� MySQLProtocolF� N) � CMySQLCursor�CMySQLCursorRaw�CMySQLCursorBuffered�CMySQLCursorBufferedRaw�CMySQLCursorPrepared�CMySQLCursorDict�CMySQLCursorBufferedDict�CMySQLCursorNamedTuple� CMySQLCursorBufferedNamedTuple)�MySQLInterfaceErrorz6MySQL Connector/Python C Extension not available ({0})Tc s� e Zd ZdZ� fdd�Zdd� Zedd� �ZdYd d �Zed d � �Z e j d d � �Z edd� �Z e j dd� �Z edd� �Z dd� Z dd� ZeZdd� ZdZd d �Zd d � Zd d!� Zed"d#� �Zd[d%d&�Zd\d'd(�Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d]d7d8�Z e Z d^d9d:�Z ed;d<� �Z!ed=d>� �Z"ed?d@� �Z#edAdB� �Z$edCdD� �Z%dEdF� Z&dGdH� Z'd_dKdL�Z(dMdN� Z)dOdP� Z*d`dQdR�Z+dSdT� Z,dUdV� Z-dWdX� Z.� Z/S )a�CMySQLConnectionz5Class initiating a MySQL Connection using Connector/Cc sR t s td��d| _d| _g | _d| _tt| �jf |� t |�dkrN| j f |� dS )ZInitializationz0MySQL Connector/Python C Extension not availableN� r ) � HAVE_CMYSQL� RuntimeError�_cmysql�_connection_timeout�_columns� converter�superr �__init__�len�connect)�self�kwargs�� __class__� �t/opt/hc_python/lib64/python3.8/site-packages/../../../lib/python3.8/site-packages/mysql/connector/connection_cext.pyr = s � zCMySQLConnection.__init__c C sF | j �� | j �� | j �� dddd| j �� d�| _| �| jd �| _dS )z<Gather information of the MySQL server before authenticationN)�protocol�server_version_originalZserver_threadid�charset� server_status� auth_pluginZ auth_dataZ capabilitiesr* )r Zget_proto_infoZget_server_info� thread_idZst_server_capabilitiesZ _handshakeZ_check_server_versionZ_server_version�r# r' r' r( � _do_handshakeK s � �z CMySQLConnection._do_handshakec C s | j �� S )z6Returns the server status attribute of MYSQL structure)r Zst_server_statusr/ r' r' r( �_server_status\ s z CMySQLConnection._server_statusTc C s. || _ | jr| j�|� | jr*| j�|� dS )zuToggle unicode mode Set whether we return string fields as unicode or not. Default is True. N)� _use_unicoder � use_unicoder � set_unicode�r# �valuer' r' r( r4 a s z CMySQLConnection.set_unicodec C s | � d�d }|dkrdS dS )z#Get whether autocommit is on or offzSELECT @@session.autocommitr r TF�� info_queryr5 r' r' r( � autocommitm s zCMySQLConnection.autocommitc C sT z| j �|� || _W n8 tk rN } ztj|j|j|jd��W 5 d}~X Y nX dS )zToggle autocommit��msg�errno�sqlstateN) r r9 Z _autocommitr r �get_mysql_exceptionr; r< r= )r# r6 �excr' r' r( r9 s s �c C s | � d�d S )zGet the current databasezSELECT DATABASE()r r7 r/ r' r' r( �database} s zCMySQLConnection.databasec C s | j �|� dS )zSet the current databaseN)r � select_dbr5 r' r' r( r@ � s c C s | j tj@ S )z'MySQL session has started a transaction)r1 r ZSTATUS_IN_TRANSr/ r' r' r( �in_transaction� s z CMySQLConnection.in_transactionc C s� t �| j�d }tj| j| j|t| jp(d�| j | j d�| _ | j | j | j| j| j| j| j| �tj�d�}| �tj�r�|�| jd | jd | jd | jd d �� z| j jf |� W n8 tk r� } ztj |j |j |j d ��W 5 d }~X Y nX | � � d S ) Nr � )�buffered�raw� charset_nameZconnection_timeoutr3 r- )�host�user�passwordr@ �portZ client_flagsZ unix_socket�compress�ca�cert�keyZ verify_cert)Zssl_caZssl_certZssl_keyZssl_verify_certr: )!r �get_info� _charset_id�_mysql_connector�MySQL� _buffered�_raw�intr r2 Z _auth_pluginr �_host�_userZ _passwordZ _database�_portZ _client_flags� _unix_socketZisset_client_flagr ZCOMPRESSZSSL�update�_sslr" r r r> r; r< r= r0 )r# rF Z cnx_kwargsr? r' r' r( �_open_connection� s@ � � � �z!CMySQLConnection._open_connectionc C sX | j rTz| j �� W n8 tk rL } ztj|j|j|jd��W 5 d}~X Y nX d| _ dS )z Disconnect from the MySQL serverr: N)r �closer r r> r; r< r= )r# r? r' r' r( r] � s �zCMySQLConnection.closec C s | j r| j �� S dS )z;Reports whether the connection to MySQL Server is availableF)r �pingr/ r' r' r( � is_connected� s z CMySQLConnection.is_connectedFr r c C sR d}z| j �� }W n tk r& Y n X |r0dS |rD| j||d� n t�|��dS )a� Check availability of the MySQL server When reconnect is set to True, one or more attempts are made to try to reconnect to the MySQL server using the reconnect()-method. delay is the number of seconds to wait between each retry. When the connection is not available, an InterfaceError is raised. Use the is_connected()-method if you just want to check the connection without raising an error. Raises InterfaceError on errors. z$Connection to MySQL is not availableN)�attempts�delay)r r^ �AttributeError� reconnectr �InterfaceError)r# rc r` ra �errmsg� connectedr' r' r( r^ � s zCMySQLConnection.pingc C s | j �|� dS )zDSets the default character set name for current connection. N)r Zset_character_set)r# r+ r' r' r( �set_character_set_name� s z'CMySQLConnection.set_character_set_namec C sN | j �|� d}| j jr@| j �� }| j �� r@| j �� t�d��| j �� |S )z%Send a query which only returns 1 rowr' z'Query should not return more than 1 row)r �query�have_result_set� fetch_row� free_resultr rd )r# rh � first_rowr' r' r( r8 � s � zCMySQLConnection.info_queryc C s& z | j �� W S tk r Y nX dS )zMySQL connection IDN)r r. r r/ r' r' r( � connection_id� s z CMySQLConnection.connection_idNc C s� | j r | jst�d��g }|dk r2|dkr2td��d}z�| j �� }|r�| jr�t|�}t|�D ]$\}}| j� | j | || �||< q\t |�}|� |� |d7 }|r�||kr�q�| j �� }qBW n@ t k r� } z"| �� tj| j| j| jd��W 5 d} ~ X Y nX |S )z8Get all or a subset of rows returned by the MySQL serverzNo result set availableNr z$count should be 1 or higher, or Noner r: )r � unread_resultr � InternalErrorrb rj r �list� enumerateZ to_pythonr �tuple�appendr rk r> r; r< r= ) r# �count�binary�columns�rows�counter�row�i�_r? r' r' r( �get_rows� s6 � �zCMySQLConnection.get_rowsc C s2 z| j d||d�d W S tk r, Y dS X dS )z.Get the next rows returned by the MySQL serverr )rt ru rv r N)r| � IndexError)r# ru rv r' r' r( �get_row s zCMySQLConnection.get_rowc C s | j r| j �� | j �� S dS )zReads the next resultN)r �consume_result� next_resultr/ r' r' r( r� $ s z CMySQLConnection.next_resultc C s | j r| j �� dS )zFrees the resultN)r rk r/ r' r' r( rk + s z CMySQLConnection.free_resultc C s | j r| j �� dS )zCommit current transactionN)r �commitr/ r' r' r( r� 0 s zCMySQLConnection.commitc C s | j r| j �� | j �� dS )z Rollback current transactionN)r r �rollbackr/ r' r' r( r� 5 s zCMySQLConnection.rollbackc C sN z| j �|� W n8 tk rH } ztj|j|j|jd��W 5 d}~X Y nX dS )zChange the current databaser: N)r rA r r r> r; r< r= )r# r@ r? r' r' r( � cmd_init_db; s �z CMySQLConnection.cmd_init_dbc C s� | j jst�d��| j �� }g | _|D ]D}| j�|d t|d �ddddt|d � tj @ t|d �f� q&| j | j � � d�| jd�S )z Fetch EOF and column information� No result set� � N� )Z status_flag� warning_count)�eofrv ) r ri r rd Z fetch_fieldsr rs rU r ZNOT_NULLr1 �st_warning_count)r# �fields�colr' r' r( �fetch_eof_columnsC s( � ��z"CMySQLConnection.fetch_eof_columnsc C s4 | j r0| j �� | j �� | j �� | j �� | jd�S dS )z Fetch EOF and status information)r� Z field_count� insert_id� affected_rowsr, N)r r� Zst_field_countr� r� r1 r/ r' r' r( �fetch_eof_status^ s �z!CMySQLConnection.fetch_eof_statusc C s� | � � z,t|t�s |�d�}| jj||||d� W n~ tk rl } ztj|j |j |j d��W 5 d}~X Y nH t k r� | j r�| j }n| jd t| j� }tjd|dfd��Y nX g | _| jjs�| �� S | �� S ) z Send a query to the MySQL serverzutf-8)rE rD � raw_as_string)r; r= N�:i zConnection not available.)r< �values)�handle_unread_result� isinstance�bytes�encoder rh r r r> r< r; r= rb rY rV �strrX �OperationalErrorr ri r� r� )r# rh rE rD r� r? �addrr' r' r( � cmd_queryk s2 � � � zCMySQLConnection.cmd_queryc s | � � | �� st�d��|dk r>t|t�s6t�d��|| �S |pF| j}|pP| j}d�|dkrf�dO �|dkrv�dO �|dkr��dO �|dkr��d O �|dkr��d O �t t t t t ttttd � }z|� | �W S tk �r d � td d�� �fdd�td�D �� ��Y nX dS )a� Instantiates and returns a cursor using C Extension By default, CMySQLCursor is returned. Depending on the options while connecting, a buffered and/or raw cursor is instantiated instead. Also depending upon the cursor options, rows can be returned as dictionary or named tuple. Dictionary and namedtuple based cursors are available with buffered output but not raw. It is possible to also give a custom cursor through the cursor_class parameter, but it needs to be a subclass of mysql.connector.cursor_cext.CMySQLCursor. Raises ProgrammingError when cursor_class is not a subclass of CursorBase. Raises ValueError when cursor is not available. Returns instance of CMySQLCursor or subclass. :param buffered: Return a buffering cursor :param raw: Return a raw cursor :param prepared: Return a cursor which uses prepared statements :param cursor_class: Use a custom cursor class :param dictionary: Rows are returned as dictionary :param named_tuple: Rows are returned as named tuple :return: Subclass of CMySQLCursor :rtype: CMySQLCursor or subclass z MySQL Connection not available.Nz=Cursor class needs be to subclass of cursor_cext.CMySQLCursorr Tr r r� r� � ) r r r � r� � r� r� r� )rD rE � dictionary� named_tuple�preparedz*Cursor not available with given criteria: z, c s$ g | ] }�d |> @ dkr� | �qS )r r r' )�.0rz ��argsZ cursor_typer' r( � <listcomp>� s �z+CMySQLConnection.cursor.<locals>.<listcomp>r� )r� r_ r r� � issubclassr ZProgrammingErrorrS rT r r r r r r r r r �KeyError� ValueError�join�range)r# rD rE r� Z cursor_classr� r� �typesr' r� r( �cursor� sN � � �zCMySQLConnection.cursorc C s | j jst�d��| j �� S )z,Returns number of rows of current result setr� )r ri r rd �num_rowsr/ r' r' r( r� � s zCMySQLConnection.num_rowsc C s | j s dS | j �� S )zReturns number of warningsr )r r� r/ r' r' r( r� � s z CMySQLConnection.warning_countc C s | j s dS | j jS )z"Check if a result set is availableF)r ri r/ r' r' r( �result_set_available� s z%CMySQLConnection.result_set_availablec C s | j S )z)Check if there are unread results or rows)r� r/ r' r' r( rn � s z CMySQLConnection.unread_resultc C s | j �� S )z Check if there are more results)r � more_resultsr/ r' r' r( r� � s z CMySQLConnection.more_resultsc C s^ t |ttf�r | jj|� }n>t |t�rRi }|�� D ] \}}| j�|�d ||< q2ntd��|S )z�Prepare parameters for statements This method is use by cursors to prepared parameters found in the list (or tuple) params. Returns dict. r z Could not process parameters)r� rp rr r Zconvert_to_mysql�dict�itemsr� )r# �params�resultrN r6 r' r' r( �prepare_for_mysql� s z"CMySQLConnection.prepare_for_mysqlc C s | j �� dS )zlConsume the current result This method consume the result by reading (consuming) all rows. N)r r r/ r' r' r( �consume_results s z CMySQLConnection.consume_results� �! c C s` z| j �|||� W n8 tk rL } ztj|j|j|jd��W 5 d}~X Y nX || _| � � dS )z!Change the current logged in userr: N) r Z change_userr r r> r; r< r= rP Z_post_connection)r# �usernamerI r@ r+ r? r' r' r( �cmd_change_user s �z CMySQLConnection.cmd_change_userc C sR z| j �|� W n8 tk rH } ztj|j|j|jd��W 5 d}~X Y nX | �� S )z,Send the Refresh command to the MySQL serverr: N) r �refreshr r r> r; r< r= r� )r# �optionsr? r' r' r( � cmd_refresh s �z CMySQLConnection.cmd_refreshc C s | � � dS )z,Close the current connection with the serverN)r] r/ r' r' r( �cmd_quit$ s zCMySQLConnection.cmd_quitc C s� | j st�d��|r.t�|�s(t�d��|}ntj}z| j �|� W n8 tk r| } ztj |j |j |j d��W 5 d}~X Y nX | � � dS )zShut down the MySQL Serverz MySQL Connection not availablezInvalid shutdown typer: N)r r r� r rO rd ZSHUTDOWN_DEFAULT�shutdownr r> r; r< r= r] )r# Z shutdown_type�levelr? r' r' r( � cmd_shutdown( s �z CMySQLConnection.cmd_shutdownc C sh | � � z | j�� }t� j|dd�W S ttjfk rb } ztj|j |j |j d��W 5 d}~X Y nX dS )z'Return statistics from the MySQL serverF)Z with_headerr: N) r� r �statr Zparse_statisticsr r rd r> r; r< r= )r# r� r? r' r' r( �cmd_statistics< s �z CMySQLConnection.cmd_statisticsc C s& t |t�std��| �d�|�� dS )zKill a MySQL processzMySQL PID must be intzKILL {0}N)r� r r� r8 �format)r# Z mysql_pidr' r' r( �cmd_process_killG s z!CMySQLConnection.cmd_process_killc C s$ | j r| �� n| jr t�d��dS )z'Check whether there is an unread resultzUnread result foundN)Zcan_consume_resultsr� rn r ro r/ r' r' r( r� M s z%CMySQLConnection.handle_unread_result)T)Fr r )NFN)FN)FFF)NNNNNN)r� r� r� r� )N)0�__name__� __module__� __qualname__�__doc__r r0 �propertyr1 r4 r9 �setterr@ rB r\ r] Z disconnectr_ r^ rg r8 rm r| r~ r� rk r� r� r� r� r� r� Z_execute_queryr� r� r� r� rn r� r� r� r� r� r� r� r� r� r� � __classcell__r' r' r% r( r 9 sz % � J � r )"r� r� r Zcatch23r � constantsr r r r r Z abstractsr r r) r r rQ Z cursor_cextr r r r r r r r r r � ImportErrorr? r� r� r r' r' r' r( �<module> s$ ,��