D7net
Home
Console
Upload
information
Create File
Create Folder
About
Tools
:
/
proc
/
3206095
/
root
/
lib
/
python3.6
/
site-packages
/
pymysql
/
__pycache__
/
Filename :
protocol.cpython-36.pyc
back
Copy
3 ���]'/ � @ s� d dl mZ ddlmZ ddlmZmZ ddlmZm Z ddl m Z ddl m Z d dlZd dlZd Zd Zd Zd Zd Zd Zdd� ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�Z dS )� )�print_function� )�MBLENGTH)�PY2� range_type)� FIELD_TYPE� SERVER_STATUS)�err)�byte2intNF� � � � c s� dd� �yTt dt� �� x4tdd�D ]&}tj|�}t d||jj|jf � q$W t dd � W n tk rp Y nX � fd d �t d t t� �d �d �D �}xP|D ]H}t dj dd� |D ��dd t|� d dj �fdd�|D �� � q�W t dd � t � d S )Nc S s6 dt | � kodk n r2t| t�r.t| �S | S dS )N� � �.)r � isinstance�int�chr)�data� r � /usr/lib/python3.6/protocol.py� printable s z dump_packet.<locals>.printablezpacket length:r � zcall[%d]: %s (line %d)�-�B c s g | ]}� ||d � �qS )� r )�.0�i)r r r � <listcomp>( s z dump_packet.<locals>.<listcomp>r � r � c s s | ]}d j t|��V qdS )z{:02X}N)�formatr )r �xr r r � <genexpr>* s z dump_packet.<locals>.<genexpr>z � � c 3 s | ]}� |�V qd S )Nr )r r# )r r r r$ , s z ) �print�len�range�sys� _getframe�f_code�co_name�f_lineno� ValueErrorr �min�join)r r �fZ dump_data�dr )r r r � dump_packet s $ & r4 c @ s� e Zd ZdZd<Zdd� Zdd� Zdd � Zd d � Zd d � Z d=dd�Z d>dd�Z e rZdd� Z ndd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;S )?� MysqlPacketzrRepresentation of a MySQL response packet. Provides an interface for reading/parsing the packet results. � _position�_datac C s d| _ || _d S )Nr )r6 r7 )�selfr �encodingr r r �__init__8 s zMysqlPacket.__init__c C s | j S )N)r7 )r8 r r r � get_all_data< s zMysqlPacket.get_all_datac C sl | 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)r7 r6 r( �DEBUGr' �dump�AssertionError)r8 �size�result�errorr r r �read? s zMysqlPacket.readc C s | j | jd� }d| _|S )z`Read all remaining data in the packet. (Subsequent read() will return errors.) N)r7 r6 )r8 r@ r r r �read_allM s zMysqlPacket.read_allc C s: | j | }|dk s |t| j�kr0td||f ��|| _ dS )z1Advance the cursor in data buffer 'length' bytes.r z4Invalid advance amount (%s) for cursor. Position=%sN)r6 r( r7 � Exception)r8 �lengthZ new_positionr r r �advanceV s zMysqlPacket.advancer c C s, |dk s|t | j�kr"td| ��|| _dS )z9Set the position of the data buffer cursor to 'position'.r z)Invalid position to rewind cursor to: %s.N)r( r7 rD r6 )r8 �positionr r r �rewind^ s zMysqlPacket.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! )r7 )r8 rG rE r r r � get_bytesd s zMysqlPacket.get_bytesc C s" t | j| j �}| jd7 _|S )Nr )�ordr7 r6 )r8 r@ r r r � read_uint8p s zMysqlPacket.read_uint8c C s | j | j }| jd7 _|S )Nr )r7 r6 )r8 r@ r r r rK u s c C s( t jd| j| j�d }| jd7 _|S )Nz<Hr r% )�struct� unpack_fromr7 r6 )r8 r@ r r r � read_uint16z s zMysqlPacket.read_uint16c C s0 t jd| j| j�\}}| jd7 _||d> S )Nz<HB� r )rL rM r7 r6 )r8 ZlowZhighr r r � read_uint24 s zMysqlPacket.read_uint24c C s( t jd| j| j�d }| jd7 _|S )Nz<Ir � )rL rM r7 r6 )r8 r@ r r r � read_uint32� s zMysqlPacket.read_uint32c C s( t jd| j| j�d }| jd7 _|S )Nz<Qr � )rL rM r7 r6 )r8 r@ r r r � read_uint64� s zMysqlPacket.read_uint64c C s: | j jd| j�}|dk r d S | j | j|� }|d | _|S )N� r r )r7 �findr6 )r8 Zend_posr@ r r r � read_string� s zMysqlPacket.read_stringc C sT | j � }|tkrdS |tk r |S |tkr0| j� S |tkr@| j� S |tkrP| j� S dS )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 �cr r r �read_length_encoded_integer� s z'MysqlPacket.read_length_encoded_integerc C s | j � }|dkrdS | 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^ rB )r8 rE r r r �read_length_coded_string� s z$MysqlPacket.read_length_coded_stringc C s. t j|�}|j| j| j�}| j|j7 _|S )N)rL �StructrM r7 r6 r? )r8 Zfmt�sr@ r r r � read_struct� s zMysqlPacket.read_structc C s | j dd� dko t| j �dkS )Nr r rU r )r7 r( )r8 r r r � is_ok_packet� s zMysqlPacket.is_ok_packetc C s | j dd� dko t| j �dk S )Nr r � �� )r7 r( )r8 r r r � is_eof_packet� s zMysqlPacket.is_eof_packetc C s | j dd� dkS )Nr r rd )r7 )r8 r r r �is_auth_switch_request� s z"MysqlPacket.is_auth_switch_requestc C s | j dd� dkS )Nr r � )r7 )r8 r r r �is_extra_auth_data� s z MysqlPacket.is_extra_auth_datac C s* t | jdd� �}d| ko$dkS S )Nr r � )rJ r7 )r8 Z field_countr r r �is_resultset_packet� s z MysqlPacket.is_resultset_packetc C s | j dd� dkS )Nr r � �)r7 )r8 r r r �is_load_local_packet� s z MysqlPacket.is_load_local_packetc C s | j dd� dkS )Nr r � �)r7 )r8 r r r �is_error_packet� s zMysqlPacket.is_error_packetc C s | j � r| j� d S )N)ro �raise_for_error)r8 r r r � check_error� s zMysqlPacket.check_errorc C s8 | j � | jd� | j� }tr(td|� tj| j� d S )Nr zerrno =)rH rF rN r< r' r Zraise_mysql_exceptionr7 )r8 �errnor r r rp � s zMysqlPacket.raise_for_errorc C s t | j� d S )N)r4 r7 )r8 r r r r= � s zMysqlPacket.dumpN)r6 r7 )r )r ) �__name__� __module__� __qualname__�__doc__� __slots__r: r; rB rC rF rH rI r rK rN rP rR rT rW r^ r_ rb rc rf rg ri rk rm ro rq rp r= r r r r r5 1 s< r5 c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) �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. c C s t j| ||� | j|� d S )N)r5 r: �_parse_field_descriptor)r8 r r9 r r r r: � s z FieldDescriptorPacket.__init__c C sv | 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)r8 r9 r r r ry � s z-FieldDescriptorPacket._parse_field_descriptorc C s* | j | jd| j� | j� | j| jd dkfS )zBProvides a 7-item tuple compatible with the Python PEP249 DB Spec.Nr% r )r} r �get_column_lengthr� r� )r8 r r r � description s z!FieldDescriptorPacket.descriptionc C s* | j tjkr$tj| jd�}| j| S | jS )Nr )r r Z VAR_STRINGr �getr~ rE )r8 Zmblenr r r r� s z'FieldDescriptorPacket.get_column_lengthc C s d| j | j| j| j| j| jf S )Nz %s %r.%r.%r, type=%s, flags=%x)� __class__rz r| r} r r� )r8 r r r �__str__ s z FieldDescriptorPacket.__str__N) rs rt ru rv r: ry r� r� r� r r r r rx � s rx c @ s e Zd ZdZdd� 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. c C sz |j � s tdt| jj� d ��|| _| jjd� | jj� | _| jj� | _ | j d�\| _ | _ | jj � | _| j tj@ | _d S )NzCannot create z object from invalid packet typer z<HH)rc r/ �strr� rs �packetrF r^ Z affected_rowsZ insert_idrb � server_status� warning_countrC �messager �SERVER_MORE_RESULTS_EXISTS�has_next)r8 � from_packetr r r r: s zOKPacketWrapper.__init__c C s t | j|�S )N)�getattrr� )r8 �keyr r r � __getattr__, s zOKPacketWrapper.__getattr__N)rs rt ru rv r: r� r r r r r� s r� c @ s e Zd ZdZdd� 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. c C sT |j � stdj| j���|| _| jjd�\| _| _trBt d| j� | jt j @ | _ d S )Nz3Cannot create '{0}' object from invalid packet typez<xhhzserver_status=) rf r/ r" r� r� rb r� r� r< r' r r� r� )r8 r� r r r r: 7 s zEOFPacketWrapper.__init__c C s t | j|�S )N)r� r� )r8 r� r r r r� B s z EOFPacketWrapper.__getattr__N)rs rt ru rv r: r� r r r r r� 0 s r� c @ s e Zd ZdZdd� 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. c C sF |j � stdj| j���|| _| jj� dd � | _trBtd| j� d S )Nz3Cannot create '{0}' object from invalid packet typer z filename=) rm r/ r" r� r� r; �filenamer< r' )r8 r� r r r r: M s z LoadLocalPacketWrapper.__init__c C s t | j|�S )N)r� r� )r8 r� r r r r� W s z"LoadLocalPacketWrapper.__getattr__N)rs rt ru rv r: r� r r r r r� F s r� ) Z __future__r �charsetr Z_compatr r Z constantsr r r& r �utilr rL r* r< rX rY rZ r[ r\ r4 �objectr5 rx r� r� r� r r r r �<module> s( 52