3 (hp@sdZddlZddlZddlZddlmZmZddlmZddl m Z m Z m Z m Z mZddlmZmZddlmZd ZyDddlZdd lmZmZmZmZmZmZmZmZmZdd lm Z Wn4e!k rZ"ze!d j#e$e"WYddZ"["XnXd ZGdddeZ%dS)z'Connection class using the C Extension N)errorsversion) INT_TYPES) CharacterSet FieldFlag ServerFlag ShutdownType ClientFlag)MySQLConnectionAbstractMySQLCursorAbstract) MySQLProtocolF) CMySQLCursorCMySQLCursorRawCMySQLCursorBufferedCMySQLCursorBufferedRawCMySQLCursorPreparedCMySQLCursorDictCMySQLCursorBufferedDictCMySQLCursorNamedTupleCMySQLCursorBufferedNamedTuple)MySQLInterfaceErrorz6MySQL Connector/Python C Extension not available ({0})TcseZdZdZfddZddZddZedd Zd d Z did dZ eddZ e j ddZ eddZ e j ddZ eddZddZddZeZddZdjd d!Zd"d#Zd$d%Zed&d'Zdkd)d*Zdld+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zdmd7d8Zd9d:Zd;d<Z d=d>Z!d?d@Z"dAdBZ#dndCdDZ$e$Z%dodEdFZ&edGdHZ'edIdJZ(edKdLZ)edMdNZ*edOdPZ+dQdRZ,dSdTZ-dpdWdXZ.dYdZZ/d[d\Z0d]d^Z1dqd_d`Z2dadbZ3dcddZ4drdedfZ5dsdgdhZ6Z7S)tCMySQLConnectionz5Class initiating a MySQL Connection using Connector/Cc shts tdd|_g|_tjjtjjtjjt j dd|_ d|_ t t|jf||rd|jf|dS)ZInitializationz0MySQL Connector/Python C Extension not availableNZmysqlvendor) HAVE_CMYSQL RuntimeError_cmysql_columnsospathjoindirnameabspath_mysql_connector__file__ _plugin_dir convertersuperr__init__connect)selfkwargs) __class__S/tmp/pip-install-q3hcpn_q/mysql-connector-python/mysql/connector/connection_cext.pyr(Gs zCMySQLConnection.__init__cCsZtjjd}|ddkrd}nd}|jjd|djdd tjdd Dtjd d S) z!Add default connection attributes rZGPLv2zGPL-2.0Z Commercialzmysql-connector-python.cSsg|] }t|qSr-)str).0xr-r-r. dsz.)Z_connector_nameZ_connector_licenseZ_connector_versionZ _source_hostN) rLICENSEsplit _conn_attrsupdater VERSIONsocket gethostname)r*Zlicense_chunksZclient_licenser-r-r._add_default_conn_attrsXs  z(CMySQLConnection._add_default_conn_attrsc CsF|jj|jj|jjdddd|jjd|_|j|jd|_dS)z}ztjt|WYdd}~XnXg|_|jsZd|_|jSd|_|j |S)zExecutes the prepared statementNFT) Z stmt_executerrr|r1rrrrr)r*rargsrr-r-r.cmd_stmt_executes z!CMySQLConnection.cmd_stmt_executecCs|jrtjd|jdS)zCloses the prepared statementzUnread result foundN)rrrZ stmt_close)r*rr-r-r.cmd_stmt_closes zCMySQLConnection.cmd_stmt_closecCs|jrtjd|jdS)zResets the prepared statementzUnread result foundN)rrrZ stmt_reset)r*rr-r-r.cmd_stmt_resets zCMySQLConnection.cmd_stmt_resetcCs|j|dkr|j}y,t|ts,|jd}|jj||||dWn~tk rz}ztj |j |j |j dWYdd}~XnHt k r|jr|j}n|jdt|j}tjd|dfdYnXg|_|jjs|jS|jS) z Send a query to the MySQL serverNzutf-8)rWrV raw_as_string)rNrP:izConnection not available.)rOvalues)handle_unread_resultri isinstancebytesencoderrrrrQrOrNrPrzrorjr1rnrrrrr)r*rrWrVrrRaddrr-r-r. cmd_querys,    zCMySQLConnection.cmd_queryc s|j||jstjd|dk r@t|ts8tjd||S|pH|j}|pR|j}d|dkrhdO|dkrxdO|dkrdO|dkrd O|dkrd Ot t t t t ttttd }y ||Stk r dtddjfddtdDYnXdS)aInstantiates and returns a cursor using C Extension By default, CMySQLCursor is returned. Depending on the options while connecting, a buffered and/or raw cursor is instantiated instead. Also depending upon the cursor options, rows can be returned as dictionary or named tuple. Dictionary and namedtuple based cursors are available with buffered output but not raw. It is possible to also give a custom cursor through the cursor_class parameter, but it needs to be a subclass of mysql.connector.cursor_cext.CMySQLCursor. Raises ProgrammingError when cursor_class is not a subclass of CursorBase. Raises ValueError when cursor is not available. Returns instance of CMySQLCursor or subclass. :param buffered: Return a buffering cursor :param raw: Return a raw cursor :param prepared: Return a cursor which uses prepared statements :param cursor_class: Use a custom cursor class :param dictionary: Rows are returned as dictionary :param named_tuple: Rows are returned as named tuple :return: Subclass of CMySQLCursor :rtype: CMySQLCursor or subclass zMySQL Connection not available.Nz=Cursor class needs be to subclass of cursor_cext.CMySQLCursorrTrrr) rrrr5rrrrrVrW dictionary named_tuplepreparedz*Cursor not available with given criteria: z, cs$g|]}d|>@dkr|qS)rrr-)r2r)r cursor_typer-r.r4Usz+CMySQLConnection.cursor..r)rVrWrrr)rrwrr issubclassr ZProgrammingErrorrhrirrrrrrrrrKeyError ValueErrorr range)r*rVrWrZ cursor_classrrtypesr-)rrr.cursorsH      zCMySQLConnection.cursorcCs|jjstjd|jjS)z,Returns number of rows of current result setz No result set)rrrr|num_rows)r*r-r-r.rXs zCMySQLConnection.num_rowscCs|js dS|jjS)zReturns number of warningsr)rr)r*r-r-r.r`szCMySQLConnection.warning_countcCs|js dS|jjS)z"Check if a result set is availableF)rr)r*r-r-r.result_set_availablehsz%CMySQLConnection.result_set_availablecCs|jS)z)Check if there are unread results or rows)r)r*r-r-r.rpszCMySQLConnection.unread_resultcCs |jjS)zCheck if there are more results)r more_results)r*r-r-r.ruszCMySQLConnection.more_resultscCsbt|ttfr|jj|}nBt|trVi}x2|jD]\}}|jj|d||<q4Wntd|S)zPrepare parameters for statements This method is use by cursors to prepared parameters found in the list (or tuple) params. Returns dict. rzCould not process parameters)rrrrZconvert_to_mysqldictitemsr)r*paramsresultrerKr-r-r.prepare_for_mysqlzs z"CMySQLConnection.prepare_for_mysqlcCs|jjdS)zlConsume the current result This method consume the result by reading (consuming) all rows. N)rr)r*r-r-r.consume_resultssz CMySQLConnection.consume_results-cCs`y|jj|||Wn8tk rL}ztj|j|j|jdWYdd}~XnX||_|j dS)z!Change the current logged in user)rNrOrPN) rZ change_userrrrQrNrOrPrg_post_connection)r*usernamer[rSr@rRr-r-r.cmd_change_users z CMySQLConnection.cmd_change_usercCsh|jdkrtjdy|jjWn8tk rZ}ztj|j|j|j dWYdd}~XnX|j dS)zpResets the session state without re-authenticating Works only for MySQL server 5.7.3 or later. rr5zFMySQL version 5.7.2 and earlier does not support COM_RESET_CONNECTION.)rNrOrPN)rrr5) rDrNotSupportedErrorrZreset_connectionrrQrNrOrPr)r*rRr-r-r.cmd_reset_connections   z%CMySQLConnection.cmd_reset_connectioncCsRy|jj|Wn8tk rH}ztj|j|j|jdWYdd}~XnX|jS)z,Send the Refresh command to the MySQL server)rNrOrPN) rZrefreshrrrQrNrOrPr)r*optionsrRr-r-r. cmd_refreshs  zCMySQLConnection.cmd_refreshcCs |jdS)z,Close the current connection with the serverN)ru)r*r-r-r.cmd_quitszCMySQLConnection.cmd_quitcCs|jstjd|r.tj|s(tjd|}ntj}y|jj|Wn8tk r|}ztj |j |j |j dWYdd}~XnX|j dS)zShut down the MySQL ServerzMySQL Connection not availablezInvalid shutdown type)rNrOrPN)rrrr rfr|ZSHUTDOWN_DEFAULTshutdownrrQrNrOrPru)r*Z shutdown_typelevelrRr-r-r. cmd_shutdowns    zCMySQLConnection.cmd_shutdowncCsf|jy|jj}tj|ddSttjfk r`}ztj|j |j |j dWYdd}~XnXdS)z'Return statistics from the MySQL serverF)Z with_header)rNrOrPN) rrstatr Zparse_statisticsrrr|rQrNrOrP)r*rrRr-r-r.cmd_statisticss  zCMySQLConnection.cmd_statisticscCs&t|tstd|jdj|dS)zKill a MySQL processzMySQL PID must be intzKILL {0}N)rrrrLformat)r*Z mysql_pidr-r-r.cmd_process_kills z!CMySQLConnection.cmd_process_killcCs6|dkr|jn|j}|jr$|jn|r2tjddS)z'Check whether there is an unread resultTzUnread result foundN)rrZcan_consume_resultsrrr)r*rrr-r-r.rs  z%CMySQLConnection.handle_unread_resultc Cs|jstjdy |jWnZtjtfk rx|jrFtjdn.|jd kr\tjdn|j|j |j |j |j YnX|s|r|j }|rx(|jD]\}}|jdj||fqW|rx(|jD]\}}|jdj||fqW|jd S) aClears the current active session This method resets the session state, if the MySQL server is 5.7.3 or later active session will be reset without re-authenticating. For other server versions session will be reset by re-authenticating. It is possible to provide a sequence of variables and their values to be set after clearing the session. This is possible for both user defined variables and session variables. This method takes two arguments user_variables and session_variables which are dictionaries. Raises OperationalError if not connected, InternalError if there are unread results and InterfaceError on errors. zMySQL Connection not available.zZReset session is not supported with compression for MySQL server version 5.7.2 or earlier.rrr5zJReset session is not supported with MySQL server version 5.7.2 or earlier.zSET @`{0}` = %szSET SESSION `{0}` = %sN)rrr5)rwrrrrNotImplementedErrorZ _compressrDrrkrlrmrgrrexecuterru)r*Zuser_variablesZsession_variablescurrerKr-r-r. reset_sessions,    zCMySQLConnection.reset_session)T)Frr)NFNNN)FNNN)N)NFF)NNNNNN)rrrr)N)F)NN)8__name__ __module__ __qualname____doc__r(r=rEpropertyrFrGrJrMsetterrSrUrsruZ disconnectrwrvrrLrrrrrtrrrrrrrrrrZ_execute_queryrrrrrrrrrrrrrrrrr __classcell__r-r-)r,r.rCsn     >    4      I          r)&rrr; sysconfigrrrZcatch23r constantsrrrr r abstractsr r r>r rr#Z cursor_cextrrrrrrrrrr ImportErrorrRrr1rr-r-r-r.s&  ,