HEX

File: //usr/lib/python3.6/site-packages/pymysql/__pycache__/protocol.cpython-36.opt-1.pyc
3

���]'/�@s�ddlmZddlmZddlmZmZddlmZm	Z	ddl
mZddlm
Z
ddlZddlZd	Zd
Zd
ZdZdZd
Zdd�ZGdd�de�ZGdd�de�ZGdd�de�ZGdd�de�ZGdd�de�ZdS)�)�print_function�)�MBLENGTH)�PY2�
range_type)�
FIELD_TYPE�
SERVER_STATUS)�err)�byte2intNF����cs�dd��yTtdt���x4tdd�D]&}tj|�}td||jj|jf�q$Wtdd�Wntk
rpYnX�fd	d
�t	dt
t��d�d
�D�}xP|D]H}tdjdd�|D��dd
t|�ddj�fdd�|D���q�Wtdd�t�dS)NcSs6dt|�kodknr2t|t�r.t|�S|SdS)N� ��.)r
�
isinstance�int�chr)�data�r�/usr/lib/python3.6/protocol.py�	printables

zdump_packet.<locals>.printablezpacket length:r�zcall[%d]: %s (line %d)�-�Bcsg|]}�||d��qS)�r)�.0�i)rrr�
<listcomp>(szdump_packet.<locals>.<listcomp>r�r� css|]}djt|��VqdS)z{:02X}N)�formatr
)r�xrrr�	<genexpr>*szdump_packet.<locals>.<genexpr>z   ��c3s|]}�|�VqdS)Nr)rr#)rrrr$,sz  )�print�len�range�sys�	_getframe�f_code�co_name�f_lineno�
ValueErrorr�min�join)rr�fZ	dump_data�dr)rrr�dump_packets 
$
& r4c@s�eZdZdZd<Zdd�Zdd�Zdd	�Zd
d�Zdd
�Z	d=dd�Z
d>dd�ZerZdd�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�Zd7d8�Zd9d:�Zd;S)?�MysqlPacketzrRepresentation of a MySQL response packet.

    Provides an interface for reading/parsing the packet results.
    �	_position�_datacCsd|_||_dS)Nr)r6r7)�selfr�encodingrrr�__init__8szMysqlPacket.__init__cCs|jS)N)r7)r8rrr�get_all_data<szMysqlPacket.get_all_datacCsl|j|j|j|�}t|�|krZd|t|�|jt|j�f}trRt|�|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)r7r6r(�DEBUGr'�dump�AssertionError)r8�size�result�errorrrr�read?szMysqlPacket.readcCs|j|jd�}d|_|S)z`Read all remaining data in the packet.

        (Subsequent read() will return errors.)
        N)r7r6)r8r@rrr�read_allMszMysqlPacket.read_allcCs:|j|}|dks |t|j�kr0td||f��||_dS)z1Advance the cursor in data buffer 'length' bytes.rz4Invalid advance amount (%s) for cursor.  Position=%sN)r6r(r7�	Exception)r8�lengthZnew_positionrrr�advanceVs

zMysqlPacket.advancercCs,|dks|t|j�kr"td|��||_dS)z9Set the position of the data buffer cursor to 'position'.rz)Invalid position to rewind cursor to: %s.N)r(r7rDr6)r8�positionrrr�rewind^szMysqlPacket.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!
        )r7)r8rGrErrr�	get_bytesds	zMysqlPacket.get_bytescCs"t|j|j�}|jd7_|S)Nr)�ordr7r6)r8r@rrr�
read_uint8pszMysqlPacket.read_uint8cCs|j|j}|jd7_|S)Nr)r7r6)r8r@rrrrKuscCs(tjd|j|j�d}|jd7_|S)Nz<Hrr%)�struct�unpack_fromr7r6)r8r@rrr�read_uint16zszMysqlPacket.read_uint16cCs0tjd|j|j�\}}|jd7_||d>S)Nz<HB�r)rLrMr7r6)r8ZlowZhighrrr�read_uint24szMysqlPacket.read_uint24cCs(tjd|j|j�d}|jd7_|S)Nz<Ir�)rLrMr7r6)r8r@rrr�read_uint32�szMysqlPacket.read_uint32cCs(tjd|j|j�d}|jd7_|S)Nz<Qr�)rLrMr7r6)r8r@rrr�read_uint64�szMysqlPacket.read_uint64cCs:|jjd|j�}|dkrdS|j|j|�}|d|_|S)N�rr)r7�findr6)r8Zend_posr@rrr�read_string�s
zMysqlPacket.read_stringcCsT|j�}|tkrdS|tkr |S|tkr0|j�S|tkr@|j�S|tkrP|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)	rK�NULL_COLUMN�UNSIGNED_CHAR_COLUMN�UNSIGNED_SHORT_COLUMNrN�UNSIGNED_INT24_COLUMNrP�UNSIGNED_INT64_COLUMNrT)r8�crrr�read_length_encoded_integer�sz'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^rB)r8rErrr�read_length_coded_string�sz$MysqlPacket.read_length_coded_stringcCs.tj|�}|j|j|j�}|j|j7_|S)N)rL�StructrMr7r6r?)r8Zfmt�sr@rrr�read_struct�s
zMysqlPacket.read_structcCs |jdd�dkot|j�dkS)NrrrUr)r7r()r8rrr�is_ok_packet�szMysqlPacket.is_ok_packetcCs |jdd�dkot|j�dkS)Nrr���	)r7r()r8rrr�
is_eof_packet�szMysqlPacket.is_eof_packetcCs|jdd�dkS)Nrrrd)r7)r8rrr�is_auth_switch_request�sz"MysqlPacket.is_auth_switch_requestcCs|jdd�dkS)Nrr�)r7)r8rrr�is_extra_auth_data�szMysqlPacket.is_extra_auth_datacCs*t|jdd��}d|ko$dkSS)Nrr�)rJr7)r8Zfield_countrrr�is_resultset_packet�szMysqlPacket.is_resultset_packetcCs|jdd�dkS)Nrr��)r7)r8rrr�is_load_local_packet�sz MysqlPacket.is_load_local_packetcCs|jdd�dkS)Nrr��)r7)r8rrr�is_error_packet�szMysqlPacket.is_error_packetcCs|j�r|j�dS)N)ro�raise_for_error)r8rrr�check_error�szMysqlPacket.check_errorcCs8|j�|jd�|j�}tr(td|�tj|j�dS)Nrzerrno =)rHrFrNr<r'r	Zraise_mysql_exceptionr7)r8�errnorrrrp�s

zMysqlPacket.raise_for_errorcCst|j�dS)N)r4r7)r8rrrr=�szMysqlPacket.dumpN)r6r7)r)r) �__name__�
__module__�__qualname__�__doc__�	__slots__r:r;rBrCrFrHrIrrKrNrPrRrTrWr^r_rbrcrfrgrirkrmrorqrpr=rrrrr51s<	


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

    Parsing is automatically done and the results are exported via public
    attributes on the class such as: db, table_name, name, length, type_code.
    cCstj|||�|j|�dS)N)r5r:�_parse_field_descriptor)r8rr9rrrr:�szFieldDescriptorPacket.__init__cCsv|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_namerb�	charsetnrrE�	type_code�flags�scale)r8r9rrrry�s

z-FieldDescriptorPacket._parse_field_descriptorcCs*|j|jd|j�|j�|j|jddkfS)zBProvides a 7-item tuple compatible with the Python PEP249 DB Spec.Nr%r)r}r�get_column_lengthr�r�)r8rrr�descriptionsz!FieldDescriptorPacket.descriptioncCs*|jtjkr$tj|jd�}|j|S|jS)Nr)rrZ
VAR_STRINGr�getr~rE)r8Zmblenrrrr�s
z'FieldDescriptorPacket.get_column_lengthcCs d|j|j|j|j|j|jfS)Nz%s %r.%r.%r, type=%s, flags=%x)�	__class__rzr|r}rr�)r8rrr�__str__szFieldDescriptorPacket.__str__N)	rsrtrurvr:ryr�r�r�rrrrrx�srxc@s eZdZdZdd�Zdd�ZdS)�OKPacketWrapperz�
    OK Packet Wrapper. It uses an existing packet object, and wraps
    around it, exposing useful variables while still providing access
    to the original packet objects variables and methods.
    cCsz|j�s tdt|jj�d��||_|jjd�|jj�|_|jj�|_	|j
d�\|_|_|jj
�|_|jtj@|_dS)NzCannot create z  object from invalid packet typerz<HH)rcr/�strr�rs�packetrFr^Z
affected_rowsZ	insert_idrb�
server_status�
warning_countrC�messager�SERVER_MORE_RESULTS_EXISTS�has_next)r8�from_packetrrrr:szOKPacketWrapper.__init__cCst|j|�S)N)�getattrr�)r8�keyrrr�__getattr__,szOKPacketWrapper.__getattr__N)rsrtrurvr:r�rrrrr�sr�c@s eZdZdZdd�Zdd�ZdS)�EOFPacketWrapperz�
    EOF Packet Wrapper. It uses an existing packet object, and wraps
    around it, exposing useful variables while still providing access
    to the original packet objects variables and methods.
    cCsT|j�stdj|j���||_|jjd�\|_|_trBt	d|j�|jt
j@|_dS)Nz3Cannot create '{0}' object from invalid packet typez<xhhzserver_status=)
rfr/r"r�r�rbr�r�r<r'rr�r�)r8r�rrrr:7s
zEOFPacketWrapper.__init__cCst|j|�S)N)r�r�)r8r�rrrr�BszEOFPacketWrapper.__getattr__N)rsrtrurvr:r�rrrrr�0sr�c@s eZdZdZdd�Zdd�ZdS)�LoadLocalPacketWrapperz�
    Load Local Packet Wrapper. It uses an existing packet object, and wraps
    around it, exposing useful variables while still providing access
    to the original packet objects variables and methods.
    cCsF|j�stdj|j���||_|jj�dd�|_trBtd|j�dS)Nz3Cannot create '{0}' object from invalid packet typerz	filename=)	rmr/r"r�r�r;�filenamer<r')r8r�rrrr:Ms
zLoadLocalPacketWrapper.__init__cCst|j|�S)N)r�r�)r8r�rrrr�Wsz"LoadLocalPacketWrapper.__getattr__N)rsrtrurvr:r�rrrrr�Fs
r�)Z
__future__r�charsetrZ_compatrrZ	constantsrrr&r	�utilr
rLr*r<rXrYrZr[r\r4�objectr5rxr�r�r�rrrr�<module>s(52