File: //proc/thread-self/root/var/opt/nydus/ops/mysql/connector/__pycache__/conversion.cpython-39.pyc
a
�,�hqf � @ s� d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z m
Z
m Z m Z m
Z
mZmZmZ ddlmZmZmZmZ ddlmZ ddlmZmZmZm Z m Z dd l m Z d
Z G d d � d �Z!G d
d� de!�Z"dS )z"Converting MySQL and Python types
� N)�Decimal)�Any�Callable�Dict�List�Optional�Set�Tuple�Union� )�MYSQL_VECTOR_TYPE_CODE� CharacterSet� FieldFlag� FieldType)�
HexLiteral)�DescriptionType�MySQLConvertibleType�MySQLProducedType�PythonProducedType�
StrOrBytes)�
NUMERIC_TYPESz.Could not convert '{value}' to python {pytype}c @ s� e Zd ZdZdee eedd�dd�Zdee ee dd �d
d �Z d edd �d
d�Z
e e e e
f d �dd�Zeee ed�dd�Zed eee ed�dd��Zeeed�dd��ZdS ) �MySQLConverterBasez�Base class for conversion classes
All class dealing with converting to and from MySQL data types must
be a subclass of this class.
�utf8TFN��charset� use_unicode� str_fallback�returnc C s@ t � | _d | _d | _d | _d| _| �|� || _|| _i | _ d S )Nr )
r
�_character_setZ python_typesZ mysql_typesr �
charset_id� set_charsetr r �_cache_field_types��selfr r r � r$ �M/opt/nydus/tmp/pip-target-wkfpz8uv/lib64/python/mysql/connector/conversion.py�__init__> s
�zMySQLConverterBase.__init__)r �
character_setr c C sD |dv r d}|dur || _ nd| _ |r,|| _| j�| j �d | _dS )zSet character set)Zutf8mb4Zutf8mb3r Nr )r r �get_charset_infor )r# r r' r$ r$ r% r Q s z MySQLConverterBase.set_charset��valuer c C s
|| _ dS )zSet whether to use UnicodeN)r �r# r* r$ r$ r% � set_unicodeb s z MySQLConverterBase.set_unicodec C sD |j j�� }z t| d|� d��|�}|W S ty> | Y S 0 dS )�!Convert Python data type to MySQL�_� _to_mysqlN)� __class__�__name__�lower�getattr�AttributeError�r# r* � type_nameZ convertedr$ r$ r% �to_mysqlf s � zMySQLConverterBase.to_mysql��vtyper* r c C s� |dks|du r"|d t jkr"dS | jszi | _t j�� D ]@\}}z$t| d|�� � d��| j|d < W q8 tyv Y q80 q8|du r�dS z| j|d ||�W S ty� | Y S 0 dS )�!Convert MySQL data type to Python� Nr r. �
_to_pythonr ) r �BITr! �desc�itemsr3 r2 r4 �KeyError)r# r9 r* �name�infor$ r$ r% � to_pythons s" � z MySQLConverterBase.to_python�r* �sql_moder c C s | S )z"Escape buffer for sending to MySQLr$ �r* rE r$ r$ r% �escape� s zMySQLConverterBase.escape��bufr c C s t | �S )z!Quote buffer for sending to MySQL��str�rI r$ r$ r% �quote� s zMySQLConverterBase.quote)r TF)N)T)N)r1 �
__module__� __qualname__�__doc__r rK �boolr& r
r r, r r
r r7 r �bytesr rC � staticmethodr rG r rM r$ r$ r$ r% r 7 s> �� � �
� � ��r c s� e Zd ZdZdRee eedd�� fdd�
ZedSe ee e d�d d
��Z
eee e e
eeef ed �d d
��Zeed�dd�Zeee ed�dd�Zee
e
d�dd��Zee
e
d�dd��Zee ee d�dd��Zee eef d�dd�Zee eef d�d d �Z eeed�d d ��Z ee ed�d d!��Z eee
d�d"d#��Z eddd�d$d%��Z!ee"j"ed�d&d'��Z#ee"j$ed�d(d)��Z%ee"j&ed�d*d+��Z'ee&j(ed�d,d-��Z)ee"j*ed�d.d/��Z+eeee d�d0d1��Z,e-ed2f e.e e-ed2f d3�d4d5�Z/edTeee e d6�d7d8��Z0e0Z1edUeee e
d6�d9d:��Z2e2Z3e2Z4e2Z5e2Z6e2Z7dVeee ed6�d;d<�Z8e8Z9edWeee ed6�d=d>��Z:edXeee e
d?�d@dA��Z;edYeee ee"j$ d?�dBdC��Z<e<Z=edZeee e"j*d?�dDdE��Z>ed[eee ee"j" d?�dFdG��Z?e?Z@ed\eee e
d?�dHdI��ZAd]eee eBe d?�dJdK�ZCd^eee e eDeBe f d?�dLdM�ZEeEZFeEZGd_eee e eDeBe f d?�dNdO�ZHed`ee ee eeIjI d6�dPdQ��ZJeHZKeHZLeHZM� ZNS )a�MySQLConvertera� Default conversion class for MySQL Connector/Python.
o escape method: for escaping values send to MySQL
o quoting method: for quoting values send to MySQL in statements
o conversion mapping: maps Python and MySQL data types to
function for converting them.
Whenever one needs to convert values differently, a converter_class
argument can be given while instantiating a new connection like
cnx.connect(converter_class=CustomMySQLConverterClass).
NTFr c s t � �|||� i | _d S )N)�superr& r! r" �r0 r$ r% r& � s �zMySQLConverter.__init__rD c C s� t | ttf�rl|dkr"| �dd�S | �dd�} | �dd�} | �dd �} | �dd
�} | �d d �} | �d
d�} npt | t�r�t | t�s�|dkr�| �dd�S | �dd�} | �dd�} | �dd�} | �dd�} | �dd�} | �dd�} | S ) z�
Escapes special characters as they are expected to by when MySQL
receives them.
As found in MySQL source mysys/charset.c
Returns the value if not a string, or the escaped string.
ZNO_BACKSLASH_ESCAPES� 's ''� \s \\�
s \n�
s \rs \'� "s \"� s \�'z''�\z\\�
z\n�
z\rz\'�"z\"�z\)�
isinstancerR � bytearray�replacerK r rF r$ r$ r% rG � s& zMySQLConverter.escaperH c C s> t | t�rt| ��d�S t | td��r.td�S td| d �S )a
Quote the parameters for commands. General rules:
o numbers are returns as bytes using ascii codec
o None is returned as bytearray(b'NULL')
o Everything else is single quoted '<buf>'
Returns a bytearray object.
�asciiNs NULLrW )rc r rK �encode�typerd rL r$ r$ r% rM � s
zMySQLConverter.quoter) c C sh |j j�� }z t| d|� d��|�}|W S tyb | jrLt|��� Y S td|� d��d�Y n0 dS )r- r. r/ zPython 'z%' cannot be converted to a MySQL typeN) r0 r1 r2 r3 r4 r rK rg � TypeErrorr5 r$ r$ r% r7 � s �
��zMySQLConverter.to_mysqlr8 c C sT |dkr|d t jkrdS |du r&dS | js~i | _t j�� D ]@\}}z$t| d|�� � d��| j|d < W q< tyz Y q<0 q<z| j|d ||�W S ty� z|� d�W Y S t
y� | Y Y S 0 Y n~ t �y } z&t |� d|d � d��|�W Y d}~nHd}~0 t �yN } z&t |� d|d � d��|�W Y d}~n
d}~0 0 dS ) r: r r Nr. r< �utf-8� (field �))
r r= r! r> r? r3 r2 r4 r@ �decode�UnicodeDecodeError�
ValueErrorri )r# r9 r* rA rB �errr$ r$ r% rC � s0 � .zMySQLConverter.to_pythonc C s t | �S )zConvert value to int��int�r* r$ r$ r% �
_int_to_mysql s z MySQLConverter._int_to_mysqlc C s t | �S )z�Convert value to int
Note: There is no type "long" in Python 3 since integers are of unlimited size.
Since Python 2 is no longer supported, this method should be deprecated.
rq rs r$ r$ r% �_long_to_mysql s z MySQLConverter._long_to_mysqlc C s t �| �rdS t| �S )zConvert value to floatN)�math�isnan�floatrs r$ r$ r% �_float_to_mysql$ s
z MySQLConverter._float_to_mysqlc C s
| � |�S )zConvert value to string)�_unicode_to_mysqlr+ r$ r$ r% �
_str_to_mysql+ s z MySQLConverter._str_to_mysqlc C sT | j }| j}|dkr(d}| j�|�d }|�|�}|| jjv rPd|v rPt||�S |S )zConvert unicode�binaryr r rX )r r r r( rg Zslash_charsetsr )r# r* r r �encodedr$ r$ r% rz / s
z MySQLConverter._unicode_to_mysqlc C s | S �zConvert value to bytesr$ rs r$ r$ r% �_bytes_to_mysql< s z MySQLConverter._bytes_to_mysqlc C s t | �S r~ )rR rs r$ r$ r% �_bytearray_to_mysqlA s z"MySQLConverter._bytearray_to_mysqlc C s | rdS dS )zConvert value to booleanr r r$ rs r$ r$ r% �_bool_to_mysqlF s z MySQLConverter._bool_to_mysqlc C s dS )z�
This would return what None would be in MySQL, but instead we
leave it None and return it right away. The actual conversion
from None to NULL happens in the quoting functionality.
Return None.
Nr$ rs r$ r$ r% �_nonetype_to_mysqlK s z!MySQLConverter._nonetype_to_mysqlc C s^ | j r4d}|�| j| j| j| j| j| j| j ��d�S d}|�| j| j| j| j| j| j��d�S )z�
Converts a datetime instance to a string suitable for MySQL.
The returned string has format: %Y-%m-%d %H:%M:%S[.%f]
If the instance isn't a datetime.datetime type, it return None.
Returns a bytes.
z7{0:04d}-{1:02d}-{2:02d} {3:02d}:{4:02d}:{5:02d}.{6:06d}rf z/{0:04d}-{1:02d}-{2:02d} {3:02d}:{4:02d}:{5:02d}) � microsecond�format�year�month�day�hour�minute�secondrg )r* �fmtr$ r$ r% �_datetime_to_mysqlV s0
��
��z!MySQLConverter._datetime_to_mysqlc C s& | j d�d| jd�d| jd���d�S )z�
Converts a date instance to a string suitable for MySQL.
The returned string has format: %Y-%m-%d
If the instance isn't a datetime.date type, it return None.
Returns a bytes.
Z04d�-Z02drf )r� r� r� rg rs r$ r$ r% �_date_to_mysqlv s
z MySQLConverter._date_to_mysqlc C s& | j r| �d��d�S | �d��d�S )z�
Converts a time instance to a string suitable for MySQL.
The returned string has format: %H:%M:%S[.%f]
If the instance isn't a datetime.time type, it return None.
Returns a bytes.
z %H:%M:%S.%frf z%H:%M:%S)r� �strftimerg rs r$ r$ r% �_time_to_mysql� s
z MySQLConverter._time_to_mysqlc C s t �d| ��d�S )z�
Converts a time.struct_time sequence to a string suitable
for MySQL.
The returned string has format: %Y-%m-%d %H:%M:%S
Returns a bytes or None when not valid.
z%Y-%m-%d %H:%M:%Srf )�timer� rg rs r$ r$ r% �_struct_time_to_mysql� s z$MySQLConverter._struct_time_to_mysqlc C s� t | jd | j �}| jrDd}| jdk r<d| j }|d8 }qH| j}nd}| jdk rZd| }t|d�\}}t|d �\}}| jr�|�||||�}n|�|||�}|�d
�S ) z�
Converts a timedelta instance to a string suitable for MySQL.
The returned string has format: %H:%M:%S
Returns a bytes.
i�Q z {0:02d}:{1:02d}:{2:02d}.{3:06d}r i@B r z{0:02d}:{1:02d}:{2:02d}r� i �< rf )�abs�days�seconds� microseconds�divmodr� rg ) r* r� r� �mcs�hours� remainder�mins�secs�resultr$ r$ r% �_timedelta_to_mysql� s
z"MySQLConverter._timedelta_to_mysqlc C s t | t�rt| ��d�S dS )z�
Converts a decimal.Decimal instance to a string suitable for
MySQL.
Returns a bytes or None when not valid.
rf N)rc r rK rg rs r$ r$ r% �_decimal_to_mysql� s
z MySQLConverter._decimal_to_mysql.)�row�fieldsr c
C sb d}dgt |� }| jsji | _tj�� D ]@\}}z$t| d|�� � d��| j|d < W q( tyf Y q(0 q(|D ]�}|d }|| dkr�|tjks�|| du r�|d7 }qnz | j| || |�||< W n� t �y z|| �
d�||< W n t �y || ||< Y n0 Y nB t t
f�yN } z$| � d|d � d�| _� W Y d} ~ n
d} ~ 0 0 |d7 }qnt|�S ) a Convert a MySQL text result row to Python types
The row argument is a sequence containing text result returned
by a MySQL server. Each value of the row is converted to the
using the field type information in the fields argument.
Returns a tuple.
r Nr. r< r rj rk rl )�lenr! r r> r? r3 r2 r4 r= r@ rm rn ro ri �message�tuple)
r# r� r� �ir� rA rB �fieldZ
field_typerp r$ r$ r% �
row_to_python� s8 � "
z MySQLConverter.row_to_python)r* r> r c C s t | �S )z.
Returns value as float type.
)rx �r* r> r$ r$ r% �_float_to_python� s z MySQLConverter._float_to_pythonc C s t | �S )z,
Returns value as int type.
rq r� r$ r$ r% �_int_to_python s z MySQLConverter._int_to_pythonc C s |� | j�}t|�S )z5
Returns value as a decimal.Decimal.
)rm r r )r# r* r> �valr$ r$ r% �_decimal_to_python s z!MySQLConverter._decimal_to_pythonc C s t | �S )z,
Returns value as str type.
rJ r� r$ r$ r% �_str s zMySQLConverter._str)r* �dscr c C s8 | }t |�dk r$ddt |� | }tt�d|�d �S )z!Returns BIT columntype as integer� r; z>Qr )r� rr �struct�unpack)r* r� Zint_valr$ r$ r% �_bit_to_python$ s z MySQLConverter._bit_to_pythonc C s� t | tj�r| S zt| �d�}t|�dkr@td|� dt|�� ���z*t�t|d �t|d �t|d ��W W S ty� Y W dS 0 W n, ttfy� td t| �� d
��d�Y n0 dS ) z�Converts TIME column MySQL to a python datetime.datetime type.
Raises ValueError if the value can not be converted.
Returns DATE column type as datetime.date type.
� -� �invalid datetime format: � len: r r � NzCould not convert z to python datetime.timedelta) rc �datetime�date�splitr� ro rr �
IndexError�repr)r* r� �partsr$ r$ r% �_date_to_python, s
* ��z MySQLConverter._date_to_pythonc
C s� d}z"| � d�\}}t|�dd��}W n ttfyD | }d}Y n0 z\dd� |� d�D �\}}}| d d ksz| d d
kr�| | | }}}tj||||d �W S tttfy� ttj | d d
��d�Y n0 dS )a Converts TIME column value to python datetime.time value type.
Converts the TIME column MySQL type passed as bytes to a python
datetime.datetime type.
Raises ValueError if the value can not be converted.
Returns datetime.timedelta type.
N� .� � 0r c S s g | ] }t |��qS r$ rq )�.0�dr$ r$ r% �
<listcomp>\ � z2MySQLConverter._time_to_python.<locals>.<listcomp>� :�- r� )r� �minutesr� r� �datetime.timedelta�r* Zpytype)
r� rr �ljustri ro r� � timedeltar� �
CONVERT_ERRORr� )r* r� r� �hmsr� r� r� r$ r$ r% �_time_to_pythonG s.
�
� ��z MySQLConverter._time_to_pythonc C s t | tj�r| S d}d}z�| �d�\}}t|�dkrT|�d�\}}t|�dd��}n|}d}dd � |�d
�D �d d � |�d �D � |g }t|�dk r�td
|� dt|�� ���ztj|� }W n ty� Y W dS 0 W n, ttf�y tt j
| dd��d�Y n0 |S )z�Converts DATETIME column value to python datetime.time value type.
Converts the DATETIME column MySQL type passed as bytes to a python
datetime.datetime type.
Returns: datetime.datetime type.
N� r� r� r� r� r c S s g | ] }t |��qS r$ rq �r� r� r$ r$ r% r� � r� z6MySQLConverter._datetime_to_python.<locals>.<listcomp>r� c S s g | ] }t |��qS r$ rq r� r$ r$ r% r� � r� r� r� r� r� r� ) rc r� r� r� rr r� ro r� ri r� r� )r* r� Z datetime_valr� Zdate_Ztime_r� Zdtvalr$ r$ r% �_datetime_to_pythonk s> ���� ��z"MySQLConverter._datetime_to_pythonc
C sL z t | �}W n: tyF } z"tdt| �� d��|�W Y d}~n
d}~0 0 |S )z#Returns YEAR column type as integerz Failed converting YEAR to int (rl N)rr ro r� )r* r� r� rp r$ r$ r% �_year_to_python� s
,z MySQLConverter._year_to_pythonc
C sl d}|� | j�}|st� S zt|�d��}W n: tyf } z"tdt|�� d��|�W Y d}~n
d}~0 0 |S )z�Returns SET column type as set
Actually, MySQL protocol sees a SET as a string type field. So this
code isn't called directly, but used by STRING_to_python() method.
Returns SET column type as a set.
N�,zCould not convert set z to a sequence)rm r �setr� ro r� )r# r* r� �set_typer� rp r$ r$ r% �_set_to_python� s
��z MySQLConverter._set_to_pythonc C s� | j dkr|S |dur`|d tjkr6| jr6|�| j �S |d tj@ rP| �||�S |d dkr`|S t|t t
f�r�| jr�z|�| j �W S t y� | Y S 0 |S )z�
Note that a SET is a string too, but using the FieldFlag we can see
whether we have to split it.
Returns string typed columns as string type.
r| Nr � r� �? ) r r ZJSONr rm r ZSETr� rc rR rd rn �r# r* r� r$ r$ r% �_string_to_python� s
z MySQLConverter._string_to_pythonc C sD |dur8|d t j@ r8|d t j@ r8|d dkr8t|�S | �||�S )z!Convert BLOB data type to Python.Nr� r� r� )r ZBLOBZBINARYrR r� r� r$ r$ r% �_blob_to_python� s � �
�z MySQLConverter._blob_to_pythonc C sH | du st | tj�r| S t | ttf�r2t�t| �S td| jj� ���dS )z�
Converts a MySQL VECTOR value to a Python array.array type.
Returns an array of floats if `value` isn't `None`, otherwise `None`.
NzGot unsupported type )rc �arrayrR rd r ri r0 r1 r� r$ r$ r% �_vector_to_python� s
z MySQLConverter._vector_to_python)NTF)N)N)N)N)N)N)N)N)N)N)N)N)N)N)Or1 rN rO rP r rK rQ r& rS r rG r
rx rr r r rR rM r r r7 r r rC rt ru ry r{ rz r rd r� r� r� r� r� r� r� r� r� � struct_timer� r� r� r� r r r� r� Z_double_to_pythonr� Z_tiny_to_pythonZ_short_to_pythonZ_int24_to_pythonZ_long_to_pythonZ_longlong_to_pythonr� Z_newdecimal_to_pythonr� r� r� Z_NEWDATE_to_pythonr� r� Z_timestamp_to_pythonr� r r� r r� Z_var_string_to_pythonZ_json_to_pythonr� r� r� Z_long_blob_to_pythonZ_medium_blob_to_pythonZ_tiny_blob_to_python�
__classcell__r$ r$ rV r% rT � s� �� $� %
� 3 �� �� ��# ��.
�� �� �� � �rT )#rP r� r� rv r� r� �decimalr �typingr r r r r r r r
� constantsr r
r r Z custom_typesr �typesr r r r r �utilsr r� r rT r$ r$ r$ r% �<module> s ( d