Your IP : 216.73.216.213

Current Path : /opt/alt/python35/lib/python3.5/site-packages/pymysql/__pycache__/
Current File : //opt/alt/python35/lib/python3.5/site-packages/pymysql/__pycache__/connections.cpython-35.opt-1.pyc


1�X��@sddlmZddlmZmZmZmZmZmZddl	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,yddl-Z-d
Z.Wne/k
r�dZ-dZ.YnXyddl0Z0e0j1�Z2[0Wne/e3fk
r�dZ2YnXdZ4ej5dd�Z6e6ddfkoer.ddl7m8Z8dd�Z9nCe6ddfkreGdd�de:�Z;dd�Z9ndd�Z9e<ej=ej>ej?ej@ejAejBejCejDejEg	�ZFeejGd�ZHdZIdZJdZKdZLdZMdZNdddZOd 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��)�SocketIOcCstjt||��S)N)�io�BufferedReaderr)�sock�mode�r#�D/opt/alt/python35/lib/python3.5/site-packages/pymysql/connections.py�	_makefile8sr%�c@s(eZdZdd�Zdd�ZdS)�SockFilecCs
||_dS)N)�_sock)�selfr!r#r#r$�__init__>szSockFile.__init__cCsy|jj|�}t|�|kr(|SxJ|jj|t|��}|sQ|S||7}t|�|kr+|Sq+WdS)N)r(Zrecv�len)r)�n�read�datar#r#r$r-As
z
SockFile.readN)�__name__�
__module__�__qualname__r*r-r#r#r#r$r'=sr'cCs
t|�S)N)r')r!r"r#r#r$r%MscCs
|j|�S)N)�makefile)r!r"r#r#r$r%RsZsha1�����latin1�csGdd��yntdt���xFtdd�D]5}tj|�}td||jj|jf�q2Wtdd�Wntk
r�YnX�fd	d
�t	dt
t��d�d
�D�}xh|D]`}tdjtdd�|��dd
t|�ddjt�fdd�|���q�Wtdd�t�dS)NcSsCdt|�kodknr?t|t�r;t|�S|SdS)N�A�z�.)r�
isinstance�int�chr)r.r#r#r$�is_asciios
"
zdump_packet.<locals>.is_asciizpacket length:rr&zcall[%d]: %s (line %d)�-�Bcs$g|]}�||d��qS)�r#)�.0�i)r.r#r$�
<listcomp>~s	zdump_packet.<locals>.<listcomp>r�rB� cSsdjt|��S)Nz{:02X})�formatr)�xr#r#r$�<lambda>�szdump_packet.<locals>.<lambda>z   r�csdj�|��S)Nz{})rH)rI)r?r#r$rJ�sz  )
�printr+�range�sys�	_getframe�f_code�co_name�f_lineno�
ValueErrorr�min�join�map)r.rD�fZ	dump_data�dr#)r.r?r$�dump_packetns $
4
1*rYcCs�|s
dStr$tdt|��t|�j�}t|�j�}t�}|j|�|j|�|j�}t||�S)N�z	password=)�DEBUGrL�str�sha_newZdigest�update�	_my_crypt)�password�messageZstage1Zstage2�s�resultr#r#r$�	_scramble�s	

rdcCs�t|�}d}xpt|�D]b}tjd|||d��dtjd|||d��dA}|tjd|�7}qW|S)NrZ�Brr)r+r�struct�unpack�pack)Zmessage1Zmessage2�lengthrcrDrIr#r#r$r_�s!%r_�c@s(eZdZdd�Zdd�ZdS)�RandStruct_323cCs-d|_||j|_||j|_dS)Ni���?)�	max_value�seed1�seed2)r)rmrnr#r#r$r*�s	zRandStruct_323.__init__cCsV|jd|j|j|_|j|jd|j|_t|j�t|j�S)N��!)rmrnrl�float)r)r#r#r$�my_rnd�szRandStruct_323.my_rndN)r/r0r1r*rrr#r#r#r$rk�srkcCs8t|�}t|dt��}tjd|�}tjd|�}t|d|dA|d|dA�}tj�}xJtttt	|���D]-}|j
tt|j
�d�d��q�Wtt|j
�d��}	|j�}
tj�}x1|
D])}|j
tt|�t|	�A��qW|j�S)Nz>LLrr��@)�_hash_password_323�SCRAMBLE_LENGTH_323rfrgrkr�BytesIOrrTr+�writerr=rr�getvaluer)r`raZ	hash_passZhash_messageZhash_pass_nZhash_message_nZrand_stZoutbuf�_Zextra�out�cr#r#r$�
_scramble_323�s"+
'r}cCs�d}d}d}xadd�|D�D]L}||d@|||d>d@N}||d>|Ad@}||d@}q&W|d
@}|d@}tjd||�S)Ni5W0PriqV4cSs(g|]}|dkrt|��qS)rG�	� �	)rGr~rr�)r)rCrIr#r#r$rE�s	z&_hash_password_323.<locals>.<listcomp>�?rjl��rrsz>LLli���li���)rfrh)r`Znr�addZnr2r|Zr1Zr2r#r#r$ru�s"

rucCstjd|�dd�S)Nz<Iro)rfrh)r,r#r#r$�
pack_int24�sr�cCs�|dkrtd|��n�|dkr5t|�S|dkrUdtjd|�S|dkrd	tjd
|�dd�S|dkr�d
tjd|�Std|df��dS)NrzFEncoding %d is less than 0 - no representation in LengthEncodedIntegerr3rrBs�z<Hr8s�z<Irorts�z<QzIEncoding %x is larger than %x - no representation in LengthEncodedIntegeriill)rSrrfrh)rDr#r#r$�
lenenc_int�s
r�c@sWeZdZdZd8Zdd�Zdd�Zdd	�Zd
d�Zdd
�Z	ddd�Z
ddd�Zer�dd�Z
ndd�Z
dd�Zdd�Zdd�Zdd�Zdd �Zd!d"�Zd#d$�Zd%d&�Zd'd(�Zd)d*�Zd+d,�Zd-d.�Zd/d0�Zd1d2�Zd3d4�Zd5d6�Zd7S)9�MysqlPacketzrRepresentation of a MySQL response packet.

    Provides an interface for reading/parsing the packet results.
    �	_position�_datacCsd|_||_dS)Nr)r�r�)r)r.�encodingr#r#r$r*�s	zMysqlPacket.__init__cCs|jS)N)r�)r)r#r#r$�get_all_data�szMysqlPacket.get_all_datacCs�|j|j|j|�}t|�|kr}d|t|�|jt|j�f}trqt|�|j�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[rL�dump�AssertionError)r)�sizerc�errorr#r#r$r-�s%

zMysqlPacket.readcCs#|j|jd�}d|_|S)z`Read all remaining data in the packet.

        (Subsequent read() will return errors.)
        N)r�r�)r)rcr#r#r$�read_all�s	zMysqlPacket.read_allcCsQ|j|}|dks.|t|j�krDtd||f��||_dS)z1Advance the cursor in data buffer 'length' bytes.rz4Invalid advance amount (%s) for cursor.  Position=%sN)r�r+r��	Exception)r)riZnew_positionr#r#r$�advances

!zMysqlPacket.advancercCs>|dks!|t|j�kr1td|��||_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$�rewinds!zMysqlPacket.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�rir#r#r$�	get_bytess	zMysqlPacket.get_bytescCs)t|j|j�}|jd7_|S)Nr)�ordr�r�)r)rcr#r#r$�
read_uint8!szMysqlPacket.read_uint8cCs#|j|j}|jd7_|S)Nr)r�r�)r)rcr#r#r$r�&scCs2tjd|j|j�d}|jd7_|S)Nz<Hrr)rf�unpack_fromr�r�)r)rcr#r#r$�read_uint16+szMysqlPacket.read_uint16cCs<tjd|j|j�\}}|jd7_||d>S)Nz<HBrorB)rfr�r�r�)r)ZlowZhighr#r#r$�read_uint240s!zMysqlPacket.read_uint24cCs2tjd|j|j�d}|jd7_|S)Nz<Ir�)rfr�r�r�)r)rcr#r#r$�read_uint325szMysqlPacket.read_uint32cCs2tjd|j|j�d}|jd7_|S)Nz<Qrrj)rfr�r�r�)r)rcr#r#r$�read_uint64:szMysqlPacket.read_uint64cCsO|jjd|j�}|dkr(dS|j|j|�}|d|_|S)Nsrr)r��findr�)r)Zend_posrcr#r#r$�read_string?s
zMysqlPacket.read_stringcCsr|j�}|tkrdS|tkr,|S|tkrB|j�S|tkrX|j�S|tkrn|j�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_integerGs

z'MysqlPacket.read_length_encoded_integercCs)|j�}|dkrdS|j|�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�r-)r)rir#r#r$�read_length_coded_stringYsz$MysqlPacket.read_length_coded_stringcCs=tj|�}|j|j|j�}|j|j7_|S)N)rfZStructr�r�r�r�)r)Zfmtrbrcr#r#r$�read_structeszMysqlPacket.read_structcCs,|jdd�dko+t|j�dkS)Nrrsr)r�r+)r)r#r#r$�is_ok_packetkszMysqlPacket.is_ok_packetcCs,|jdd�dko+t|j�dkS)Nrrs�r�)r�r+)r)r#r#r$�
is_eof_packetoszMysqlPacket.is_eof_packetcCs|jdd�dkS)Nrrs�)r�)r)r#r#r$�is_auth_switch_requestusz"MysqlPacket.is_auth_switch_requestcCs1t|jdd��}d|ko.dkSS)Nrr�)r�r�)r)�field_countr#r#r$�is_resultset_packetyszMysqlPacket.is_resultset_packetcCs|jdd�dkS)Nrrs�)r�)r)r#r#r$�is_load_local_packet}sz MysqlPacket.is_load_local_packetcCs|jdd�dkS)Nrrs�)r�)r)r#r#r$�is_error_packet�szMysqlPacket.is_error_packetcCsV|j�rR|j�|jd�|j�}trBtd|�tj|j�dS)Nrzerrno =)	r�r�r�r�r[rLrZraise_mysql_exceptionr�)r)�errnor#r#r$�check_error�s


zMysqlPacket.check_errorcCst|j�dS)N)rYr�)r)r#r#r$r��szMysqlPacket.dumpN)r�r�)r/r0r1�__doc__�	__slots__r*r�r-r�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@sReZdZdZdd�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.
    cCs$tj|||�|j|�dS)N)r�r*�_parse_field_descriptor)r)r.r�r#r#r$r*�szFieldDescriptorPacket.__init__cCs�|j�|_|j�|_|j�j|�|_|j�j|�|_|j�j|�|_|j�j|�|_|jd�\|_	|_
|_|_|_
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��	charsetnrri�	type_code�flags�scale)r)r�r#r#r$r��sz-FieldDescriptorPacket._parse_field_descriptorcCs;|j|jd|j�|j�|j|jddkfS)zBProvides a 7-item tuple compatible with the Python PEP249 DB Spec.Nrr)r�r��get_column_lengthr�r�)r)r#r#r$�description�s		z!FieldDescriptorPacket.descriptioncCs9|jtjkr2tj|jd�}|j|S|jS)Nr)r�r�
VAR_STRINGr�getr�ri)r)Zmblenr#r#r$r��sz'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�)r)r#r#r$�__str__�szFieldDescriptorPacket.__str__N)	r/r0r1r�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.
    cCs�|j�s,tdt|jj�d��||_|jjd�|jj�|_|jj�|_	|j
d�\|_|_|jj
�|_|jtj@|_dS)NzCannot create z  object from invalid packet typerz<HH)r�rSr\r�r/�packetr�r��
affected_rows�	insert_idr��
server_status�
warning_countr�rar�SERVER_MORE_RESULTS_EXISTS�has_next)r)�from_packetr#r#r$r*�s
	zOKPacketWrapper.__init__cCst|j|�S)N)�getattrr�)r)�keyr#r#r$�__getattr__�szOKPacketWrapper.__getattr__N)r/r0r1r�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.
    cCsx|j�s$tdj|j���||_|jjd�\|_|_trat	d|j�|jt
j@|_dS)Nz3Cannot create '{0}' object from invalid packet typez<xhhzserver_status=)
r�rSrHr�r�r�r�r�r[rLrr�r�)r)r�r#r#r$r*�s	zEOFPacketWrapper.__init__cCst|j|�S)N)r�r�)r)r�r#r#r$r��szEOFPacketWrapper.__getattr__N)r/r0r1r�r*r�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.
    cCsc|j�s$tdj|j���||_|jj�dd�|_tr_td|j�dS)Nz3Cannot create '{0}' object from invalid packet typerz	filename=)	r�rSrHr�r�r��filenamer[rL)r)r�r#r#r$r*�s	zLoadLocalPacketWrapper.__init__cCst|j|�S)N)r�r�)r)r�r#r#r$r�sz"LoadLocalPacketWrapper.__getattr__N)r/r0r1r�r*r�r#r#r#r$r��s
r�c @s�eZdZdZdZdZdZddddddddddddedddddddddddd_didddd	d
�Zdd�Z	d
d�Z
edd��Zdd�Z
e
Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd �Zd!d"�Zd#d$�Zdd%d&�Zd'd(�Zd)d*�Zdd+d,�Zd-d.�Zd/d0�Zdd1d2�Zdd3d4�Zd5d6�Z d7d8�Z!d9d:d;�Z"d<d=�Z#dd>d?�Z$d@dA�Z%e&dBdC�Z'dDdE�Z(dFdG�Z)ddHdI�Z*dJdK�Z+dLdM�Z,dNdO�Z-dPdQ�Z.dRdS�Z/dTdU�Z0dVdW�Z1dXdY�Z2dZd[�Z3d\d]�Z4e5j6Z6e5j7Z7e5j8Z8e5j9Z9e5j:Z:e5j;Z;e5j<Z<e5j=Z=e5j>Z>e5j?Z?dS)`�
Connectionz�
    Representation of a socket with a mysql server.

    The proper way to get an instance of this class is to call
    connect().
    NrKFr�
rBic"s�|dk	rtjdt�|dkrAtjddkrAd}|dk	r_|dkr_|}|dk	rx|rx|}|s�|r�td��t|�|_|jr�|tj	O}�r�|	r�tj
jd�r�d}	nd	}	|	r�s�d
�t���j
tjj|	����fdd�}|d
|�}|d|�}|d|�}|d|�}|d|�}t|d|��}|d|�}|d|�}|s�i}t|t�rxIdddddgD]2} |d| |j| ��}!|!r�|!|| <q�Wd|_|rZts2td��d|_|tjO}|j|�|_|pcd|_|prd|_|p�t|_|p�d|_||_||_ ||_!d|ko�d kns�t"d!��|p�d|_#|dk	r|dkrt"d"��||_$|dk	r9|dkr9t"d#��||_%|r]||_&d|_'nt(|_&d|_'|dk	r�||_'t)|j&�j*|_*|tj+O}|jr�|tj,O}||_-|
|_.d|_/d|_0d$|_1||_2|
dkrt3}
td%d&�|
j4�D��|_5td'd&�|
j4�D��|_6||_7||_8||_9||_:|r~d|_;n
|j<�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.cnfZclientcs;|r
|Sy�j�|�SWntk
r6|SYnXdS)N)r�r�)r��arg)�cfg�read_default_groupr#r$�_configls
z$Connection.__init__.<locals>._config�userr`�host�database�socket�portzbind-addresszdefault-character-set�ca�capath�certr��cipherzssl-Fzssl module not found�	localhosti�rKi�3�z+connect_timeout should be >0 and <=31536000zread_timeout should be >= 0zwrite_timeout should be >= 0z
Not connectedcSs4g|]*\}}t|�tk	r||f�qSr#)�typer=)rC�k�vr#r#r$rE�s	z'Connection.__init__.<locals>.<listcomp>cSs4g|]*\}}t|�tkr||f�qSr#)r�r=)rCr�r�r#r#r$rE�s	)=�warnings�warn�DeprecationWarningrN�version_info�NotImplementedError�bool�
_local_infilerZLOCAL_FILES�platform�
startswithrr-�os�path�
expanduserr=r<�dictr��ssl�SSL_ENABLED�SSL�_create_ssl_ctx�ctxr�r��DEFAULT_USERr�r`r��unix_socket�bind_addressrS�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�r`r�r�r�r�rZread_default_fileZconvr�rrrr�r�r��compressZ
named_pipeZno_delay�
autocommitr�ZpasswdZlocal_infiler
Z
defer_connectZauth_plugin_mapZread_timeoutZ
write_timeoutr�r�r��valuer#)r�r�r$r*s�<	

				
									
	
						""				zConnection.__init__cCst|tj�r|S|jd�}|jd�}|dkoI|dk}tjd|d|�}|oz|jdd�|_|r�tjntj|_d|kr�|j	|dd|jd��d	|kr�|j
|d	�|jtjO_|jtj
O_|S)
Nr�r�Zcafile�check_hostnameTr�Zkeyfiler�r�)r<r�Z
SSLContextr�Zcreate_default_contextrZ	CERT_NONEZ
CERT_REQUIREDZverify_modeZload_cert_chainZset_ciphers�optionsZOP_NO_SSLv2ZOP_NO_SSLv3)r)Zsslpr�r�Zhasnocar�r#r#r$r��s zConnection._create_ssl_ctxcCs�|jrtjd��d|_|jdkr4dStjddtj�}z*y|j|�Wnt	k
rtYnXWd|j
�XdS)z*Send the quit message and close the socketzAlready closedTNz<iBr)�_closedr�Errorr(rfrhrZCOM_QUIT�_write_bytesr��_force_close)r)�	send_datar#r#r$�close�s		
	zConnection.closecCs
|jdk	S)N)r()r)r#r#r$�open�szConnection.openc	Cs;|jr%y|jj�WnYnXd|_d|_dS)z%Close connection without QUIT messageN)r(r�_rfile)r)r#r#r$r�s		zConnection._force_closecCs5t|�|_|j�}||kr1|j�dS)N)r�r�get_autocommit�_send_autocommit_mode)r)rZcurrentr#r#r$r�szConnection.autocommitcCst|jtj@�S)N)r�r�rZSERVER_STATUS_AUTOCOMMIT)r)r#r#r$r�s	zConnection.get_autocommitcCsF|j�}|j�s*tjdd��t|�}|j|_|S)Ni�zCommand Out of Sync)�_read_packetr�r�OperationalErrorr�r�)r)�pkt�okr#r#r$�_read_ok_packet�szConnection._read_ok_packetcCs1|jtjd|j|j��|j�dS)z2Set whether or not to commit after every execute()zSET AUTOCOMMIT = %sN)�_execute_commandr�	COM_QUERY�escaperr#)r)r#r#r$rsz Connection._send_autocommit_modecCs!|jtjd�|j�dS)zBegin transaction.ZBEGINN)r$rr%r#)r)r#r#r$�beginszConnection.begincCs!|jtjd�|j�dS)z Commit changes to stable storageZCOMMITN)r$rr%r#)r)r#r#r$�commitszConnection.commitcCs!|jtjd�|j�dS)z!Roll back the current transactionZROLLBACKN)r$rr%r#)r)r#r#r$�rollbackszConnection.rollbackcCs0|jtjd�t|�}|j�|jS)z
SHOW WARNINGS)r$rr%�MySQLResultr-�rows)r)rcr#r#r$�
show_warningss
zConnection.show_warningscCs!|jtj|�|j�dS)zSet current dbN)r$rZCOM_INIT_DBr#)r)r�r#r#r$�	select_db"szConnection.select_dbcCs:t|t�r$d|j|�dSt||jd|�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)�objr/r#r#r$r&'szConnection.escapecCs|j||j�S)zrAlias for escape()
        
        Non-standard, for internal use; do not use this in your applications.
        )r&r	)r)r0r#r#r$�literal0szConnection.literalcCs*|jtj@r |jdd�St|�S)Nr.z'')r�rZ"SERVER_STATUS_NO_BACKSLASH_ESCAPES�replacer)r)rbr#r#r$r7s
zConnection.escape_stringcCs|r||�S|j|�S)z+Create a new cursor to execute queries with)r)r)�cursorr#r#r$r3=s
zConnection.cursorcCs
|j�S)z%Context manager that returns a Cursor)r3)r)r#r#r$�	__enter__CszConnection.__enter__cCs!|r|j�n
|j�dS)z2On successful exit, commit. On exception, rollbackN)r)r()r)�excr�	tracebackr#r#r$�__exit__Gs
zConnection.__exit__cCs{t|t�rLtptrLtr7|j|j�}n|j|jd�}|jtj	|�|j
d|�|_|jS)N�surrogateescape�
unbuffered)r<rrr	r�encoder�r$rr%�_read_query_resultr)r)�sqlr9r#r#r$�queryOszConnection.querycCs|jd|�|_|jS)Nr9)r;r)r)r9r#r#r$�next_result[szConnection.next_resultcCs|jS)N)r)r)r#r#r$r�_szConnection.affected_rowscCs/tjd|�}|jtj|�|j�S)Nz<I)rfrhr$rZCOM_PROCESS_KILLr#)r)�	thread_idr�r#r#r$�killbszConnection.killTcCs�|jdkr7|r(|j�d}ntjd��y!|jtjd�|j�SWn2tk
r�|r�|j�|j	d�S�YnXdS)zCheck if the server is aliveNFzAlready closedrK)
r(rrrr$rZCOM_PINGr#r��ping)r)Z	reconnectr#r#r$rAgs
	


zConnection.pingcCsOt|�j}|jtjd|j|��|j�||_||_dS)NzSET NAMES %s)rr�r$rr%r&rr�)r)r�r�r#r#r$�set_charsetys
 
	zConnection.set_charsetcCsd|_yI|dkr�|jr�|jdkr�tjtjtj�}|j|j�|j|j�d|_	t
rdtd�n�i}|jdk	r�|jdf|d<xsy)tj
|j|jf|j|�}PWq�ttfk
r}z|jtjkrw��WYdd}~Xq�Xq�Wd|j|jf|_	t
rKtd	�|jtjtjd
�|jd�|jtjtjd
�||_t|d�|_d|_|j�|j�|jdk	r�|j�}|j d|jf�|j!dk	r2|j�}|j |j!�|j"�|j#�|j$dk	rQ|j%|j$�Wn�t&k
r}z�d|_|dk	r�y|j"�WnYnXt'|tttj(f�rt)j*d
d|j|f�}||_+t,j-�|_,t
r�t|j,�|��WYdd}~XnXdS)NFr��	127.0.0.1zLocalhost via UNIX socketzconnected using unix_socketrZsource_addresszsocket %s:%dzconnected using socketr�rbzSET sql_mode=%si�z(Can't connect to MySQL server on %r (%s))r�rC).rr�r�r�ZAF_UNIXZSOCK_STREAM�
settimeoutr�rrr[rLr�Zcreate_connectionr��OSError�IOErrorr��EINTRZ
setsockoptZIPPROTO_TCPZTCP_NODELAYZ
SOL_SOCKETZSO_KEEPALIVEr(r%r�_next_seq_id�_get_server_information�_request_authenticationrr3�executerrr(rr�
BaseExceptionr<r�rr Zoriginal_exceptionr6�
format_exc)r)r!�kwargs�er|r5r#r#r$r�s|		
	

		



		
zConnection.connectcCsXtt|��t|j�|}tr3t|�|j|�|jdd|_dS)zwWrites an entire "mysql packet" in its entirety to the network
        addings its length and sequence number.
        rrFN)r�r+rrIr[rYr)r)Zpayloadr.r#r#r$�write_packet�s
#

zConnection.write_packetc
Cs$d}x�|jd�}tr(t|�tjd|�\}}}||d>}||jkr�|j�|dkr�tjt	j
d��tjd||jf��|jdd	|_|j|�}tr�t|�||7}|d
kr�q	|tkr	Pq	W|||j
�}	|	j�|	S)z�Read an entire "mysql packet" in its entirety from the network
        and return a MysqlPacket type that represents the results.
        rZr�z<HBBrBrz,Lost connection to MySQL server during queryz1Packet sequence number wrong - got %d expected %drrFi���)�_read_bytesr[rYrfrgrIrrr r�CR_SERVER_LOST�
InternalError�MAX_PACKET_LENr�r�)
r)Zpacket_typeZbuffZ
packet_headerZbtrlZbtrhZ
packet_numberZ
bytes_to_readZ	recv_datar�r#r#r$r�s8

	


zConnection._read_packetcCs�|jj|j�x�y|jj|�}PWqttfk
r�}z@|jtjkr]w|j	�t
jtj
d|f��WYdd}~XqXqWt|�|kr�|j	�t
jtj
d��|S)Nz1Lost connection to MySQL server during query (%s)z,Lost connection to MySQL server during query)r(rEr�rr-rGrFr�rHrrr rrSr+)r)Z	num_bytesr.rPr#r#r$rR�s"
'
zConnection._read_bytescCsy|jj|j�y|jj|�WnKtk
rt}z+|j�tjtj	d|f��WYdd}~XnXdS)NzMySQL server has gone away (%r))
r(rEr�ZsendallrGrrr rZCR_SERVER_GONE_ERROR)r)r.rPr#r#r$rs
zConnection._write_bytesc	Cs�|rCyt|�}|j�WqYd|_d|_�YqYXnt|�}|j�||_|jdk	r}|j|_|jS)NF)r*�init_unbuffered_query�unbuffered_active�
connectionr-rr�r�)r)r9rcr#r#r$r;s		
	zConnection._read_query_resultcCs|jr|jjSdSdS)Nr)rr�)r)r#r#r$r�'s	
zConnection.insert_idcCs|jstjd��|jdk	rs|jjrMtjd�|jj�x|jjri|j	�qPWd|_t
|t�r�|j|j
�}ttt|�d�}tjd||�}||d|d�}|j|�tr�t|�d|_|tkrdS||dd�}xTttt|��}|j|d|��||d�}|r'|tkr'Pq'WdS)Nz(0, '')z.Previous unbuffered result was left incompleterz<iB)r(r�InterfaceErrorrrWr�r��_finish_unbuffered_queryr�r>r<rr:r�rTrUr+rfrhrr[rYrIrQ)r)Zcommandr<�packet_sizeZpreluder�r#r#r$r$-s6	

	

	zConnection._execute_commandcCs�t|jjdd�d�dkr7|jtjO_|jdkrRtd��t|j	�j
}t|jt�r�|jj
|j�|_tjd|jd|d�}|jr|jtj@r|j|�|jj|jd|j�|_t|jd	�|_||jd
}d}|jdkrLt|jj
d
�|j�}|jtj@ry|t t!|��|7}nA|jtj"@r�|tjdt!|��|7}n||d
7}|j#r|jtj$@rt|j#t�r�|j#j
|j�|_#||j#d
7}|jtj%@rS|j}t|t�rE|j
d�}||d
7}|j|�|j&�}|j'�r�|j(�|j)�}|jtj%@r�|dk	r�|j*||�}n;t+|jj
d
�|j�d
}|j|�|j&�}dS)Nr;rr�zDid not specify a usernamez<iIB23srZZserver_hostnamerDsrK�mysql_native_passwordr7re�ascii)rKr]),r=�server_version�splitrrZ
MULTI_RESULTSr�rSrr��idr<rr:r�rfrhr��server_capabilitiesr�rQr�Zwrap_socketr(r�r%r�_auth_plugin_namerdr`�saltZPLUGIN_AUTH_LENENC_CLIENT_DATAr�r+ZSECURE_CONNECTIONr�r�PLUGIN_AUTHrr�r�r��
_process_authr})r)Z
charset_idZ	data_initr.Zauthrespr��auth_packet�plugin_namer#r#r$rKSsP%
!#	

"
z"Connection._request_authenticationcCs�|jj|�}|s3|jj|jd��}|r�y||�}|j|�SWq�tk
r�|dkr�tjdd||f��Yq�tk
r�tjdd||f��Yq�Xnd}|dkr�t|j	j
d�|j��d}n�|d	kr1t|j	j
d�|j��d}n�|d
krV|j	j
d�d}nv|dkr�|}xG|j
�}|d@dk}|d
@d
k}	|j�}
|
dkr�|j|j	j
d�d�n�|rid}y'|j||
�}|j|d�Wq�tk
r2tjdd||f��Yq�tk
retjdd||||
f��Yq�Xntjdd||f��|j�}|j�|j�s�|	rkPqkW|Stjdd|��|j|�|j�}|j�|S)Nr^sdialogizGAuthentication plugin '%s' not loaded: - %r missing authenticate methodzXAuthentication plugin '%s' not loaded: - %r cannot be constructed with connection objectsmysql_native_passwordr7ssmysql_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)rr�r�Zauthenticate�AttributeErrorrr �	TypeErrorrdr`r:r�r}r�rQ�promptrr�r�)r)rhrgZplugin_classZhandlerr.r!ZflagZechoZlastrkZrespr#r#r$rf�sh

(( 




zConnection._process_authcCs|jdS)Nr)�server_thread_id)r)r#r#r$r?�szConnection.thread_idcCs|jS)N)r�)r)r#r#r$�character_set_name�szConnection.character_set_namecCs|jS)N)r)r)r#r#r$�
get_host_info�szConnection.get_host_infocCs|jS)N)�protocol_version)r)r#r#r$�get_proto_info�szConnection.get_proto_infoc	Cs�d}|j�}|j�}tr.t|�t|||d��|_|d7}|jd|�}|||�jd�|_|d}t	j
d|||d��|_|d7}|||d�|_|d7}t	j
d	|||d
��d|_
|d
7}t|�|dkr�t	j
d|||d��\}}}}|d7}||_t|�j|_||_tr�td
|�|j
|d>O_
tr�td|�td|d�}|d7}t|�||kr|j||||�7_||7}|d7}|j
tj@r�t|�|kr�|jd|�}|dkry||d�jd�|_n|||�jd�|_dS)Nrrsr7z<Ir�rjr�z<Hrr&z<BHHBzserver_status: %xrBz	salt_len:�r�)rr�r[rYrror�r�r_rfrgrlrdrbr+Zserver_languager
r�Zserver_charsetr�rL�maxrrerc)	r)rDr�r.Z
server_endZlang�statZcap_hZsalt_lenr#r#r$rJ�sL


#

'
,
		



"z"Connection._get_server_informationcCs|jS)N)r_)r)r#r#r$�get_server_infoszConnection.get_server_infoi@i)@r/r0r1r�r(rcrrr*r�r�propertyrr�__del__rrr#rr'r(r)r,r-r&r1rr3r4r7r=r>r�r@rArBrrQr�rrRrr;r�r$rKrfr?rmrnrprJrtr�WarningrrYZ
DatabaseErrorZ	DataErrorr ZIntegrityErrorrTZProgrammingErrorZNotSupportedErrorr#r#r#r$r�s�							�
		H$
&=?:									r�c@s�eZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�ZdS)r*cCsg||_d|_d|_d|_d|_d|_d|_d|_d|_d|_	d|_
dS)z.
        :type connection: Connection
        NrF)rXr�r�r�r�rar�r�r+r�rW)r)rXr#r#r$r*&s										zMySQLResult.__init__cCs|jr|j�dS)N)rWrZ)r)r#r#r$rv6s	zMySQLResult.__del__cCsizX|jj�}|j�r.|j|�n)|j�rJ|j|�n
|j|�Wdd|_XdS)N)rXrr�r#r��_read_load_local_packet�_read_result_packet)r)�first_packetr#r#r$r-:szMySQLResult.readcCs�d|_|jj�}|j�rF|j|�d|_d|_nP|j�rt|j|�d|_d|_n"|j�|_|j	�d|_
dS)NTFl����)rWrXrr�r#r�rxr�r��_get_descriptionsr�)r)rzr#r#r$rVGs	
	
	
z!MySQLResult.init_unbuffered_querycCsXt|�}|j|_|j|_|j|_|j|_|j|_|j|_dS)N)r�r�r�r�r�rar�)r)rz�	ok_packetr#r#r$r#\szMySQLResult._read_ok_packetc	Cs�|jjstd��t|�}t|j|j�}y|j�Wn|jj��YnX|jj�}|j�s�t	j
dd��|j|�dS)NzF**WARN**: Received LOAD_LOCAL packet but local_infile option is false.i�zCommands Out of Sync)rXr��RuntimeErrorr��
LoadLocalFiler�rrr�rr r#)r)rzZload_packetZsenderr|r#r#r$rxes	
z#MySQLResult._read_load_local_packetcCs8|j�sdSt|�}|j|_|j|_dS)NFT)r�r�r�r�)r)r�Zwpr#r#r$�_check_packet_is_eofvsz MySQLResult._check_packet_is_eofcCs'|j�|_|j�|j�dS)N)r�r�r{�_read_rowdata_packet)r)rzr#r#r$ry�s
zMySQLResult._read_result_packetcCsr|js
dS|jj�}|j|�rJd|_d|_d|_dS|j|�}d|_|f|_|S)NFr)rWrXrrr+�_read_row_from_packetr�)r)r��rowr#r#r$�_read_rowdata_packet_unbuffered�s					z+MySQLResult._read_rowdata_packet_unbufferedcCsDx=|jr?|jj�}|j|�rd|_d|_qWdS)NF)rWrXrr)r)r�r#r#r$rZ�s
	z$MySQLResult._finish_unbuffered_querycCsmg}xB|jj�}|j|�r1d|_P|j|j|��q	Wt|�|_t|�|_dS)z:Read a rowdata packet for each data row in the result set.N)	rXrr�appendr�r+r��tupler+)r)r+r�r#r#r$r��s	z MySQLResult._read_rowdata_packetcCs�g}x�|jD]�\}}y|j�}Wntk
rAPYnX|dk	r�|dk	ri|j|�}tr|td|�|dk	r�||�}|j|�qWt|�S)NzDEBUG: DATA = )�
convertersr��
IndexErrorr�r[rLr�r�)r)r�r�r��	converterr.r#r#r$r��s

z!MySQLResult._read_row_from_packetc
Cs]g|_g|_|jj}|jj}g}xt|j�D]�}|jjt�}|jj	|�|j	|j
��|j}|r�|tj
kr�|}q�|tkr�|jdkr�d}q�|}q�d}nd}|jjj|�}|tkrd}tr!tdj||��|jj	||f�q@W|jj�}	t|�|_
dS)z>Read a column descriptor packet for each column in the result.r�Nr^zDEBUG: field={}, converter={})Zfieldsr�rXr�r�rr�rr�r�r�r�rZJSON�
TEXT_TYPESr�r
r�rr[rLrHr�)
r)r�Z
conn_encodingr�rDZfieldZ
field_typer�r�Z
eof_packetr#r#r$r{�s6							zMySQLResult._get_descriptionsN)r/r0r1r*rvr-rVr#rxrryr�rZr�r�r{r#r#r#r$r*$s
	

r*c@s(eZdZdd�Zdd�ZdS)r~cCs||_||_dS)N)r�rX)r)r�rXr#r#r$r*�s	zLoadLocalFile.__init__cCs�|jjstjd��|j}z�y\t|jd��C}t|jd	�}x'|j|�}|sjP|j	|�qTWWdQRXWn0t
k
r�tjddj|j���YnXWd|j	d�XdS)
z3Send data packets from the local file to the serverz(0, '')rDrBiNi�zCan't find file '{0}'rZi@)
rXr(rrYrr�rTr
r-rQrGr rH)r)ZconnZ	open_filer[�chunkr#r#r$r�s	
'zLoadLocalFile.send_dataN)r/r0r1r*rr#r#r#r$r~�sr~)aZ
__future__rZ_compatrrrrrr	r��	functoolsr
Zhashlibrr�r�rfrNr6r�r�rrr
Z	constantsrrrrrr�rrrrrZcursorsrZ
optionfiler�utilrrrKrr�r��ImportErrorZgetpassZgetuserr��KeyErrorr[r�Z_py_versionZ	_socketiorr%�objectr'�setZBITZBLOBZ	LONG_BLOBZMEDIUM_BLOB�STRINGZ	TINY_BLOBr�ZVARCHARZGEOMETRYr��newr]r�r�r�r�r�r�rUrYrdr_rvrkr}rur�r�r�r�r�r�r�r�r*r~r#r#r#r$�<module>s�.("



�2���"�