o hV@sxdZddlZddlZddlmZddlmZmZmZddl m Z m Z m Z ddl mZGdd d eZGd d d eZdS) z"Converting MySQL and Python types N)Decimal) FieldType FieldFlag CharacterSet)PY2 NUMERIC_TYPES struct_unpack) HexLiteralc@sLeZdZdZdddZddZddd Zd d Zd d ZddZ ddZ dS)MySQLConverterBasezBase class for conversion classes All class dealing with converting to and from MySQL data types must be a subclass of this class. utf8TcCs<d|_d|_d|_d|_d|_||||i|_dS)Nr) python_types mysql_typescharset charset_id use_unicode set_charset set_unicode_cache_field_typesselfrrrL/var/www/html/env/lib/python3.10/site-packages/mysql/connector/conversion.py__init__0s   zMySQLConverterBase.__init__cCs8|dkrd}|dur||_nd|_t|jd|_dS)zSet character setutf8mb4r Nr)rrget_charset_infor)rrrrrr:s zMySQLConverterBase.set_charsetcCs ||_dS)zSet whether to use UnicodeN)rrvaluerrrrEs zMySQLConverterBase.set_unicodecCs:|jj}z t|d||WSty|YSw)!Convert Python data type to MySQL _{0}_to_mysql) __class____name__lowergetattrformatAttributeErrorrr type_namerrrto_mysqlIs  zMySQLConverterBase.to_mysqlc Cs|dks|dur|dtjkrdS|js9i|_tjD]\}}zt|d||j|d<Wqty8Yqwz |j|d||WStyO|YSw)!Convert MySQL data type to PythonNr_{0}_to_pythonr) rBITrdescitemsr#r$r%KeyError)rvtypernameinforrr to_pythonQs"   zMySQLConverterBase.to_pythoncC|S)z"Escape buffer for sending to MySQLrrrrrescapegzMySQLConverterBase.escapecCt|S)z!Quote buffer for sending to MySQLstrrbufrrrquotekzMySQLConverterBase.quoteN)r T)T) r! __module__ __qualname____doc__rrrr(r3r5r<rrrrr )s   r c@speZdZdZdHddZddZdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-ZdId.d/ZeZdId0d1ZeZeZeZeZeZ dId2d3Z!e!Z"dId4d5Z#dId6d7Z$dId8d9Z%e%Z&dId:d;Z'dIdd?Z*dId@dAZ+dIdBdCZ,dIdDdEZ-e-Z.dIdFdGZ/e,Z0e/Z1e/Z2dS)JMySQLConverteraDefault 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). NTcCst|||i|_dSN)r rrrrrrr~s zMySQLConverter.__init__cCs|dur|St|tr |St|ttfr:|dd}|dd}|dd}|dd }|d d }|d d }|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. N\s\\ s\n s\r's\'"s\"s\\z\\ z\n z\r'z\'"z\"z\) isinstancerbytes bytearrayreplacerrrrr5s&            zMySQLConverter.escapecCs\t|trtrt|trt|St|St|dSt|tdr&tdStd|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 '' Returns a bytearray object. asciiNsNULLrF) rOrrfloatreprr9encodetyperQr:rrrr<s  zMySQLConverter.quotecCs@|jj}z t|d||WStytd|w)rrz0Python '{0}' cannot be converted to a MySQL type)r r!r"r#r$r% TypeErrorr&rrrr(s  zMySQLConverter.to_mysqlc Cs"|dkr |dtjkr dS|durdS|js;i|_tjD]\}}zt|d||j|d<Wqty:Yqwz |j|d||WStycz| dWYSt yb|YYSwt yx}z t d||dfd}~wt y}z t d||dfd}~w)r)rrNr+utf-8z %s (field %s)) rr,rr-r.r#r$r%r/decodeUnicodeDecodeError ValueErrorrX)rr0rr1r2errrrrr3s<     zMySQLConverter.to_pythoncCr7zConvert value to intintrrrr _int_to_mysqlr=zMySQLConverter._int_to_mysqlcCr7r^r_rrrr_long_to_mysqlr=zMySQLConverter._long_to_mysqlcCr7)zConvert value to floatrTrrrr_float_to_mysqlr=zMySQLConverter._float_to_mysqlcCstrt|S||S)zConvert value to string)rr9_unicode_to_mysqlrrrr _str_to_mysqls zMySQLConverter._str_to_mysqlcCsP|j}|j}|dkrd}t|d}||}|tjvr&d|vr&t||S|S)zConvert unicodebinaryr rrC)rrrrrVslash_charsetsr )rrrrencodedrrrres   z MySQLConverter._unicode_to_mysqlcCr4zConvert value to bytesrrrrr_bytes_to_mysqlr6zMySQLConverter._bytes_to_mysqlcCr7rj)rPrrrr_bytearray_to_mysqlr=z"MySQLConverter._bytearray_to_mysqlcCs |rdSdS)zConvert value to booleanrrrrrrr_bool_to_mysqlszMySQLConverter._bool_to_mysqlcCsdS)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. Nrrrrr_nonetype_to_mysqlsz!MySQLConverter._nonetype_to_mysqlc Cs^|jrd}||j|j|j|j|j|j|jdSd}||j|j|j|j|j|jdS)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}rSz-{0:d}-{1:02d}-{2:02d} {3:02d}:{4:02d}:{5:02d}) microsecondr$yearmonthdayhourminutesecondrV)rrfmtrrr_datetime_to_mysqls      z!MySQLConverter._datetime_to_mysqlcCsd|j|j|jdS)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}rS)r$rprqrrrVrrrr_date_to_mysql's zMySQLConverter._date_to_mysqlcCs&|jr |ddS|ddS)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.%frSz%H:%M:%S)rostrftimerVrrrr_time_to_mysql3s zMySQLConverter._time_to_mysqlcCstd|dS)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:%SrS)timeryrVrrrr_struct_time_to_mysql@sz$MySQLConverter._struct_time_to_mysqlc Cst|jd|j}|jr"d}|jdkrd|j}|d8}n|j}nd}|jdkr-d|}t|d\}}t|d \}}|jrG|||||} n||||} trR| 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. iQz{0:02d}:{1:02d}:{2:02d}.{3:06d}ri@Brz{0:02d}:{1:02d}:{2:02d}-i<rS)absdaysseconds microsecondsdivmodr$rrV) rrrrvmcshours remainderminssecsresultrrr_timedelta_to_mysqlJs$     z"MySQLConverter._timedelta_to_mysqlcCst|tr t|dSdS)z Converts a decimal.Decimal instance to a string suitable for MySQL. Returns a bytes or None when not valid. rSN)rOrr9rVrrrr_decimal_to_mysqlls z MySQLConverter._decimal_to_mysqlc CsHd}dgt|}|js1i|_tjD]\}}zt|d||j|d<Wqty0Yqw|D]l}|d}||dkrD|tjksJ||durO|d7}q3z|j||||||<Wn=t yz || d||<Wnt y~||||<YnwYnt t fy} z dt| |d| _d} ~ ww|d7}q3t|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. rNr+rrYz{0} (field {1}))lenrrr-r.r#r$r%r,r/rZr[r\rXr9messagetuple) rrowfieldsirr1r2field field_typer]rrr row_to_pythonxsB   "   zMySQLConverter.row_to_pythoncCr7)z. Returns value as float type. rcrrr-rrr_FLOAT_to_pythonzMySQLConverter._FLOAT_to_pythoncCr7)z, Returns value as int type. r_rrrr_INT_to_pythonrzMySQLConverter._INT_to_pythoncCs||j}t|S)z5 Returns value as a decimal.Decimal. )rZrr)rrr-valrrr_DECIMAL_to_pythons z!MySQLConverter._DECIMAL_to_pythoncCr7)z, Returns value as str type. r8rrrr_strrzMySQLConverter._strcCs2|}t|dkrddt||}td|dS)z!Returns BIT columntype as integerr*z>Qr)rr )rrdscint_valrrr_BIT_to_pythons zMySQLConverter._BIT_to_pythoncCsXt|tjr|Sz|d}tt|dt|dt|dWSty+YdSw)zA Returns DATE column type as datetime.date type. -rrN)rOdatetimedatesplitr`r\)rrrpartsrrr_DATE_to_pythons  ( zMySQLConverter._DATE_to_pythonc Csd}z|d\}}t|dd}Wn ty |}d}Ynwz0dd|dD\}}}|dd ks;|dd krF| | | }}}tj||||d }W|Sty_td |w) zA Returns TIME column type as datetime.time type. N.0rcSg|]}t|qSrr_).0drrr z2MySQLConverter._TIME_to_python..:-r})rminutesrrz2Could not convert {0} to python datetime.timedelta)rr`ljustr\r timedeltar$) rrrtime_valhmsrrrrrrr_TIME_to_pythons.  zMySQLConverter._TIME_to_pythonc Cst|tjr|Sd}z@|d\}}t|dkr(|d\}}t|dd}n|}d}dd |d Dd d |d D|g}tj|}W|StyVd}Y|Sw) zI Returns DATETIME column type as datetime.datetime type. N rrrrrcSrrr_rrrrrrrz6MySQLConverter._DATETIME_to_python..rcSrrr_rrrrrrr)rOrrrr`rr\) rrr datetime_valdate_time_rrdtvalrrr_DATETIME_to_pythons*    z"MySQLConverter._DATETIME_to_pythoncCs*zt|}W|Stytd|w)z#Returns YEAR column type as integerz"Failed converting YEAR to int (%s))r`r\)rrr-rprrr_YEAR_to_pythons   zMySQLConverter._YEAR_to_pythoncCsJd}||j}|s tSz t|d}W|Sty$td|w)zReturns 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.)rZrsetrr\)rrrset_typerrrr_SET_to_pythons  zMySQLConverter._SET_to_pythonc Csdzt|}|rt|WS|WStyYnw|dkr dS|dkr&dS|ddkrs|ddkrs|dd}||}|d urC|S||}|d urN|Sz||}|d ur[|WSWn tyeYnwt|tt frs| |j S|d ur|d t j @r|||S|d t j@r|j d krt|tsz| |j WSttfy|YSw|S|||S) zbReturns JSON column type as python type Returns JSON column type as python type. strueTsfalseFr"rNrg)rT is_integerr`r\rrrrOrPrQrZrrSETrBINARYr9 LookupErrorr[_STRING_to_python)rrrnumvalue_nqvalue_datetime value_date value_timerrr_JSON_to_python*sT          zMySQLConverter._JSON_to_pythonc Cs|dur9|dtj@r|||S|dtj@r9|jdkr7t|ts7z||jWStt fy6|YSw|S|jdkr@|St|t t frP|j rP||jS|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. Nrrg) rrrrrrOr9rZrr[rPrQrrrrrrrras    z MySQLConverter._STRING_to_pythoncCsH|dur|dtj@rtr|St|trt||jSt|S|||S)z Convert BLOB data type to PythonNr)rrrrOr9rPrrrrrr_BLOB_to_python~s   zMySQLConverter._BLOB_to_python)NTrB)3r!r>r?r@rr5r<r(r3rarbrdrfrerkrlrmrnrwrxrzr|rrrr_DOUBLE_to_pythonr_TINY_to_python_SHORT_to_python_INT24_to_python_LONG_to_python_LONGLONG_to_pythonr_NEWDECIMAL_to_pythonrrr_NEWDATE_to_pythonrr_TIMESTAMP_to_pythonrrrr_VAR_STRING_to_pythonr_LONG_BLOB_to_python_MEDIUM_BLOB_to_python_TINY_BLOB_to_pythonrrrrrApsb  !     " .          7  rA)r@rr{decimalr constantsrrrcatch23rrr custom_typesr objectr rArrrrs  G