U e @spdZddlZddlZddlZddlmZmZmZmZm Z m Z m Z m Z ddl mZddlmZddlmZmZmZmZmZddlmZdd lmZmZmZmZmZdd lm Z dd l!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(d Z)zLddl*Z*dd l*m+Z+m,Z,ddl-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7Wn6e8k rVZ9ze8de9de9W5dZ9[9XYnXdZ)GdddeZ:dS)z'Connection class using the C Extension.N)AnyDictListOptionalSequenceTupleTypeUnion)version)MySQLConnectionAbstract) CharacterSet ClientFlag FieldFlag ServerFlag ShutdownType)MySQLConverter)InterfaceError InternalErrorOperationalErrorProgrammingErrorget_mysql_exception) MySQLProtocol)CextEofPacketTypeCextResultTypeDescriptionTypeParamsSequenceOrDictTypeRowTypeStatsPacketType StrOrBytesF)MySQLInterfaceError MySQLPrepStmt) CMySQLCursorCMySQLCursorBufferedCMySQLCursorBufferedDictCMySQLCursorBufferedNamedTupleCMySQLCursorBufferedRawCMySQLCursorDictCMySQLCursorNamedTupleCMySQLCursorPreparedCMySQLCursorPreparedDictCMySQLCursorRawz2MySQL Connector/Python C Extension not available ()Tc seZdZdZeddfdd ZddddZddd d Zee dd d Z e dd ddZ de ddddZee dddZeje ddddZee dddZeje ddddZee dddZddddZddddZeZe dd d!Ze dd"d#Zde e e dd'd(d)Ze dd*d+d,Zeeed-d.d/Zeee dd0d1Zdee e eee ee ee!e"eeee#fd2d3d4Z$de eee ee ee!e"eee#fd5d6d7Z%ee dd8d9Z&ddd:d;Z'dddd?Z)e dd@dAdBZ*dee!e+dCdDdEZ,ee#ddFdGZ-e.e!dHdIdJZ/e!eee0e#e+fdKdLdMZ1e!ddNdOdPZ2e!ddNdQdRZ3deee e e ee0e#e+fdSdTdUZ4e4Z5dee ee ee ee6e7ee ee e7dVdWdXZ8ee ddYdZZ9ee dd[d\Z:ee dd]d^Z;ee dd_d`Ze0e?e.e@e e.ffdcdddeZAdddfdgZBde e e e e e e ee ddj dkdlZCe ddmdnZDe ee#dodpdqZEdddrdsZFdee ddtdudvZGeHddwdxZIe ddydzd{ZJedd|d}ZKedd~dZLeedddZMeeeedddZNde ddddZOdee@e efee@e efddddZPZQS)CMySQLConnectionz6Class initiating a MySQL Connection using Connector/C.N)kwargsreturnc sts tdd|_g|_tjtjtjt j ddd|_ t dkrntj|j sntjdrhdnd|_ d|_t|r|jf|dS) ZInitializationz0MySQL Connector/Python C Extension not availableNmysqlvendorpluginLinuxz/usr/lib64/mysql/pluginz/usr/lib/mysql/plugin) HAVE_CMYSQL RuntimeError_cmysql_columnsospathjoindirnameabspath_mysql_connector__file__ _plugin_dirplatformsystemexists convertersuper__init__connect)selfr. __class__`/var/www/html/assets/scripts/venv/lib/python3.8/site-packages/mysql/connector/connection_cext.pyrE[s*   zCMySQLConnection.__init__)r/c CsZtjd}|ddkrd}nd}|jd|ddd tjdd Dtd d S) z!Add default connection attributes rZGPLv2zGPL-2.0Z Commercialzmysql-connector-python.cSsg|] }t|qSrJ)str).0xrJrJrK sz.)Z_connector_nameZ_connector_licenseZ_connector_versionZ _source_hostN) r LICENSEsplit _conn_attrsupdater:VERSIONsocket gethostname)rGZlicense_chunksZclient_licenserJrJrK_add_default_conn_attrsxs  z(CMySQLConnection._add_default_conn_attrsc CsR|j|j|jdddd|jd|_||jd|_t |jdS)z|r`|tt|||t|j|j|jd |W5d}~XYnX|| fS) z8Get all or a subset of rows returned by the MySQL serverzNo result set availableNrz$count should be 1 or higher, or NoneTr eofFrm)r unread_resultr6rrrrrCrwlist enumerateZ to_pythonr7tupleappendfetch_eof_columnsr_unread_resultr rrNrrnrorp)rGrrrrwrrrowscounterrrowi__eofrrrJrJrKget_rowss`       zCMySQLConnection.get_rows)rrrwrr/cCsPz4|jd||||d\}}|r,|d|fWSd|fWStk rJYdSXdS)z.Get the next rows returned by the MySQL serverr )rrrrwrrN)NN)r IndexError)rGrrrwrrrrJrJrKget_rows  zCMySQLConnection.get_rowcCs|jr|j|jSdS)zReads the next resultN)r6consume_result next_resultrarJrJrKrs  zCMySQLConnection.next_resultcCs|jr|jdS)zFrees the resultN)r6rrarJrJrKrszCMySQLConnection.free_resultcCs|jr||jdS)zCommit current transactionN)r6rcommitrarJrJrKrszCMySQLConnection.commitcCs|jr|j|jdS)zRollback current transactionN)r6rrollbackrarJrJrKrs zCMySQLConnection.rollback)rsr/c CsNz|j|Wn8tk rH}zt|j|j|jd|W5d}~XYnXdS)zChange the current databasermNrt)rGrsrrrJrJrK cmd_init_dbszCMySQLConnection.cmd_init_db)rr/cCs|r |jn|jj}|std|r*|n|j}g|_|D]N}|j|dt|dddddt|dtj@t|dt|df q>|j |j d|jdS) z Fetch EOF and column information No result setN )Z status_flag warning_count)rr) rr6rZ fetch_fieldsr7rintrZNOT_NULLrcst_warning_count)rGrrfieldscolrJrJrKrs0   z"CMySQLConnection.fetch_eof_columnscCs4|jr0|j|j|j|j|jdSdS)z Fetch EOF and status information)rZ field_count insert_id affected_rowsr^N)r6rZst_field_countrrrcrarJrJrKfetch_eof_statussz!CMySQLConnection.fetch_eof_status) statementr/c Cs^|jstdz|j|}|j|_|WStk rX}ztt||W5d}~XYnXdS)zPrepares the SQL statementMySQL Connection not availableN)r6rZ stmt_preparerrr rrN)rGrstmtrrrJrJrKcmd_stmt_prepare(s z!CMySQLConnection.cmd_stmt_prepare) statement_idargsr/c Gsjz|j|Wn0tk r>}ztt||W5d}~XYnXg|_|jsZd|_|Sd|_||S)zExecutes the prepared statementNFT) Z stmt_executer rrNr7rrrr)rGrrrrrJrJrKcmd_stmt_execute4s z!CMySQLConnection.cmd_stmt_execute)rr/cCs|jrtd|dS)zCloses the prepared statementUnread result foundN)rrZ stmt_closerGrrJrJrKcmd_stmt_closeFszCMySQLConnection.cmd_stmt_closecCs|jrtd|dS)zResets the prepared statementrN)rrZ stmt_resetrrJrJrKcmd_stmt_resetLszCMySQLConnection.cmd_stmt_reset)rrwrv raw_as_stringr/c Cs||dkr|j}z0t|ts,|d}|jj|||||jdWntk r~}zt |j |j |j d|W5d}~XYnRt k r}z4|jr|jn|jd|j}td|dfd|W5d}~XYnXg|_|jjs|S|S) z Send a query to the MySQL serverNzutf-8)rwrvrZ query_attrs)rnrp:izConnection not available.)rovalues)rr isinstancebytesencoder6rZ _query_attrsr rrornrprrrrrr7rrr)rGrrwrvrrraddrrJrJrK cmd_queryRsD   zCMySQLConnection.cmd_query)rvrwprepared cursor_class dictionary named_tupler/c s|||std|dk r@dkr|qS)r rrJ)rOrrZ cursor_typerJrKrQsz+CMySQLConnection.cursor..r)rrr issubclassr"rrrr#r+r&r'r$r(r%r)r*KeyError ValueErrorr:range)rGrvrwrrrrtypesrJrrKcursor|sX%     zCMySQLConnection.cursorcCs|jjstd|jS)z,Returns number of rows of current result setr)r6rrnum_rowsrarJrJrKrszCMySQLConnection.num_rowscCs|js dS|jS)zReturns number of warningsr)r6rrarJrJrKrszCMySQLConnection.warning_countcCs|js dS|jjS)z"Check if a result set is availableF)r6rrarJrJrKresult_set_availablesz%CMySQLConnection.result_set_availablecCs|jS)z)Check if there are unread results or rows)rrarJrJrKrszCMySQLConnection.unread_resultcCs |jS)zCheck if there are more results)r6 more_resultsrarJrJrKrszCMySQLConnection.more_results)paramsr/csg}t|ttfr:jr,fdd|D}qƈjj|}nt|tri}jr|D](\}}jj j |||<qVq|D]\}}j|d||<qnt dt |j d|d|S)zPrepare parameters for statements This method is use by cursors to prepared parameters found in the list (or tuple) params. Returns dict. c s(g|] }jjj|qSrJ)rCquoteescapeto_mysql)rOrerarJrKrQsz6CMySQLConnection.prepare_for_mysql..rzCould not process parameters: (z)), it must be of type list, tuple or dict)rrrrCr6Zconvert_to_mysqldictitemsrrrrtype__name__)rGrresultrrerJrarKprepare_for_mysqls(    z"CMySQLConnection.prepare_for_mysqlcCs|jdS)zlConsume the current result This method consume the result by reading (consuming) all rows. N)r6rrarJrJrKconsume_resultssz CMySQLConnection.consume_results-) usernamer{rsr]r|r}r~rr/c Cshz|j|||||||Wn8tk rT} zt| j| j| jd| W5d} ~ XYnX||_|dS)z!Change the current logged in userrmN) r6Z change_userr rrnrorpr_post_connection) rGr r{rsr]r|r}r~rrrrJrJrKcmd_change_users(  z CMySQLConnection.cmd_change_usercCs|j}|r||S)zResets the session state without re-authenticating Reset command only works on MySQL server 5.7.3 or later. The result is True for a successful reset otherwise False. Returns bool )r6Zreset_connectionr)rGresrJrJrKcmd_reset_connection>s z%CMySQLConnection.cmd_reset_connection)optionsr/c CsZz||j|Wn8tk rP}zt|j|j|jd|W5d}~XYnX|S)z,Send the Refresh command to the MySQL serverrmN) rr6refreshr rrnrorpr)rGrrrrJrJrK cmd_refreshKszCMySQLConnection.cmd_refreshcCs |dS)z,Close the current connection with the serverN)rrarJrJrKcmd_quitWszCMySQLConnection.cmd_quit) shutdown_typer/c Cs|jstd|r*t|s$td|}ntj}z|j|Wn8tk rx}zt|j |j |j d|W5d}~XYnX| dS)zShut down the MySQL ServerrzInvalid shutdown typermN) r6rrrrZSHUTDOWN_DEFAULTshutdownr rrnrorpr)rGrlevelrrrJrJrK cmd_shutdown[s$ zCMySQLConnection.cmd_shutdownc Csf|z|j}tj|ddWSttfk r`}zt|j|j |j d|W5d}~XYnXdS)z'Return statistics from the MySQL serverF)Z with_headerrmN) rr6statrZparse_statisticsr rrrnrorp)rGrrrrJrJrKcmd_statisticsos zCMySQLConnection.cmd_statistics) mysql_pidr/cCs&t|tstd|d|dS)zKill a MySQL processzMySQL PID must be intzKILL N)rrrrk)rGrrJrJrKcmd_process_kill{s z!CMySQLConnection.cmd_process_killcCstdS)zSend the DEBUG commandNNotImplementedErrorrarJrJrK cmd_debugszCMySQLConnection.cmd_debugcCstdS)zSend the PING commandNrrarJrJrKcmd_pingszCMySQLConnection.cmd_ping) statementsr/cCstdS)z/Send one or more statements to the MySQL serverNr)rGr"rJrJrKcmd_query_iterszCMySQLConnection.cmd_query_iter)rparam_iddatar/cCstdS)zSend data for a columnNr)rGrr$r%rJrJrKcmd_stmt_send_long_datasz(CMySQLConnection.cmd_stmt_send_long_data)rr/cCs4|dkr|jn|j}|jr$|n |r0tddS)z'Check whether there is an unread resultTrN)rrZcan_consume_resultsr r)rGrrrJrJrKrs  z%CMySQLConnection.handle_unread_result)user_variablessession_variablesr/c Cs|std|sbz,||j|j|j|j|j|j |j |j Wnt k r`| YnX|sj|r|}|r|D]\}}|d|d|fq~|r|D]\}}|d|d|fq|dS)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. rzSET @`z` = %sz SET SESSION `N)rrrrrrrrrrrrrrrrexecuter)rGr'r(currrerJrJrK reset_sessions2 zCMySQLConnection.reset_session)T)Fr r)NFNNN)FNNN)N)NFF)NNNNNN)r r r r r r r N)N)F)NN)Rr __module__ __qualname____doc__rrErZrbpropertyrrcrNrdboolrhrlsetterrsrurrZ disconnectrrrrrrrrkrrrr!rrrrrrrrrrrrrrr rrrrZ_execute_queryrr"rrrrrrrrrr r rrrrrrrrr r!r#r&rr+ __classcell__rJrJrHrKr-Xs.    Q    A    #     (  S '       r-);r.r8r@rXtypingrrrrrrrr r r Z abstractsr constantsr rrrr conversionrerrorsrrrrrr[rrrrrrrrrr4r=r r!Z cursor_cextr"r#r$r%r&r'r(r)r*r+ ImportErrorexcr-rJrJrJrKs0(    $ 4