3 (h{U@s|dZddlZddlZddlmZddlmZmZmZddl m Z m Z m Z ddl mZdZGd d d eZGd d d eZdS) z"Converting MySQL and Python types N)Decimal) FieldType FieldFlag CharacterSet)PY2 NUMERIC_TYPES struct_unpack) HexLiteralz.Could not convert '{value}' to python {pytype}c@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|_|j||j|i|_dS)Nr)Z python_typesZ mysql_typescharset charset_id use_unicode set_charset set_unicode_cache_field_types)selfr rrN/tmp/pip-install-q3hcpn_q/mysql-connector-python/mysql/connector/conversion.py__init__2s  zMySQLConverterBase.__init__cCs8|dkr d}|dk r||_nd|_tj|jd|_dS)zSet character setZutf8mb4r Nr)r rget_charset_infor)rr rrrr<s zMySQLConverterBase.set_charsetcCs ||_dS)zSet whether to use UnicodeN)r)rvaluerrrrGszMySQLConverterBase.set_unicodec Cs:|jjj}yt|dj||Stk r4|SXdS)z!Convert Python data type to MySQLz _{0}_to_mysqlN) __class____name__lowergetattrformatAttributeError)rr type_namerrrto_mysqlKs  zMySQLConverterBase.to_mysqlc Cs|dks|dkr"|dtjkr"dS|jszi|_xJtjjD]<\}}yt|dj||j|d<Wq:tk rtYq:Xq:Wy|j|d||Stk r|SXdS)z!Convert MySQL data type to PythonNrz_{0}_to_pythonr) rBITrdescitemsrrrKeyError)rvtypernameinforrr to_pythonSs zMySQLConverterBase.to_pythoncCs|S)z"Escape buffer for sending to MySQLr)rrrrrescapeiszMySQLConverterBase.escapecCst|S)z!Quote buffer for sending to MySQL)str)rbufrrrquotemszMySQLConverterBase.quoteN)r T)T) r __module__ __qualname____doc__rrrr r)r*r-rrrrr +s  r c@sjeZdZdZdFddZddZdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-ZdGd.d/ZeZdHd0d1ZeZeZeZeZeZ dId2d3Z!e!Z"dJd4d5Z#dKd6d7Z$dLd8d9Z%e%Z&dMd:d;Z'dNdd?Z*dPd@dAZ+dQdBdCZ,e,Z-e,Z.dRdDdEZ/e/Z0e/Z1e/Z2dS)SMySQLConverteraDefault 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). NTcCstj|||i|_dS)N)r rr)rr rrrrrszMySQLConverter.__init__cCs|dkr |St|tr|St|ttfrr|jdd}|jdd}|jdd}|jdd }|jd d }|jd d }nH|jdd}|jdd}|jdd}|jdd}|jdd}|jdd}|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 bytearrayreplace)rrrrrr*s$            zMySQLConverter.escapecCs\t|tr6tr(t|tr t|St|St|jdSt|tdrLtdStd|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. asciiNsNULLr5) r>rrfloatreprr+encodetyper@)rr,rrrr-s  zMySQLConverter.quotec CsH|jjj}yt|dj||Stk rBtdj|YnXdS)z!Convert Python data type to MySQLz _{0}_to_mysqlz0Python '{0}' cannot be converted to a MySQL typeN)rrrrrr TypeError)rrrrrrr s  zMySQLConverter.to_mysqlcCsD|dkr|dtjkrdS|dkr&dS|js~i|_xJtjjD]<\}}yt|dj||j|d<Wq>tk rxYq>Xq>Wy|j|d||Stk ry |j dSt k r|SXYnzt k r}zt d||dfWYdd}~XnDt k r2}zt d||dfWYdd}~XnYnXdS)z!Convert MySQL data type to PythonrrNz_{0}_to_pythonzutf-8z %s (field %s)) rr"rr#r$rrrr%decodeUnicodeDecodeError ValueErrorrG)rr&rr'r(errrrrr)s2   $$zMySQLConverter.to_pythoncCst|S)zConvert value to int)int)rrrrr _int_to_mysqlszMySQLConverter._int_to_mysqlcCst|S)zConvert value to int)rL)rrrrr_long_to_mysqlszMySQLConverter._long_to_mysqlcCst|S)zConvert value to float)rC)rrrrr_float_to_mysqlszMySQLConverter._float_to_mysqlcCstr t|S|j|S)zConvert value to string)rr+_unicode_to_mysql)rrrrr _str_to_mysqlszMySQLConverter._str_to_mysqlcCsP|j}|j}|dkr&d}tj|d}|j|}|tjkrLd|krLt||S|S)zConvert unicodebinaryr rr2)r rrrrEZslash_charsetsr )rrr rencodedrrrrPs   z MySQLConverter._unicode_to_mysqlcCs|S)zConvert value to bytesr)rrrrr_bytes_to_mysqlszMySQLConverter._bytes_to_mysqlcCst|S)zConvert value to bytes)r?)rrrrr_bytearray_to_mysqlsz"MySQLConverter._bytearray_to_mysqlcCs |rdSdS)zConvert value to booleanrrr)rrrrr_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. Nr)rrrrr_nonetype_to_mysql sz!MySQLConverter._nonetype_to_mysqlcCs^|jr4d}|j|j|j|j|j|j|j|jjdSd}|j|j|j|j|j|j|jjdS)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}rBz/{0:04d}-{1:02d}-{2:02d} {3:02d}:{4:02d}:{5:02d}) microsecondryearmonthdayhourminutesecondrE)rrfmtrrr_datetime_to_mysqls    z!MySQLConverter._datetime_to_mysqlcCsdj|j|j|jjdS)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:04d}-{1:02d}-{2:02d}rB)rrYrZr[rE)rrrrr_date_to_mysql)s zMySQLConverter._date_to_mysqlcCs&|jr|jdjdS|jdjdS)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.%frBz%H:%M:%S)rXstrftimerE)rrrrr_time_to_mysql5s zMySQLConverter._time_to_mysqlcCstjd|jdS)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:%SrB)timerbrE)rrrrr_struct_time_to_mysqlBsz$MySQLConverter._struct_time_to_mysqlc Cst|jd|j}|jrDd}|jdkrrr+rE)rrrrr_decimal_to_mysqlns z MySQLConverter._decimal_to_mysqlc Cshd}dgt|}|jsji|_xJtjjD]<\}}yt|dj||j|d<Wq*tk rdYq*Xq*Wx|D]}|d}||dkr|tjks||dkr|d7}qpy|j||||||<Wnt k ry||j d||<Wn"t k r||||<YnXYn@t t fk rR} zdjt| |d| _WYdd} ~ XnX|d7}qpWt|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. rNz_{0}_to_pythonrzutf-8z{0} (field {1}))lenrrr#r$rrrr"r%rHrIrJrGr+messagetuple) rrowfieldsirrr'r(fieldZ field_typerKrrr row_to_pythonzs6   " zMySQLConverter.row_to_pythoncCst|S)z. Returns value as float type. )rC)rrr#rrr_FLOAT_to_pythonszMySQLConverter._FLOAT_to_pythoncCst|S)z, Returns value as int type. )rL)rrr#rrr_INT_to_pythonszMySQLConverter._INT_to_pythoncCs|j|j}t|S)z5 Returns value as a decimal.Decimal. )rHr r)rrr#valrrr_DECIMAL_to_pythons z!MySQLConverter._DECIMAL_to_pythoncCst|S)z, Returns value as str type. )r+)rrr#rrr_strszMySQLConverter._strcCs2|}t|dkr$ddt||}td|dS)z!Returns BIT columntype as integerr!z>Qr)rur )rrdscZint_valrrr_BIT_to_pythons zMySQLConverter._BIT_to_pythoncCst|tjr|Syj|jd}t|dkrdatetimedatesplitrurJrrL IndexError)rrrpartsrrr_DATE_to_pythons   & zMySQLConverter._DATE_to_pythoncCsy"|jd\}}t|jdd}Wn ttfk rB|}d}YnXyZdd|jdD\}}}|ddksx|dd kr| | | }}}tj||||d Stttfk rttj |d d YnXd S)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.time type. .0rcSsg|] }t|qSr)rL).0drrr sz2MySQLConverter._TIME_to_python..:-rf)rnminutesrjrkzdatetime.timedelta)rpytypeN) rrLljustrGrJr timedeltar CONVERT_ERRORr)rrrhmsrmrnrprqrrr_TIME_to_pythons   zMySQLConverter._TIME_to_pythonc Cst|tjr|Sd}y|jd\}}t|dkrP|jd\}}t|jdd}n|}d}dd |jd Dd d |jd D|g}t|dkrtd j|t|n$ytj|}Wntk rdSXWn*tt fk rtt j|ddYnX|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 rrrrrcSsg|] }t|qSr)rL)rrzrrrrsz6MySQLConverter._DATETIME_to_python..rcSsg|] }t|qSr)rL)rrzrrrrsrz#invalid datetime format: {} len: {}zdatetime.timedelta)rr) r>rrrurLrrJrrrGr) rrrZ datetime_valZdate_Ztime_rrmZdtvalrrr_DATETIME_to_pythons,  .  z"MySQLConverter._DATETIME_to_pythonc Cs2y t|}Wn tk r,td|YnX|S)z#Returns YEAR column type as integerz"Failed converting YEAR to int (%s))rLrJ)rrr#rYrrr_YEAR_to_python.s  zMySQLConverter._YEAR_to_pythonc CsRd}|j|j}|stSyt|jd}Wn tk rLtd|YnX|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,z'Could not convert set %s to a sequence.)rHr setrrJ)rrrset_typerrrr_SET_to_python7s zMySQLConverter._SET_to_pythonc Cs|dk rr|dtj@r"|j||S|dtj@rr|jdkrnt|t rny |j|jStt fk rj|SXn|S|jdkr|St|t t fr|j r|j|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. NrR) rZSETrBINARYr r>r+rH LookupErrorrIr?r@r)rrrrrr_STRING_to_pythonIs    z MySQLConverter._STRING_to_pythoncCs8|dk r,|dtj@r,|dtj@r,t|S|j||S)z!Convert BLOB data type to Python.Nr)rZBLOBrr?r)rrrrrr_BLOB_to_pythongszMySQLConverter._BLOB_to_python)NT)N)N)N)N)N)N)N)N)N)N)N)N)3rr.r/r0rr*r-r r)rMrNrOrQrPrTrUrVrWr`rarcrersrtr|r}Z_DOUBLE_to_pythonr~Z_TINY_to_pythonZ_SHORT_to_pythonZ_INT24_to_pythonZ_LONG_to_pythonZ_LONGLONG_to_pythonrZ_NEWDECIMAL_to_pythonrrrZ_NEWDATE_to_pythonrrZ_TIMESTAMP_to_pythonrrrZ_VAR_STRING_to_pythonZ_JSON_to_pythonrZ_LONG_BLOB_to_pythonZ_MEDIUM_BLOB_to_pythonZ_TINY_BLOB_to_pythonrrrrr1rs`   !     " .        '   r1)r0rrddecimalr constantsrrrZcatch23rrr Z custom_typesr robjectr r1rrrrs  G