U
|
·ý°dS. ã @ s¤ d dl mZ d dlmZmZ d dlmZ ddlZddlZdZ dZ
|
dZ dZ d Z d
|
Zd d ZG d d dZG dd deZG dd dZG dd dZG dd dZdS )é )ÚMBLENGTH)Ú
|
FIELD_TYPEÚ SERVER_STATUS)Úerré NFéû éü éý éþ c sî dd zLt dt tddD ]&}t |¡}t d||jj|jf q"t d W n tk
|
rh Y nX fdd td
|
t t d d D }|D ]H}t d
|
dd |D ¡dd t| d d
|
fdd|D ¡ qt d t d S )Nc S s$ d| krdk r n nt | S dS )Né é Ú.)Úchr©Údata© r úGd:\z\workplace\vscode\pyvenv\venv\Lib\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># s z dump_packet.<locals>.<listcomp>r é r ú c s s | ]}d |¡V qdS )z{:02X}N)Úformat©r Úxr r r Ú <genexpr>& s z dump_packet.<locals>.<genexpr>z z Ú c 3 s | ]} |V qd S ©Nr r )r r r r ) s ) ÚprintÚlenÚrangeÚsysÚ _getframeÚf_codeÚco_nameÚf_linenoÚ
|
ValueErrorÚminÚjoin)r r ÚfZ dump_dataÚdr )r r r Ú dump_packet s,
|
$ÿþýÿr. c @ sè e Zd ZdZdZdd Zdd Zdd Zd d
|
Zd d Z d:ddZ
|
d;ddZ dd Z dd 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 d9S )<Ú MysqlPacketzrRepresentation of a MySQL response packet.
|
|
Provides an interface for reading/parsing the packet results.
|
©Ú _positionÚ_datac C s d| _ || _d S )Nr r0 ©Úselfr Úencodingr r r Ú__init__7 s zMysqlPacket.__init__c C s | j S r ©r2 ©r4 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)r2 r1 r" ÚDEBUGr! ÚdumpÚAssertionError)r4 Ú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©r2 r1 ©r4 r> r r r Úread_allN s zMysqlPacket.read_allc C s: | j | }|dk s |t| jkr0td||f || _ dS )z1Advance the cursor in data buffer 'length' bytes.r z4Invalid advance amount (%s) for cursor. Position=%sN)r1 r" r2 Ú Exception)r4 ÚlengthZ new_positionr r r ÚadvanceW s
|
ÿÿzMysqlPacket.advancer c C s, |dk s|t | jkr"td| || _dS )z9Set the position of the data buffer cursor to 'position'.r z)Invalid position to rewind cursor to: %s.N)r" r2 rD r1 )r4 Úpositionr r r Úrewinda 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 )r4 rG rE r r r Ú get_bytesg s zMysqlPacket.get_bytesc C s | j | j }| jd7 _|S ©Nr rA rB r r r Ú
|
read_uint8r s zMysqlPacket.read_uint8c C s( t d| j| j¡d }| jd7 _|S )Nz<Hr é ©ÚstructÚ unpack_fromr2 r1 rB r r r Ú read_uint16w s zMysqlPacket.read_uint16c C s0 t d| j| j¡\}}| jd7 _||d> S )Nz<HBé r rM )r4 ÚlowÚhighr r r Ú read_uint24| s zMysqlPacket.read_uint24c C s( t d| j| j¡d }| jd7 _|S )Nz<Ir é rM rB r r r Ú read_uint32 s zMysqlPacket.read_uint32c C s( t d| j| j¡d }| jd7 _|S )Nz<Qr é rM rB 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 )r2 Úfindr1 )r4 Zend_posr> 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) rK Ú NULL_COLUMNÚUNSIGNED_CHAR_COLUMNÚUNSIGNED_SHORT_COLUMNrP ÚUNSIGNED_INT24_COLUMNrT ÚUNSIGNED_INT64_COLUMNrX )r4 Úcr r r Úread_length_encoded_integer s z'MysqlPacket.read_length_encoded_integerc C s | ¡ }|dkrdS | |¡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)rb r@ )r4 rE r r r Úread_length_coded_string¥ s z$MysqlPacket.read_length_coded_stringc C s. t |¡}| | j| j¡}| j|j7 _|S r )rN ÚStructrO r2 r1 r= )r4 ÚfmtÚsr> r r r Ú read_struct± s
|
zMysqlPacket.read_structc C s | j d dkot| j dkS )Nr r ©r2 r" r8 r r r Ú is_ok_packet· s zMysqlPacket.is_ok_packetc C s | j d dkot| j dk S )Nr r
|
é rh r8 r r r Ú is_eof_packet» s zMysqlPacket.is_eof_packetc C s | j d dkS )Nr r
|
r7 r8 r r r Úis_auth_switch_requestÁ s z"MysqlPacket.is_auth_switch_requestc C s | j d dkS )Nr r r7 r8 r r r Úis_extra_auth_dataÅ s z MysqlPacket.is_extra_auth_datac C s" | j d }d| ko dkS S )Nr r éú r7 )r4 Z field_countr r r Úis_resultset_packetÉ s
|
z MysqlPacket.is_resultset_packetc C s | j d dkS )Nr r r7 r8 r r r Úis_load_local_packetÍ s z MysqlPacket.is_load_local_packetc C s | j d dkS )Nr éÿ r7 r8 r r r Úis_error_packetÐ s zMysqlPacket.is_error_packetc C s | ¡ r| ¡ d S r )rr Úraise_for_errorr8 r r r Ú check_errorÓ s zMysqlPacket.check_errorc C s8 | ¡ | d¡ | ¡ }tr(td| t | j¡ d S )Nr zerrno =)rH rF rP r: r! r Zraise_mysql_exceptionr2 )r4 Úerrnor r r rs × s
|
|
zMysqlPacket.raise_for_errorc C s t | j d S r )r. r2 r8 r r r r; ß s zMysqlPacket.dumpN)r )r ) Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__Ú __slots__r6 r9 r@ rC rF rH rI rK rP rT rV rX r[ rb rc rg ri rk rl rm ro rp rr rt rs r; r r r r r/ / s8
|
|
|
r/ 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 )r/ r6 Ú_parse_field_descriptorr3 r r r r6 ê 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)rc ÚcatalogÚdbÚdecodeÚ
|
table_nameZ org_tableÚnameZorg_namerg Ú charsetnrrE Ú type_codeÚflagsÚscale)r4 r5 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.NrL r )r r Úget_column_lengthr
r r8 r r r Ú description s ùz!FieldDescriptorPacket.descriptionc C s* | j tjkr$t | jd¡}| j| S | jS rJ )r r Z
|
VAR_STRINGr Úgetr rE )r4 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 r8 r r r Ú__str__ s úz FieldDescriptorPacket.__str__N) rv rw rx ry r6 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)ri r) Ústrr rv ÚpacketrF rb Z affected_rowsZ insert_idrg Ú server_statusÚ warning_countrC Úmessager ÚSERVER_MORE_RESULTS_EXISTSÚhas_next©r4 Z from_packetr r r r6 ' s
|
ÿþÿ zOKPacketWrapper.__init__c C s t | j|S r ©Úgetattrr ©r4 Úkeyr r r Ú __getattr__8 s zOKPacketWrapper.__getattr__N©rv rw rx ry r6 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 sV | ¡ std| j d|| _| j d¡\| _| _trDtd| j | jt j
|
@ | _ d S )NúCannot create 'ú!' object from invalid packet typez<xhhzserver_status=) rk r) r r rg r r r: r! r r r r r r r r6 C s ÿ zEOFPacketWrapper.__init__c C s t | j|S r r r r r r r O s z EOFPacketWrapper.__getattr__Nr r r r r r < 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 sH | ¡ std| j d|| _| j ¡ dd
| _trDtd| j d S )Nr r r z filename=)rp r) r r r9 Úfilenamer: r! r r r r r6 Z s ÿz LoadLocalPacketWrapper.__init__c C s t | j|S r r r r r r r e s z"LoadLocalPacketWrapper.__getattr__Nr r r r r r S s r )Úcharsetr Ú constantsr r r r rN r$ r: r\ r] r^ r_ r` r. r/ r{ r r r r r r r Ú<module> s" 5=
|
