3 K^(m@sdZddlZddlZddlmZddlmZmZmZm Z m Z ddl m Z m Z ddlmZddlmZmZdd l mZGd d d eZdS) z,Implements the MySQL Client/Server protocol N)Decimal) FieldFlag ServerCmd FieldType ClientFlagMAX_MYSQL_TABLE_COLUMNS)errorsutils)get_auth_plugin)PY2 struct_unpack) get_exceptionc@seZdZdZddZddZdAd d ZdBd dZdCddZdDddZ ddZ ddZ ddZ dEddZ ddZdFdd ZdGd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-ZdHd.d/Zd0d1Zd2d3Zd4d5Zd6d7Zd8d9Zffddd:fd;d<Zd=d>Zd?d@ZdS)I MySQLProtocolzRImplements MySQL client/server protocol Create and parses MySQL packets. cCs |tj@r|r|jddSdS)z.Prepare database string for handshake responseutf8)rZCONNECT_WITH_DBencode)self client_flagsdatabaser_/tmp/pip-unpacked-wheel-jty9dxb9/mysql_connector-2.2.9.data/purelib/mysql/connector/protocol.py_connect_with_db-szMySQLProtocol._connect_with_dbc Cs|sdSy"t||||||d}|j} Wn<ttjfk rf} ztjdjt| WYdd} ~ XnX|tj @rt | } t j d| | } n| d} | S)z#Prepare the authentication responser)usernamepasswordr ssl_enabledzFailed authentication: {0}Nz%sz2MySQLProtocol.parse_statistics..:rczutf-8z{0} ({1}:{2}).)splitr rrKintrr ) rr8Z with_headererrmsgrNpairspairZlblvalrrrparse_statisticss(    zMySQLProtocol.parse_statisticsrc Cs^g}d}d}d}|d k}x:|s(||kr*P|j} | jdr| ddg} |j} x(| jdrz| j| dd|j} qTW| j| ddtjtdj| }nj| ddkr| ddkr|j| }d}nB|r| ddkr| dd kr|j| }d}nd}tj| dd}|dkr.|dk r.|j|n|dkrJ|dkrJt | |d 7}qW||fS) zRead MySQL text result Reads all or given number of rows from the socket. Returns a tuple with 2 elements: a list with all rows and the EOF packet. NrrBsrQr-rarbr)rBrtrB) recvrMappendr Zread_lc_string_list bytearrayjoinrgrWr) rsockversionrYrowseofZrowdataiZeof57r8Zdatasrrrread_text_result5s>        zMySQLProtocol.read_text_resultcCs|dtjkrd}d}nL|dtjkr0d}d}n4|dtjtjfkrNd}d}n|dtjkrdd}d}|d tj@rz|j}||d t ||d |d fS) z%Parse an integer from a binary packetrbhrcr}rQqrGrtNr) rTINYSHORTINT24LONGLONGLONGrZUNSIGNEDupperr )rr8fieldformat_lengthrrr_parse_binary_integer_sz#MySQLProtocol._parse_binary_integercCsB|dtjkrd}d}nd}d}||dt||d|dfS)z)Parse a float/double from a binary packetrrGdrQfNr)rDOUBLEr )rr8rrrrrr_parse_binary_floatss z!MySQLProtocol._parse_binary_floatc Cs|d}d}|dkr@tjtd|ddd|d|dd}nl|dkrd}|d krntd |d |dd}tjtd|ddd|d|d|d |d |d|d}||dd|fS)z&Parse a timestamp from a binary packetrNrQHrrd)yearmonthdayrt IrGrB)rrrhourminutesecond microsecond)datetimedater )rr8rrvaluemcsrrr_parse_binary_timestamp~s(z%MySQLProtocol._parse_binary_timestampcCs|d}|d|d}d}|dkr:td|ddd}td|ddd}|ddkrd|d 9}tj||d||d|dd }||dd|fS) z'Parse a time value from a binary packetrrrGrNrBrtr)daysseconds microsecondsminuteshoursrI)r r timedelta)rr8rrr?rrtmprrr_parse_binary_times  z MySQLProtocol._parse_binary_timec Cstt|ddd}dd|d|D}||d}g}x.t|D] \}}|t|ddd|dd>@r|jdqFqF|dtjtjtjtjtj fkr|j ||\}}|j|qF|dtj tj fkr|j ||\}}|j|qF|dtjtjtjfkr$|j||\}}|j|qF|dtjkrP|j||\}}|j|qFtj|\}}|j|qFWt|S) z(Parse values from a binary result packetrtrcrGcSsg|] }t|qSr)rn)rir}rrrrksz6MySQLProtocol._parse_binary_values..rNr)r# enumeraternrvrrrrrrrrFLOATrDATETIMEDATEZ TIMESTAMPrTIMErr rUtuple) rfieldsr8Znull_bitmap_length null_bitmapvaluesposrrrrr_parse_binary_valuess4 $      z"MySQLProtocol._parse_binary_valuesc Csg}d}d}d}x|dk rP||kr&P|j}|ddkrJ|j|}d}n$|ddkrnd}|j||dd}|dkr|dk r|j|n|dkr|dkrt||d7}qW||fS)zwRead MySQL binary protocol result Reads all or given number of binary resultset rows from the socket. NrrQrarBr)rurgrrvr) rrycolumnsrYr{r|rr}r8rrrread_binary_results,     z MySQLProtocol.read_binary_resultc Cs|ddkstjdi}yhtj|ddd\}|d<tj|d\}|d<tj|d\}|d <|d d}tj|d\}|d <Wntk rtjdYnX|S) z'Parse a MySQL Binary Protocol OK packetrQrzFailed parsing Binary OK packetrBN statement_idrcZ num_columnsZ num_paramsrrS)r rr Zread_intrV)rr8Zok_pktrrrparse_binary_prepare_oks   z%MySQLProtocol.parse_binary_prepare_okcCsd}d}|dkrX|dkr$d}tj}q|dkr8d}tj}q|dkrLd}tj}qd }tj}nJd}|d krpd }tj}n2|d krd }tj}n|dkrd}tj}n tj}d}tj||||fS)z0Prepare an integer for the MySQL binary protocolNrrirlr}rBirlrQiii)rrrrrr$r%)rrr^r_rrrr_prepare_binary_integers6z%MySQLProtocol._prepare_binary_integercCst|tjrtj}nt|tjr(tj}ntdtj|j tj |j tj |j }t|tjr|tj |j tj |jtj |j}|jdkr|tj|j7}tj t||}||fS)aPrepare a timestamp object for the MySQL binary protocol This method prepares a timestamp of type datetime.datetime or datetime.date for sending over the MySQL binary protocol. A tuple is returned with the prepared value and field type as elements. Raises ValueError when the argument value is of invalid type. Returns a tuple. z2Argument must a datetime.datetime or datetime.dater) isinstancerrrrrrVr int2storerr;rrrrrrr:r#)rrr^packedrrr_prepare_binary_timestamps    z'MySQLProtocol._prepare_binary_timestampc Cst|tjtjfstdtj}d}d}d}t|tjr|jdkrFd}t|j d\}}t|d\}} |t j t |jt j |t j |t j | 7}|j}n8|t j dt j |jt j |jt j |j7}|j}|r|t j |7}t j ||}t j t||}||fS)aPrepare a time object for the MySQL binary protocol This method prepares a time object of type datetime.timedelta or datetime.time for sending over the MySQL binary protocol. A tuple is returned with the prepared value and field type as elements. Raises ValueError when the argument value is of invalid type. Returns a tuple. z3Argument must a datetime.timedelta or datetime.timerNr-ri<)rrrtimerVrrrdivmodrr r:absr;rrrrrr#) rrr^negativerrr remainderZminsZsecsrrr_prepare_binary_time8s2   " z"MySQLProtocol._prepare_binary_timecCstj|tj||}|S)zMPrepare long data for prepared statements Returns a string. )r r:r)rZ statementparamr?r8rrr_prepare_stmt_send_long_dataesz*MySQLProtocol._prepare_stmt_send_long_datarcCs6d}dgt|dd}g} g} d} |dkr2i}|o8|rt|t|krVtjdxlt|D]^\} } || }d}|dkr|| dd| d>O<| jtjtjtj|qbn| |kr|| drtj }ntj }nt |t r|j |\} }}| j| nt |trftr:| jtjt||n"|j|}| jtjt||tj}nBt |tr| jtjt||tj }ntrt |tr|j|}| jtjt||tj}nt |tr| jtjtt|j|t|j|tj}nt |tr6| jtjd|tj}nrt |tjtjfrd|j|\} }| j| nDt |tjtjfr|j |\} }| j| ntj!d j"|j#j$d | jtj|tj|qbWtj%|tj|tj%|dj&d d |Dtjd}x| D]}||7}qWx| D]}||7}q W|S) z6Make a MySQL packet with the Statement Execute commandrrrtrGr-NzTFailed executing prepared statement: data values does not match number of parametersrz:MySQL binary protocol can not handle '{classname}' objects) classnamecSsg|]}tjd|qS)r)r$r%)ribitrrrrksz3MySQLProtocol.make_stmt_execute..)'r#r rrrvr r;rZNULLZBLOBSTRINGrrnrr!r Zlc_intrZVARCHARbytesunicoderZDECIMALfloatr$r%rrrrrrrrr __class____name__r:rx)rrr? parametersr_Zlong_data_usedr5Ziteration_countrrtypesrrr\rr^r8Za_typeZa_valuerrrmake_stmt_executeps                  0   zMySQLProtocol.make_stmt_executecCsZ|ddkstjdtj|dddd\}}|rL|d d krL|dd }|jd |fS) z&Parse a MySQL AuthSwitchRequest-packetrQraz'Failed parsing AuthSwitchRequest packetrBNr)rDrrrrIrI)r rr rJrK)rr8Z plugin_namerrrparse_auth_switch_requests  z'MySQLProtocol.parse_auth_switch_requestcCs"|ddkstjd|ddS)z!Parse a MySQL AuthMoreData-packetrQrz"Failed parsing AuthMoreData packetrBN)r r)rr8rrrparse_auth_more_datas z"MySQLProtocol.parse_auth_more_data)NNNr+rr,FN)r+rr,)N)NNNr+rFN)r[)T)r)r)r __module__ __qualname____doc__rr*r9r<r@rArPrWrZr`rgrsr~rrrrrrrrrrrrrrrrrrr'sF "   ",   *  !"- Y r)rr$rdecimalr constantsrrrrrr r Zauthenticationr Zcatch23r r robjectrrrrrs