o –¨ÊhŒÚã@sàdZzddlZdZWndZYddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZmZmZddlmZmZmZmZmZdd lmZmZmZmZdd lmZdd lmZm Z dd l!m"Z"m#Z#dd l$m%Z%m&Z&ddl'm(Z(m)Z)m*Z*ddl+m,Z,m-Z-m.Z.ddl/m0Z0dZ1dZ2dZ3dZ4dZ5dZ6e  7d¡Z8dZ9dZ:e ;d¡ZƒZ?dd„Z@Gd d!„d!e>ƒZAGd"d#„d#eAƒZBGd$d%„d%ejCƒZDGd&d'„d'e>ƒZEGd(d)„d)e>ƒZFGd*d+„d+e>ƒZGdS),z4Implementation of communication for MySQL X servers.éNTF©Úwrapsé)ÚMySQL41AuthPluginÚPlainAuthPluginÚSha256MemoryAuthPlugin)ÚInterfaceErrorÚOperationalErrorÚ PoolErrorÚProgrammingErrorÚ TimeoutError)ÚPY3Ú STRING_TYPESÚ UNICODE_TYPESÚqueue©ÚSchema)ÚSSLModeÚAuth)ÚescapeÚget_item_or_attr)ÚProtocolÚMessageReaderWriter)ÚResultÚ RowResultÚ DocResult)Ú SqlStatementÚ AddStatementÚquote_identifier)ÚProtobufi'zDROP DATABASE IF EXISTS {0}z!CREATE DATABASE IF NOT EXISTS {0}zLSELECT SCHEMA_NAME FROM INFORMATION_SCHEMA.SCHEMATA WHERE SCHEMA_NAME = '{}'écéxz[^a-zA-Z0-9._:\-*$#]i›Ä Úmysqlxc KsLg}dD]}z | t||ƒ¡WqtyYqw|s!tdƒ‚d |¡S)abGenerate a pool name. This function takes keyword arguments, usually the connection arguments and tries to generate a name for the pool. Args: **kwargs: Arbitrary keyword arguments with the connection arguments. Raises: PoolError: If the name can't be generated. Returns: str: The generated pool name. )ÚhostÚportÚuserÚdatabaseÚ client_idz.Failed generating pool name; specify pool_nameÚ_)ÚappendÚstrÚKeyErrorr Újoin)ÚkwargsÚpartsÚkey©r0úC/var/www/html/env/lib/python3.10/site-packages/mysqlx/connection.pyÚgenerate_pool_nameNs ÿ r2c@sleZdZdZdd„Zefdd„Zdd„Zdd „Zd d „Z d d „Z dd„Z dd„Z dd„Z dd„Zdd„ZdS)Ú SocketStreamzImplements a socket stream.cCsd|_d|_d|_d|_dS)NF)Ú_socketÚ_is_sslÚ _is_socketÚ_host©Úselfr0r0r1Ú__init__ls zSocketStream.__init__c CsŠ|dur|d}zt ||¡|_|d|_WdStyDzt tj¡|_|j |¡|j |¡d|_WYdSt yCt dƒ‚ww)zÇConnects to a TCP service. Args: params (tuple): The connection parameters. Raises: :class:`mysqlx.InterfaceError`: If Unix socket is not supported. NéèrTzUnix socket unsupported) ÚsocketÚcreate_connectionr4r7Ú ValueErrorÚAF_UNIXÚ settimeoutÚconnectr6ÚAttributeErrorr)r9ÚparamsÚconnect_timeoutr0r0r1rArs     ÿúzSocketStream.connectcCsb|jdur tdƒ‚g}|dkr,|j |¡}|dkrtdƒ‚| |¡|t|ƒ8}|dksd |¡S)z“Receive data from the socket. Args: count (int): Buffer size. Returns: bytes: The data received. NúMySQLx Connection not availablerózUnexpected connection close)r4r ÚrecvÚ RuntimeErrorr)Úlenr,)r9ÚcountÚbufÚdatar0r0r1Úread‰s    û zSocketStream.readcCs"|jdur tdƒ‚|j |¡dS)z_Send data to the socket. Args: data (bytes): The data to be sent. NrE)r4r Úsendall)r9rLr0r0r1rNs zSocketStream.sendallcCsF|jsdSz|j tj¡|j ¡Wn tjyYnwd|_dS)zClose the socket.N)r4Úshutdownr<Ú SHUT_RDWRÚcloseÚerrorr8r0r0r1rQ§sþ zSocketStream.closecCs | ¡dS©N)rQr8r0r0r1Ú__del__³s zSocketStream.__del__c Csts | ¡tdƒ‚t tj¡}| ¡|rz ˆ|g|¢Ri|¤ŽWStjtfy| ¡tdƒ‚w)zWrapper function.zCannot connect to host)r<rRrHÚ disconnectr)r9Úargsr-©Úfuncr0r1Úwrapper$s þz(catch_network_exception..wrapperr)r‹rŒr0rŠr1Úcatch_network_exceptionsrc@seZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dd„Zedd„ƒZedd„ƒZedd„ƒZedd„ƒZed d!„ƒZed"d#„ƒZed$d%„ƒZed&d'„ƒZed(d)„ƒZed*d+„ƒZed,d-„ƒZd.d/„Zd0d1„Zd2d3„Zd4d5„Zd6d7„Zd8S)9Ú ConnectionzkConnection to a MySQL Server. Args: settings (dict): Dictionary with connection settings. cCsÐ||_tƒ|_d|_d|_| d¡|_| d¡|_| d¡|_d|_ | dg¡|_ d|vrA|drA|j   | d¡| dd¡dœ¡d|_ d |_ | ¡|j jd d „d d | d t¡|_|jdkrfd|_dSdS)Nr%ÚpasswordÚschemaÚroutersr#r$)r#r$éÿÿÿÿTcSó|dS©NÚpriorityr0©Úxr0r0r1ÚIóz%Connection.__init__..©r/Úreverseúconnect-timeoutr)Úsettingsr3ÚstreamÚ reader_writerÚprotocolÚgetÚ_userÚ _passwordÚ_schemaÚ_active_resultÚ_routersr)Ú _cur_routerÚ _can_failoverÚ_ensure_prioritiesÚsortr‡Ú_connect_timeout)r9rr0r0r1r:5s0    þÿ  ýzConnection.__init__cCs"|jdur|j ¡d|_dSdS)zFetch active result.N)r¥Ú fetch_allr8r0r0r1Úfetch_active_resultQs   þzConnection.fetch_active_resultcCs ||_dS)aSet active result. Args: `Result`: It can be :class:`mysqlx.Result`, :class:`mysqlx.BufferingResult`, :class:`mysqlx.RowResult`, :class:`mysqlx.SqlResult` or :class:`mysqlx.DocResult`. N)r¥©r9Úresultr0r0r1Úset_active_resultWó zConnection.set_active_resultcCs‚d}d}|jD]"}| dd¡}|dur|d7}||d<n |dkr%tddƒ‚|d8}qd|kr7t|jƒkr?ndStdd ƒ‚dS) zuEnsure priorities. Raises: :class:`mysqlx.ProgrammingError`: If priorities are invalid. rédr•Nrz(The priorities must be between 0 and 100i§z\You must either assign no priority to any of the routers or give a priority for every routeri )r¦r¡r rI)r9Úpriority_countr•ÚrouterÚprir0r0r1r©bs"   ÿ  þÿzConnection._ensure_prioritiescCsä|js#d|_d|jvr|jd|j dd¡fSd|jvr!|jdSdS|jr+|jdkr;d|_d|_|jD]}d|d <q4|jd 7_|j|jd}|j|jd}|jd krad|j|jd d <|jt|jƒd krnd|_||fS) zkReturns the connection parameters. Returns: tuple: The connection parameters. Fr#r$é$r<)rVr¶r’TÚ availablerr)r¦r¨rr¡r§rI)r9r´r#r$r0r0r1Ú_get_connection_paramszs(      z!Connection._get_connection_paramsc Csôd}|jr?z!|j | ¡|j¡t|jƒ|_t|jƒ|_|  ¡|  ¡WdSt j y;}z|}WYd}~nd}~ww|js|durgt |t jƒsP|jdvrgtsgt|jƒdkr_td |j¡ƒ‚td |j¡ƒ‚t|jƒdkrutd |¡ƒ‚tddƒ‚) a Attempt to connect to the MySQL server. Raises: :class:`mysqlx.InterfaceError`: If fails to connect to the MySQL server. :class:`mysqlx.TimeoutError`: If connect timeout was exceeded. N)é ésrzLConnection attempt to the server was aborted. Timeout of {0} ms was exceededzdAll server connection attempts were aborted. Timeout of {0} ms was exceeded for each selected serverzCannot connect to host: {0}z'Failed to connect to any of the routersi¡)r¨ržrAr¸r«rrŸrr Ú_handle_capabilitiesÚ _authenticater<rRÚ isinstanceÚtimeoutÚerrnor rIr¦r rar)r9rRrtr0r0r1rAšs8  ÿ  €ÿ÷þþ zConnection.connectc Cs:|j d¡tjkr dS|j ¡r|j d¡rt d¡dS|j  ¡j }|r1t |ddƒ  ¡dks;n  |  ¡tdƒ‚d}t ¡d kret ¡\}}}zd |voW| d ¡dd k}Wn tydd}Ynwtjd krt|st|  ¡tdƒ‚|jjdd|j |j dtj¡|j d¡|j d¡|j d¡|j d¡¡dS)a1Handle capabilities. Raises: :class:`mysqlx.OperationalError`: If SSL is not enabled at the server. :class:`mysqlx.RuntimeError`: If support for SSL is not available in Python. zssl-modeNz(SSL not required when using Unix socket.rrhÚtlsFzSSL not enabled at serverÚLinuxz Oracle LinuxÚ.Ú7)ééé zPLAIN authentication is not allowed via unencrypted connection)ræN) ržrrrr¢r£r rârãrærç)r9rér0r0r1rÜs  ÿzConnection._authenticate_plaincCrá)zBAuthenticate with the MySQL server using `Sha256MemoryAuthPlugin`.N) rr¢r£r rârãrärårærçrèr0r0r1rÞrëz&Connection._authenticate_sha256_memorycGsd|jdur tdƒ‚t|tƒstdƒ‚ts(t|tƒr(|j dtt |dƒƒ|¡dS|j d||¡dS)aExecute a SQL statement. Args: sql (str): The SQL statement. *args: Arbitrary arguments. Raises: :class:`mysqlx.ProgrammingError`: If the SQL statement is not a valid string. NrEz'The SQL statement is not a valid stringÚsqlzutf-8) r r r½rr r rÚsend_execute_statementÚbytesÚ bytearray)r9rìr‰r0r0r1Úsend_sql#s  ÿzConnection.send_sqlcCs<|jdur tdƒ‚|j |¡d}t|tƒr|j}t||ƒS)a Send an insert statement. Args: statement (`Statement`): It can be :class:`mysqlx.InsertStatement` or :class:`mysqlx.AddStatement`. Returns: :class:`mysqlx.Result`: A result object. NrE)r r Ú send_insertr½rÚidsr)r9Ú statementròr0r0r1rñ9s    zConnection.send_insertcCs$|j |¡| ¡rt|ƒSt|ƒS)aBSend an find statement. Args: statement (`Statement`): It can be :class:`mysqlx.ReadStatement` or :class:`mysqlx.FindStatement`. Returns: `Result`: It can be class:`mysqlx.DocResult` or :class:`mysqlx.RowResult`. )r Ú send_findÚ is_doc_basedrr©r9rór0r0r1ÚfindLs zConnection.findcCó|j |¡t|ƒS)aSend an delete statement. Args: statement (`Statement`): It can be :class:`mysqlx.RemoveStatement` or :class:`mysqlx.DeleteStatement`. Returns: :class:`mysqlx.Result`: The result object. )r Ú send_deleterrör0r0r1Údelete[ó zConnection.deletecCrø)aSend an delete statement. Args: statement (`Statement`): It can be :class:`mysqlx.ModifyStatement` or :class:`mysqlx.UpdateStatement`. Returns: :class:`mysqlx.Result`: The result object. )r Ú send_updaterrör0r0r1ÚupdateirûzConnection.updatecGs6z |j |||¡t|ƒWSty|r‚YdSw)azExecute a non query command. Args: namespace (str): The namespace. cmd (str): The command. raise_on_fail (bool): `True` to raise on fail. *args: Arbitrary arguments. Raises: :class:`mysqlx.OperationalError`: On errors. Returns: :class:`mysqlx.Result`: The result object. N)r rírr )r9Ú namespaceÚcmdÚ raise_on_failr‰r0r0r1Úexecute_nonqueryws  ÿÿzConnection.execute_nonquerycGs>|j d||¡t|ƒ}| ¡|jdkrtdƒ‚|ddS)aExecute a SQL scalar. Args: sql (str): The SQL statement. *args: Arbitrary arguments. Raises: :class:`mysqlx.InterfaceError`: If no data found. Returns: :class:`mysqlx.Result`: The result. rìrz No data found)r rírr¬rJr)r9rìr‰r¯r0r0r1Úexecute_sql_scalarŽs   zConnection.execute_sql_scalarcGs|j d||¡t|ƒS)zÇReturns the row result. Args: cmd (str): The command. *args: Arbitrary arguments. Returns: :class:`mysqlx.RowResult`: The result object. Úxplugin)r rír)r9rÿr‰r0r0r1Úget_row_result£s zConnection.get_row_resultcCó |j |¡S)zdRead row. Args: result (:class:`mysqlx.RowResult`): The result object. )r Úread_rowr®r0r0r1r±ó zConnection.read_rowcCó|j |¡dS)zeClose result. Args: result (:class:`mysqlx.Result`): The result object. N)r Ú close_resultr®r0r0r1r ºszConnection.close_resultcCr)zlGet column metadata. Args: result (:class:`mysqlx.Result`): The result object. )r Úget_column_metadatar®r0r0r1r ÃrzConnection.get_column_metadatacCs |j ¡S)zgCheck if connection is open. Returns: bool: `True` if connection is open. )ržrr8r0r0r1rÌr‚zConnection.is_opencCs| ¡sdS|j ¡dS)zDisconnect from server.N)rržrQr8r0r0r1rˆÔszConnection.disconnectc Cs | ¡sdSzCz|jdur|j ¡|j ¡|j ¡Wnttfy9}zt  d  |¡¡WYd}~n d}~wwW|j   ¡dSW|j   ¡dS|j   ¡w)z*Close a sucessfully authenticated session.NzGWarning: An error occurred while attempting to close the connection: {}) rr¥r¬r Ú send_closeÚread_okrr rÈrÉraržrQ©r9rtr0r0r1Ú close_sessionÚs$    €ÿ€ÿ€ûzConnection.close_sessionc Csl| ¡sdS|jdur|j ¡z|j ¡WdSttfy5}zt d  |¡¡WYd}~dSd}~ww)z*Reset a sucessfully authenticated session.NzDWarning: An error occurred while attempting to reset the session: {}) rr¥r¬r Ú send_resetrr rÈrÉrar r0r0r1Ú reset_sessionës  ÿ€ÿzConnection.reset_sessioncCsB| ¡sdS|jdur|j ¡|j ¡|j ¡|j ¡dS)z€Announce to the server that the client wants to close the connection. Discards any session state of the server. N)rr¥r¬r Úsend_connection_closer ržrQr8r0r0r1rÍ÷s    zConnection.close_connectionN) rƒr„r…r†r:r­r°r©r¸rAr»r¼ràrÜrÞrrðrñr÷rúrýrrrrr r rrˆrrrÍr0r0r0r1rŽ/sP  &,             rŽcsLeZdZdZ‡fdd„Z‡fdd„Zdd„Zdd „Zd d „Zd d „Z ‡Z S)ÚPooledConnectionaClass to hold :class:`Connection` instances in a pool. PooledConnection is used by :class:`ConnectionPool` to facilitate the connection to return to the pool once is not required, more specifically once the close_session() method is invoked. It works like a normal Connection except for methods like close() and sql(). The close_session() method will add the connection back to the pool rather than disconnecting from the MySQL server. The sql() method is used to execute sql statements. Args: pool (ConnectionPool): The pool where this connection must return. .. versionadded:: 8.0.13 csFt|tƒs tdƒ‚tt|ƒ |j¡||_|jd|_|jd|_ dS)Nz&pool should be a ConnectionPool objectr#r$) r½ÚConnectionPoolrBÚsuperrr:Ú cnx_configÚpoolr#r$)r9r©Ú __class__r0r1r:s  zPooledConnection.__init__cstt|ƒ ¡dS)zGCloses the connection. This method closes the socket. N)rrrr8rr0r1rÍsz!PooledConnection.close_connectioncCs|j |¡dS)a“Do not close, but add connection back to pool. The close_session() method does not close the connection with the MySQL server. The connection is added back to the pool so it can be reused. When the pool is configured to reset the session, the session state will be cleared by re-authenticating the user once the connection is get from the pool. N)rÚadd_connectionr8r0r0r1r%s zPooledConnection.close_sessioncCs |jdur |j ¡| ¡dS)z#Reconnect this connection. N)r¥r¬r¼r8r0r0r1Ú reconnect2s   zPooledConnection.reconnectcCs | ¡dS)zQReset the connection. Resets the connection by re-authenticate. N)rr8r0r0r1Úreset9s zPooledConnection.resetcCó t||ƒS)zÓCreates a :class:`mysqlx.SqlStatement` object to allow running the SQL statement on the target MySQL Server. Returns: :class:`mysqlx.SqlStatement`: The sql statement object. )r)r9Ú sql_statementr0r0r1rì@s zPooledConnection.sql) rƒr„r…r†r:rÍrrrrìÚ __classcell__r0r0rr1rs   rc@sVeZdZdZdd„Zdd„Zedd„ƒZdd d „Zd d „Z d d„Z dd„Z dd„Z dS)raµThis class represents a pool of connections. Initializes the Pool with the given name and settings. Args: name (str): The name of the pool, used to track a single pool per combination of host and user. **kwargs: max_size (int): The maximun number of connections to hold in the pool. reset_session (bool): If the connection should be reseted when is taken from the pool. max_idle_time (int): The maximum number of milliseconds to allow a connection to be idle in the queue before being closed. Zero value means infinite. queue_timeout (int): The maximum number of milliseconds a request will wait for a connection to become available. A zero value means infinite. priority (int): The router priority, to choose this pool over other with lower priority. Raises: :class:`mysqlx.PoolError` on errors. .. versionadded:: 8.0.13 cKs| |¡d|_g|_| dd¡|_tj ||j¡| dd¡|_| dd¡|_ ||_ | dd¡|_ | dd¡|_ ||_ |d |_|d |_dS) NrÚmax_sizeérTÚ max_idle_timeÚ queue_timeoutr•r#r$)Ú_set_pool_nameÚ_open_sessionsÚ_connections_opennedr¡Ú pool_max_sizerÚQueuer:rr!rr"Ú _priorityrr#r$)r9rhr-r0r0r1r:fs  zConnectionPool.__init__cCs<t |¡r td |¡ƒ‚t|ƒtkrtd |¡ƒ‚||_dS)aƒSet the name of the pool. This method checks the validity and sets the name of the pool. Args: pool_name (str): The pool name. Raises: AttributeError: If the pool_name contains illegal characters ([^a-zA-Z0-9._\-*$#]) or is longer than connection._CNX_POOL_MAX_NAME_SIZE. z+Pool name '{0}' contains illegal characterszPool name '{0}' is too longN)Ú_CNX_POOL_NAME_REGEXÚsearchrBrarIÚ_CNX_POOL_MAX_NAME_SIZErh)r9Ú pool_namer0r0r1r#vs ÿ ÿ zConnectionPool._set_pool_namecCs t|jƒS)zMReturns the number of open connections that can return to this pool. )rIr%r8r0r0r1Úopen_connections‹s zConnectionPool.open_connectionsNcCs¬|jstdƒ‚| ¡rtdƒ‚|sFt|ƒ}| d¡ ¡ ¡dd}tdd„| d¡d d ¡Dƒƒd kr?| d   d¡¡ ¡|j   |¡n t |tƒsOtd ƒ‚|  |¡d S)a=Adds a connection to this pool. This method instantiates a Connection using the configuration passed when initializing the ConnectionPool instance or using the set_config() method. If cnx is a Connection instance, it will be added to the queue. Args: cnx (PooledConnection): The connection object. Raises: PoolError: If no configuration is set, if no more connection can be added (maximum reached) or if the connection can not be instantiated. z&Connection configuration not availableú'Failed adding connection; queue is fullúshow variables like "version"rrcSóg|]}t|ƒ‘qSr0©Úint©Ú.0Únr0r0r1Ú ¬óz1ConnectionPool.add_connection..ú-r©éé úset mysqlx_wait_timeout = {}ú2Connection instance not subclass of PooledSession.N)rr ÚfullrrìÚexecuter¬ÚtuplerÑrar%r)r½Úqueue_connection)r9ÚcnxÚverr0r0r1r‘s( ÿÿ& ÿzConnectionPool.add_connectioncCsTt|tƒs tdƒ‚|jr| ¡z |j|ddWdStjy)tdƒYdSw)aGPut connection back in the queue: This method is putting a connection back in the queue. It will not acquire a lock as the methods using _queue_connection() will have it set. Args: PooledConnection: The connection object. Raises: PoolError: On errors. r=F)Úblockr.N)r½rr rÚputrÚFull©r9rBr0r0r1rA·s ÿÿzConnectionPool.queue_connectioncCr)zNTracks connection in order of close it when client.close() is invoke. N)r%r))r9Ú connectionr0r0r1Útrack_connectionÐszConnectionPool.track_connectioncCr{rS)rhr8r0r0r1Ú__str__ÕszConnectionPool.__str__cCs|jD]}| ¡qdS)z#Empty this ConnectionPool. N)r%rÍrGr0r0r1rQØs  ÿzConnectionPool.closerS) rƒr„r…r†r:r#Úpropertyr-rrArIrJrQr0r0r0r1rJs  & rc@sPeZdZdZdZiZdd„Zdd„Zdd„Zd d „Z d d „Z d d„Z dd„Z dS)Ú PoolsManagerz–Manages a pool of connections for a host or hosts in routers. This class handles all the pools of Connections. .. versionadded:: 8.0.13 NcCs"tjdurt |¡t_it_tjSrS)rLÚ_PoolsManager__instanceÚobjectÚ__new__Ú_PoolsManager__pools)Úclsr0r0r1rOés  zPoolsManager.__new__cCs,|j |g¡}|D] }|j|krdSq dS)zþVerifies if a pool exists with the given name. Args: client_id (str): The client id. pool_name (str): The name of the pool. Returns: bool: Returns `True` if the pool exists otherwise `False`. TF)rPr¡rh)r9r'r,Úpoolsrr0r0r1Ú _pool_existsïs   ÿzPoolsManager._pool_existscCs^g}g}| |¡}|D] \}}| |¡q |j | dd¡g¡D] }|j|vr,| |¡q |S)záRetrieves a list of pools that shares the given settings. Args: settings (dict): the configuration of the pool. Returns: list: A list of pools that shares the given settings. r'úNo id)Ú_get_connections_settingsr)rPr¡rh)r9rÚavailable_poolsÚ pool_namesÚconnections_settingsÚ router_namer(rr0r0r1Ú _get_poolsÿs      €zPoolsManager._get_poolscCs¸| ¡}| dg¡}g}d|vr| d¡d|vr*d|vr*| d|d|ddœ¡|jdd„dd |D]$}| ¡}|d|d<|d|d<|d |d <| td i|¤Ž|f¡q5|S) aHGenerates a list of separated connection settings for each host. Gets a list of connection settings for each host or router found in the given settings. Args: settings (dict): The configuration for the connections. Returns: list: A list of connections settings r‘r#r$r)r•r#r$cSr“r”r0r–r0r0r1r˜,r™z8PoolsManager._get_connections_settings..Tršr•Nr0)Úcopyr¡Úpopr)rªr2)r9rÚ pool_settingsr‘rXr´Úconnection_settingsr0r0r1rUs,  þ    ÿÿz&PoolsManager._get_connections_settingscCs‚| |¡}| dd¡|jvrg|j| dd¡<|D]%\}}| | dd¡|¡r(q|j | dd¡g¡}| t|fi|¤Ž¡qdS)a#Creates a `ConnectionPool` instance to hold the connections. Creates a `ConnectionPool` instance to hold the connections only if no other pool exists with the same configuration. Args: cnx_settings (dict): The configuration for the connections. r'rTN)rUr¡rPrSr)r)r9Ú cnx_settingsrXrYrrr0r0r1Ú create_pool7s  ÿÿùzPoolsManager.create_poolc Csh| |¡}t|ƒ}t|ƒD]#}||}z| ¡dkr{t ¡Rz |jd|jd}Wn tj y7t dƒ‚w|  ¡|  d¡  ¡ ¡dd}tdd„| d ¡d d ¡Dƒƒd krf|  d  d¡¡  ¡|Wd ƒWS1suwYn¡|j|jkr½t|ƒ}| |¡| ¡|  d¡  ¡ ¡dd}tdd„| d ¡d d ¡Dƒƒd kr¸|  d  d¡¡  ¡|WSt ¡RzC|jd|jd}|  ¡|  d¡  ¡ ¡dd}tdd„| d ¡d d ¡Dƒƒd krù|  d  d¡¡  ¡|WWd ƒWStj yt dƒ‚w1swYWq ttfy1||dkr/‚Yq wd S)ažGet a connection from the pool. This method returns an `PooledConnection` instance which has a reference to the pool that created it, and can be used as a normal Connection. When the MySQL connection is not connected, a reconnect is attempted. Raises: :class:`PoolError`: On errors. Returns: PooledConnection: A pooled connection object. rT)rDr¾z)Failed getting connection; pool exhaustedr/rcSr0r0r1r3r0r0r1r6rr7z/PoolsManager.get_connection..r8rÂr9r<NcSr0r0r1r3r0r0r1r6r7cSr0r0r1r3r0r0r1r6Žr7zpool max size has been reached)rZrIÚrangeÚqsizeÚ threadingÚRLockr¡r"rÚEmptyr rrìr?r¬r@rÑrar-r&rrIrArr )r9rrRÚ num_poolsÚ pool_numberrrBrCr0r0r1Úget_connectionPsž    ÿÿÿ ÿÿÿÿ(ñ   ÿÿÿÿ  ÿ ÿÿÿÿô ÿó€üÌzPoolsManager.get_connectioncCsX| |¡}|D] }| ¡| dd¡dur'|j | d¡¡}||vr'| |¡qt|ƒS)zjCloses the connections in the pools Returns: int: The number of closed pools r'N)rZrQr¡rPÚremoverI)r9r_rRrÚ client_poolsr0r0r1Ú close_pool›s  €zPoolsManager.close_pool) rƒr„r…r†rMrPrOrSrZrUr`rhrkr0r0r0r1rLßs! KrLc@s´eZdZdZdd„Zedd„ƒZejdd„ƒZdd„Zd d „Z d d „Z d d„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zd(dd„Zd d!„Zd"d#„Zd$d%„Zd&d'„ZdS))ÚSessionaWEnables interaction with a X Protocol enabled MySQL Product. The functionality includes: - Accessing available schemas. - Schema management operations. - Enabling/disabling warning generation. - Retrieval of connection information. Args: settings (dict): Connection data used to connect to the database. c Csà| dtj¡|_||_d|vr,|dr,tƒ |¡tƒ |¡|_|jdur+td|d‚n t |jƒ|_|j  ¡|j d¡}|rnz|  d  t |ƒ¡¡ ¡WdStym}z|jdkr^|jnd  |¡}t||jƒ‚d}~wwdS) Nzuse-pureÚpoolingz.connection could not be retrieve from pool. %s)ÚvaluesrzUSE {}iú#Default schema '{}' does not exists)r¡rÚuse_pureÚ _settingsrLr`rhÚ _connectionr rŽrArìrarr?r r¿Úmsgr)r9rrrtÚerrmsgr0r0r1r:¹s0  ÿÿ   ÿ €üýzSession.__init__cCstjS)zAbool: `True` to use pure Python Protobuf implementation. )rrpr8r0r0r1rpÑszSession.use_purecCs t|tƒs tdƒ‚t |¡dS)Nz)'use_pure' option should be True or False)r½Úboolr rÚ set_use_pure)r9Úvaluer0r0r1rp×s cCs |jj ¡S)zzReturns `True` if the session is open. Returns: bool: Returns `True` if the session is open. )rrržrr8r0r0r1rÝr€zSession.is_opencCs t|j|ƒS)z}Creates a :class:`mysqlx.SqlStatement` object to allow running the SQL statement on the target MySQL Server. )rrr)r9rìr0r0r1rìås z Session.sqlcCr{)z~Returns the underlying connection. Returns: mysqlx.connection.Connection: The connection object. )rrr8r0r0r1rhër}zSession.get_connectioncCs | d¡ ¡}dd„| ¡DƒS)z°Returns the list of schemas in the current session. Returns: `list`: The list of schemas in the current session. .. versionadded:: 8.0.12 zSHOW DATABASEScSsg|]}|d‘qS)rr0)r4Úrowr0r0r1r6ür7z'Session.get_schemas..)rìr?r¬r®r0r0r1Ú get_schemasószSession.get_schemascCr)zöRetrieves a Schema object from the current session by it's name. Args: name (string): The name of the Schema object to be retrieved. Returns: mysqlx.Schema: The Schema object with the given name. r©r9rhr0r0r1Ú get_schemaþr±zSession.get_schemacCsr|jj d¡}|r7| t t|ƒ¡¡ ¡ ¡}z|dd|kr&t ||ƒWSWdSt y6t d |¡ƒ‚wdS)aRetrieves a Schema object from the current session by the schema name configured in the connection settings. Returns: mysqlx.Schema: The Schema object with the given name at connect time. None: In case the default schema was not provided with the initialization data. Raises: :class:`mysqlx.ProgrammingError`: If the provided default schema does not exists. rrroN) rrrr¡rìÚ_SELECT_SCHEMA_NAME_QUERYrarr?r¬rrÒr )r9rÚresr0r0r1Úget_default_schema s$ ÿþ ÿ ýÿÿzSession.get_default_schemacCs|j dt t|ƒ¡d¡dS)z‹Drops the schema with the specified name. Args: name (string): The name of the Schema object to be retrieved. rìTN)rrrÚ_DROP_DATABASE_QUERYrarrzr0r0r1Ú drop_schema$sÿzSession.drop_schemacCs$|j dt t|ƒ¡d¡t||ƒS)z¬Creates a schema on the database and returns the corresponding object. Args: name (string): A string value indicating the schema name. rìT)rrrÚ_CREATE_DATABASE_QUERYrarrrzr0r0r1Ú create_schema-sÿ zSession.create_schemacCó|j ddd¡dS)z+Starts a transaction context on the server.rìzSTART TRANSACTIONTN©rrrr8r0r0r1Ústart_transaction8szSession.start_transactioncCrƒ)zXCommits all the operations executed after a call to startTransaction(). rìÚCOMMITTNr„r8r0r0r1Úcommit<ózSession.commitcCrƒ)zYDiscards all the operations executed after a call to startTransaction(). rìÚROLLBACKTNr„r8r0r0r1ÚrollbackBrˆzSession.rollbackNcCsX|dur d t ¡¡}nt|tƒrt| ¡ƒdkrtdƒ‚|j  dd t |ƒ¡d¡|S)aCreates a transaction savepoint. If a name is not provided, one will be generated using the uuid.uuid1() function. Args: name (Optional[string]): The savepoint name. Returns: string: The savepoint name. Nz{0}rúInvalid SAVEPOINT namerìz SAVEPOINT {0}T) raÚuuidÚuuid1r½rrIÚstripr rrrrrzr0r0r1Ú set_savepointHs   þzSession.set_savepointcCó@t|tƒr t| ¡ƒdkrtdƒ‚|j dd t|ƒ¡d¡dS)zRollback to a transaction savepoint with the given name. Args: name (string): The savepoint name. rr‹rìzROLLBACK TO SAVEPOINT {0}TN© r½rrIrŽr rrrrarrzr0r0r1Ú rollback_to]ó   þzSession.rollback_tocCr)z{Release a transaction savepoint with the given name. Args: name (string): The savepoint name. rr‹rìzRELEASE SAVEPOINT {0}TNr‘rzr0r0r1Úrelease_savepointir“zSession.release_savepointcCs|j ¡t|jƒ|_dS)zCloses the session.N)rrrrŽrqr8r0r0r1rQus z Session.closecCs|j ¡dS)z8Closes all underliying connections as pooled connectionsN)rrrÍr8r0r0r1Úclose_connections{szSession.close_connectionsrS)rƒr„r…r†r:rKrpÚsetterrrìrhryr{r~r€r‚r…r‡rŠrr’r”rQr•r0r0r0r1rl¬s.          rlc@sJeZdZdZddd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dd„Z dS)ÚClienta7Class defining a client, it stores a connection configuration. Args: connection_dict (dict): The connection information to connect to a MySQL server. options_dict (dict): The options to configure this client. .. versionadded:: 8.0.13 NcCs’||_|dur i}g|_t ¡|_| | dd¡¡| | dd¡¡| | dd¡¡|  | dd¡¡|j |jd<|j |jd<|j|jd <dS) Nrr r!rr"ÚenabledTrmr') rÚsessionsrŒÚuuid4r'Ú_set_pool_sizer¡Ú_set_max_idle_timeÚ_set_queue_timeoutÚ_set_pool_enabledÚpooling_enabledr)r9Úconnection_dictÚ options_dictr0r0r1r:Šs   zClient.__init__cCsFt|tƒst|tƒr|dkstd |¡ƒ‚|dkrt|_dS||_dS)a“Set the size of the pool. This method sets the size of the pool but it will not resize the pool. Args: pool_size (int): An integer equal or greater than 0 indicating the pool size. Raises: :class:`AttributeError`: If the pool_size value is not an integer greater or equal to 0. rzWPool max_size value must be an integer greater than 0, the given value {} is not valid.N)r½rur2rBraÚ_CNX_POOL_MAXSIZEr)r9Ú pool_sizer0r0r1r››s  þzClient._set_pool_sizecCsRt|tƒst|tƒr|dkstd |¡ƒ‚||_|dkrtnt|dƒ|jd<dS)a}Set the max idle time. This method sets the max idle time. Args: max_idle_time (int): An integer equal or greater than 0 indicating the max idle time. Raises: :class:`AttributeError`: If the max_idle_time value is not an integer greater or equal to 0. r’ziConnection max_idle_time value must be an integer greater or equal to 0, the given value {} is not valid.rr;r!N)r½rur2rBrar!Ú_CNX_POOL_MAX_IDLE_TIMEr)r9r!r0r0r1rœ°s ÿþÿ  ÿzClient._set_max_idle_timecCst|tƒs tdƒ‚||_dS)aSet if the pool is enabled. This method sets if the pool is enabled. Args: enabled (bool): True if to enabling the pool. Raises: :class:`AttributeError`: If the value of enabled is not a bool type. z*The enabled value should be True or False.N)r½rurBrŸ)r9r˜r0r0r1ržÇs  zClient._set_pool_enabledcCslt|tƒst|tƒr|dkstd |¡ƒ‚||_|dkrtnt|dƒ|jd<d|jvr4|j|jd<dSdS)a}Set the queue timeout. This method sets the queue timeout. Args: queue_timeout (int): An integer equal or greater than 0 indicating the queue timeout. Raises: :class:`AttributeError`: If the queue_timeout value is not an integer greater or equal to 0. r’ziConnection queue_timeout value must be an integer greater or equal to 0, the given value {} is not valid.rr;r"rœN)r½rur2rBrar"Ú_CNX_POOL_QUEUE_TIMEOUTr)r9r"r0r0r1rÖs ÿþÿ ÿ ÿzClient._set_queue_timeoutcCst|jƒ}|j |¡|S)z~Creates a Session instance using the provided connection data. Returns: Session: Session object. )rlrr™r)©r9Úsessionr0r0r1Ú get_sessionñs  zClient.get_sessioncCs&tƒ |j¡|jD]}| ¡q dS)z3Closes the sessions opened by this client. N)rLrkrr™r•r¦r0r0r1rQûs  ÿz Client.closerS) rƒr„r…r†r:r›rœržrr¨rQr0r0r0r1r—€s  r—)Hr†rXrWrÓr<ÚloggingrŒrÎrgÚrercÚ functoolsrÚauthenticationrrrÚerrorsrr r r r Úcompatr rrrÚcrudrÚ constantsrrÚhelpersrrr rrr¯rrrrórrrÚprotobufrr‡rrr|r¢r+Úcompiler)r¤r¥Ú getLoggerrÈr2rNr3rrŽrr'rrLrlr—r0r0r0r1Úsf     4XFNU