| Current File : //lib/python3.9/site-packages/pymysql/__pycache__/protocol.cpython-39.opt-1.pyc |
a
���]'/ � @ 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� �zLt dt� �� tdd�D ]&}t�|�}t d||jj|jf � q"t d� W n tyf Y n0 � fdd �t d
t
t� �d �d �D �}|D ]H}t d
� dd� |D ��dd t|� d d� �fdd�|D �� � q�t d� t � d S )Nc S s6 dt | � krdk r2n nt| t�r.t| �S | S dS )N� � �.)r
�
isinstance�int�chr��data� r �4/usr/lib/python3.9/site-packages/pymysql/protocol.py� printable s
z dump_packet.<locals>.printablezpacket length:r � zcall[%d]: %s (line %d)zB------------------------------------------------------------------c s g | ]}� ||d � �qS )� r )�.0�ir r r �
<listcomp>( � z dump_packet.<locals>.<listcomp>r � r � c s s | ]}d � t|��V qdS )z{:02X}N)�formatr
�r �xr r r � <genexpr>* r z dump_packet.<locals>.<genexpr>z z � c 3 s | ]}� |�V qd S �Nr r# )r r r r% , r ) �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(
$���r5 c @ s� e Zd ZdZdZdd� 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.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z d8d9� 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 r7 ��selfr �encodingr r r �__init__8 s zMysqlPacket.__init__c C s | j S r' �r9 �r; r r r � get_all_data< s zMysqlPacket.get_all_datac C sl | j | j| j| � }t|�|krZd|t|�| jt| j �f }trRt|� | �� 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)r9 r8 r) �DEBUGr( �dump�AssertionError)r; �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�r9 r8 �r; rE 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)r8 r) r9 � Exception)r; �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) r9 rK r8 )r; �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!
r> )r; rN rL r r r � get_bytesd s zMysqlPacket.get_bytesc C s" t | j| j �}| jd7 _|S �Nr )�ordr9 r8 rI r r r �
read_uint8p s zMysqlPacket.read_uint8c C s | j | j }| jd7 _|S rQ rH rI r r r rS u s c C s( t �d| j| j�d }| jd7 _|S )Nz<Hr � ��struct� unpack_fromr9 r8 rI r r r � read_uint16z s zMysqlPacket.read_uint16c C s0 t �d| j| j�\}}| jd7 _||d> S )Nz<HB� r rU )r; ZlowZhighr r r � read_uint24 s zMysqlPacket.read_uint24c C s( t �d| j| j�d }| jd7 _|S )Nz<Ir � rU rI r r r � read_uint32� s zMysqlPacket.read_uint32c C s( t �d| j| j�d }| jd7 _|S )Nz<Qr � rU rI r r r � read_uint64� s zMysqlPacket.read_uint64c C s: | j �d| j�}|dk r d S | j | j|� }|d | _|S )N� r r )r9 �findr8 )r; Zend_posrE r r r � read_string� s
zMysqlPacket.read_stringc C sT | � � }|tkrdS |tk r |S |tkr0| �� S |tkr@| �� S |tkrP| �� 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) rS � NULL_COLUMN�UNSIGNED_CHAR_COLUMN�UNSIGNED_SHORT_COLUMNrX �UNSIGNED_INT24_COLUMNrZ �UNSIGNED_INT64_COLUMNr^ )r; �cr r r �read_length_encoded_integer� s z'MysqlPacket.read_length_encoded_integerc C s | � � }|du rdS | �|�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)rh rG )r; rL r r r �read_length_coded_string� s z$MysqlPacket.read_length_coded_stringc C s. t �|�}|�| j| j�}| j|j7 _|S r' )rV �StructrW r9 r8 rD )r; Zfmt�srE r r r � read_struct� s
zMysqlPacket.read_structc C s | j dd� dko t| j �dkS )Nr r r_ r �r9 r) r? r r r � is_ok_packet� s zMysqlPacket.is_ok_packetc C s | j dd� dko t| j �dk S )Nr r � �� rm r? r r r �
is_eof_packet� s zMysqlPacket.is_eof_packetc C s | j dd� dkS )Nr r ro r> r? r r r �is_auth_switch_request� s z"MysqlPacket.is_auth_switch_requestc C s | j dd� dkS )Nr r � r> r? r r r �is_extra_auth_data� s z MysqlPacket.is_extra_auth_datac C s* t | jdd� �}d| ko$dkS S )Nr r � )rR r9 )r; Z field_countr r r �is_resultset_packet� s z MysqlPacket.is_resultset_packetc C s | j dd� dkS )Nr r � �r> r? r r r �is_load_local_packet� s z MysqlPacket.is_load_local_packetc C s | j dd� dkS )Nr r � �r> r? r r r �is_error_packet� s zMysqlPacket.is_error_packetc C s | � � r| �� d S r' )rz �raise_for_errorr? r r r � check_error� s zMysqlPacket.check_errorc C s8 | � � | �d� | �� }tr(td|� t�| j� d S )Nr zerrno =)rO rM rX rA r( r Zraise_mysql_exceptionr9 )r; �errnor r r r{ � s
zMysqlPacket.raise_for_errorc C s t | j� d S r' )r5 r9 r? r r r rB � s zMysqlPacket.dumpN)r )r ) �__name__�
__module__� __qualname__�__doc__� __slots__r= r@ rG rJ rM rO rP r rS rX rZ r\ r^ ra rh ri rl rn rq rr rt rv rx rz r| r{ rB r r r r r6 1 s<
r6 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 �| ||� | �|� d S r' )r6 r= �_parse_field_descriptorr: r r r r= � s z FieldDescriptorPacket.__init__c C sv | � � | _| � � | _| � � �|�| _| � � �|�| _| � � �|�| _| � � �|�| _| �d�\| _ | _
| _ | _ | _
dS )z�Parse the 'Field Descriptor' (Metadata) packet.
This is compatible with MySQL 4.1+ (not compatible with MySQL 4.0).
z <xHIBHBxxN)ri �catalog�db�decode�
table_nameZ org_table�nameZorg_namerl � charsetnrrL � type_code�flags�scale)r; r< r r r r� � s
�z-FieldDescriptorPacket._parse_field_descriptorc C s* | j | jd| �� | �� | j| jd dkfS )zBProvides a 7-item tuple compatible with the Python PEP249 DB Spec.NrT r )r� r� �get_column_lengthr� r� r? r r r � description s �z!FieldDescriptorPacket.descriptionc C s* | j tjkr$t�| jd�}| j| S | jS rQ )r� r Z
VAR_STRINGr �getr� rL )r; 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__r� r� r� r� r� r? r r r �__str__ s
��z FieldDescriptorPacket.__str__N) r~ r r� r� r= r� r� r� r� r r r r r� � s r� 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 |� � s tdt| jj� d ��|| _| j�d� | j�� | _| j�� | _ | �
d�\| _ | _ | j�
� | _| j tj@ | _d S )NzCannot create z object from invalid packet typer z<HH)rn r0 �strr� r~ �packetrM rh Z
affected_rowsZ insert_idrl �
server_status�
warning_countrJ �messager �SERVER_MORE_RESULTS_EXISTS�has_next�r; Z from_packetr r r r= s � zOKPacketWrapper.__init__c C s t | j|�S r' ��getattrr� �r; �keyr r r � __getattr__, s zOKPacketWrapper.__getattr__N�r~ r r� r� 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 |� � std�| j���|| _| j�d�\| _| _trBt d| j� | jt
j @ | _ d S )N�3Cannot create '{0}' object from invalid packet typez<xhhzserver_status=)
rq r0 r" r� r� rl r� r� rA r( r r� r� r� r r r r= 7 s ��zEOFPacketWrapper.__init__c C s t | j|�S r' r� r� r r r r� B s z EOFPacketWrapper.__getattr__Nr� 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 |� � std�| j���|| _| j�� dd � | _trBtd| j� d S )Nr� r z filename=) rx r0 r" r� r� r@ �filenamerA r( r� r r r r= M s ��z LoadLocalPacketWrapper.__init__c C s t | j|�S r' r� r� r r r r� W s z"LoadLocalPacketWrapper.__getattr__Nr� r r r r r� F s
r� ) Z
__future__r �charsetr Z_compatr r Z constantsr r r& r �utilr
rV r+ rA rb rc rd re rf r5 �objectr6 r� r� r� r� r r r r �<module> s( 52