3
(h                 @   s  d dl mZ ddlmZmZmZmZmZmZ d dl	Z	d dl
mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZddlmZmZmZ ddlmZmZmZmZ ddlmZmZm Z m!Z" dd	l#m$Z$ dd
l%m&Z& ddl'm(Z(m)Z) ddl*m+Z+ yd dl,Z,dZ-W n e.k
r&   dZ,dZ-Y nX yd dl/Z/e/j0 Z1[/W n e.e2fk
r\   dZ1Y nX dZ3ej4dd Z5e5dAkre rddl6m7Z7 dd Z8n,e5dBkrG dd de9Z:dd Z8ndd Z8e;ej<ej=ej>ej?ej@ejAejBejCejDg	ZEeejFdZGdZHdZIdZJdZKdZLdZMdDZNd d! ZOd"d# ZPd$d% ZQd&ZRG d'd( d(e9ZSd)d* ZTd+d, ZUd-d. ZVd/d0 ZWG d1d2 d2e9ZXG d3d4 d4eXZYG d5d6 d6e9ZZG d7d8 d8e9Z[G d9d: d:e9Z\G d;d< d<e9Z]G d=d> d>e9Z^G d?d@ d@e9Z_dS )E    )print_function   )PY2
range_type	text_typestr_typeJYTHON
IRONPYTHONN)partial)MBLENGTHcharset_by_namecharset_by_id)CLIENTCOMMAND
FIELD_TYPESERVER_STATUS)escape_itemescape_stringthroughconversions)Cursor)Parser)byte2intint2byte)errTF      )SocketIOc             C   s   t jt| |S )N)ioBufferedReaderr   )sockmode r"   8/tmp/pip-install-q3hcpn_q/PyMySQL/pymysql/connections.py	_makefile8   s    r$      c               @   s   e Zd Zdd Zdd ZdS )SockFilec             C   s
   || _ d S )N)_sock)selfr    r"   r"   r#   __init__>   s    zSockFile.__init__c             C   sZ   | j j|}t||kr|S x8| j j|t| }|s:|S ||7 }t||kr|S qW d S )N)r'   recvlen)r(   nreaddatar"   r"   r#   r-   A   s    zSockFile.readN)__name__
__module____qualname__r)   r-   r"   r"   r"   r#   r&   =   s   r&   c             C   s   |dkst t| S )Nrb)AssertionErrorr&   )r    r!   r"   r"   r#   r$   M   s    c             C   s
   | j |S )N)makefile)r    r!   r"   r"   r#   r$   R   s    sha1            latin1   c                s   dd yTt dt  x4tddD ]&}tj|}t d||jj|jf  q$W t dd  W n tk
rp   Y nX  fd	d
t	dt
t ddD }xP|D ]H}t djtdd |ddt|   d djtfdd|  qW t dd  t   d S )Nc             S   s6   dt |   kodkn  r2t| tr.t| S | S dS )NA   z   .)r   
isinstanceintchr)r.   r"   r"   r#   is_asciio   s
    
zdump_packet.<locals>.is_asciizpacket length:r   r%   zcall[%d]: %s (line %d)-B   c                s   g | ]} ||d   qS )   r"   ).0i)r.   r"   r#   
<listcomp>~   s    zdump_packet.<locals>.<listcomp>r      rE    c             S   s   dj t| S )Nz{:02X})formatr   )xr"   r"   r#   <lambda>   s    zdump_packet.<locals>.<lambda>z   r    c                s   dj  | S )Nz{})rK   )rL   )rB   r"   r#   rM      s    z  )printr+   rangesys	_getframef_codeco_namef_lineno
ValueErrorr   minjoinmap)r.   rG   fZ	dump_datadr"   )r.   rB   r#   dump_packetn   s     
$
& r\   c             C   s`   | sdS t rtdt|   t| j }t|j }t }|j| |j| |j }t||S )N    z	password=)DEBUGrO   strsha_newdigestupdate	_my_crypt)passwordmessageZstage1Zstage2sresultr"   r"   r#   	_scramble   s     

rh   c             C   sj   t | }d}xXt|D ]L}tjd| ||d  d tjd|||d  d A }|tjd|7 }qW |S )Nr]   Br   r   )r+   r   structunpackpack)Zmessage1Zmessage2lengthrg   rG   rL   r"   r"   r#   rc      s    rc      c               @   s   e Zd Zdd Zdd ZdS )RandStruct_323c             C   s"   d| _ || j  | _|| j  | _d S )Ni?)	max_valueseed1seed2)r(   rq   rr   r"   r"   r#   r)      s    zRandStruct_323.__init__c             C   sD   | j d | j | j | _ | j | j d | j | _t| j t| j S )N   !   )rq   rr   rp   float)r(   r"   r"   r#   my_rnd   s    zRandStruct_323.my_rndN)r/   r0   r1   r)   rv   r"   r"   r"   r#   ro      s   ro   c             C   s   t | }t |d t }tjd|}tjd|}t|d |d A |d |d A }tj }x8tttt	|D ]"}|j
tt|j d d  qnW tt|j d }	|j }
tj }x&|
D ]}|j
tt|t|	A  qW |j S )Nz>LLr   r      @   )_hash_password_323SCRAMBLE_LENGTH_323rj   rk   ro   r   BytesIOr   rW   r+   writer   r@   rv   getvaluer   )rd   re   Z	hash_passZhash_messageZhash_pass_nZhash_message_nZrand_stZoutbuf_extraoutcr"   r"   r#   _scramble_323   s    "
r   c             C   s   d}d}d}xVdd | D D ]D}||d@ | | |d>  d@ N }||d> |A  d@ }|| d@ }qW |d@ }|d@ }t jd||S )Ni5W0Pr   iqV4c             S   s   g | ]}|dkrt |qS )rJ   	    	   )rJ   r   r   r   )r   )rF   rL   r"   r"   r#   rH      s    z&_hash_password_323.<locals>.<listcomp>?   rn   l    r   rw   z>LLl        il        i)rj   rl   )rd   nraddZnr2r   Zr1Zr2r"   r"   r#   ry      s     ry   c             C   s   t jd| d d S )Nz<Irs   )rj   rl   )r,   r"   r"   r#   
pack_int24   s    r   c             C   s   | dk rt d|  np| dk r&t| S | dk r>dtjd|  S | dk r^d	tjd
| d d  S | dk rvdtjd|  S t d| df d S )Nr   zFEncoding %d is less than 0 - no representation in LengthEncodedIntegerr6   r   rE      z<Hr;      z<Irs   rx      z<QzIEncoding %x is larger than %x - no representation in LengthEncodedIntegeri   i   l            l            )rV   r   rj   rl   )rG   r"   r"   r#   
lenenc_int   s    r   c               @   s   e Zd ZdZd8Zdd Zdd Zdd	 Zd
d Zdd Z	d9ddZ
d:ddZerZdd Zndd Zdd Zdd Zdd Zdd Zdd  Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Zd5d6 Zd7S );MysqlPacketzrRepresentation of a MySQL response packet.

    Provides an interface for reading/parsing the packet results.
    	_position_datac             C   s   d| _ || _d S )Nr   )r   r   )r(   r.   encodingr"   r"   r#   r)      s    zMysqlPacket.__init__c             C   s   | j S )N)r   )r(   r"   r"   r#   get_all_data   s    zMysqlPacket.get_all_datac             C   sl   | j | j| j|  }t||krZd|t|| jt| j f }trRt| | 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^   rO   dumpr3   )r(   sizerg   errorr"   r"   r#   r-      s    zMysqlPacket.readc             C   s   | j | jd }d| _|S )z`Read all remaining data in the packet.

        (Subsequent read() will return errors.)
        N)r   r   )r(   rg   r"   r"   r#   read_all   s    zMysqlPacket.read_allc             C   s:   | j | }|dk s |t| jkr0td||f || _ dS )z1Advance the cursor in data buffer 'length' bytes.r   z4Invalid advance amount (%s) for cursor.  Position=%sN)r   r+   r   	Exception)r(   rm   Znew_positionr"   r"   r#   advance  s
    
zMysqlPacket.advancer   c             C   s,   |dk s|t | jkr"td| || _dS )z9Set the position of the data buffer cursor to 'position'.r   z)Invalid position to rewind cursor to: %s.N)r+   r   r   r   )r(   positionr"   r"   r#   rewind  s    zMysqlPacket.rewindr   c             C   s   | 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   rm   r"   r"   r#   	get_bytes  s    	zMysqlPacket.get_bytesc             C   s"   t | j| j }|  jd7  _|S )Nr   )ordr   r   )r(   rg   r"   r"   r#   
read_uint8!  s    zMysqlPacket.read_uint8c             C   s   | j | j }|  jd7  _|S )Nr   )r   r   )r(   rg   r"   r"   r#   r   &  s    c             C   s(   t jd| j| jd }|  jd7  _|S )Nz<Hr   r   )rj   unpack_fromr   r   )r(   rg   r"   r"   r#   read_uint16+  s    zMysqlPacket.read_uint16c             C   s0   t jd| j| j\}}|  jd7  _||d>  S )Nz<HBrs   rE   )rj   r   r   r   )r(   lowhighr"   r"   r#   read_uint240  s    zMysqlPacket.read_uint24c             C   s(   t jd| j| jd }|  jd7  _|S )Nz<Ir      )rj   r   r   r   )r(   rg   r"   r"   r#   read_uint325  s    zMysqlPacket.read_uint32c             C   s(   t jd| j| jd }|  jd7  _|S )Nz<Qr   rn   )rj   r   r   r   )r(   rg   r"   r"   r#   read_uint64:  s    zMysqlPacket.read_uint64c             C   s:   | j jd| j}|dk rd S | j | j| }|d | _|S )N    r   r   )r   findr   )r(   Zend_posrg   r"   r"   r#   read_string?  s    
zMysqlPacket.read_stringc             C   sT   | j  }|tkrdS |tk r |S |tkr0| j S |tkr@| j S |tkrP| j S dS )zRead 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_integerG  s    z'MysqlPacket.read_length_encoded_integerc             C   s   | j  }|dkrdS | j|S )a  Read 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(   rm   r"   r"   r#   read_length_coded_stringY  s    z$MysqlPacket.read_length_coded_stringc             C   s.   t j|}|j| j| j}|  j|j7  _|S )N)rj   Structr   r   r   r   )r(   fmtrf   rg   r"   r"   r#   read_structe  s    
zMysqlPacket.read_structc             C   s    | j dd dkot| j dkS )Nr   r   r   r   )r   r+   )r(   r"   r"   r#   is_ok_packetk  s    zMysqlPacket.is_ok_packetc             C   s    | j dd dkot| j dk S )Nr   r   r   r   )r   r+   )r(   r"   r"   r#   is_eof_packeto  s    zMysqlPacket.is_eof_packetc             C   s   | j dd dkS )Nr   r   r   )r   )r(   r"   r"   r#   is_auth_switch_requestu  s    z"MysqlPacket.is_auth_switch_requestc             C   s*   t | jdd }d|  ko$dkS   S )Nr   r      )r   r   )r(   field_countr"   r"   r#   is_resultset_packety  s    zMysqlPacket.is_resultset_packetc             C   s   | j dd dkS )Nr   r      )r   )r(   r"   r"   r#   is_load_local_packet}  s    z MysqlPacket.is_load_local_packetc             C   s   | j dd dkS )Nr   r      )r   )r(   r"   r"   r#   is_error_packet  s    zMysqlPacket.is_error_packetc             C   s@   | j  r<| j  | jd | j }tr0td| tj| j d S )Nr   zerrno =)	r   r   r   r   r^   rO   r   Zraise_mysql_exceptionr   )r(   errnor"   r"   r#   check_error  s    
 
zMysqlPacket.check_errorc             C   s   t | j d S )N)r\   r   )r(   r"   r"   r#   r     s    zMysqlPacket.dumpN)r   r   )r   )r   )r/   r0   r1   __doc__	__slots__r)   r   r-   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r"   r"   r"   r#   r      s8   	


r   c               @   s8   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d ZdS )FieldDescriptorPacketzA 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.
    c             C   s   t j| || | j| d S )N)r   r)   _parse_field_descriptor)r(   r.   r   r"   r"   r#   r)     s    zFieldDescriptorPacket.__init__c             C   sv   | j  | _| j  | _| j  j|| _| j  j|| _| j  j|| _| j  j|| _| jd\| _	| _
| _| _| _dS )zParse the 'Field Descriptor' (Metadata) packet.

        This is compatible with MySQL 4.1+ (not compatible with MySQL 4.0).
        z	<xHIBHBxxN)r   catalogdbdecode
table_nameZ	org_tablenameZorg_namer   	charsetnrrm   	type_codeflagsscale)r(   r   r"   r"   r#   r     s    

z-FieldDescriptorPacket._parse_field_descriptorc             C   s*   | j | jd| j | j | j| jd dkfS )zBProvides a 7-item tuple compatible with the Python PEP249 DB Spec.Nr   r   )r   r   get_column_lengthr   r   )r(   r"   r"   r#   description  s    z!FieldDescriptorPacket.descriptionc             C   s*   | j tjkr$tj| jd}| j| S | jS )Nr   )r   r   
VAR_STRINGr   getr   rm   )r(   Zmblenr"   r"   r#   r     s    
z'FieldDescriptorPacket.get_column_lengthc             C   s    d| j | j| j| j| j| jf S )Nz%s %r.%r.%r, type=%s, flags=%x)	__class__r   r   r   r   r   )r(   r"   r"   r#   __str__  s    zFieldDescriptorPacket.__str__N)	r/   r0   r1   r   r)   r   r   r   r   r"   r"   r"   r#   r     s   r   c               @   s    e Zd ZdZdd Zdd ZdS )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.
    c             C   sz   |j  s tdt| jj d || _| jjd | jj | _| jj | _	| j
d\| _| _| jj | _| jtj@ | _d S )NzCannot create z  object from invalid packet typer   z<HH)r   rV   r_   r   r/   packetr   r   affected_rows	insert_idr   server_statuswarning_countr   re   r   SERVER_MORE_RESULTS_EXISTShas_next)r(   from_packetr"   r"   r#   r)     s    zOKPacketWrapper.__init__c             C   s   t | j|S )N)getattrr   )r(   keyr"   r"   r#   __getattr__  s    zOKPacketWrapper.__getattr__N)r/   r0   r1   r   r)   r   r"   r"   r"   r#   r     s   r   c               @   s    e Zd ZdZdd Zdd ZdS )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.
    c             C   sT   |j  stdj| j|| _| jjd\| _| _trBt	d| j | jt
j@ | _d S )Nz3Cannot create '{0}' object from invalid packet typez<xhhzserver_status=)r   rV   rK   r   r   r   r   r   r^   rO   r   r   r   )r(   r   r"   r"   r#   r)     s    
 zEOFPacketWrapper.__init__c             C   s   t | j|S )N)r   r   )r(   r   r"   r"   r#   r     s    zEOFPacketWrapper.__getattr__N)r/   r0   r1   r   r)   r   r"   r"   r"   r#   r     s   r   c               @   s    e Zd ZdZdd Zdd ZdS )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.
    c             C   sF   |j  stdj| j|| _| jj dd  | _trBtd| j d S )Nz3Cannot create '{0}' object from invalid packet typer   z	filename=)	r   rV   rK   r   r   r   filenamer^   rO   )r(   r   r"   r"   r#   r)     s    
 zLoadLocalPacketWrapper.__init__c             C   s   t | j|S )N)r   r   )r(   r   r"   r"   r#   r     s    z"LoadLocalPacketWrapper.__getattr__N)r/   r0   r1   r   r)   r   r"   r"   r"   r#   r     s   
r   c               @   s  e Zd ZdZdZdZddddddddddddedddddddddddd^di ddfdd	Zd
d Zdd Z	e
dd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zd d! Zd"d# Zd_d$d%Zd&d' Zd(d) Zd`d*d+Zd,d- Zd.d/ Zdad0d1Zdbd2d3Zd4d5 Zd6d7 Zdcd9d:Z d;d< Z!ddd=d>Z"d?d@ Z#e$fdAdBZ%dCdD Z&dEdF Z'dedGdHZ(dIdJ Z)dKdL Z*dMdN Z+dOdP Z,dQdR Z-dSdT Z.dUdV Z/dWdX Z0dYdZ Z1d[d\ Z2e3j4Z4e3j5Z5e3j6Z6e3j7Z7e3j8Z8e3j9Z9e3j:Z:e3j;Z;e3j<Z<e3j=Z=dS )f
Connectionz
    Representation of a socket with a mysql server.

    The proper way to get an instance of this class is to call
    connect().
    NrN   r   FrE   i   c                s  |dk	rt jdt |dkr.tjd dkr.d}|dk	rB|dkrB|}|dk	rT| rT|}|s\|rdtd|rr|tjO }d| _|rt	stdd| _|tj
O }| j|| _r|	 rtjjd	rd
}	nd}	|	rBsdt   jtjj|	  fdd}|d|}|d|}|d|}|d|}|d|}t|d|}|d|}|pJd| _|pVd| _|pbt| _|pnd| _|| _|| _|dk	r|dkrtd|| _|dk	r|dkrtd|| _|r|| _d| _ nt!| _d| _ |dk	r|| _ t"| jj#| _#|tj$O }| jr|tj%O }|| _&|| _'|| _(d| _)d| _*d| _+|| _,|
dkrTt-}
t.dd |
j/ D | _0t.dd |
j/ D | _1|| _2|| _3|| _4|| _5|rd| _6n| j7  dS )aM
  
        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)
        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.
        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)
        Nzno_delay option is deprecatedr   r   Tz3compress and named_pipe arguments are not supportedFzssl module not foundwinz	c:\my.iniz/etc/my.cnfclientc                s.   |r|S y j | S  tk
r(   |S X d S )N)r   r   )r   arg)cfgread_default_groupr"   r#   _configm  s    z$Connection.__init__.<locals>._configuserrd   hostdatabasesocketportzdefault-character-set	localhosti  rN   zread_timeout should be >= 0zwrite_timeout should be >= 0zNot connectedc             S   s$   g | ]\}}t |tk	r||fqS r"   )typer@   )rF   kvr"   r"   r#   rH     s    z'Connection.__init__.<locals>.<listcomp>c             S   s$   g | ]\}}t |tkr||fqS r"   )r   r@   )rF   r   r   r"   r"   r#   rH     s    )8warningswarnDeprecationWarningrQ   version_infoNotImplementedErrorr   ZLOCAL_FILESsslSSL_ENABLEDSSL_create_ssl_ctxctxplatform
startswithr   r-   ospath
expanduserr@   r   r   DEFAULT_USERr   rd   r   unix_socketrV   _read_timeout_write_timeoutcharsetuse_unicodeDEFAULT_CHARSETr   r   ZCAPABILITIESCONNECT_WITH_DBclient_flagcursorclassconnect_timeout_result_affected_rows	host_infoautocommit_mode_convdictitemsencodersdecoderssql_modeinit_commandmax_allowed_packet_auth_plugin_mapr'   connect)r(   r   r   rd   r   r   r  r  r  Zread_default_fileconvr  r  r	  r  r
  r   r   compressZ
named_pipeZno_delay
autocommitr   passwdZlocal_infiler  Zdefer_connectZauth_plugin_mapread_timeoutZwrite_timeoutr   r"   )r   r   r#   r)     s    7












zConnection.__init__c             C   s   t |tjr|S |jd}|jd}|d ko2|d k}tj||d}| oR|jdd|_|r`tjntj|_d|kr|j	|d |jdd d	|kr|j
|d	  | jtjO  _| jtjO  _|S )
Ncacapath)cafiler  check_hostnameTcertr   )keyfilecipher)r?   r   
SSLContextr   create_default_contextr!  	CERT_NONECERT_REQUIREDverify_modeload_cert_chainset_ciphersoptionsOP_NO_SSLv2OP_NO_SSLv3)r(   Zsslpr  r  Zhasnocar   r"   r"   r#   r     s    

zConnection._create_ssl_ctxc             C   sn   | j dkrtjdtjddtj}z(y| j| W n tk
rH   Y nX W d| j }d| _ d| _	|j
  X dS )z*Send the quit message and close the socketNzAlready closedz<iBr   )r'   r   Errorrj   rl   r   ZCOM_QUIT_write_bytesr   _rfileclose)r(   	send_datar    r"   r"   r#   r2    s    


zConnection.closec             C   s
   | j d k	S )N)r'   )r(   r"   r"   r#   open  s    zConnection.openc          	   C   s2   | j r"y| j j  W n   Y nX d | _ d | _d S )N)r'   r2  r1  )r(   r"   r"   r#   __del__  s    zConnection.__del__c             C   s&   t || _| j }||kr"| j  d S )N)boolr  get_autocommit_send_autocommit_mode)r(   valuecurrentr"   r"   r#   r    s    
zConnection.autocommitc             C   s   t | jtj@ S )N)r6  r   r   ZSERVER_STATUS_AUTOCOMMIT)r(   r"   r"   r#   r7    s    zConnection.get_autocommitc             C   s0   | j  }|j stjddt|}|j| _|S )Ni  zCommand Out of Sync)_read_packetr   r   OperationalErrorr   r   )r(   pktokr"   r"   r#   _read_ok_packet  s    zConnection._read_ok_packetc             C   s&   | j tjd| j| j  | j  dS )z2Set whether or not to commit after every execute()zSET AUTOCOMMIT = %sN)_execute_commandr   	COM_QUERYescaper  r?  )r(   r"   r"   r#   r8    s    
z Connection._send_autocommit_modec             C   s   | j tjd | j  dS )zBegin transaction.ZBEGINN)r@  r   rA  r?  )r(   r"   r"   r#   begin  s    zConnection.beginc             C   s   | j tjd | j  dS )z Commit changes to stable storageZCOMMITN)r@  r   rA  r?  )r(   r"   r"   r#   commit  s    zConnection.commitc             C   s   | j tjd | j  dS )z!Roll back the current transactionZROLLBACKN)r@  r   rA  r?  )r(   r"   r"   r#   rollback  s    zConnection.rollbackc             C   s$   | j tjd  t| }|j  |jS )zSHOW WARNINGS)r@  r   rA  MySQLResultr-   rows)r(   rg   r"   r"   r#   show_warnings  s    zConnection.show_warningsc             C   s   | j tj| | j  dS )zSet current dbN)r@  r   ZCOM_INIT_DBr?  )r(   r   r"   r"   r#   	select_db  s    zConnection.select_dbc             C   s,   t |trd| j| d S t|| j|dS )zEscape whatever value you pass to it.
        
        Non-standard, for internal use; do not use this in your applications.
        ')mapping)r?   r   r   r   r  )r(   objrK  r"   r"   r#   rB    s    
zConnection.escapec             C   s   | j || jS )zrAlias for escape()
        
        Non-standard, for internal use; do not use this in your applications.
        )rB  r  )r(   rL  r"   r"   r#   literal  s    zConnection.literalc             C   s    | j tj@ r|jddS t|S )NrJ  z'')r   r   Z"SERVER_STATUS_NO_BACKSLASH_ESCAPESreplacer   )r(   rf   r"   r"   r#   r   "  s    zConnection.escape_stringc             C   s   |r|| S | j | S )z+Create a new cursor to execute queries with)r	  )r(   cursorr"   r"   r#   rO  (  s    zConnection.cursorc             C   s   | j  S )z%Context manager that returns a Cursor)rO  )r(   r"   r"   r#   	__enter__.  s    zConnection.__enter__c             C   s   |r| j   n| j  dS )z2On successful exit, commit. On exception, rollbackN)rE  rD  )r(   excr9  	tracebackr"   r"   r#   __exit__2  s    
zConnection.__exit__c             C   sV   t |tr4tpt r4tr&|j| j}n|j| jd}| jtj	| | j
|d| _| jS )Nsurrogateescape)
unbuffered)r?   r   r   r	   r   encoder   r@  r   rA  _read_query_resultr  )r(   sqlrU  r"   r"   r#   query:  s    zConnection.queryc             C   s   | j |d| _| jS )N)rU  )rW  r  )r(   rU  r"   r"   r#   next_resultF  s    zConnection.next_resultc             C   s   | j S )N)r  )r(   r"   r"   r#   r   J  s    zConnection.affected_rowsc             C   s"   t jd|}| jtj| | j S )Nz<I)rj   rl   r@  r   ZCOM_PROCESS_KILLr?  )r(   	thread_idr   r"   r"   r#   killM  s    zConnection.killTc             C   sn   | j dkr&|r| j  d}n
tjdy| jtjd | j S  tk
rh   |rb| j  | j	dS  Y nX dS )zCheck if the server is aliveNFzAlready closedrN   )
r'   r  r   r/  r@  r   ZCOM_PINGr?  r   ping)r(   Z	reconnectr"   r"   r#   r]  R  s    


zConnection.pingc             C   s:   t |j}| jtjd| j|  | j  || _|| _d S )NzSET NAMES %s)r   r   r@  r   rA  rB  r;  r  )r(   r  r   r"   r"   r#   set_charsetd  s
    
zConnection.set_charsetc             C   s:  y|d kr | j rZ| jdkrZtjtjtj}|j| j |j| j  d| _t	rt
d nxXytj| j| jf| j}P W q\ ttfk
r } z|jtjkrw\ W Y d d }~X q\X q\W d| j| jf | _t	rt
d |jtjtjd |jd  |jtjtjd || _t|d| _d	| _| j  | j  | jd k	rL| j }|jd
| jf | jd k	r|| j }|j| j |j   | j!  | j"d k	r| j#| j" W n t$k
r4 } zd | _|d k	ry|j   W n   Y nX t%|tttj&fr"t'j(dd| j|f }||_)t*j+ |_*t	rt
|j* | W Y d d }~X nX d S )Nr   	127.0.0.1zLocalhost via UNIX socketzconnected using unix_socketzsocket %s:%dzconnected using socketr   r2   r   zSET sql_mode=%si  z(Can't connect to MySQL server on %r (%s))r   r_  ),r  r   r   AF_UNIXSOCK_STREAM
settimeoutr
  r  r  r^   rO   create_connectionr   OSErrorIOErrorr   EINTR
setsockoptIPPROTO_TCPTCP_NODELAY
SOL_SOCKETSO_KEEPALIVEr'   r$   r1  _next_seq_id_get_server_information_request_authenticationr  rO  executer  r2  rD  r  r  BaseExceptionr?   r   r   r<  Zoriginal_exceptionrR  
format_exc)r(   r    er   rQ  r"   r"   r#   r  m  sr    
 
 


 
zConnection.connectc             C   sD   t t|t| j | }tr&t| | j| | jd d | _dS )zwWrites an entire "mysql packet" in its entirety to the network
        addings its length and sequence number.
        r   rI   N)r   r+   r   rl  r^   r\   r0  )r(   payloadr.   r"   r"   r#   write_packet  s
     
zConnection.write_packetc       
      C   s   d}x| j d}trt| tjd|\}}}||d>  }|| jkrXtjd|| jf | jd d | _| j |}tr~t| ||7 }|dkrq|tk rP qW ||| j	}	|	j
  |	S )	zRead an entire "mysql packet" in its entirety from the network
        and return a MysqlPacket type that represents the results.
        r]   r   z<HBBrE   z1Packet sequence number wrong - got %d expected %dr   rI   i )_read_bytesr^   r\   rj   rk   rl  r   InternalErrorMAX_PACKET_LENr   r   )
r(   Zpacket_typeZbuffZpacket_headerZbtrlZbtrhZpacket_numberZbytes_to_readZ	recv_datar   r"   r"   r#   r;    s,    
 

 zConnection._read_packetc             C   s   | j j| j x^y| jj|}P W q ttfk
rh } z&|jtjkrFwt	j
dd|f W Y d d }~X qX qW t||k rt	j
dd|S )Ni  z1Lost connection to MySQL server during query (%s)z,Lost connection to MySQL server during query)r'   rb  r  r1  r-   re  rd  r   rf  r   r<  r+   )r(   	num_bytesr.   rr  r"   r"   r#   ru    s    "zConnection._read_bytesc             C   sX   | j j| j y| j j| W n4 tk
rR } ztjdd|f W Y d d }~X nX d S )Ni  zMySQL server has gone away (%r))r'   rb  r  sendallre  r   r<  )r(   r.   rr  r"   r"   r#   r0    s
    zConnection._write_bytesc          	   C   sd   |r6yt | }|j  W qF   d|_d |_ Y qFX nt | }|j  || _|jd k	r^|j| _|jS )NF)rF  init_unbuffered_queryunbuffered_active
connectionr-   r  r   r   )r(   rU  rg   r"   r"   r#   rW    s    

zConnection._read_query_resultc             C   s   | j r| j jS dS d S )Nr   )r  r   )r(   r"   r"   r#   r     s    zConnection.insert_idc             C   s  | j stjd| jd k	rR| jjr6tjd | jj  x| jjrJ| j	  q8W d | _t
|trh|j| j}ttt|d }tjd||}||d |d   }| j| trt| d| _|tk rd S ||d d  }x@ttt|}| j|d |  ||d  }| r|tk rP qW d S )Nz(0, '')z.Previous unbuffered result was left incompleter   z<iB)r'   r   InterfaceErrorr  r{  r   r   _finish_unbuffered_queryr   rZ  r?   r   rV  r   rW   rw  r+   rj   rl   r0  r^   r\   rl  rt  )r(   commandrX  packet_sizeZpreluder   r"   r"   r#   r@    s6    






 zConnection._execute_commandc             C   s8  t | jjddd dkr*|  jtjO  _| jd kr<tdt| j	j
}t| jtrd| jj| j| _tjd| jd|d}| jr| jtj@ r| j| | jj| j| jd| _t| jd	| _|| j d
 }d}| jdkrt| jjd| j}| jtj@ r|t t!|| 7 }n4| jtj"@ r4|tjdt!|| 7 }n||d
 7 }| j#r| jtj$@ rt| j#trt| j#j| j| _#|| j#d
 7 }| jtj%@ r| j}t|tr|jd}||d
 7 }| j| | j& }|j' r4|j(  |j) }| jtj%@ r
|d k	r
| j*||}n*t+| jjd| jd
 }| j| | j& }d S )Nr>   r   r      zDid not specify a usernamez<iIB23sr]   )server_hostnamer2   r   rN   mysql_native_passwordr:   ri   ascii)rN   r  ),r@   server_versionsplitr  r   ZMULTI_RESULTSr   rV   r   r  idr?   r   rV  r   rj   rl   r   server_capabilitiesr   rt  r   wrap_socketr'   r   r$   r1  _auth_plugin_namerh   rd   saltZPLUGIN_AUTH_LENENC_CLIENT_DATAr   r+   ZSECURE_CONNECTIONr   r  PLUGIN_AUTHr;  r   r   r   _process_authr   )r(   Z
charset_idZ	data_initr.   Zauthrespr   auth_packetplugin_namer"   r"   r#   rn  -  sP    






z"Connection._request_authenticationc             C   sL  | j j|}|s"| j j|jd}|ry|| }|j|S  tk
rh   |dkrdtjdd||f Y q tk
r   tjdd||f Y qX nd }|dkrt| j	j
d|j d }nr|d	krt| j	j
d|j d }nL|d
kr | j	j
dd }n.|dkr|}x|j }|d@ dk}|d@ dk}	|j }
|
dkr\| j| j	j
dd  n|rd}y|j||
}| j|d  W nV tk
r   tjdd||f Y n. tk
r   tjdd||||
f Y nX ntjdd||f | j }|j  |j s|	rP qW |S tjdd| | j| | j }|j  |S )Nr  s   dialogi  zGAuthentication plugin '%s' not loaded: - %r missing authenticate methodzXAuthentication plugin '%s' not loaded: - %r cannot be constructed with connection objects   mysql_native_passwordr:   r   s   mysql_old_passwords   mysql_clear_passwordr%   r   r   s
   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   ZauthenticateAttributeErrorr   r<  	TypeErrorrh   rd   rV  r   r   r   rt  promptr;  r   r   )r(   r  r  Zplugin_classhandlerr.   r=  flagZecholastr  respr"   r"   r#   r  j  sh    




zConnection._process_authc             C   s
   | j d S )Nr   )server_thread_id)r(   r"   r"   r#   r[    s    zConnection.thread_idc             C   s   | j S )N)r  )r(   r"   r"   r#   character_set_name  s    zConnection.character_set_namec             C   s   | j S )N)r  )r(   r"   r"   r#   get_host_info  s    zConnection.get_host_infoc             C   s   | j S )N)protocol_version)r(   r"   r"   r#   get_proto_info  s    zConnection.get_proto_infoc       	      C   s   d}| j  }|j }tr t| t|||d  | _|d7 }|jd|}||| jd| _|d }t	j
d|||d  | _|d7 }|||d  | _|d7 }t	j
d	|||d
  d | _|d
7 }t||d kr\t	j
d|||d  \}}}}|d7 }|| _t|j| _|| _tr,td|  |  j|d> O  _trNtd| td|d }|d7 }t||| kr|  j||||  7  _||7 }|d7 }| jtj@ rt||kr|jd|}|dk r||d  jd| _n||| jd| _d S )Nr   r   r   r:   z<Ir   rn   r   z<Hr   r%   z<BHHBzserver_status: %xrE   z	salt_len:   
   )r;  r   r^   r\   r   r  r   r   r  rj   rk   r  r  r  r+   Zserver_languager   r   Zserver_charsetr   rO   maxr   r  r  )	r(   rG   r   r.   Z
server_endlangstatZcap_hZsalt_lenr"   r"   r#   rm    sL        

z"Connection._get_server_informationc             C   s   | j S )N)r  )r(   r"   r"   r#   get_server_info  s    zConnection.get_server_infoi @  i   )N)N)F)F)T)N)F)>r/   r0   r1   r   r'   r  r   r)   r   r2  propertyr4  r5  r  r7  r?  r8  rC  rD  rE  rH  rI  rB  rM  r   rO  rP  rS  rY  rZ  r   r\  r]  r^  r  rt  r   r;  ru  r0  rW  r   r@  rn  r  r[  r  r  r  rm  r  r   Warningr/  r}  ZDatabaseErrorZ	DataErrorr<  ZIntegrityErrorrv  ZProgrammingErrorZNotSupportedErrorr"   r"   r"   r#   r     s    	
	



	
C
&=?:r   c               @   st   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd ZdS )rF  c             C   sF   || _ d| _d| _d| _d| _d| _d| _d| _d| _d| _	d| _
dS )z.
        :type connection: Connection
        Nr   F)r|  r   r   r   r   re   r   r   rG  r   r{  )r(   r|  r"   r"   r#   r)      s    zMySQLResult.__init__c             C   s   | j r| j  d S )N)r{  r~  )r(   r"   r"   r#   r5    s    zMySQLResult.__del__c             C   sN   z@| j j }|j r | j| n|j r4| j| n
| j| W d d | _ X d S )N)r|  r;  r   r?  r   _read_load_local_packet_read_result_packet)r(   first_packetr"   r"   r#   r-     s    
zMySQLResult.readc             C   sl   d| _ | jj }|j r0| j| d| _ d | _n8|j rP| j| d| _ d | _n|j | _| j	  d| _
d S )NTFl    )r{  r|  r;  r   r?  r   r  r   r   _get_descriptionsr   )r(   r  r"   r"   r#   rz  !  s    



z!MySQLResult.init_unbuffered_queryc             C   s<   t |}|j| _|j| _|j| _|j| _|j| _|j| _d S )N)r   r   r   r   r   re   r   )r(   r  	ok_packetr"   r"   r#   r?  6  s    zMySQLResult._read_ok_packetc          	   C   sh   t |}t|j| j}y|j  W n   | jj   Y nX | jj }|j sZtjdd| j	| d S )Ni  zCommands Out of Sync)
r   LoadLocalFiler   r|  r3  r;  r   r   r<  r?  )r(   r  Zload_packetZsenderr  r"   r"   r#   r  ?  s    

z#MySQLResult._read_load_local_packetc             C   s(   |j  sdS t|}|j| _|j| _dS )NFT)r   r   r   r   )r(   r   Zwpr"   r"   r#   _check_packet_is_eofM  s    z MySQLResult._check_packet_is_eofc             C   s   |j  | _| j  | j  d S )N)r   r   r  _read_rowdata_packet)r(   r  r"   r"   r#   r  Y  s    
zMySQLResult._read_result_packetc             C   sP   | j s
d S | jj }| j|r4d| _ d | _d | _d S | j|}d| _|f| _|S )NFr   )r{  r|  r;  r  rG  _read_row_from_packetr   )r(   r   rowr"   r"   r#   _read_rowdata_packet_unbuffered^  s    


z+MySQLResult._read_rowdata_packet_unbufferedc             C   s0   x*| j r*| jj }| j|rd| _ d | _qW d S )NF)r{  r|  r;  r  )r(   r   r"   r"   r#   r~  p  s
    

z$MySQLResult._finish_unbuffered_queryc             C   sN   g }x0| j j }| j|r"d| _ P |j| j| qW t|| _t|| _dS )z:Read a rowdata packet for each data row in the result set.N)	r|  r;  r  appendr  r+   r   tuplerG  )r(   rG  r   r"   r"   r#   r  z  s    


z MySQLResult._read_rowdata_packetc             C   s   g }xx| j D ]n\}}y|j }W n tk
r6   P Y nX |d k	rp|d k	rR|j|}tr`td| |d k	rp||}|j| qW t|S )NzDEBUG: DATA = )
convertersr   
IndexErrorr   r^   rO   r  r  )r(   r   r  r   	converterr.   r"   r"   r#   r    s    
 
z!MySQLResult._read_row_from_packetc       
      C   s  g | _ g | _| jj}| jj}g }xt| jD ]}| jjt}| j j	| |j	|j
  |j}|r|tjkrp|}q|tkr|jdkrd}q|}qd}nd}| jjj|}|tkrd}trtdj|| | jj	||f q,W | jj }	|	j stdt|| _
dS )z>Read a column descriptor packet for each column in the result.r   Nr  zDEBUG: field={}, converter={}zProtocol error, expecting EOF)fieldsr  r|  r  r   r   r   r;  r   r  r   r   r   ZJSON
TEXT_TYPESr   r  r   r   r^   rO   rK   r   r3   r  )
r(   r  Zconn_encodingr   rG   fieldZ
field_typer   r  Z
eof_packetr"   r"   r#   r    s8    

 
zMySQLResult._get_descriptionsN)r/   r0   r1   r)   r5  r-   rz  r?  r  r  r  r  r~  r  r  r  r"   r"   r"   r#   rF    s   	
rF  c               @   s   e Zd Zdd Zdd ZdS )r  c             C   s   || _ || _d S )N)r   r|  )r(   r   r|  r"   r"   r#   r)     s    zLoadLocalFile.__init__c             C   s   | j jstjd| j }zvyHt| jd2}t|jd	}x|j|}|sHP |j	| q8W W dQ R X W n( t
k
r   tjddj| jY nX W d|j	d X dS )
z3Send data packets from the local file to the serverz(0, '')r2   rE   i   Ni  zCan't find file '{0}'r]   i @  )r|  r'   r   r}  r4  r   rW   r  r-   rt  re  r<  rK   )r(   conn	open_filer  chunkr"   r"   r#   r3    s    

zLoadLocalFile.send_dataN)r/   r0   r1   r)   r3  r"   r"   r"   r#   r    s   r  )r   r   )r   r%   i   i )`
__future__r   _compatr   r   r   r   r   r	   r   	functoolsr
   hashlibr   r   r   rj   rQ   rR  r   r  r   r   r   	constantsr   r   r   r   r  r   r   r   r   r  Zcursorsr   Z
optionfiler   utilr   r   rN   r   r   r   ImportErrorgetpassgetuserr   KeyErrorr^   r   Z_py_versionZ	_socketior   r$   objectr&   setZBITZBLOBZ	LONG_BLOBZMEDIUM_BLOBSTRINGZ	TINY_BLOBr   ZVARCHARZGEOMETRYr  newr`   r   r   r   r   r   r  rw  r\   rh   rc   rz   ro   r   ry   r   r   r   r   r   r   r   r   rF  r  r"   r"   r"   r#   <module>   s    






 .2     ~ G