U ea@sdZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z m Z m Z mZmZddlmZmZmZddlmZddlmZmZmZmZmZmZdd lmZd ZGd d d Z Gd dde Z!dS)z"Converting MySQL and Python types N)Decimal)AnyCallableDictListOptionalSetTupleUnion) CharacterSet FieldFlag FieldType) HexLiteral)DescriptionTypeRowType StrOrBytesToMysqlInputTypesToMysqlOutputTypesToPythonOutputTypes) NUMERIC_TYPESz.Could not convert '{value}' to python {pytype}c@seZdZdZdeeeeddddZeedd d d Zdedd d dZ e e e e fd ddZ eeeedddZeeed ddZeeedddZdS)MySQLConverterBasezBase class for conversion classes All class dealing with converting to and from MySQL data types must be a subclass of this class. utf8TFNcharset use_unicode str_fallbackreturncCs8d|_d|_d|_d|_||||_||_i|_dS)Nr)Z python_typesZ mysql_typesr charset_id set_charsetrr_cache_field_typesselfrrrr#[/var/www/html/assets/scripts/venv/lib/python3.8/site-packages/mysql/connector/conversion.py__init__>s zMySQLConverterBase.__init__)rrcCs8|dkr d}|dk r||_nd|_t|jd|_dS)zSet character set)Zutf8mb4Zutf8mb3rNr)rr get_charset_infor)r"rr#r#r$rPs zMySQLConverterBase.set_charsetvaluercCs ||_dS)zSet whether to use UnicodeN)rr"r(r#r#r$ set_unicode[szMySQLConverterBase.set_unicodecCsF|jj}zt|d|d|}|WStk r@|YSXdS)!Convert Python data type to MySQL_ _to_mysqlN) __class____name__lowergetattrAttributeErrorr"r( type_nameZ convertedr#r#r$to_mysql_s zMySQLConverterBase.to_mysqlvtyper(rc Cs|dks|dkr"|dtjkr"dS|js|i|_tjD]B\}}z$t|d|d|j|d<Wq8tk rxYq8Xq8|dkrdSz|j|d||WStk r|YSXdS)!Convert MySQL data type to PythonNr r, _to_pythonr) rBITr descitemsr1r0r2KeyError)r"r7r(nameinfor#r#r$ to_pythonls$zMySQLConverterBase.to_pythoncCs|S)z"Escape buffer for sending to MySQLr#r(r#r#r$escapeszMySQLConverterBase.escapebufrcCst|S)z!Quote buffer for sending to MySQLstrrEr#r#r$quoteszMySQLConverterBase.quote)rTF)T)r/ __module__ __qualname____doc__rrGboolr%rr*rr rr5rbytesrrA staticmethodrrCrrIr#r#r#r$r7s.    rc@seZdZdZdOeeeeddddZee e dd d Z eee e e eeefed d d ZeedddZeeeedddZee e dddZee e dddZee ee dddZee eefdddZee eefdddZeeedddZeeeddd Zeee dd!d"Z edddd#d$Z!ee"j"edd%d&Z#ee"j$edd'd(Z%ee"j&edd)d*Z'ee&j(edd+d,Z)ee"j*edd-d.Z+eeeedd/d0Z,e-ed1fe.ee/d2d3d4Z0edPeeee d5d6d7Z1e1Z2edQeeee d5d8d9Z3e3Z4e3Z5e3Z6e3Z7e3Z8dReeeed5d:d;Z9e9Z:edSeeeed5dd?d@ZdAdBZ=e=Z>edVeeee"j*d>dCdDZ?edWeeeee"j"d>dEdFZ@e@ZAedXeeee d>dGdHZBdYeeeeCed>dIdJZDdZeeee eEeCefd>dKdLZFeFZGeFZHd[eeee eEeCefd>dMdNZIeIZJeIZKeIZLdS)\MySQLConverteraDefault 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). NTFrcCst||||i|_dS)N)rr%r r!r#r#r$r%szMySQLConverter.__init__r'cCst|ttfrX|dd}|dd}|dd}|dd}|d d }|d d }n\t|trt|ts|d d}|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. \s\\ s\n s\r's\'"s\"s\\z\\ z\n z\r'z\'"z\"z\) isinstancerN bytearrayreplacerGrrBr#r#r$rCs            zMySQLConverter.escaperDcCs>t|trt|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. asciiNsNULLrT)r]rrGencodetyper^rHr#r#r$rIs zMySQLConverter.quotecCsj|jj}zt|d|d|}|WStk rd|jrNt|YStd|ddYnXdS)r+r,r-zPython 'z%' cannot be converted to a MySQL typeN) r.r/r0r1r2rrGra TypeErrorr3r#r#r$r5s  zMySQLConverter.to_mysqlr6c CsV|dkr|dtjkrdS|dkr&dS|jsi|_tjD]B\}}z$t|d|d|j|d<Wq<tk r|Yqt k rP}zt |d|dd|W5d}~XYnXdS) r8rr Nr,r:utf-8 (field )) rr;r r<r=r1r0r2r>decodeUnicodeDecodeError ValueErrorrc)r"r7r(r?r@errr#r#r$rAs2*zMySQLConverter.to_pythoncCst|S)zConvert value to intintrBr#r#r$ _int_to_mysqlszMySQLConverter._int_to_mysqlcCst|S)zConvert value to int Note: there is not type "long" in Python 3 since integers (int) are of unlimited size. Since Python 2 is no longer supported, this method should be deprecated. rkrBr#r#r$_long_to_mysql szMySQLConverter._long_to_mysqlcCst|rdSt|S)zConvert value to floatN)mathisnanfloatrBr#r#r$_float_to_mysqls zMySQLConverter._float_to_mysqlcCs ||S)zConvert value to string)_unicode_to_mysqlr)r#r#r$ _str_to_mysqlszMySQLConverter._str_to_mysqlcCsP|j}|j}|dkr&d}t|d}||}|tjkrLd|krLt||S|S)zConvert unicodebinaryrrrQ)rrr r&raZslash_charsetsr)r"r(rrencodedr#r#r$rs!s   z MySQLConverter._unicode_to_mysqlcCs|SzConvert value to bytesr#rBr#r#r$_bytes_to_mysql.szMySQLConverter._bytes_to_mysqlcCst|Srw)rNrBr#r#r$_bytearray_to_mysql3sz"MySQLConverter._bytearray_to_mysqlcCs |rdSdS)zConvert value to booleanr rr#rBr#r#r$_bool_to_mysql8szMySQLConverter._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. Nr#rBr#r#r$_nonetype_to_mysql=s z!MySQLConverter._nonetype_to_mysqlc Cs^|jr4d}||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. z7{0:04d}-{1:02d}-{2:02d} {3:02d}:{4:02d}:{5:02d}.{6:06d}r`z/{0:04d}-{1:02d}-{2:02d} {3:02d}:{4:02d}:{5:02d}) microsecondformatyearmonthdayhourminutesecondra)r(fmtr#r#r$_datetime_to_mysqlHs0  z!MySQLConverter._datetime_to_mysqlcCs&|jdd|jdd|jddS)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-02dr`)r~rrrarBr#r#r$_date_to_mysqlhs 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.%fr`z%H:%M:%S)r|strftimerarBr#r#r$_time_to_mysqlts 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:%Sr`)timerrarBr#r#r$_struct_time_to_mysqls z$MySQLConverter._struct_time_to_mysqlc Cst|jd|j}|jrDd}|jdkrrgrhrircmessagetuple) r"rrirr?r@fieldZ field_typerjr#r#r$ row_to_pythons: " zMySQLConverter.row_to_python)r(r<rcCst|S)z. Returns value as float type. )rqr(r<r#r#r$_float_to_pythonszMySQLConverter._float_to_pythoncCst|S)z, Returns value as int type. rkrr#r#r$_int_to_pythonszMySQLConverter._int_to_pythoncCs||j}t|S)z5 Returns value as a decimal.Decimal. )rgrr)r"r(r<valr#r#r$_decimal_to_pythons z!MySQLConverter._decimal_to_pythoncCst|S)z, Returns value as str type. rFrr#r#r$_strszMySQLConverter._str)r(dscrcCs8|}t|dkr$ddt||}ttd|dS)z!Returns BIT columntype as integerr9z>Qr)rrlstructunpack)r(rZint_valr#r#r$_bit_to_pythons zMySQLConverter._bit_to_pythonc Cst|tjr|Szv|d}t|dkr@td|dt|z*tt|dt|dt|dWWStk rYWdSXWn.ttfk rtd t|d dYnXdS) zConverts 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: rr NzCould not convert z to python datetime.timedelta) r]datetimedatesplitrrirl IndexErrorrepr)r(rpartsr#r#r$_date_to_pythons   *zMySQLConverter._date_to_pythonc Csd}z"|d\}}t|dd}Wn ttfk rF|}d}YnXz\dd|dD\}}}|dd ks||dd kr| | | }}}tj||||d WStttfk rttj |d d dYnXdS)aConverts 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.0rcSsg|] }t|qSr#rk).0dr#r#r$ Nsz2MySQLConverter._time_to_python..:-r)rminutesrrdatetime.timedeltar(Zpytype) rrlljustrcrir timedeltar CONVERT_ERRORr})r(rrhmsrrrr#r#r$_time_to_python9s4    zMySQLConverter._time_to_pythonc Cs t|tjr|Sd}d}z|d\}}t|dkrT|d\}}t|dd}n|}d}dd |d Dd d |d D|g}t|dkrtd |dt|ztj|}Wntk rYWdSXWn.ttfk rtt j |dddYnX|S)zConverts 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 rrrrrcSsg|] }t|qSr#rkrrr#r#r$rusz6MySQLConverter._datetime_to_python..rcSsg|] }t|qSr#rkrr#r#r$rvsrrrrr) r]rrrrlrrirrcrr})r(rZ datetime_valrZdate_Ztime_rZdtvalr#r#r$_datetime_to_python]s>    z"MySQLConverter._datetime_to_pythonc CsJz t|}Wn8tk rD}ztdt|d|W5d}~XYnX|S)z#Returns YEAR column type as integerzFailed converting YEAR to int (rfN)rlrir)r(rr~rjr#r#r$_year_to_pythons  (zMySQLConverter._year_to_pythonc Csjd}||j}|stSzt|d}Wn8tk rd}ztdt|d|W5d}~XYnX|S)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,zCould not convert set z to a sequence)rgrsetrrir)r"r(rset_typerrjr#r#r$_set_to_pythons  zMySQLConverter._set_to_pythoncCs|jdkr|S|dk r`|dtjkr6|jr6||jS|dtj@rP|||S|ddkr`|St|t t fr|jrz||jWSt k r|YSX|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. ruNr r?) rrJSONrrgr ZSETrr]rNr^rhr"r(rr#r#r$_string_to_pythons     z MySQLConverter._string_to_pythoncCsD|dk r8|dtj@r8|dtj@r8|ddkr8t|S|||S)z!Convert BLOB data type to Python.Nrrr)r ZBLOBBINARYrNrrr#r#r$_blob_to_pythons   zMySQLConverter._blob_to_python)NTF)N)N)N)N)N)N)N)N)N)N)N)N)Mr/rJrKrLrrGrMr%rOrrCr rqrlrrrNrIrrr5rrrArmrnrrrtrsrxr^ryrzr{rrrrrr struct_timerrrrr rrrrZ_double_to_pythonrZ_tiny_to_pythonZ_short_to_pythonZ_int24_to_pythonZ_long_to_pythonZ_longlong_to_pythonrZ_newdecimal_to_pythonrrrZ_NEWDATE_to_pythonrrZ_timestamp_to_pythonrrrrrZ_var_string_to_pythonZ_json_to_pythonrZ_long_blob_to_pythonZ_medium_blob_to_pythonZ_tiny_blob_to_pythonr#r#r#r$rPs $ %        3 #.    rP)"rLrrorrdecimalrtypingrrrrrrr r constantsr r rZ custom_typesrtypesrrrrrrutilsrrrrPr#r#r#r$s (   Z