
    Yh$.                         d dl mZ d dlmZmZ d dlmZ ddlZddlZdZ	dZ
dZdZd	Zd
Zd Z G d d      Z G d de      Z G d d      Z G d d      Z G d d      Zy)   )MBLENGTH)
FIELD_TYPESERVER_STATUS)err    NF            c           	      `   d 	 t        dt        |              t        dd      D ]F  }t        j                  |      }t        d||j
                  j                  |j                  fz         H t        d       t        dt        t        |       d      d	      D cg c]
  }| ||d	z     }}|D ]S  }t        d
j                  d |D              dd	t        |      z
  z  z   dz   dj                  fd|D              z          U t        d       t                y # t        $ r Y w xY wc c}w )Nc                 4    d| cxk  rdk  rt        |       S  yy)N       .)chr)datas    a/var/www/django-portfolio.reubendavern.link/venv/lib/python3.12/site-packages/pymysql/protocol.py	printablezdump_packet.<locals>.printable   s!    t9     zpacket length:r      zcall[%d]: %s (line %d)zB------------------------------------------------------------------r          c              3   $   K   | ]  }|d  
 yw)02XN ).0xs     r   	<genexpr>zdump_packet.<locals>.<genexpr>&   s     +A#wZ+s   z   z   c              3   .   K   | ]  } |        y wNr   )r   r   r   s     r   r   zdump_packet.<locals>.<genexpr>)   s     .qil.s   )printlenrangesys	_getframef_codeco_namef_lineno
ValueErrorminjoin)r   if	dump_datadr   s        @r   dump_packetr2      s$   
D	*q! 	PAa A*a1A1A1::-NNO	P 	h ,1CD	34G+LMaa!b&!MIM 
HH+++rCF{#$ gg.A../	

 
(O	G  Ms   A5D D+	D('D(c                       e Zd ZdZdZd Zd Zd Zd Zd Z	ddZ
dd	Zd
 Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zy) MysqlPacketzrRepresentation of a MySQL response packet.

    Provides an interface for reading/parsing the packet results.
    	_position_datac                      d| _         || _        y )Nr   r5   selfr   encodings      r   __init__zMysqlPacket.__init__7   s    
r   c                     | j                   S r"   r7   r:   s    r   get_all_datazMysqlPacket.get_all_data;   s    zzr   c           
      X   | j                   | j                  | j                  |z    }t        |      |k7  r`d| dt        |       d| j                   dt        | j                          }t        rt	        |       | j                          t        |      | xj                  |z  c_        |S )zCRead the first 'size' bytes in packet and advance cursor past them.z-Result length not requested length:
Expected=z
.  Actual=z.  Position: z.  Data Length: )r7   r6   r$   DEBUGr#   dumpAssertionError)r:   sizeresulterrors       r   readzMysqlPacket.read>   s    DNNdnnt.CEv;$ 6CK=dnnEUUefijnjtjtfuevx  e		 ''$r   c                 F    | j                   | j                  d }d| _        |S )z`Read all remaining data in the packet.

        (Subsequent read() will return errors.)
        Nr7   r6   r:   rF   s     r   read_allzMysqlPacket.read_allM   s%    
 DNN,-r   c                     | j                   |z   }|dk  s|t        | j                        kD  rt        d| d|       || _         y)z1Advance the cursor in data buffer 'length' bytes.r   zInvalid advance amount (z) for cursor.  Position=N)r6   r$   r7   	Exception)r:   lengthnew_positions      r   advancezMysqlPacket.advanceV   sN    ~~.!|c$**o=*6(2J<.Y  &r   c                 h    |dk  s|t        | j                        kD  rt        d|z        || _        y)z9Set the position of the data buffer cursor to 'position'.r   z)Invalid position to rewind cursor to: %s.N)r$   r7   rN   r6   )r:   positions     r   rewindzMysqlPacket.rewind_   s0    a<8c$**o5G(RSS!r   c                 &    | j                   |||z    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:   rS   rO   s      r   	get_byteszMysqlPacket.get_bytese   s     zz(h&799r   c                 b    | j                   | j                     }| xj                  dz  c_        |S Nr   rJ   rK   s     r   
read_uint8zMysqlPacket.read_uint8p   s'    DNN+!r   c                     t        j                  d| j                  | j                        d   }| xj                  dz  c_        |S )Nz<Hr      structunpack_fromr7   r6   rK   s     r   read_uint16zMysqlPacket.read_uint16u   6    ##D$**dnnEaH!r   c                     t        j                  d| j                  | j                        \  }}| xj                  dz  c_        ||dz  z   S )Nz<HB   r   r\   )r:   lowhighs      r   read_uint24zMysqlPacket.read_uint24z   s?    &&udjj$..I	T!dbj!!r   c                     t        j                  d| j                  | j                        d   }| xj                  dz  c_        |S )Nz<Ir      r\   rK   s     r   read_uint32zMysqlPacket.read_uint32   r`   r   c                     t        j                  d| j                  | j                        d   }| xj                  dz  c_        |S )Nz<Qr      r\   rK   s     r   read_uint64zMysqlPacket.read_uint64   r`   r   c                     | j                   j                  d| j                        }|dk  ry | j                   | j                  | }|dz   | _        |S )N    r   r   )r7   findr6   )r:   end_posrF   s      r   read_stringzMysqlPacket.read_string   sH    **//%8Q;DNNW5 1r   c                     | j                         }|t        k(  ry|t        k  r|S |t        k(  r| j	                         S |t
        k(  r| j                         S |t        k(  r| j                         S y)zRead 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)	rY   NULL_COLUMNUNSIGNED_CHAR_COLUMNUNSIGNED_SHORT_COLUMNr_   UNSIGNED_INT24_COLUMNre   UNSIGNED_INT64_COLUMNrk   )r:   cs     r   read_length_encoded_integerz'MysqlPacket.read_length_encoded_integer   sv     OO##H''##%%''##%%''##%% (r   c                 J    | j                         }|y| 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)rx   rH   )r:   rO   s     r   read_length_coded_stringz$MysqlPacket.read_length_coded_string   s)     113>yy  r   c                     t        j                  |      }|j                  | j                  | j                        }| xj                  |j
                  z  c_        |S r"   )r]   Structr^   r7   r6   rE   )r:   fmtsrF   s       r   read_structzMysqlPacket.read_struct   s?    MM#tzz4>>:!&& r   c                 Z    | j                   d   dk(  xr t        | j                         dk\  S )Nr   r   r7   r$   r?   s    r   is_ok_packetzMysqlPacket.is_ok_packet   s'    zz!}!:c$**o&::r   c                 Z    | j                   d   dk(  xr t        | j                         dk  S )Nr   r   	   r   r?   s    r   is_eof_packetzMysqlPacket.is_eof_packet   s)     zz!}$<TZZ1)<<r   c                 &    | j                   d   dk(  S )Nr   r   r>   r?   s    r   is_auth_switch_requestz"MysqlPacket.is_auth_switch_request   s    zz!}$$r   c                 &    | j                   d   dk(  S )Nr   r   r>   r?   s    r   is_extra_auth_datazMysqlPacket.is_extra_auth_data   s    zz!}!!r   c                 @    | j                   d   }d|cxk  xr dk  S c S )Nr   r      r>   )r:   field_counts     r   is_resultset_packetzMysqlPacket.is_resultset_packet   s%    jjmK&3&&&&r   c                 &    | j                   d   dk(  S )Nr   r   r>   r?   s    r   is_load_local_packetz MysqlPacket.is_load_local_packet       zz!}$$r   c                 &    | j                   d   dk(  S )Nr      r>   r?   s    r   is_error_packetzMysqlPacket.is_error_packet   r   r   c                 F    | j                         r| j                          y y r"   )r   raise_for_errorr?   s    r   check_errorzMysqlPacket.check_error   s    !  " "r   c                     | j                          | j                  d       | j                         }t        rt	        d|       t        j                  | j                         y )Nr   zerrno =)rT   rQ   r_   rB   r#   r   raise_mysql_exceptionr7   )r:   errnos     r   r   zMysqlPacket.raise_for_error   sC    Q  ")U#!!$**-r   c                 .    t        | j                         y r"   )r2   r7   r?   s    r   rC   zMysqlPacket.dump   s    DJJr   N)r   )r   )__name__
__module____qualname____doc__	__slots__r<   r@   rH   rL   rQ   rT   rV   rY   r_   re   rh   rk   rp   rx   rz   r   r   r   r   r   r   r   r   r   r   rC   r   r   r   r4   r4   /   s    
 'I&"	:

"


&$
!;=%"'%%#. r   r4   c                   .    e Zd ZdZd Zd Zd Zd Zd Zy)FieldDescriptorPacketzA 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                 T    t         j                  | ||       | j                  |       y r"   )r4   r<   _parse_field_descriptorr9   s      r   r<   zFieldDescriptorPacket.__init__   s"    T42$$X.r   c                    | j                         | _        | j                         | _        | j                         j                  |      | _        | j                         j                  |      | _        | j                         j                  |      | _        | j                         j                  |      | _        | j                  d      \  | _	        | _
        | _        | _        | _        y)zParse the 'Field Descriptor' (Metadata) packet.

        This is compatible with MySQL 4.1+ (not compatible with MySQL 4.0).
        z	<xHIBHBxxN)rz   catalogdbdecode
table_name	org_tablenameorg_namer   	charsetnrrO   	type_codeflagsscale)r:   r;   s     r   r   z-FieldDescriptorPacket._parse_field_descriptor   s    
 446//1779@@J668??I113::8D	557>>xH [)	
NKNJJr   c                     | j                   | j                  d| j                         | j                         | j                  | j                  dz  dk(  fS )zBProvides a 7-item tuple compatible with the Python PEP249 DB Spec.Nr[   r   )r   r   get_column_lengthr   r   r?   s    r   descriptionz!FieldDescriptorPacket.description  sN     IINN""$""$JJJJNa
 	
r   c                     | j                   t        j                  k(  r/t        j                  | j
                  d      }| j                  |z  S | j                  S rX   )r   r   
VAR_STRINGr   getr   rO   )r:   mblens     r   r   z'FieldDescriptorPacket.get_column_length  sA    >>Z222LL3E;;%''{{r   c                     dj                  | j                  | j                  | j                  | j                  | j
                  | j                        S )Nz&{} {!r}.{!r}.{!r}, type={}, flags={:x})format	__class__r   r   r   r   r   r?   s    r   __str__zFieldDescriptorPacket.__str__  s?    7>>NNGGOOIINNJJ
 	
r   N)	r   r   r   r   r<   r   r   r   r   r   r   r   r   r      s     /**


r   r   c                       e Zd ZdZd Zd Zy)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                    |j                         s.t        dt        | j                  j                        z   dz         || _        | j
                  j                  d       | j
                  j                         | _        | j
                  j                         | _	        | j                  d      \  | _        | _        | j
                  j                         | _        | j                  t        j                   z  | _        y )NzCannot create z  object from invalid packet typer   z<HH)r   r+   strr   r   packetrQ   rx   affected_rows	insert_idr   server_statuswarning_countrL   messager   SERVER_MORE_RESULTS_EXISTShas_nextr:   from_packets     r   r<   zOKPacketWrapper.__init__%  s    '') dnn--./45  "A![[DDF@@B151A1A%1H.D.{{++-**]-U-UUr   c                 .    t        | j                  |      S r"   getattrr   r:   keys     r   __getattr__zOKPacketWrapper.__getattr__6      t{{C((r   Nr   r   r   r   r<   r   r   r   r   r   r     s    V")r   r   c                       e Zd ZdZd Zd Zy)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                 0   |j                         st        d| j                   d      || _        | j                  j	                  d      \  | _        | _        t        rt        d| j                         | j                  t        j                  z  | _        y )NCannot create '!' object from invalid packet typez<xhhzserver_status=)r   r+   r   r   r   r   r   rB   r#   r   r   r   r   s     r   r<   zEOFPacketWrapper.__init__A  s    ((*!$..!11RS  "151H1H1P.D."D$6$67**]-U-UUr   c                 .    t        | j                  |      S r"   r   r   s     r   r   zEOFPacketWrapper.__getattr__M  r   r   Nr   r   r   r   r   r   :  s    
V)r   r   c                       e Zd ZdZd Zd Zy)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                     |j                         st        d| j                   d      || _        | j                  j	                         dd  | _        t        rt        d| j
                         y y )Nr   r   r   z	filename=)r   r+   r   r   r@   filenamerB   r#   r   s     r   r<   zLoadLocalPacketWrapper.__init__X  sg    //1!$..!11RS  "002126+t}}- r   c                 .    t        | j                  |      S r"   r   r   s     r   r   z"LoadLocalPacketWrapper.__getattr__c  r   r   Nr   r   r   r   r   r   Q  s    	.)r   r   )charsetr   	constantsr   r   r    r   r]   r&   rB   rr   rs   rt   ru   rv   r2   r4   r   r   r   r   r   r   r   <module>r      sw     0   
 	    4o  o d:
K :
z) )8) ).) )r   