관리-도구
편집 파일: conversion.cpython-38.pyc
U )�e M � @ sx d Z ddlZddlZddlmZ ddlmZmZmZ ddl m Z m Z m Z ddl mZ G dd � d e�ZG d d � d e�ZdS ) z"Converting MySQL and Python types � N)�Decimal� )� FieldType� FieldFlag� CharacterSet)�PY2� NUMERIC_TYPES� struct_unpack)� HexLiteralc @ sL e Zd ZdZddd�Zdd� Zddd �Zd d � Zd d � Zdd� Z dd� Z dS )�MySQLConverterBasez�Base class for conversion classes All class dealing with converting to and from MySQL data types must be a subclass of this class. �utf8Tc C s< d | _ d | _d | _d| _d | _| �|� | �|� i | _d S )Nr )Z python_typesZ mysql_types�charset� charset_id� use_unicode� set_charset� set_unicode�_cache_field_types��selfr r � r �o/opt/hc_python/lib64/python3.8/site-packages/../../../lib/python3.8/site-packages/mysql/connector/conversion.py�__init__+ s zMySQLConverterBase.__init__c C s8 |dkr d}|dk r || _ nd| _ t�| j �d | _dS )zSet character setZutf8mb4r Nr )r r �get_charset_infor )r r r r r r 5 s z MySQLConverterBase.set_charsetc C s || _ dS )zSet whether to use UnicodeN)r �r �valuer r r r @ s z MySQLConverterBase.set_unicodec C s@ |j j�� }zt| d�|��|�W S tk r: | Y S X dS )�!Convert Python data type to MySQL� _{0}_to_mysqlN)� __class__�__name__�lower�getattr�format�AttributeError�r r � type_namer r r �to_mysqlD s zMySQLConverterBase.to_mysqlc C s� |dks|dkr"|d t jkr"dS | jsvi | _t j�� D ]<\}}z t| d�|��| j|d < W q8 tk rr Y q8X q8z| j|d ||�W S tk r� | Y S X dS )�!Convert MySQL data type to Python� Nr �_{0}_to_pythonr ) r �BITr �desc�itemsr r! r" �KeyError)r �vtyper �name�infor r r � to_pythonL s �z MySQLConverterBase.to_pythonc C s |S )z"Escape buffer for sending to MySQLr �r �bufr r r �escapeb s zMySQLConverterBase.escapec C s t |�S )z!Quote buffer for sending to MySQL��strr1 r r r �quotef s zMySQLConverterBase.quoteN)r T)T) r � __module__� __qualname__�__doc__r r r r% r0 r3 r6 r r r r r $ s r c @ sf e Zd ZdZdFdd�Zdd� Zdd � Zd d � Zd d � Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� ZdGd.d/�ZeZdHd0d1�ZeZ eZ eZ eZ eZ dId2d3�Z!e!Z"dJd4d5�Z#dKd6d7�Z$dLd8d9�Z%e%Z&dMd:d;�Z'dNd<d=�Z(e(Z)dOd>d?�Z*dPd@dA�Z+dQdBdC�Z,e,Z-dRdDdE�Z.e.Z/e.Z0e.Z1dS )S�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). NTc C s t �| ||� i | _d S )N)r r r r r r r r y s zMySQLConverter.__init__c C s� |dkr |S t |t�r|S t |ttf�rr|�dd�}|�dd�}|�dd�}|�dd �}|�d d �}|�d d �}nH|�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. N� \s \\� s \n� s \r� 's \'� "s \"� s \�\z\\� z\n� z\r�'z\'�"z\"�z\)� isinstancer �bytes� bytearray�replacer r r r r3 } s$ zMySQLConverter.escapec C sd t |t�r:tr*t |t�r t|�S t|�S q`t|��d�S n&t |td��rPtd�S td| d �S dS )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 NULLr> ) rG r r �float�reprr5 �encode�typerI r1 r r r r6 � s zMySQLConverter.quotec C sJ |j j�� }zt| d�|��|�W S tk rD td�|���Y nX dS )r r z0Python '{0}' cannot be converted to a MySQL typeN)r r r r r! r" � TypeErrorr# r r r r% � s �zMySQLConverter.to_mysqlc C sP |dkr|d t jkrdS |dkr&dS | jszi | _t j�� D ]<\}}z t| d�|��| j|d < W q< tk rv Y q<X q<z| j|d ||�W S tk r� z|� d�W Y S t k r� | Y Y S X Y nz t k �r } zt d||d f ��W 5 d}~X Y nD t k �r> } zt d||d f ��W 5 d}~X Y n � Y nX dS )r& r r Nr( �utf-8z %s (field %s)) r r) r r* r+ r r! r" r, �decode�UnicodeDecodeError� ValueErrorrP )r r- r r. r/ �errr r r r0 � s6 �$$zMySQLConverter.to_pythonc C s t |�S �zConvert value to int��intr r r r � _int_to_mysql� s z MySQLConverter._int_to_mysqlc C s t |�S rV rW r r r r �_long_to_mysql� s z MySQLConverter._long_to_mysqlc C s t |�S )zConvert value to float�rL r r r r �_float_to_mysql� s z MySQLConverter._float_to_mysqlc C s t r t|�S | �|�S )zConvert value to string)r r5 �_unicode_to_mysqlr r r r � _str_to_mysql� s z MySQLConverter._str_to_mysqlc C sP | j }| j}|dkr&d}t�|�d }|�|�}|tjkrLd|krLt||�S |S )zConvert unicode�binaryr r r; )r r r r rN Zslash_charsetsr )r r r r �encodedr r r r] � s z MySQLConverter._unicode_to_mysqlc C s |S �zConvert value to bytesr r r r r �_bytes_to_mysql� s z MySQLConverter._bytes_to_mysqlc C s t |�S ra )rH r r r r �_bytearray_to_mysql� s z"MySQLConverter._bytearray_to_mysqlc C s |rdS dS dS )zConvert value to booleanr r Nr r r r r �_bool_to_mysql 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 r r r r �_nonetype_to_mysql 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. z5{0:d}-{1:02d}-{2:02d} {3:02d}:{4:02d}:{5:02d}.{6:06d}rK z-{0:d}-{1:02d}-{2:02d} {3:02d}:{4:02d}:{5:02d}) � microsecondr! �year�month�day�hour�minute�secondrN )r r �fmtr r r �_datetime_to_mysql s0 �� ��z!MySQLConverter._datetime_to_mysqlc C s d� |j|j|j��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. z{0:d}-{1:02d}-{2:02d}rK )r! rg rh ri rN r r r r �_date_to_mysql& 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.%frK z%H:%M:%S)rf �strftimerN r r r r �_time_to_mysql2 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:%SrK )�timerp rN r 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|�|||�} tr�| S | �d �S 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}�-i �< rK N)�abs�days�seconds� microseconds�divmodr! r rN ) r r rx rm �mcs�hours� remainder�mins�secs�resultr r r �_timedelta_to_mysqlI 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. rK N)rG r r5 rN r r r r �_decimal_to_mysqll s z MySQLConverter._decimal_to_mysqlc C s` d}dgt |� }| jsfi | _tj�� D ]<\}}z t| d�|��| j|d < W q( tk rb Y q(X q(|D ]�}|d }|| dkr�|tjks�|| dkr�|d7 }qjz | j| || |�||< W n� t k �r z|| � d�||< W n" t k �r || ||< Y nX Y n@ t t fk �rL } z d�t| �|d �| _� W 5 d} ~ X Y nX |d7 }qjt|�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 rQ z{0} (field {1}))�lenr r r* r+ r r! r" r) r, rR rS rT rP r5 �message�tuple) r �row�fields�ir� r. r/ �fieldZ field_typerU r r r � row_to_pythonx s: �" z MySQLConverter.row_to_pythonc C s t |�S )z. Returns value as float type. r[ �r r r* r r r �_FLOAT_to_python� s z MySQLConverter._FLOAT_to_pythonc C s t |�S )z, Returns value as int type. rW 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. )rR 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. r4 r� r r r �_str� s zMySQLConverter._strc C s2 |}t |�dk r$ddt |� | }td|�d S )z!Returns BIT columntype as integer� r' z>Qr )r� r )r r �dscZint_valr r r �_BIT_to_python� s z MySQLConverter._BIT_to_pythonc C sN z2|� d�}t�t|d �t|d �t|d ��W S tk rH Y dS X dS )zA Returns DATE column type as datetime.date type. � -r r � N)�split�datetime�daterX rT )r r r� �partsr r r �_DATE_to_python� s (z MySQLConverter._DATE_to_pythonc C s� d}z"|� d�\}}t|�dd��}W n tk rB |}d}Y nX z^dd� |� d�D �\}}}|d d ksx|d d kr�| | | }}}tj||||d �}W n" tk r� td �|���Y nX |S dS ) zA Returns TIME column type as datetime.time type. N� .� � 0r c S s g | ] }t |��qS r rW )�.0�dr r r � <listcomp>� s z2MySQLConverter._TIME_to_python.<locals>.<listcomp>� :�- rt )r| �minutesrx ry z2Could not convert {0} to python datetime.timedelta)r� rX �ljustrT r� � timedeltar! ) r r r� Ztime_val�hmsr{ r| r~ r r r r �_TIME_to_python� s, � �� z MySQLConverter._TIME_to_pythonc C s� d}z~|� d�\}}t|�dkr@|� d�\}}t|�dd��}n|}d}dd � |� d �D �d d � |� d �D � |g }tj|� }W n tk r� d}Y nX |S ) zI Returns DATETIME column type as datetime.datetime type. N� r� r� r� r� r c S s g | ] }t |��qS r rW �r� r� r r r r� s z6MySQLConverter._DATETIME_to_python.<locals>.<listcomp>r� c S s g | ] }t |��qS r rW r� r r r r� s r� )r� r� rX r� r� rT ) r r r� Z datetime_valZdate_Ztime_r� r{ Zdtvalr r r �_DATETIME_to_python� s" �� z"MySQLConverter._DATETIME_to_pythonc C s2 z t |�}W n tk r, td| ��Y nX |S )z#Returns YEAR column type as integerz"Failed converting YEAR to int (%s))rX rT )r r r* rg r r r �_YEAR_to_python s z MySQLConverter._YEAR_to_pythonc C sR d}|� | j�}|st� S zt|�d��}W n tk rL td| ��Y nX |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�,z'Could not convert set %s to a sequence.)rR r �setr� rT )r r r� �set_typer� r r r �_SET_to_python s z MySQLConverter._SET_to_pythonc C sf |dk r4|d t j@ r"| �||�S |d t j@ r4|S | jdkrB|S t|ttf�rb| jrb|� | j�S |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. N� r_ ) r ZSETr� �BINARYr rG rH rI r rR �r r r� r r r �_STRING_to_python& s z MySQLConverter._STRING_to_pythonc C s2 |dk r&|d t j@ r&tr |S t|�S | �||�S )z Convert BLOB data type to PythonNr� )r r� r rH r� r� r r r �_BLOB_to_python= s z MySQLConverter._BLOB_to_python)NT)N)N)N)N)N)N)N)N)N)N)N)N)2r r7 r8 r9 r r3 r6 r% r0 rY rZ r\ r^ r] rb rc rd re rn ro rq rs 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� Z_VAR_STRING_to_pythonr� Z_LONG_BLOB_to_pythonZ_MEDIUM_BLOB_to_pythonZ_TINY_BLOB_to_pythonr r r r r: k s^ ! # . r: )r9 r� rr �decimalr � constantsr r r Zcatch23r r r Z custom_typesr �objectr r: r r r r �<module> s G