o hU @sdZddlmZddlmZddlmZmZmZm Z m Z ddl m Z m Z ddlmZdZz"dd lZdd lmZmZmZmZmZmZmZmZmZdd lmZWneycZz ed e ed Z[wwd ZGddde Z!d S)z'Connection class using the C Extension )errors) INT_TYPES) CharacterSet FieldFlag ServerFlag ShutdownType ClientFlag)MySQLConnectionAbstractMySQLCursorAbstract) MySQLProtocolFN) CMySQLCursorCMySQLCursorRawCMySQLCursorBufferedCMySQLCursorBufferedRawCMySQLCursorPreparedCMySQLCursorDictCMySQLCursorBufferedDictCMySQLCursorNamedTupleCMySQLCursorBufferedNamedTuple)MySQLInterfaceErrorz6MySQL Connector/Python C Extension not available ({0})TcseZdZdZfddZddZeddZdYd d Zed d Z e j d d Z eddZ e j ddZ eddZ ddZ ddZeZddZdZddZddZd d!Zed"d#Zd[d%d&Zd\d'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd]d7d8ZeZ $ $d^d9d:Z ed;d<Z!ed=d>Z"ed?d@Z#edAdBZ$edCdDZ%dEdFZ&dGdHZ' I Jd_dKdLZ(dMdNZ)dOdPZ*d`dQdRZ+dSdTZ,dUdVZ-dWdXZ.Z/S)aCMySQLConnectionz5Class initiating a MySQL Connection using Connector/Cc sPtstdd|_g|_d|_tt|jdi||r&|jdi|dSdS)Initializationz0MySQL Connector/Python C Extension not availableN) HAVE_CMYSQL RuntimeError_cmysql_columns convertersuperr__init__connect)selfkwargs __class__rQ/var/www/html/env/lib/python3.10/site-packages/mysql/connector/connection_cext.pyr CszCMySQLConnection.__init__c CsF|j|j|jdddd|jd|_||jd|_dS)zfs  zCMySQLConnection.set_unicodecCs|dd}|dkr dSdS)z#Get whether autocommit is on or offzSELECT @@session.autocommitr rTF info_queryr?rrr& autocommitrszCMySQLConnection.autocommitc CsHz |j|||_WdSty#}z tj|j|j|jdd}~ww)zToggle autocommitmsgerrnosqlstateN) rrC _autocommitrrget_mysql_exceptionrErFrG)r"r@excrrr&rCxs   cCs|ddS)zGet the current databasezSELECT DATABASE()r rAr6rrr&databaseszCMySQLConnection.databasecC|j|dS)zSet the current databaseN)r select_dbr?rrr&rKcCs |jtj@S)z'MySQL session has started a transaction)r:rSTATUS_IN_TRANSr6rrr&in_transactions zCMySQLConnection.in_transactionc Cst|jd}tj|j|j||jpd|j|j d|_ |j |j |j |j|j|j|j|tjdd }|jsZ||jd|jd|jd|jdpMd |jd pTd |jd z |j jd i|Wnty|}z tj|j|j|jd d}~ww| dS)Nr )bufferedraw charset_nameconnection_timeoutr=r,T) hostuserpasswordrKport client_flags unix_socketcompress ssl_disabledcacertkey verify_certFverify_identity)ssl_cassl_certssl_keyssl_verify_certssl_verify_identityr\rDr)!rget_info _charset_id_mysql_connectorMySQL _buffered_raw_connection_timeoutr< _auth_pluginr_host_user _password _database_port _client_flags _unix_socketisset_client_flagrCOMPRESS _ssl_disabledupdate_sslgetr!rrrIrErFrGr7)r"rS cnx_kwargsrJrrr&_open_connectionsJ         z!CMySQLConnection._open_connectionc CsZ|jr+z ||jWnty%}z tj|j|j|jdd}~wwd|_dSdS)z Disconnect from the MySQL serverrDN) r free_resultcloserrrIrErFrG)r"rJrrr&rs  zCMySQLConnection.closecCs|jr|jSdS)z;Reports whether the connection to MySQL Server is availableF)rpingr6rrr& is_connecteds zCMySQLConnection.is_connectedFrr cCsNd}z|j}Wn tyYnw|rdS|r"|j||ddSt|)aCheck availability of the MySQL server When reconnect is set to True, one or more attempts are made to try to reconnect to the MySQL server using the reconnect()-method. delay is the number of seconds to wait between each retry. When the connection is not available, an InterfaceError is raised. Use the is_connected()-method if you just want to check the connection without raising an error. Raises InterfaceError on errors. z$Connection to MySQL is not availableN)attemptsdelay)rrAttributeError reconnectrInterfaceError)r"rrrerrmsg connectedrrr&rs  zCMySQLConnection.pingcCrL)zDSets the default character set name for current connection. N)rset_character_set)r"r*rrr&set_character_set_namerNz'CMySQLConnection.set_character_set_namecCsN|j|d}|jjr |j}|jr |jtd|j|S)z%Send a query which only returns 1 rowrz'Query should not return more than 1 row)rqueryhave_result_set fetch_rowr~rr)r"r first_rowrrr&rBs     zCMySQLConnection.info_querycCs"z|jWStyYdSw)zMySQL connection IDN)rr1rr6rrr& connection_ids   zCMySQLConnection.connection_idNc Cs8|jr|js td|dur|j}g}|dur |dkr tdd}z^|j}|rh|jsQ|jrQt|}t |D]\}} |sL|j |j |||||<q8t |}| ||d7}|ra||kran|j}|s*|su|d} |nd} W|| fSW|| fSty} z|tj| j| j| jdd} ~ ww)z8Get all or a subset of rows returned by the MySQL serverzNo result set availableNr z$count should be 1 or higher, or NonereofrD)r unread_resultr InternalErrorrlrrrlist enumerate to_pythonrtupleappendfetch_eof_columnsr~rrIrErFrG) r"countbinarycolumnsrRrowscounterrowi__eofrJrrr&get_rowssP          zCMySQLConnection.get_rowscCsHz|jd|||d\}}|r|d|fWSd|fWSty#YdSw)z.Get the next rows returned by the MySQL serverr)rrrrRr N)NN)r IndexError)r"rrrRrrrrr&get_row.s    zCMySQLConnection.get_rowcCs|jr |j|jSdS)zReads the next resultN)rconsume_result next_resultr6rrr&r:s  zCMySQLConnection.next_resultcC|jr |jdSdS)zFrees the resultN)rr~r6rrr&r~AzCMySQLConnection.free_resultcCr)zCommit current transactionN)rcommitr6rrr&rFrzCMySQLConnection.commitcCs"|jr|j|jdSdS)zRollback current transactionN)rrrollbackr6rrr&rKs zCMySQLConnection.rollbackc CsBz |j|WdSty }z tj|j|j|jdd}~ww)zChange the current databaserDN)rrMrrrIrErFrG)r"rKrJrrr& cmd_init_dbQs zCMySQLConnection.cmd_init_dbc Cs|jjs td|j}g|_|D]"}|j|dt|dddddt|dtj @t|dfq|j |j d|jdS)z Fetch EOF and column information No result setN ) status_flag warning_count)rr) rrrr fetch_fieldsrrintrNOT_NULLr:st_warning_count)r"fieldscolrrr&rYs(     z"CMySQLConnection.fetch_eof_columnscCs4|jr|j|j|j|j|jdSdS)z Fetch EOF and status information)r field_count insert_id affected_rowsr+N)rrst_field_countrrr:r6rrr&fetch_eof_statustsz!CMySQLConnection.fetch_eof_statusc Cs||dur |j}zt|ts|d}|jj||||dWn7ty8}z tj |j |j |j dd}~wt yX|jrE|j}n |jdt|j}tjd|dfdwg|_|jjsd|S|S) z Send a query to the MySQL serverNzutf-8)rRrQ raw_as_string)rErG:izConnection not available.)rFvalues)handle_unread_resultrl isinstancebytesencoderrrrrIrFrErGrrurostrrsOperationalErrorrrrr)r"rrRrQrrJaddrrrr& cmd_querys6     zCMySQLConnection.cmd_queryc s||s td|durt|tstd||S|p#|j}|p(|j}d|dur3dO|dur;dO|durCdO|durKd O|durSd Ot t t t t ttttd }z||WStyd td dfddtdDw)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.CMySQLCursorr Trrr) r rrrrrr)rQrR dictionary named_tuplepreparedz*Cursor not available with given criteria: z, cs$g|]}d|>@dkr|qS)rr r).0rargs cursor_typerr& sz+CMySQLConnection.cursor..r)rrrr issubclassr ProgrammingErrorrkrlr rrrrrrrrKeyError ValueErrorjoinrange)r"rQrRr cursor_classrrtypesrrr&cursorsP      zCMySQLConnection.cursorcCs|jjs td|jS)z,Returns number of rows of current result setr)rrrrnum_rowsr6rrr&rs  zCMySQLConnection.num_rowscCs|jsdS|jS)zReturns number of warningsr )rrr6rrr&rs zCMySQLConnection.warning_countcCs|jsdS|jjS)z"Check if a result set is availableF)rrr6rrr&result_set_availablesz%CMySQLConnection.result_set_availablecCs|jS)z)Check if there are unread results or rows)rr6rrr&rszCMySQLConnection.unread_resultcCr8)zCheck if there are more results)r more_resultsr6rrr&rr;zCMySQLConnection.more_resultscCs^t|ttfr|jj|}|St|tr+i}|D]\}}|j|d||<q|Std)zPrepare parameters for statements This method is use by cursors to prepared parameters found in the list (or tuple) params. Returns dict. r zCould not process parameters)rrrrconvert_to_mysqldictitemsr)r"paramsresultr_r@rrr&prepare_for_mysql s  z"CMySQLConnection.prepare_for_mysqlcCs|jdS)zlConsume the current result This method consume the result by reading (consuming) all rows. N)rrr6rrr&consume_resultssz CMySQLConnection.consume_results-c CsVz |j|||Wnty!}z tj|j|j|jdd}~ww||_| dS)z!Change the current logged in userrDN) r change_userrrrIrErFrGrh_post_connection)r"usernamerWrKr*rJrrr&cmd_change_user%s  z CMySQLConnection.cmd_change_userc CsFz |j|W|Sty"}z tj|j|j|jdd}~ww)z,Send the Refresh command to the MySQL serverrDN) rrefreshrrrIrErFrGr)r"optionsrJrrr& cmd_refresh1s zCMySQLConnection.cmd_refreshcCs |dS)z,Close the current connection with the serverN)rr6rrr&cmd_quit;s zCMySQLConnection.cmd_quitc Cs|jstd|rt|std|}ntj}z|j|Wnty9}z tj |j |j |j dd}~ww| dS)zShut down the MySQL ServerzMySQL Connection not availablezInvalid shutdown typerDN)rrrrrgrSHUTDOWN_DEFAULTshutdownrrIrErFrGr)r" shutdown_typelevelrJrrr& cmd_shutdown?s      zCMySQLConnection.cmd_shutdownc CsZ|z|j}tj|ddWSttjfy,}z tj|j |j |j dd}~ww)z'Return statistics from the MySQL serverF) with_headerrDN) rrstatr parse_statisticsrrrrIrErFrG)r"rrJrrr&cmd_statisticsSs  zCMySQLConnection.cmd_statisticscCs&t|ts td|d|dS)zKill a MySQL processzMySQL PID must be intzKILL {0}N)rrrrBformat)r" mysql_pidrrr&cmd_process_kill^s z!CMySQLConnection.cmd_process_killcCs&|jr |dS|jrtddS)z'Check whether there is an unread resultzUnread result foundN)can_consume_resultsrrrrr6rrr&rds   z%CMySQLConnection.handle_unread_result)T)Frr )NFNN)FNN)NFF)NNNNNN)rrrr)N)0__name__ __module__ __qualname____doc__r r7propertyr:r>rCsetterrKrPr}r disconnectrrrrBrrrrr~rrrrrr_execute_queryrrrrrrrrrrrrrrr __classcell__rrr$r&r?sz         (    (   J         r)"rrrcatch23r constantsrrrrr abstractsr r r'r rri cursor_cextr rrrrrrrrr ImportErrorrJrrrrrrr&s*   ,