U
    Ýëd3È  ã                	   @   sš  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 ddl	m
Z
mZ ddlmZmZmZmZmZ ddlmZ ddlmZ ddlmZ dd	lmZmZmZmZmZmZ dd
lmZmZ zd dl Z dZ!W n e"k
rð   dZ dZ!Y nX zd dl#Z#e# $¡ Z%[#W n e"e&fk
r&   dZ%Y nX dZ'ej(ej)ej*ej+ej,ej-ej.ej/ej0h	Z1dZ2dZ3dd„ Z4dd„ Z5G dd„ dƒZ6G dd„ dƒZ7G dd„ dƒZ8dS )é    Né   )Ú_auth)Úcharset_by_nameÚcharset_by_id)ÚCLIENTÚCOMMANDÚCRÚ
FIELD_TYPEÚSERVER_STATUS)Ú
converters)ÚCursor)ÚParser)Údump_packetÚMysqlPacketÚFieldDescriptorPacketÚOKPacketWrapperÚEOFPacketWrapperÚLoadLocalPacketWrapper)ÚerrÚVERSION_STRINGTFZutf8mb4éÿÿÿ c                 C   s   t  d| ¡d d… S )Nú<Ié   )ÚstructÚpack)Ún© r   ú7/tmp/pip-unpacked-wheel-zrp_b4p0/pymysql/connections.pyÚ_pack_int24C   s    r   c                 C   sŒ   | dk rt d|  ƒ‚nr| dk r(t| gƒS | dk r@dt d| ¡ S | dk r`dt d	| ¡d d
…  S | dk rxdt d| ¡ S t d| df ƒ‚d S )Nr   zFEncoding %d is less than 0 - no representation in LengthEncodedIntegeréû   i   ó   üú<Hé   ó   ýr   r   l            ó   þz<QzIEncoding %x is larger than %x - no representation in LengthEncodedInteger)Ú
ValueErrorÚbytesr   r   )Úir   r   r   Ú_lenenc_intH   s"    ÿ
ÿÿr(   c                '   @   s&  e Zd Zd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ddddddddddddd	œ&d
d„Z	dd„ Z
dd„ Zdd„ Zdd„ Zedd„ ƒZdd„ Ze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dbd*d+„Zd,d-„ Zd.d/„ Zd0d1„ Zdcd2d3„Zddd4d5„Z ded6d7„Z!d8d9„ Z"d:d;„ Z#dfd<d=„Z$d>d?„ Z%dgd@dA„Z&dBdC„ Z'e(fdDdE„Z)dFdG„ Z*dHdI„ Z+dhdJdK„Z,dLdM„ Z-dNdO„ Z.dPdQ„ Z/dRdS„ Z0dTdU„ Z1dVdW„ Z2dXdY„ Z3dZd[„ Z4d\d]„ Z5d^d_„ Z6d`da„ Z7e8j9Z9e8j:Z:e8j;Z;e8j<Z<e8j=Z=e8j>Z>e8j?Z?e8j@Z@e8jAZAe8jBZBdS )iÚ
ConnectionaP  
    Representation of a socket with a mysql server.

    The proper way to get an instance of this class is to call
    connect().

    Establish a connection to the MySQL database. Accepts several
    arguments:

    :param host: Host where the database server is located
    :param user: Username to log in as
    :param password: Password to use.
    :param database: Database to use, None to not use a particular one.
    :param port: MySQL port to use, default is usually OK. (default: 3306)
    :param 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.
    :param unix_socket: Optionally, you can use a unix socket rather than TCP/IP.
    :param read_timeout: The timeout for reading from the connection in seconds (default: None - no timeout)
    :param write_timeout: The timeout for writing to the connection in seconds (default: None - no timeout)
    :param charset: Charset you want to use.
    :param sql_mode: Default SQL_MODE to use.
    :param read_default_file:
        Specifies  my.cnf file to read these parameters from under the [client] section.
    :param conv:
        Conversion dictionary to use instead of the default one.
        This is used to provide custom marshalling and unmarshalling of types.
        See converters.
    :param use_unicode:
        Whether or not to default to unicode strings.
        This option defaults to true.
    :param client_flag: Custom flags to send to MySQL. Find potential values in constants.CLIENT.
    :param cursorclass: Custom cursor class to use.
    :param init_command: Initial SQL statement to run when connection is established.
    :param connect_timeout: Timeout before throwing an exception when connecting.
        (default: 10, min: 1, max: 31536000)
    :param ssl:
        A dict of arguments similar to mysql_ssl_set()'s parameters.
    :param ssl_ca: Path to the file that contains a PEM-formatted CA certificate
    :param ssl_cert: Path to the file that contains a PEM-formatted client certificate
    :param ssl_disabled: A boolean value that disables usage of TLS
    :param ssl_key: Path to the file that contains a PEM-formatted private key for the client certificate
    :param ssl_verify_cert: Set to true to check the validity of server certificates
    :param ssl_verify_identity: Set to true to check the server's identity
    :param read_default_group: Group to read from in the configuration file.
    :param autocommit: Autocommit mode. None means use server default. (default: False)
    :param local_infile: Boolean to enable the use of LOAD DATA LOCAL command. (default: False)
    :param 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).
    :param defer_connect: Don't explicitly connect on construction - wait for connect call.
        (default: False)
    :param 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)
    :param server_public_key: SHA256 authentication plugin public key value. (default: None)
    :param binary_prefix: Add _binary prefix on bytes and bytearray. (default: False)
    :param compress: Not supported
    :param named_pipe: Not supported
    :param db: **DEPRECATED** Alias for database.
    :param passwd: **DEPRECATED** Alias for password.

    See `Connection <https://www.python.org/dev/peps/pep-0249/#connection-objects>`_ in the
    specification.
    NÚ Fr   Té
   r"   )&ÚuserÚpasswordÚhostÚdatabaseÚunix_socketÚportÚcharsetÚsql_modeÚread_default_fileÚconvÚuse_unicodeÚclient_flagÚcursorclassÚinit_commandÚconnect_timeoutÚread_default_groupÚ
autocommitÚlocal_infileÚmax_allowed_packetÚdefer_connectÚauth_plugin_mapÚread_timeoutÚwrite_timeoutÚbind_addressÚbinary_prefixÚprogram_nameÚserver_public_keyÚsslÚssl_caÚssl_certÚssl_disabledÚssl_keyÚssl_verify_certÚssl_verify_identityÚcompressÚ
named_pipeÚpasswdÚdbc       &   *         s¼  |&d k	r|d kr|&}|%d k	r$|s$|%}|#s,|$r4t dƒ‚t|ƒ| _| jrN|tjO }ˆrl|	sltj d¡rhd}	nd}	|	r6ˆszd‰tƒ ‰ ˆ  	t
j |	¡¡ ‡ ‡fdd„}'|'d|ƒ}|'d	|ƒ}|'d
|ƒ}|'d|ƒ}|'d|ƒ}t|'d|ƒƒ}|'d|ƒ}|'d|ƒ}|süi }t|tƒr6dD ](}(|'d|( | |(¡ƒ})|)r|)||(< qd| _|sÒ|s`|s`| s`|!s`|"r¢|t|"ƒ|!d k	rv|!nddœ}|d k	r||d< | d k	r¢| |d< |rÒts¶t dƒ‚d| _|tjO }|  |¡| _|pÚd| _|pæd| _t| jƒtk	rtdƒ‚|p
t| _|pd| _t| jtƒr6| j d¡| _|| _|| _ || _!d|  k r`dksjn tdƒ‚|prd | _"|d k	r’|dkr’td ƒ‚|| _#|d k	r´|dkr´td!ƒ‚|| _$|pÂt%| _&|| _'t(| j&ƒj)| _)|tj*O }| jrö|tj+O }|| _,|| _-d | _.d| _/d"| _0|| _1|
d kr*t2j3}
d#d$„ |
 4¡ D ƒ| _5d%d$„ |
 4¡ D ƒ| _6|| _7|| _8|| _9|pli | _:|| _;|| _<d&tt
 =¡ ƒt>d'œ| _?|r¢|| j?d(< |r°d | _@n|  A¡  d S ))Nz3compress and named_pipe arguments are not supportedÚwinz	c:\my.iniz/etc/my.cnfÚclientc                    s4   |r|S zˆ   ˆ| ¡W S  tk
r.   | Y S X d S ©N)ÚgetÚ	Exception)ÚkeyÚarg©Úcfgr;   r   r   Ú_configò   s    z$Connection.__init__.<locals>._configr,   r-   r.   r/   Úsocketr1   zbind-addresszdefault-character-set)ÚcaÚcapathÚcertrW   Úcipherzssl-F)r]   Úcheck_hostnameÚverify_moder_   rW   zssl module not foundTÚ	localhostiê  zport should be of type intó    Úlatin1r   i€3áz+connect_timeout should be >0 and <=31536000zread_timeout should be > 0zwrite_timeout should be > 0zNot connectedc                 S   s"   i | ]\}}t |ƒtk	r||“qS r   ©ÚtypeÚint©Ú.0ÚkÚvr   r   r   Ú
<dictcomp>L  s       z'Connection.__init__.<locals>.<dictcomp>c                 S   s"   i | ]\}}t |ƒtkr||“qS r   rf   ri   r   r   r   rm   M  s       Zpymysql)Z_client_nameZ_pidZ_client_versionrE   )BÚNotImplementedErrorÚboolÚ_local_infiler   ZLOCAL_FILESÚsysÚplatformÚ
startswithr   ÚreadÚosÚpathÚ
expanduserrh   Ú
isinstanceÚdictrU   rG   ÚSSL_ENABLEDÚSSLÚ_create_ssl_ctxÚctxr.   r1   rg   r%   ÚDEFAULT_USERr,   r-   ÚstrÚencoderQ   r0   rC   r:   Ú_read_timeoutÚ_write_timeoutÚDEFAULT_CHARSETr2   r6   r   ÚencodingZCAPABILITIESÚCONNECT_WITH_DBr7   r8   Ú_resultÚ_affected_rowsÚ	host_infoÚautocommit_moder   ZconversionsÚitemsÚencodersÚdecodersr3   r9   r>   Ú_auth_plugin_mapÚ_binary_prefixrF   Úgetpidr   Ú_connect_attrsÚ_sockÚconnect)*Úselfr,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   r[   rW   Úvaluer   rY   r   Ú__init__¥   sÔ    *ÿ








ÿû






ý
zConnection.__init__c                 C   s   | S rT   r   ©r“   r   r   r   Ú	__enter__c  s    zConnection.__enter__c                 G   s   ~|   ¡  d S rT   )Úclose)r“   Úexc_infor   r   r   Ú__exit__f  s    zConnection.__exit__c                 C   sR  t |tjƒr|S | d¡}| d¡}|d ko2|d k}tj||d}| oR| dd¡|_| d¡}|d kr||rrtjntj|_nxt |t	ƒrš|rtjntj|_nZt |t
ƒr¬| ¡ }|dkr¾tj|_n6|dkrÐtj|_n$|d	krâtj|_n|rìtjntj|_d
|kr|j|d
 | d¡d d|kr.| |d ¡ | jtjO  _| jtjO  _|S )Nr]   r^   )Úcafiler^   ra   Trb   )ÚnoneÚ0ÚfalseÚnoÚoptional)ÚrequiredÚ1ÚtrueÚyesr_   rW   )Úkeyfiler`   )rx   rG   Ú
SSLContextrU   Úcreate_default_contextra   Ú	CERT_NONEÚCERT_REQUIREDrb   ro   r   ÚlowerÚCERT_OPTIONALÚload_cert_chainÚset_ciphersÚoptionsÚOP_NO_SSLv2ÚOP_NO_SSLv3)r“   Zsslpr]   r^   Zhasnocar}   Zverify_mode_valuer   r   r   r|   j  s8    









zConnection._create_ssl_ctxc                 C   sl   | j rt d¡‚d| _ | jdkr$dS t ddtj¡}z(z|  	|¡ W n t
k
rX   Y nX W 5 |  ¡  X dS )z÷
        Send the quit message and close the socket.

        See `Connection.close() <https://www.python.org/dev/peps/pep-0249/#Connection.close>`_
        in the specification.

        :raise Error: If the connection is already closed.
        úAlready closedTNú<iBr   )Ú_closedr   ÚErrorr‘   r   r   r   ZCOM_QUITÚ_force_closeÚ_write_bytesrV   )r“   Ú	send_datar   r   r   r˜   Š  s    	


zConnection.closec                 C   s
   | j dk	S )z%Return True if the connection is openN)r‘   r–   r   r   r   Úopen   s    zConnection.openc                 C   s2   | j r"z| j  ¡  W n   Y nX d| _ d| _dS )z%Close connection without QUIT messageN)r‘   r˜   Ú_rfiler–   r   r   r   rµ   ¥  s    zConnection._force_closec                 C   s&   t |ƒ| _|  ¡ }||kr"|  ¡  d S rT   )ro   r‰   Úget_autocommitÚ_send_autocommit_mode)r“   r”   Úcurrentr   r   r   r<   ±  s    
zConnection.autocommitc                 C   s   t | jtj@ ƒS rT   )ro   Úserver_statusr
   ZSERVER_STATUS_AUTOCOMMITr–   r   r   r   rº   ·  s    zConnection.get_autocommitc                 C   s0   |   ¡ }| ¡ st dd¡‚t|ƒ}|j| _|S )NéÞ  zCommand Out of Sync)Ú_read_packetÚis_ok_packetr   ÚOperationalErrorr   r½   )r“   ÚpktÚokr   r   r   Ú_read_ok_packetº  s    zConnection._read_ok_packetc                 C   s&   |   tjd|  | 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   r»   Â  s
     ÿz Connection._send_autocommit_modec                 C   s   |   tjd¡ |  ¡  dS )zBegin transaction.ZBEGINN©rÅ   r   rÆ   rÄ   r–   r   r   r   ÚbeginÉ  s    zConnection.beginc                 C   s   |   tjd¡ |  ¡  dS )z¨
        Commit changes to stable storage.

        See `Connection.commit() <https://www.python.org/dev/peps/pep-0249/#commit>`_
        in the specification.
        ZCOMMITNrÈ   r–   r   r   r   ÚcommitÎ  s    zConnection.commitc                 C   s   |   tjd¡ |  ¡  dS )z­
        Roll back the current transaction.

        See `Connection.rollback() <https://www.python.org/dev/peps/pep-0249/#rollback>`_
        in the specification.
        ZROLLBACKNrÈ   r–   r   r   r   ÚrollbackØ  s    zConnection.rollbackc                 C   s$   |   tjd¡ t| ƒ}| ¡  |jS )z%Send the "SHOW WARNINGS" SQL command.zSHOW WARNINGS)rÅ   r   rÆ   ÚMySQLResultrt   Úrows)r“   Úresultr   r   r   Úshow_warningsâ  s    zConnection.show_warningsc                 C   s   |   tj|¡ |  ¡  dS )zI
        Set current db.

        :param db: The name of the db.
        N)rÅ   r   ZCOM_INIT_DBrÄ   )r“   rQ   r   r   r   Ú	select_dbé  s    zConnection.select_dbc                 C   sX   t |tƒrd|  |¡ d S t |ttfƒrF|  |¡}| jrBd| }|S tj|| j	|dS )z}Escape whatever value you pass to it.

        Non-standard, for internal use; do not use this in your applications.
        ú'Ú_binary)Úmapping)
rx   r   Úescape_stringr&   Ú	bytearrayÚ_quote_bytesrŽ   r   Zescape_itemr2   )r“   ÚobjrÓ   Úretr   r   r   rÇ   ò  s    

zConnection.escapec                 C   s   |   || j¡S )zjAlias for escape()

        Non-standard, for internal use; do not use this in your applications.
        )rÇ   r‹   )r“   r×   r   r   r   Úliteral   s    zConnection.literalc                 C   s"   | j tj@ r| dd¡S t |¡S )NrÑ   z'')r½   r
   Ú"SERVER_STATUS_NO_BACKSLASH_ESCAPESÚreplacer   rÔ   ©r“   Úsr   r   r   rÔ     s    zConnection.escape_stringc                 C   s0   | j tj@ r&d| dd¡ dd¡f S t |¡S )Nz'%s'ó   's   ''ÚasciiÚsurrogateescape)r½   r
   rÚ   rÛ   Údecoder   Zescape_bytesrÜ   r   r   r   rÖ     s    zConnection._quote_bytesc                 C   s   |r|| ƒS |   | ¡S )a	  
        Create a new cursor to execute queries with.

        :param cursor: The type of cursor to create; one of :py:class:`Cursor`,
            :py:class:`SSCursor`, :py:class:`DictCursor`, or :py:class:`SSDictCursor`.
            None means use Cursor.
        )r8   )r“   Úcursorr   r   r   râ     s    zConnection.cursorc                 C   s:   t |tƒr| | jd¡}|  tj|¡ | j|d| _| jS )Nrà   ©Ú
unbuffered)	rx   r   r€   r„   rÅ   r   rÆ   Ú_read_query_resultr‡   )r“   Úsqlrä   r   r   r   Úquery  s
    
zConnection.queryc                 C   s   | j |d| _| jS )Nrã   )rå   r‡   )r“   rä   r   r   r   Únext_result'  s    zConnection.next_resultc                 C   s   | j S rT   )r‡   r–   r   r   r   Úaffected_rows+  s    zConnection.affected_rowsc                 C   s"   t  d|¡}|  tj|¡ |  ¡ S )Nr   )r   r   rÅ   r   ZCOM_PROCESS_KILLrÄ   )r“   Ú	thread_idrX   r   r   r   Úkill.  s    zConnection.killc                 C   st   | j dkr&|r|  ¡  d}n
t d¡‚z|  tjd¡ |  ¡  W n. tk
rn   |rh|  ¡  |  	d¡ n‚ Y nX dS )z¹
        Check if the server is alive.

        :param reconnect: If the connection is closed, reconnect.
        :raise Error: If the connection is closed and reconnect=False.
        NFr±   r*   )
r‘   r’   r   r´   rÅ   r   ZCOM_PINGrÄ   rV   Úping)r“   Z	reconnectr   r   r   rì   3  s    

zConnection.pingc                 C   s:   t |ƒj}|  tjd|  |¡ ¡ |  ¡  || _|| _d S )NzSET NAMES %s)r   r„   rÅ   r   rÆ   rÇ   r¿   r2   )r“   r2   r„   r   r   r   Úset_charsetJ  s
    
zConnection.set_charsetc              
   C   s`  d| _ z´|d kr&| jr\t tjtj¡}| | j¡ | | j¡ d| _d| _	t
rZtdƒ nÀi }| jd k	rx| jdf|d< z"tj| j| jf| jf|Ž}W qÚW qx ttfk
rÖ } z|jtjkrÄW Y ¢qx‚ W 5 d }~X Y qxX qxd| j| jf | _t
røtdƒ | tjtjd	¡ | tjtjd	¡ | d ¡ || _| d
¡| _d| _|  ¡  |  ¡  | jd k	rr|   ¡ }| !d| jf¡ | j"d k	r¢|   ¡ }| !| j"¡ | #¡  |  $¡  | j%d k	rº|  &| j%¡ W nž t'k
rZ } z~d | _|d k	rúz| #¡  W n   Y nX t(|tttj)fƒrHt* +dd| j|f ¡}||_,t- .¡ |_-t
rDt|j-ƒ |‚‚ W 5 d }~X Y nX d S )NFzLocalhost via UNIX socketTzconnected using unix_socketr   Úsource_addresszsocket %s:%dzconnected using socketr   ÚrbzSET sql_mode=%siÓ  z(Can't connect to MySQL server on %r (%s))/r³   r0   r\   ÚAF_UNIXÚSOCK_STREAMÚ
settimeoutr:   r’   rˆ   Ú_secureÚDEBUGÚprintrC   Úcreate_connectionr.   r1   ÚOSErrorÚIOErrorÚerrnoÚEINTRÚ
setsockoptÚIPPROTO_TCPÚTCP_NODELAYÚ
SOL_SOCKETÚSO_KEEPALIVEr‘   Úmakefiler¹   Ú_next_seq_idÚ_get_server_informationÚ_request_authenticationr3   râ   Úexecuter9   r˜   rÊ   r‰   r<   ÚBaseExceptionrx   Úerrorr   rÁ   Zoriginal_exceptionÚ	tracebackÚ
format_exc)r“   ÚsockÚkwargsÚeÚcÚexcr   r   r   r’   S  s‚    



 ÿÿ

 ÿ

zConnection.connectc                 C   sF   t t|ƒƒt| jgƒ | }tr(t|ƒ |  |¡ | jd d | _dS )zvWrites an entire "mysql packet" in its entirety to the network
        adding its length and sequence number.
        r   é   N)r   Úlenr&   r  rô   r   r¶   )r“   ÚpayloadÚdatar   r   r   Úwrite_packetŸ  s
    
zConnection.write_packetc           
      C   sð   t ƒ }|  d¡}t d|¡\}}}||d>  }|| jkrj|  ¡  |dkrVt tj	d¡‚t 
d|| jf ¡‚| jd d | _|  |¡}trt|ƒ ||7 }|d	kr¢q|tk rq®q|t|ƒ| jƒ}	|	 ¡ rì| jd
k	rä| jjdkräd| j_|	 ¡  |	S )a!  Read an entire "mysql packet" in its entirety from the network
        and return a MysqlPacket type that represents the results.

        :raise OperationalError: If the connection to the MySQL server is lost.
        :raise InternalError: If the packet sequence number is wrong.
        é   z<HBBé   r   ú,Lost connection to MySQL server during queryz1Packet sequence number wrong - got %d expected %dr   r  r   NTF)rÕ   Ú_read_bytesr   Úunpackr  rµ   r   rÁ   r   ÚCR_SERVER_LOSTÚInternalErrorrô   r   ÚMAX_PACKET_LENr&   r„   Zis_error_packetr†   Úunbuffered_activeZraise_for_error)
r“   Zpacket_typeZbuffZpacket_headerZbtrlZbtrhZpacket_numberZbytes_to_readZ	recv_dataÚpacketr   r   r   r¿   «  s>    

þÿÿ
zConnection._read_packetc              
   C   s¾   | j  | j¡ z| j |¡}W q˜W q ttfk
rx } z4|jtjkrLW Y ¢"q|  	¡  t
 tjd|f ¡‚W 5 d }~X Y q tk
r”   |  	¡  ‚ Y qX qt|ƒ|k rº|  	¡  t
 tjd¡‚|S )Nz1Lost connection to MySQL server during query (%s)r  )r‘   rò   r   r¹   rt   rø   r÷   rù   rú   rµ   r   rÁ   r   r  r  r  )r“   Ú	num_bytesr  r  r   r   r   r  Ø  s,    þ
 ÿzConnection._read_bytesc              
   C   sb   | j  | j¡ z| j  |¡ W n> tk
r\ } z |  ¡  t tj	d|f ¡‚W 5 d }~X Y nX d S )NzMySQL server has gone away (%r))
r‘   rò   r‚   Úsendallrø   rµ   r   rÁ   r   ZCR_SERVER_GONE_ERROR)r“   r  r  r   r   r   r¶   ñ  s     ÿzConnection._write_bytesc                 C   sj   d | _ |r<zt| ƒ}| ¡  W qL   d|_d |_‚ Y qLX nt| ƒ}| ¡  || _ |jd k	rd|j| _|jS ©NF)r†   rÌ   Úinit_unbuffered_queryr  Ú
connectionrt   r½   ré   )r“   rä   rÎ   r   r   r   rå   û  s    

zConnection._read_query_resultc                 C   s   | j r| j jS dS d S ©Nr   )r†   Ú	insert_idr–   r   r   r   r#    s    zConnection.insert_idc                 C   s  | j st dd¡‚| jdk	rP| jjr8t d¡ | j ¡  | jjrJ|  	¡  q8d| _t
|tƒrf| | j¡}ttt|ƒd ƒ}t d||¡}||d|d …  }|  |¡ tr°t|ƒ d| _|tk rÂdS ||d d… }ttt|ƒƒ}|  |d|… ¡ ||d… }|sÒ|tk rÒqqÒdS )z~
        :raise InterfaceError: If the connection is closed.
        :raise ValueError: If no username was specified.
        r   r*   Nz.Previous unbuffered result was left incompleter   r²   )r‘   r   ÚInterfaceErrorr†   r  ÚwarningsÚwarnÚ_finish_unbuffered_queryÚhas_nextrè   rx   r   r€   r„   Úminr  r  r   r   r¶   rô   r   r  r  )r“   Úcommandræ   Úpacket_sizeZpreluder  r   r   r   rÅ     s4    





zConnection._execute_commandc           
      C   sÔ  t | 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¡| _t d| jt|d¡}| jr¾| jtj@ r¾|  |¡ | jj| j| jd| _| j d	¡| _d
| _|| j d }d}d }| jdkrôd}t | j | j!¡}n®| jdkrd}t | j | j!¡}nŒ| jdkr^d}| j rNt"r<t#dƒ t $| j | j!¡}nt"r¢t#dƒ nD| jdkr¢d}| jr| jtj@ r| j d }n| j ržd}nd}| jtj%@ rÆ|t&t'|ƒƒ| 7 }n4| jtj(@ rî|t dt'|ƒ¡| 7 }n||d 7 }| j)r<| jtj*@ r<t| j)tƒr.| j) | j¡| _)|| j)d 7 }| jtj+@ r\||pTdd 7 }| jtj,@ ràd}| j- .¡ D ]N\}}| d¡}|t dt'|ƒ¡| 7 }| d¡}|t dt'|ƒ¡| 7 }qx|t dt'|ƒ¡| 7 }|  |¡ |  /¡ }	|	 0¡ rht"r
t#dƒ |	 1¡  |	 2¡ }| jtj+@ r@|d k	r@|  3||	¡}	n&t 4| j | j!¡d }|  |¡ |  /¡ }	nZ|	 5¡ rÂt"r€t#dƒ | jdkršt 6| |	¡}	n(| jdkr´t 7| |	¡}	nt8 9d| j¡‚t"rÐt#dƒ d S )NÚ.r   r   é   zDid not specify a usernamez<iIB23srd   )Úserver_hostnamerï   Tó    r*   Zmysql_native_passwordó   mysql_native_passwordZcaching_sha2_passwordó   caching_sha2_passwordzcaching_sha2: trying fast pathzcaching_sha2: empty passwordZsha256_passwordó   sha256_passwordó   ÚBúutf-8zreceived auth switchzreceived extra dataz(Received extra packet for auth method %rzSucceed to auth):rh   Úserver_versionÚsplitr7   r   ZMULTI_RESULTSr,   r%   r   r2   Úidrx   r   r€   r„   r   r   r  rG   Úserver_capabilitiesr{   r  r}   Úwrap_socketr‘   r.   r   r¹   ró   Ú_auth_plugin_namer   Úscramble_native_passwordr-   Úsaltrô   rõ   Zscramble_caching_sha2ZPLUGIN_AUTH_LENENC_CLIENT_DATAr(   r  ZSECURE_CONNECTIONrQ   r…   ÚPLUGIN_AUTHZCONNECT_ATTRSr   rŠ   r¿   Zis_auth_switch_requestÚ
read_uint8Úread_stringÚ_process_authÚscramble_old_passwordZis_extra_auth_dataÚcaching_sha2_password_authÚsha256_password_authr   rÁ   )
r“   Z
charset_idZ	data_initr  ZauthrespÚplugin_nameZconnect_attrsrk   rl   Úauth_packetr   r   r   r  >  s´    
    ÿ







ÿþ


 ÿz"Connection._request_authenticationc                 C   s.  |   |¡}|rPz| |¡W S  tk
rN   |dkrJt dd|t|ƒf ¡‚Y nX |dkrdt | |¡S |dkrxt | |¡S |dkr–t 	| j
| ¡ ¡}nz|dkr´t | j
| ¡ ¡}n\|dkrÖt | j
| ¡ ¡d	 }n:|d
krì| j
d	 }n$|dkr |}| ¡ }|d@ dk}|d@ dk}| ¡ }	|	dkr>|  | j
d	 ¡ n–|rÀd}
z| ||	¡}
|  |
d	 ¡ W nV tk
r   t dd||f ¡‚Y n. tk
r¼   t dd|||
|	f ¡‚Y nX nt dd||f ¡‚|  ¡ }| ¡  | ¡ sü| rúqü qú|S t dd| ¡‚|  |¡ |  ¡ }| ¡  |S )Ns   dialogé  zGAuthentication plugin '%s' not loaded: - %r missing authenticate methodr1  r2  r0  s   client_ed25519s   mysql_old_passwordr/  s   mysql_clear_passwordé   é   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)Ú_get_auth_plugin_handlerZauthenticateÚAttributeErrorr   rÁ   rg   r   rC  rD  r<  r-   Zread_allZed25519_passwordrB  r?  r  ÚpromptÚ	TypeErrorr¿   Zcheck_errorrÀ   )r“   rE  rF  Úhandlerr  rÂ   ÚflagZechoÚlastrL  Úrespr   r   r   rA  ¯  s’    

þþ
ÿÿ

þþ
þþÿþ ÿ
zConnection._process_authc              	   C   sp   | j  |¡}|s,t|tƒr,| j  | d¡¡}|rhz|| ƒ}W ql tk
rd   t dd||f ¡‚Y qlX nd }|S )Nrß   rG  zXAuthentication plugin '%s' not loaded: - %r cannot be constructed with connection object)r   rU   rx   r&   rá   rM  r   rÁ   )r“   rE  Zplugin_classrN  r   r   r   rJ  þ  s    þþz#Connection._get_auth_plugin_handlerc                 C   s
   | j d S r"  )Úserver_thread_idr–   r   r   r   rê     s    zConnection.thread_idc                 C   s   | j S rT   )r2   r–   r   r   r   Úcharacter_set_name  s    zConnection.character_set_namec                 C   s   | j S rT   )rˆ   r–   r   r   r   Úget_host_info  s    zConnection.get_host_infoc                 C   s   | j S rT   )Úprotocol_versionr–   r   r   r   Úget_proto_info  s    zConnection.get_proto_infoc           	      C   s
  d}|   ¡ }| ¡ }|| | _|d7 }| d|¡}|||…  d¡| _|d }t d|||d … ¡| _|d7 }|||d … | _	|d7 }t d	|||d
 … ¡d | _
|d
7 }t|ƒ|d krft d|||d … ¡\}}}}|d7 }|| _zt|ƒj| _W n tk
r   d | _Y nX || _tr6td| ƒ |  j
|d> O  _
trXtd|ƒ td|d ƒ}|d7 }t|ƒ|| kr¢|  j	|||| … 7  _	||7 }|d7 }| j
tj@ rt|ƒ|kr| d|¡}|dk rò||d …  d¡| _n|||…  d¡| _d S )Nr   r   r/  re   r   r  é   é	   r!   rI  rH  z<BHHBzserver_status: %xr  z	salt_len:é   r+   r5  )r¿   Zget_all_datarU  Úfindrá   r6  r   r  rR  r=  r9  r  Zserver_languager   ÚnameZserver_charsetÚKeyErrorr½   rô   rõ   Úmaxr   r>  r;  )	r“   r'   r  r  Z
server_endÚlangÚstatZcap_hZsalt_lenr   r   r   r    sN    
 

z"Connection._get_server_informationc                 C   s   | j S rT   )r6  r–   r   r   r   Úget_server_info^  s    zConnection.get_server_info)N)N)F)F)T)N)F)CÚ__name__Ú
__module__Ú__qualname__Ú__doc__r‘   r;  r³   ró   r   r•   r—   rš   r|   r˜   Úpropertyr¸   rµ   Ú__del__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  rA  rJ  rê   rS  rT  rV  r  r`  r   ÚWarningr´   r$  ZDatabaseErrorZ	DataErrorrÁ   ZIntegrityErrorr  ZProgrammingErrorZNotSupportedErrorr   r   r   r   r)   \   sÈ   CØ ? 



	


	

	
L-

+qOA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 )rÌ   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½   Úwarning_countÚmessageÚfield_countÚdescriptionrÍ   r(  r  )r“   r!  r   r   r   r•   n  s    zMySQLResult.__init__c                 C   s   | j r|  ¡  d S rT   )r  r'  r–   r   r   r   rf  ~  s    zMySQLResult.__del__c                 C   sN   z@| j  ¡ }| ¡ r |  |¡ n| ¡ r4|  |¡ n
|  |¡ W 5 d | _ X d S rT   )r!  r¿   rÀ   rÄ   Úis_load_local_packetÚ_read_load_local_packetÚ_read_result_packet©r“   Úfirst_packetr   r   r   rt   ‚  s    
zMySQLResult.readc                 C   sl   d| _ | j ¡ }| ¡ r0|  |¡ d| _ d| _n8| ¡ rP|  |¡ d| _ d| _n| ¡ | _|  	¡  d| _
dS )zw
        :raise OperationalError: If the connection to the MySQL server is lost.
        :raise InternalError:
        TFNl   ÿÿÿÿ )r  r!  r¿   rÀ   rÄ   rl  rm  Úread_length_encoded_integerrj  Ú_get_descriptionsré   ro  r   r   r   r     s    



z!MySQLResult.init_unbuffered_queryc                 C   s<   t |ƒ}|j| _|j| _|j| _|j| _|j| _|j| _d S rT   )r   ré   r#  r½   rh  ri  r(  )r“   rp  Ú	ok_packetr   r   r   rÄ   ¨  s    zMySQLResult._read_ok_packetc                 C   sx   | j jstdƒ‚t|ƒ}t|j| j ƒ}z| ¡  W n   | j  ¡  ‚ Y nX | j  ¡ }| ¡ sjt	 
dd¡‚|  |¡ d S )NzF**WARN**: Received LOAD_LOCAL packet but local_infile option is false.r¾   zCommands Out of Sync)r!  rp   ÚRuntimeErrorr   ÚLoadLocalFileÚfilenamer·   r¿   rÀ   r   rÁ   rÄ   )r“   rp  Zload_packetZsenderrs  r   r   r   rm  ±  s     ÿ

ÿz#MySQLResult._read_load_local_packetc                 C   s(   |  ¡ sdS t|ƒ}|j| _|j| _dS )NFT)Úis_eof_packetr   rh  r(  )r“   r  Zwpr   r   r   Ú_check_packet_is_eofÅ  s    z MySQLResult._check_packet_is_eofc                 C   s   |  ¡ | _|  ¡  |  ¡  d S rT   )rq  rj  rr  Ú_read_rowdata_packetro  r   r   r   rn  Ñ  s    
zMySQLResult._read_result_packetc                 C   sP   | j s
d S | j ¡ }|  |¡r4d| _ d | _d | _d S |  |¡}d| _|f| _|S )NFr   )r  r!  r¿   rx  rÍ   Ú_read_row_from_packetré   )r“   r  Úrowr   r   r   Ú_read_rowdata_packet_unbufferedÖ  s    


z+MySQLResult._read_rowdata_packet_unbufferedc                 C   s,   | j r(| j ¡ }|  |¡r d| _ d | _q d S r  )r  r!  r¿   rx  )r“   r  r   r   r   r'  è  s
    

z$MySQLResult._finish_unbuffered_queryc                 C   sJ   g }| j  ¡ }|  |¡r d| _ q2| |  |¡¡ qt|ƒ| _t|ƒ| _dS )z:Read a rowdata packet for each data row in the result set.N)	r!  r¿   rx  Úappendrz  r  ré   ÚtuplerÍ   )r“   rÍ   r  r   r   r   ry  ò  s    


z MySQLResult._read_rowdata_packetc              	   C   s†   g }| j D ]r\}}z| ¡ }W n tk
r8   Y  q~Y nX |d k	rr|d k	rT| |¡}trbtd|ƒ |d k	rr||ƒ}| |¡ q
t|ƒS )NzDEBUG: DATA = )r   Zread_length_coded_stringÚ
IndexErrorrá   rô   rõ   r}  r~  )r“   r  r{  r„   Ú	converterr  r   r   r   rz  ÿ  s    

z!MySQLResult._read_row_from_packetc           
      C   s  g | _ g | _| jj}| jj}g }t| jƒD ]°}| j t¡}| j  	|¡ | 	| 
¡ ¡ |j}|r’|tjkrn|}q–|tkrŒ|jdkr†d}q|}q–d}nd}| jj |¡}|tjkr²d}trÊtd|› d|› ƒ | j 	||f¡ q*| j ¡ }	|	 ¡ sötdƒ‚t|ƒ| _
dS )z>Read a column descriptor packet for each column in the result.é?   Nrß   zDEBUG: field=z, converter=zProtocol error, expecting EOF)Úfieldsr   r!  r6   r„   Úrangerj  r¿   r   r}  rk  Z	type_coder	   ZJSONÚ
TEXT_TYPESZ	charsetnrrŒ   rU   Zthroughrô   rõ   rw  ÚAssertionErrorr~  )
r“   r6   Zconn_encodingrk  r'   ÚfieldZ
field_typer„   r€  Z
eof_packetr   r   r   rr    s8    



zMySQLResult._get_descriptionsN)ra  rb  rc  r•   rf  rt   r   rÄ   rm  rx  rn  r|  r'  ry  rz  rr  r   r   r   r   rÌ   m  s   	
rÌ   c                   @   s   e Zd Zdd„ Zdd„ ZdS )ru  c                 C   s   || _ || _d S rT   )rv  r!  )r“   rv  r!  r   r   r   r•   ?  s    zLoadLocalFile.__init__c              	   C   s    | j jst dd¡‚| j }ztzDt| jdƒ.}t|jdƒ}| 	|¡}|sHqT| |¡ q8W 5 Q R X W n* t
k
rŠ   t dd| j› d¡‚Y nX W 5 | d¡ X d	S )
z3Send data packets from the local file to the serverr   r*   rd   rï   i @  iù  zCan't find file 'rÑ   N)r!  r‘   r   r$  r  r¸   rv  r)  r>   rt   rø   rÁ   )r“   ÚconnÚ	open_filer+  Úchunkr   r   r   r·   C  s      ÿ
 zLoadLocalFile.send_dataN)ra  rb  rc  r•   r·   r   r   r   r   ru  >  s   ru  )9rù   ru   r\   r   rq   r  r%  r*   r   r2   r   r   Ú	constantsr   r   r   r	   r
   r   Zcursorsr   Z
optionfiler   Úprotocolr   r   r   r   r   r   r   r   rG   rz   ÚImportErrorÚgetpassÚgetuserr~   r\  rô   ZBITZBLOBZ	LONG_BLOBZMEDIUM_BLOBÚSTRINGZ	TINY_BLOBZ
VAR_STRINGZVARCHARZGEOMETRYr„  rƒ   r  r   r(   r)   rÌ   ru  r   r   r   r   Ú<module>   sj    

÷         R