HEX

File: //opt/alt/python38/lib/python3.8/site-packages/pymysql/__pycache__/connections.cpython-38.opt-1.pyc
U

1�X��
@s�ddlmZddlmZmZmZmZmZmZddl	Z	ddl
mZddlZddl
Z
ddlZddlZddlZddlZddlZddlZddlmZmZmZddlmZmZmZmZmZddlmZm Z m!Z!m"Z#dd	l$m%Z%dd
l&m'Z'ddl(m)Z)m*Z*ddl+m,Z,zddl-Z-d
Z.Wne/k
�r*dZ-dZ.YnXzddl0Z0e0�1�Z2[0Wne/e3fk
�r`dZ2YnXdZ4ej5dd�Z6e6dk�r�e�s�ddl7m8Z8dd�Z9n,e6dk�r�Gdd�de:�Z;dd�Z9ndd�Z9e<ej=ej>ej?ej@ejAejBejCejDejEg	�ZFeejGd�ZHdZIdZJdZKdZLdZMdZNdZOd d!�ZPd"d#�ZQd$d%�ZRd&ZSGd'd(�d(e:�ZTd)d*�ZUd+d,�ZVd-d.�ZWd/d0�ZXGd1d2�d2e:�ZYGd3d4�d4eY�ZZGd5d6�d6e:�Z[Gd7d8�d8e:�Z\Gd9d:�d:e:�Z]Gd;d<�d<e:�Z^Gd=d>�d>e:�Z_Gd?d@�d@e:�Z`dS)A�)�print_function�)�PY2�
range_type�	text_type�str_type�JYTHON�
IRONPYTHONN)�partial)�MBLENGTH�charset_by_name�
charset_by_id)�CLIENT�COMMAND�CR�
FIELD_TYPE�
SERVER_STATUS)�escape_item�
escape_string�through�conversions)�Cursor)�Parser)�byte2int�int2byte)�errTF�)r�)�SocketIOcCst�t||��S�N)�io�BufferedReaderr��sock�mode�r%�D/opt/alt/python38/lib/python3.8/site-packages/pymysql/connections.py�	_makefile8sr')r�c@seZdZdd�Zdd�ZdS)�SockFilecCs
||_dSr��_sock)�selfr#r%r%r&�__init__>szSockFile.__init__cCsV|j�|�}t|�|kr|S|j�|t|��}|s8|S||7}t|�|kr|SqdSr)r+Zrecv�len)r,�n�read�datar%r%r&r0Asz
SockFile.readN)�__name__�
__module__�__qualname__r-r0r%r%r%r&r)=sr)cCs|dkst�t|�S)N�rb)�AssertionErrorr)r"r%r%r&r'MscCs
|�|�Sr)Zmakefiler"r%r%r&r'RsZsha1�����latin1��c	s�dd��zLtdt���tdd�D]&}t�|�}td||jj|jf�q"td�Wntk
rhYnX�fdd	�t	d
t
t��d�d�D�}|D]H}td
�tdd�|��ddt|�dd�t�fdd�|���q�td�t�dS)NcSs6dt|�krdkr2nnt|t�r.t|�S|SdS)N�A�z�.)r�
isinstance�int�chr�r1r%r%r&�is_asciios

zdump_packet.<locals>.is_asciizpacket length:rr(zcall[%d]: %s (line %d)zB------------------------------------------------------------------csg|]}�||d��qS)�r%)�.0�irCr%r&�
<listcomp>~szdump_packet.<locals>.<listcomp>r�rE� cSsd�t|��S)Nz{:02X})�formatr��xr%r%r&�<lambda>��zdump_packet.<locals>.<lambda>z   z  �csd��|��S)Nz{})rKrL)rDr%r&rN�rO)
�printr.�range�sys�	_getframe�f_code�co_name�f_lineno�
ValueErrorr�min�join�map)r1rG�fZ	dump_data�dr%)r1rDr&�dump_packetns(
$���r^cCs`|sdStrtdt|��t|���}t|���}t�}|�|�|�|�|��}t||�S)NrOz	password=)�DEBUGrQ�str�sha_newZdigest�update�	_my_crypt)�password�messageZstage1Zstage2�s�resultr%r%r&�	_scramble�s

rhc	Csft|�}d}t|�D]L}t�d|||d��dt�d|||d��dA}|t�d|�7}q|S)NrO�Brr)r.r�struct�unpack�pack)Zmessage1Zmessage2�lengthrgrGrMr%r%r&rc�s�rc�c@seZdZdd�Zdd�ZdS)�RandStruct_323cCs"d|_||j|_||j|_dS)Ni���?)�	max_value�seed1�seed2)r,rqrrr%r%r&r-�szRandStruct_323.__init__cCsD|jd|j|j|_|j|jd|j|_t|j�t|j�S)N��!)rqrrrp�float�r,r%r%r&�my_rnd�szRandStruct_323.my_rndN)r2r3r4r-rwr%r%r%r&ro�srocCs�t|�}t|dt��}t�d|�}t�d|�}t|d|dA|d|dA�}t��}tttt	|���D]"}|�
tt|�
�d�d��qltt|�
�d��}	|��}
t��}|
D]}|�
tt|�t|	�A��q�|��S)N�>LLrr��@)�_hash_password_323�SCRAMBLE_LENGTH_323rjrkror �BytesIOrrYr.�writerrArw�getvaluer)rdreZ	hash_passZhash_messageZhash_pass_nZhash_message_nZrand_stZoutbuf�_Zextra�out�cr%r%r&�
_scramble_323�s � r�cCs~d}d}d}dd�|D�D]D}||d@|||d>d@N}||d>|Ad@}||d@}q|d	@}|d	@}t�d
||�S)Ni5W0PriqV4cSsg|]}|dkrt|��qS))rJ�	� �	)r)rFrMr%r%r&rH�sz&_hash_password_323.<locals>.<listcomp>�?rnl��i���rx�rjrl)rdZnr�addZnr2r�Zr1Zr2r%r%r&r{�s r{cCst�d|�dd�S)N�<Irsr�)r/r%r%r&�
pack_int24�sr�cCs�|dkrtd|��np|dkr&t|�S|dkr>dt�d|�S|dkr^dt�d	|�dd
�S|dkrvdt�d
|�Std|df��dS)NrzFEncoding %d is less than 0 - no representation in LengthEncodedIntegerr7i���<H���r�rsl���<QzIEncoding %x is larger than %x - no representation in LengthEncodedInteger)rXrrjrl)rGr%r%r&�
lenenc_int�sr�c@s�eZdZdZdZdd�Zdd�Zdd�Zd	d
�Zdd�Z	d7dd�Z
d8dd�ZerZdd�Z
ndd�Z
dd�Zdd�Zdd�Zdd�Zdd�Zd d!�Zd"d#�Zd$d%�Zd&d'�Zd(d)�Zd*d+�Zd,d-�Zd.d/�Zd0d1�Zd2d3�Zd4d5�Zd6S)9�MysqlPacketzrRepresentation of a MySQL response packet.

    Provides an interface for reading/parsing the packet results.
    ��	_position�_datacCsd|_||_dS�Nrr��r,r1�encodingr%r%r&r-�szMysqlPacket.__init__cCs|jSr�r�rvr%r%r&�get_all_data�szMysqlPacket.get_all_datacCsl|j|j|j|�}t|�|krZd|t|�|jt|j�f}trRt|�|��t|��|j|7_|S)zCRead the first 'size' bytes in packet and advance cursor past them.z\Result length not requested length:
Expected=%s.  Actual=%s.  Position: %s.  Data Length: %s)r�r�r.r_rQ�dumpr6)r,�sizerg�errorr%r%r&r0�s�zMysqlPacket.readcCs|j|jd�}d|_|S)z`Read all remaining data in the packet.

        (Subsequent read() will return errors.)
        N�r�r��r,rgr%r%r&�read_all�szMysqlPacket.read_allcCs:|j|}|dks |t|j�kr0td||f��||_dS)z1Advance the cursor in data buffer 'length' bytes.rz4Invalid advance amount (%s) for cursor.  Position=%sN)r�r.r��	Exception)r,rmZnew_positionr%r%r&�advances
�zMysqlPacket.advancercCs,|dks|t|j�kr"td|��||_dS)z9Set the position of the data buffer cursor to 'position'.rz)Invalid position to rewind cursor to: %s.N)r.r�r�r�)r,�positionr%r%r&�rewindszMysqlPacket.rewindrcCs|j|||�S)a>Get 'length' bytes starting at 'position'.

        Position is start of payload (first four packet header bytes are not
        included) starting at index '0'.

        No error checking is done.  If requesting outside end of buffer
        an empty string (or string shorter than 'length') may be returned!
        r�)r,r�rmr%r%r&�	get_bytess	zMysqlPacket.get_bytescCs"t|j|j�}|jd7_|S�Nr)�ordr�r�r�r%r%r&�
read_uint8!szMysqlPacket.read_uint8cCs|j|j}|jd7_|Sr�r�r�r%r%r&r�&scCs(t�d|j|j�d}|jd7_|S)Nr�rr�rj�unpack_fromr�r�r�r%r%r&�read_uint16+szMysqlPacket.read_uint16cCs0t�d|j|j�\}}|jd7_||d>S)Nz<HBrsrEr�)r,ZlowZhighr%r%r&�read_uint240szMysqlPacket.read_uint24cCs(t�d|j|j�d}|jd7_|S)Nr�r�r�r�r%r%r&�read_uint325szMysqlPacket.read_uint32cCs(t�d|j|j�d}|jd7_|S)Nr�rrnr�r�r%r%r&�read_uint64:szMysqlPacket.read_uint64cCs:|j�d|j�}|dkrdS|j|j|�}|d|_|S)N�rr)r��findr�)r,Zend_posrgr%r%r&�read_string?s
zMysqlPacket.read_stringcCsT|��}|tkrdS|tkr |S|tkr0|��S|tkr@|��S|tkrP|��SdS)z�Read a 'Length Coded Binary' number from the data buffer.

        Length coded numbers can be anywhere from 1 to 9 bytes depending
        on the value of the first byte.
        N)	r��NULL_COLUMN�UNSIGNED_CHAR_COLUMN�UNSIGNED_SHORT_COLUMNr��UNSIGNED_INT24_COLUMNr��UNSIGNED_INT64_COLUMNr�)r,r�r%r%r&�read_length_encoded_integerGsz'MysqlPacket.read_length_encoded_integercCs|��}|dkrdS|�|�S)aRead a 'Length Coded String' from the data buffer.

        A 'Length Coded String' consists first of a length coded
        (unsigned, positive) integer represented in 1-9 bytes followed by
        that many bytes of binary data.  (For example "cat" would be "3cat".)
        N)r�r0)r,rmr%r%r&�read_length_coded_stringYsz$MysqlPacket.read_length_coded_stringcCs.t�|�}|�|j|j�}|j|j7_|Sr)rjZStructr�r�r�r�)r,Zfmtrfrgr%r%r&�read_structes
zMysqlPacket.read_structcCs |jdd�dkot|j�dkS)Nrrr�r�r�r.rvr%r%r&�is_ok_packetkszMysqlPacket.is_ok_packetcCs |jdd�dkot|j�dkS)Nrrr�r�r�rvr%r%r&�
is_eof_packetoszMysqlPacket.is_eof_packetcCs|jdd�dkS)Nrrr�r�rvr%r%r&�is_auth_switch_requestusz"MysqlPacket.is_auth_switch_requestcCs*t|jdd��}d|ko$dkSS)Nrr�)r�r�)r,�field_countr%r%r&�is_resultset_packetyszMysqlPacket.is_resultset_packetcCs|jdd�dkS)Nrr��r�rvr%r%r&�is_load_local_packet}sz MysqlPacket.is_load_local_packetcCs|jdd�dkS)Nrr��r�rvr%r%r&�is_error_packet�szMysqlPacket.is_error_packetcCs@|��r<|��|�d�|��}tr0td|�t�|j�dS)Nrzerrno =)	r�r�r�r�r_rQrZraise_mysql_exceptionr�)r,�errnor%r%r&�check_error�s

zMysqlPacket.check_errorcCst|j�dSr)r^r�rvr%r%r&r��szMysqlPacket.dumpN)r)r)r2r3r4�__doc__�	__slots__r-r�r0r�r�r�r�rr�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r%r%r%r&r��s8	


r�c@s8eZdZdZdd�Zdd�Zdd�Zdd	�Zd
d�ZdS)
�FieldDescriptorPacketz�A MysqlPacket that represents a specific column's metadata in the result.

    Parsing is automatically done and the results are exported via public
    attributes on the class such as: db, table_name, name, length, type_code.
    cCst�|||�|�|�dSr)r�r-�_parse_field_descriptorr�r%r%r&r-�szFieldDescriptorPacket.__init__cCsv|��|_|��|_|���|�|_|���|�|_|���|�|_|���|�|_|�d�\|_	|_
|_|_|_
dS)z�Parse the 'Field Descriptor' (Metadata) packet.

        This is compatible with MySQL 4.1+ (not compatible with MySQL 4.0).
        z	<xHIBHBxxN)r�Zcatalog�db�decode�
table_nameZ	org_table�nameZorg_namer��	charsetnrrm�	type_code�flags�scale)r,r�r%r%r&r��s

�z-FieldDescriptorPacket._parse_field_descriptorcCs*|j|jd|��|��|j|jddkfS)zBProvides a 7-item tuple compatible with the Python PEP249 DB Spec.Nrr)r�r��get_column_lengthr�r�rvr%r%r&�description�s�z!FieldDescriptorPacket.descriptioncCs*|jtjkr$t�|jd�}|j|S|jSr�)r�r�
VAR_STRINGr�getr�rm)r,Zmblenr%r%r&r��s
z'FieldDescriptorPacket.get_column_lengthcCs d|j|j|j|j|j|jfS)Nz%s %r.%r.%r, type=%s, flags=%x)�	__class__r�r�r�r�r�rvr%r%r&�__str__�s��zFieldDescriptorPacket.__str__N)	r2r3r4r�r-r�r�r�r�r%r%r%r&r��sr�c@s eZdZdZdd�Zdd�ZdS)�OKPacketWrapperz�
    OK Packet Wrapper. It uses an existing packet object, and wraps
    around it, exposing useful variables while still providing access
    to the original packet objects variables and methods.
    cCsz|��s tdt|jj�d��||_|j�d�|j��|_|j��|_	|�
d�\|_|_|j�
�|_|jtj@|_dS)NzCannot create z  object from invalid packet typerz<HH)r�rXr`r�r2�packetr�r��
affected_rows�	insert_idr��
server_status�
warning_countr�rer�SERVER_MORE_RESULTS_EXISTS�has_next�r,Zfrom_packetr%r%r&r-�s�zOKPacketWrapper.__init__cCst|j|�Sr��getattrr��r,�keyr%r%r&�__getattr__�szOKPacketWrapper.__getattr__N�r2r3r4r�r-r�r%r%r%r&r��sr�c@s eZdZdZdd�Zdd�ZdS)�EOFPacketWrapperz�
    EOF Packet Wrapper. It uses an existing packet object, and wraps
    around it, exposing useful variables while still providing access
    to the original packet objects variables and methods.
    cCsT|��std�|j���||_|j�d�\|_|_trBt	d|j�|jt
j@|_dS)N�3Cannot create '{0}' object from invalid packet typez<xhhzserver_status=)
r�rXrKr�r�r�r�r�r_rQrr�r�r�r%r%r&r-�s��zEOFPacketWrapper.__init__cCst|j|�Srr�r�r%r%r&r��szEOFPacketWrapper.__getattr__Nr�r%r%r%r&r��sr�c@s eZdZdZdd�Zdd�ZdS)�LoadLocalPacketWrapperz�
    Load Local Packet Wrapper. It uses an existing packet object, and wraps
    around it, exposing useful variables while still providing access
    to the original packet objects variables and methods.
    cCsF|��std�|j���||_|j��dd�|_trBtd|j�dS)Nr�rz	filename=)	r�rXrKr�r�r��filenamer_rQr�r%r%r&r-�s��zLoadLocalPacketWrapper.__init__cCst|j|�Srr�r�r%r%r&r�sz"LoadLocalPacketWrapper.__getattr__Nr�r%r%r%r&r��s
r�c@seZdZdZdZdZdZddddddddddddedddddddddddddidddfdd	�Zd
d�Z	dd
�Z
edd��Zdd�Z
e
Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�Zd d!�Zd"d#�Zd]d$d%�Zd&d'�Zd(d)�Zd^d*d+�Zd,d-�Zd.d/�Zd_d0d1�Zd`d2d3�Zd4d5�Z d6d7�Z!dad9d:�Z"d;d<�Z#dbd=d>�Z$d?d@�Z%e&fdAdB�Z'dCdD�Z(dEdF�Z)dcdGdH�Z*dIdJ�Z+dKdL�Z,dMdN�Z-dOdP�Z.dQdR�Z/dSdT�Z0dUdV�Z1dWdX�Z2dYdZ�Z3d[d\�Z4e5j6Z6e5j7Z7e5j8Z8e5j9Z9e5j:Z:e5j;Z;e5j<Z<e5j=Z=e5j>Z>e5j?Z?dS)d�
Connectionz�
    Representation of a socket with a mysql server.

    The proper way to get an instance of this class is to call
    connect().
    NrPFr�
r�c"s>|dk	rt�dt�|dkr.tjddkr.d}|dk	rB|dkrB|}|dk	rR|sR|}|sZ|rbtd��t|�|_|jr||tj	O}�r�|	s�tj
�d�r�d}	nd	}	|	�rf�s�d
�t����
tj�|	����fdd�}|d
|�}|d|�}|d|�}|d|�}|d|�}t|d|��}|d|�}|d|�}|�s,i}t|t��rfdD](} |d| |�| ��}!|!�r<|!|| <�q<d|_|�r�t�s�td��d|_|tjO}|�|�|_|�p�d|_|�p�d|_|�p�t|_|�p�d|_||_||_ ||_!d|k�r�dk�snt"d��|�pd|_#|dk	�r(|dk�r(t"d��||_$|dk	�rJ|dk�rJt"d��||_%|�rd||_&d|_'nt(|_&d|_'|dk	�r�||_'t)|j&�j*|_*|tj+O}|j�r�|tj,O}||_-|
|_.d|_/d|_0d |_1||_2|
dk�r�t3}
td!d"�|
�4�D��|_5td#d"�|
�4�D��|_6||_7||_8||_9||_:|�r2d|_;n|�<�dS)$aC
        Establish a connection to the MySQL database. Accepts several
        arguments:

        host: Host where the database server is located
        user: Username to log in as
        password: Password to use.
        database: Database to use, None to not use a particular one.
        port: MySQL port to use, default is usually OK. (default: 3306)
        bind_address: When the client has multiple network interfaces, specify
            the interface from which to connect to the host. Argument can be
            a hostname or an IP address.
        unix_socket: Optionally, you can use a unix socket rather than TCP/IP.
        charset: Charset you want to use.
        sql_mode: Default SQL_MODE to use.
        read_default_file:
            Specifies  my.cnf file to read these parameters from under the [client] section.
        conv:
            Conversion dictionary to use instead of the default one.
            This is used to provide custom marshalling and unmarshaling of types.
            See converters.
        use_unicode:
            Whether or not to default to unicode strings.
            This option defaults to true for Py3k.
        client_flag: Custom flags to send to MySQL. Find potential values in constants.CLIENT.
        cursorclass: Custom cursor class to use.
        init_command: Initial SQL statement to run when connection is established.
        connect_timeout: Timeout before throwing an exception when connecting.
            (default: 10, min: 1, max: 31536000)
        ssl:
            A dict of arguments similar to mysql_ssl_set()'s parameters.
            For now the capath and cipher arguments are not supported.
        read_default_group: Group to read from in the configuration file.
        compress; Not supported
        named_pipe: Not supported
        autocommit: Autocommit mode. None means use server default. (default: False)
        local_infile: Boolean to enable the use of LOAD DATA LOCAL command. (default: False)
        max_allowed_packet: Max size of packet sent to server in bytes. (default: 16MB)
            Only used to limit size of "LOAD LOCAL INFILE" data packet smaller than default (16KB).
        defer_connect: Don't explicitly connect on contruction - wait for connect call.
            (default: False)
        auth_plugin_map: A dict of plugin names to a class that processes that plugin.
            The class will take the Connection object as the argument to the constructor.
            The class needs an authenticate method taking an authentication packet as
            an argument.  For the dialog plugin, a prompt(echo, prompt) method can be used
            (if no authenticate method) for returning a string from the user. (experimental)
        db: Alias for database. (for compatibility to MySQLdb)
        passwd: Alias for password. (for compatibility to MySQLdb)
        Nzno_delay option is deprecatedrrTz3compress and named_pipe arguments are not supported�winz	c:\my.iniz/etc/my.cnfZclientcs4|r|Sz���|�WStk
r.|YSXdSr)r�r�)r��arg�Zcfg�read_default_groupr%r&�_configlsz$Connection.__init__.<locals>._config�userrd�host�database�socket�portzbind-addresszdefault-character-set)�ca�capath�certr��cipherzssl-Fzssl module not found�	localhosti�rPi�3�z+connect_timeout should be >0 and <=31536000zread_timeout should be >= 0zwrite_timeout should be >= 0z
Not connectedcSs$g|]\}}t|�tk	r||f�qSr%��typerA�rF�k�vr%r%r&rH�sz'Connection.__init__.<locals>.<listcomp>cSs$g|]\}}t|�tkr||f�qSr%r�rr%r%r&rH�s)=�warnings�warn�DeprecationWarningrS�version_info�NotImplementedError�bool�
_local_infilerZLOCAL_FILES�platform�
startswithrr0�os�path�
expanduserrAr@�dictr��ssl�SSL_ENABLED�SSL�_create_ssl_ctx�ctxr�r��DEFAULT_USERr�rdr��unix_socket�bind_addressrX�connect_timeout�
_read_timeout�_write_timeout�charset�use_unicode�DEFAULT_CHARSETrr�ZCAPABILITIES�CONNECT_WITH_DB�client_flag�cursorclass�_result�_affected_rows�	host_info�autocommit_mode�_conv�items�encoders�decoders�sql_mode�init_command�max_allowed_packet�_auth_plugin_mapr+�connect)"r,r�r�rdr�r�rrr*Zread_default_fileZconvrr r!r+rrr��compressZ
named_pipeZno_delay�
autocommitr�ZpasswdZlocal_infiler,Z
defer_connectZauth_plugin_mapZread_timeoutZ
write_timeoutrr�r��valuer%r�r&r-s�<













zConnection.__init__cCs�t|tj�r|S|�d�}|�d�}|dko2|dk}tj||d�}|oR|�dd�|_|r`tjntj|_d|kr�|j	|d|�d�d�d	|kr�|�
|d	�|jtjO_|jtj
O_|S)
Nr�r�)Zcafiler��check_hostnameTr�r�)Zkeyfiler�)r@rZ
SSLContextr�Zcreate_default_contextr2Z	CERT_NONEZ
CERT_REQUIREDZverify_modeZload_cert_chainZset_ciphers�optionsZOP_NO_SSLv2ZOP_NO_SSLv3)r,Zsslpr�r�Zhasnocarr%r%r&r�s

zConnection._create_ssl_ctxcCsl|jrt�d��d|_|jdkr$dSt�ddtj�}z(z|�	|�Wnt
k
rXYnXW5|��XdS)z*Send the quit message and close the socket�Already closedTN�<iBr)�_closedr�Errorr+rjrlrZCOM_QUIT�_force_close�_write_bytesr�)r,�	send_datar%r%r&�close�s


zConnection.closecCs
|jdk	Srr*rvr%r%r&�open�szConnection.opencCs2|jr"z|j��WnYnXd|_d|_dS)z%Close connection without QUIT messageN)r+r;�_rfilervr%r%r&r8�szConnection._force_closecCs&t|�|_|��}||kr"|��dSr)r	r%�get_autocommit�_send_autocommit_mode)r,r1�currentr%r%r&r0�s
zConnection.autocommitcCst|jtj@�Sr)r	r�rZSERVER_STATUS_AUTOCOMMITrvr%r%r&r>�s�zConnection.get_autocommitcCs0|��}|��st�dd��t|�}|j|_|S)N��zCommand Out of Sync)�_read_packetr�r�OperationalErrorr�r�)r,�pkt�okr%r%r&�_read_ok_packet�szConnection._read_ok_packetcCs&|�tjd|�|j��|��dS)z2Set whether or not to commit after every execute()zSET AUTOCOMMIT = %sN)�_execute_commandr�	COM_QUERY�escaper%rFrvr%r%r&r?s

�z Connection._send_autocommit_modecCs|�tjd�|��dS)zBegin transaction.ZBEGINN�rGrrHrFrvr%r%r&�beginszConnection.begincCs|�tjd�|��dS)z Commit changes to stable storageZCOMMITNrJrvr%r%r&�commitszConnection.commitcCs|�tjd�|��dS)z!Roll back the current transactionZROLLBACKNrJrvr%r%r&�rollbackszConnection.rollbackcCs$|�tjd�t|�}|��|jS)z
SHOW WARNINGS)rGrrH�MySQLResultr0�rowsr�r%r%r&�
show_warningsszConnection.show_warningscCs|�tj|�|��dS)zSet current dbN)rGrZCOM_INIT_DBrF)r,r�r%r%r&�	select_db"szConnection.select_dbcCs,t|t�rd|�|�dSt||j|d�S)z�Escape whatever value you pass to it.
        
        Non-standard, for internal use; do not use this in your applications.
        �')�mapping)r@rrrr)r,�objrSr%r%r&rI's
zConnection.escapecCs|�||j�S)zrAlias for escape()
        
        Non-standard, for internal use; do not use this in your applications.
        )rIr()r,rTr%r%r&�literal0szConnection.literalcCs |jtj@r|�dd�St|�S)NrRz'')r�rZ"SERVER_STATUS_NO_BACKSLASH_ESCAPES�replacer)r,rfr%r%r&r7s
�zConnection.escape_stringcCs|r||�S|�|�S)z+Create a new cursor to execute queries with)r!)r,�cursorr%r%r&rW=szConnection.cursorcCs|��S)z%Context manager that returns a Cursor)rWrvr%r%r&�	__enter__CszConnection.__enter__cCs|r|��n|��dS)z2On successful exit, commit. On exception, rollbackN)rMrL)r,�excr1�	tracebackr%r%r&�__exit__Gs
zConnection.__exit__cCsTt|t�r2ts2ts2tr$|�|j�}n|�|jd�}|�tj	|�|j
|d�|_|jS)N�surrogateescape��
unbuffered)r@rrr	r�encoder�rGrrH�_read_query_resultr#)r,�sqlr^r%r%r&�queryOszConnection.querycCs|j|d�|_|jS)Nr])r`r#)r,r^r%r%r&�next_result[szConnection.next_resultcCs|jSr)r#rvr%r%r&r�_szConnection.affected_rowscCs"t�d|�}|�tj|�|��S)Nr�)rjrlrGrZCOM_PROCESS_KILLrF)r,�	thread_idr�r%r%r&�killbszConnection.killTcCst|jdkr&|r|��d}n
t�d��z|�tjd�|��WStk
rn|rh|��|�	d�YS�YnXdS)zCheck if the server is aliveNFr4rP)
r+r.rr7rGrZCOM_PINGrFr��ping)r,Z	reconnectr%r%r&rfgs


zConnection.pingcCs:t|�j}|�tjd|�|��|��||_||_dS)NzSET NAMES %s)rr�rGrrHrIrBr)r,rr�r%r%r&�set_charsetys

zConnection.set_charsetc
Csdd|_�z�|dk�r*|jr`|jdkr`t�tjtj�}|�|j�|�|j�d|_	t
r^td�n�i}|jdk	r||jdf|d<z"tj
|j|jf|jf|�}Wq�Wq|ttfk
r�}z|jtjkr�WY�q|�W5d}~XYq|Xq|d|j|jf|_	t
r�td�|�tjtjd	�|�d�|�tjtjd	�||_t|d
�|_d|_|��|��|jdk	�rv|��}|� d|jf�|j!dk	�r�|��}|� |j!�|�"�|�#�|j$dk	�r�|�%|j$�Wn�t&k
�r^}z~d|_|dk	�r�z|�"�WnYnXt'|tttj(f��rLt)�*dd
|j|f�}||_+t,�-�|_,t
�rHt|j,�|��W5d}~XYnXdS)NF)r�z	127.0.0.1zLocalhost via UNIX socketzconnected using unix_socketrZsource_addresszsocket %s:%dzconnected using socketrr5zSET sql_mode=%si�z(Can't connect to MySQL server on %r (%s)).r6rr�r�ZAF_UNIXZSOCK_STREAM�
settimeoutrr.r$r_rQrZcreate_connectionr��OSError�IOErrorr��EINTRZ
setsockoptZIPPROTO_TCPZTCP_NODELAYZ
SOL_SOCKETZSO_KEEPALIVEr+r'r=�_next_seq_id�_get_server_information�_request_authenticationr*rW�executer+r;rLr%r0�
BaseExceptionr@r�rrCZoriginal_exceptionrZ�
format_exc)r,r#�kwargs�er�rYr%r%r&r.�s�



��

��

zConnection.connectcCsDtt|��t|j�|}tr&t|�|�|�|jdd|_dS)zwWrites an entire "mysql packet" in its entirety to the network
        addings its length and sequence number.
        rrIN)r�r.rrlr_r^r9)r,Zpayloadr1r%r%r&�write_packet�s

zConnection.write_packetc
Cs�d}|�d�}trt|�t�d|�\}}}||d>}||jkrt|��|dkr`t�t	j
d��t�d||jf��|jdd	|_|�|�}tr�t|�||7}|d
kr�q|tkrq�q|||j
�}	|	��|	S)z�Read an entire "mysql packet" in its entirety from the network
        and return a MysqlPacket type that represents the results.
        rOr�z<HBBrEr�,Lost connection to MySQL server during queryz1Packet sequence number wrong - got %d expected %drrIr<)�_read_bytesr_r^rjrkrlr8rrCr�CR_SERVER_LOST�
InternalError�MAX_PACKET_LENr�r�)
r,Zpacket_typeZbuffZ
packet_headerZbtrlZbtrhZ
packet_numberZ
bytes_to_readZ	recv_datar�r%r%r&rB�s<

���
zConnection._read_packetc
Cs�|j�|j�z|j�|�}Wq|Wqttfk
rx}z4|jtjkrLWY�"q|�	�t
�tj
d|f��W5d}~XYqXqt|�|kr�|�	�t
�tj
d��|S)Nz1Lost connection to MySQL server during query (%s)ru)r+rhrr=r0rjrir�rkr8rrCrrwr.)r,Z	num_bytesr1rsr%r%r&rv�s&��zConnection._read_bytesc
Csb|j�|j�z|j�|�Wn>tk
r\}z |��t�tj	d|f��W5d}~XYnXdS)NzMySQL server has gone away (%r))
r+rhrZsendallrjr8rrCrZCR_SERVER_GONE_ERROR)r,r1rsr%r%r&r9s�zConnection._write_bytescCsd|r6zt|�}|��WqFd|_d|_�YqFXnt|�}|��||_|jdk	r^|j|_|jS�NF)rN�init_unbuffered_query�unbuffered_active�
connectionr0r"r�r�)r,r^rgr%r%r&r`s

zConnection._read_query_resultcCs|jr|jjSdSdSr�)r"r�rvr%r%r&r�'szConnection.insert_idcCs|jst�d��|jdk	rN|jjr6t�d�|j��|jjrH|�	�q6d|_t
|t�rd|�|j
�}ttt|�d�}t�d||�}||d|d�}|�|�tr�t|�d|_|tkr�dS||dd�}ttt|��}|�|d|��||d�}|s�|tkrАqq�dS)N�(0, '')z.Previous unbuffered result was left incompleterr5)r+r�InterfaceErrorr"r|rr�_finish_unbuffered_queryr�rcr@rr_r�rYryr.rjrlr9r_r^rlrt)r,Zcommandra�packet_sizeZpreluder�r%r%r&rG-s4






zConnection._execute_commandcCs8t|j�dd�d�dkr*|jtjO_|jdkr<td��t|j	�j
}t|jt�rd|j�
|j�|_t�d|jd|d�}|jr�|jtj@r�|�|�|jj|j|jd�|_t|jd	�|_||jd
}d}|jdkr�t|j�
d�|j�}|jtj@�r|t t!|��|7}n4|jtj"@�r4|t�d
t!|��|7}n||d
7}|j#�r�|jtj$@�r�t|j#t��rt|j#�
|j�|_#||j#d
7}|jtj%@�r�|j}t|t��r�|�
d�}||d
7}|�|�|�&�}|�'��r4|�(�|�)�}|jtj%@�r
|dk	�r
|�*||�}n*t+|j�
d�|j�d
}|�|�|�&�}dS)Nr?rr�zDid not specify a usernamez<iIB23srO)Zserver_hostnamer5r�)rPZmysql_native_passwordr;ri�ascii),rA�server_version�splitr rZ
MULTI_RESULTSr�rXrr�idr@rr_r�rjrlr�server_capabilitiesrrtrZwrap_socketr+r�r'r=�_auth_plugin_namerhrd�saltZPLUGIN_AUTH_LENENC_CLIENT_DATAr�r.ZSECURE_CONNECTIONr�r�PLUGIN_AUTHrBr�r�r��
_process_authr�)r,Z
charset_idZ	data_initr1Zauthrespr��auth_packet�plugin_namer%r%r&rnSsP






z"Connection._request_authenticationcCsJ|j�|�}|s"|j�|�d��}|r�z||�}|�|�WStk
rj|dkrft�dd||f��Yq�tk
r�t�dd||f��Yq�Xnd}|dkr�t|j	�
d�|���d}�nn|d	kr�t|j	�
d�|���d}�nH|d
k�r|j	�
d�d}�n*|dk�r|}|�
�}|d@dk}|d
@d
k}	|��}
|
dk�rZ|�|j	�
d�d�n�|�r�d}z|�||
�}|�|d�WnVtk
�r�t�dd||f��Yn.tk
�r�t�dd||||
f��YnXnt�dd||f��|��}|��|���s|	�r�q�q|St�dd|��|�|�|��}|��|S)Nr�sdialogizGAuthentication plugin '%s' not loaded: - %r missing authenticate methodzXAuthentication plugin '%s' not loaded: - %r cannot be constructed with connection objectsmysql_native_passwordr;r�smysql_old_passwordsmysql_clear_passwordr(rrs
Password: z3no response - TypeError within plugin.prompt methodzAAuthentication plugin '%s' not loaded: - %r missing prompt methodi
zTAuthentication plugin '%s' %r didn't respond with string. Returned '%r' to prompt %rz.Authentication plugin '%s' (%r) not configuredz)Authentication plugin '%s' not configured)r-r�r�Zauthenticate�AttributeErrorrrC�	TypeErrorrhrdr_r�r�r�rt�promptrBr�r�)r,r�r�Zplugin_class�handlerr1rD�flagZechoZlastr�Zrespr%r%r&r��sn�
�


�

�
zConnection._process_authcCs
|jdSr�)�server_thread_idrvr%r%r&rd�szConnection.thread_idcCs|jSr)rrvr%r%r&�character_set_name�szConnection.character_set_namecCs|jSr)r$rvr%r%r&�
get_host_info�szConnection.get_host_infocCs|jSr)�protocol_versionrvr%r%r&�get_proto_info�szConnection.get_proto_infoc	Csd}|��}|��}tr t|�t|||d��|_|d7}|�d|�}|||��d�|_|d}t	�
d|||d��|_|d7}|||d�|_|d7}t	�
d	|||d
��d|_
|d
7}t|�|dk�r\t	�
d|||d��\}}}}|d7}||_t|�j|_||_t�r,td
|�|j
|d>O_
t�rNtd|�td|d�}|d7}t|�||k�r�|j||||�7_||7}|d7}|j
tj@�r�t|�|k�r�|�d|�}|dk�r�||d��d�|_n|||��d�|_dS)Nrrr�r;r�r�rnr�r�rr(z<BHHBzserver_status: %xrEz	salt_len:�r�)rBr�r_r^rr�r�r�r�rjrkr�r�r�r.Zserver_languager
r�Zserver_charsetr�rQ�maxrr�r�)	r,rGr�r1Z
server_endZlang�statZcap_hZsalt_lenr%r%r&rm�sL 

z"Connection._get_server_informationcCs|jSr)r�rvr%r%r&�get_server_infoszConnection.get_server_info)N)N)F)F)T)N)F)@r2r3r4r�r+r�r6rr-rr;�propertyr<r8�__del__r0r>rFr?rKrLrMrPrQrIrUrrWrXr[rbrcr�rerfrgr.rtr�rBrvr9r`r�rGrnr�rdr�r�r�rmr�r�Warningr7rZ
DatabaseErrorZ	DataErrorrCZIntegrityErrorrxZProgrammingErrorZNotSupportedErrorr%r%r%r&r�s��
4


	



	
H$

&=?:r�c@steZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�ZdS)rNcCsF||_d|_d|_d|_d|_d|_d|_d|_d|_d|_	d|_
dS)z.
        :type connection: Connection
        NrF)r}r�r�r�r�rer�r�rOr�r|)r,r}r%r%r&r-&szMySQLResult.__init__cCs|jr|��dSr)r|r�rvr%r%r&r�6szMySQLResult.__del__cCsNz@|j��}|��r |�|�n|��r4|�|�n
|�|�W5d|_XdSr)r}rBr�rFr��_read_load_local_packet�_read_result_packet�r,�first_packetr%r%r&r0:s
zMySQLResult.readcCsld|_|j��}|��r0|�|�d|_d|_n8|��rP|�|�d|_d|_n|��|_|�	�d|_
dS)NTFl����)r|r}rBr�rFr�r�r�r��_get_descriptionsr�r�r%r%r&r{Gs



z!MySQLResult.init_unbuffered_querycCs<t|�}|j|_|j|_|j|_|j|_|j|_|j|_dSr)r�r�r�r�r�rer�)r,r��	ok_packetr%r%r&rF\szMySQLResult._read_ok_packetcCsx|jjstd��t|�}t|j|j�}z|��Wn|j���YnX|j��}|��sjt	�
dd��|�|�dS)NzF**WARN**: Received LOAD_LOCAL packet but local_infile option is false.rAzCommands Out of Sync)r}r
�RuntimeErrorr��
LoadLocalFiler�r:rBr�rrCrF)r,r�Zload_packetZsenderr�r%r%r&r�es�

z#MySQLResult._read_load_local_packetcCs(|��sdSt|�}|j|_|j|_dS)NFT)r�r�r�r�)r,r�Zwpr%r%r&�_check_packet_is_eofvsz MySQLResult._check_packet_is_eofcCs|��|_|��|��dSr)r�r�r��_read_rowdata_packetr�r%r%r&r��s
zMySQLResult._read_result_packetcCsP|js
dS|j��}|�|�r4d|_d|_d|_dS|�|�}d|_|f|_|S)NFr)r|r}rBr�rO�_read_row_from_packetr�)r,r��rowr%r%r&�_read_rowdata_packet_unbuffered�s


z+MySQLResult._read_rowdata_packet_unbufferedcCs,|jr(|j��}|�|�rd|_d|_qdSrz)r|r}rBr�)r,r�r%r%r&r��s


z$MySQLResult._finish_unbuffered_querycCsJg}|j��}|�|�r d|_q2|�|�|��qt|�|_t|�|_dS)z:Read a rowdata packet for each data row in the result set.N)	r}rBr��appendr�r.r��tuplerO)r,rOr�r%r%r&r��s


z MySQLResult._read_rowdata_packetc	Cs�g}|jD]r\}}z|��}Wntk
r8Yq~YnX|dk	rr|dk	rT|�|�}trbtd|�|dk	rr||�}|�|�q
t|�S)NzDEBUG: DATA = )�
convertersr��
IndexErrorr�r_rQr�r�)r,r�r�r��	converterr1r%r%r&r��s

z!MySQLResult._read_row_from_packetc
Cs�g|_g|_|jj}|jj}g}t|j�D]�}|j�t�}|j�	|�|�	|�
��|j}|r�|tj
krn|}q�|tkr�|jdkr�d}q�|}q�d}nd}|jj�|�}|tkr�d}tr�td�||��|j�	||f�q*|j��}	|	��s�td��t|�|_
dS)z>Read a column descriptor packet for each column in the result.r�Nr�zDEBUG: field={}, converter={}zProtocol error, expecting EOF)Zfieldsr�r}rr�rr�rBr�r�r�r�rZJSON�
TEXT_TYPESr�r)r�rr_rQrKr�r6r�)
r,rZ
conn_encodingr�rGZfieldZ
field_typer�r�Z
eof_packetr%r%r&r��s8


zMySQLResult._get_descriptionsN)r2r3r4r-r�r0r{rFr�r�r�r�r�r�r�r�r%r%r%r&rN$s
	

rNc@seZdZdd�Zdd�ZdS)r�cCs||_||_dSr)r�r})r,r�r}r%r%r&r-�szLoadLocalFile.__init__c	Cs�|jjst�d��|j}zrzDt|jd��.}t|jd�}|�	|�}|sFqR|�|�q6W5QRXWn(t
k
r�t�dd�|j���YnXW5|�d�XdS)z3Send data packets from the local file to the serverr~rOr5i@i�zCan't find file '{0}'N)
r}r+rrrtr<r�rYr,r0rjrCrK)r,ZconnZ	open_filer��chunkr%r%r&r:�s

zLoadLocalFile.send_dataN)r2r3r4r-r:r%r%r%r&r��sr�)aZ
__future__rZ_compatrrrrrr	r��	functoolsr
Zhashlibr r
r�rjrSrZrrrrr
Z	constantsrrrrrr�rrrrr&ZcursorsrZ
optionfiler�utilrrrPrrr�ImportErrorZgetpassZgetuserr�KeyErrorr_rZ_py_versionZ	_socketiorr'�objectr)�setZBITZBLOBZ	LONG_BLOBZMEDIUM_BLOB�STRINGZ	TINY_BLOBr�ZVARCHARZGEOMETRYr��newrar�r�r�r�r�rryr^rhrcr|ror�r{r�r�r�r�r�r�r�r�rNr�r%r%r%r&�<module>s� 




�

.2%J