3 S^u @sdZyddlZdZWndZYnXddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZmZmZddlmZmZmZmZmZmZdd lmZmZmZmZdd lmZdd lm Z m!Z!dd l"m#Z#m$Z$dd l%m&Z&m'Z'ddl(m)Z)m*Z*m+Z+m,Z,ddl-m.Z.m/Z/m0Z0ddl1m2Z2ej3j4dddl5m6Z6ddl7m8Z8m9Z9dZ:dZ;dZdZ?e j@dZAdZBdZCejDdZEddZFGdd d eGZHd!d"ZIGd#d$d$eGZJGd%d&d&eJZKGd'd(d(ejLZMGd)d*d*eGZNGd+d,d,eGZOGd-d.d.eGZPdS)/z4Implementation of communication for MySQL X servers.NTF)wraps)MySQL41AuthPluginPlainAuthPluginSha256MemoryAuthPlugin)InterfaceErrorNotSupportedErrorOperationalError PoolErrorProgrammingError TimeoutError)PY3 STRING_TYPES UNICODE_TYPESqueue)Schema)SSLModeAuth)escapeget_item_or_attr)ProtocolMessageReaderWriter)Result RowResult SqlResult DocResult) SqlStatement AddStatementquote_identifier)Protobufz..)linux_distribution)VERSIONLICENSEi'zDROP DATABASE IF EXISTS {0}z!CREATE DATABASE IF NOT EXISTS {0}zLSELECT SCHEMA_NAME FROM INFORMATION_SCHEMA.SCHEMATA WHERE SCHEMA_NAME = '{}'cxz[^a-zA-Z0-9._:\-*$#]i Zmysqlxc KsTg}x8dD]0}y|jt||Wq tk r8Yq Xq W|sJtddj|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. hostportuserdatabase client_idz.Failed generating pool name; specify pool_name_)r%r&r'r(r))appendstrKeyErrorr join)kwargspartskeyr2C/tmp/pip-build-8app2_gc/mysql-connector-python/mysqlx/connection.pygenerate_pool_nameSs  r4c@sleZdZdZddZefddZddZdd Zd d Z d d Z ddZ ddZ ddZ ddZddZdS) SocketStreamzImplements a socket stream.cCsd|_d|_d|_d|_dS)NF)_socket_is_ssl _is_socket_host)selfr2r2r3__init__qszSocketStream.__init__cCs|dk r|d}ytj|||_|d|_Wnbtk ry0tjtj|_|jj||jj|d|_Wnt k rt dYnXYnXdS)zConnects to a TCP service. Args: params (tuple): The connection parameters. Raises: :class:`mysqlx.InterfaceError`: If Unix socket is not supported. NirTzUnix socket unsupported) socketcreate_connectionr6r9 ValueErrorAF_UNIX settimeoutconnectr8AttributeErrorr)r:paramsconnect_timeoutr2r2r3rAws    zSocketStream.connectcCs`|jdkrtdg}x>|dkrT|jj|}|dkrt tj fk r}z|jtdj|WYdd}~XnX|ry|j |tj|_Wn>t tj fk r}z|jtdj|WYdd}~XnX|rJy|j||Wn@t tj fk rH}z|jtdj|WYdd}~XnX|j|j|_|tjkr@g} tjdkr|jdks|jd krdd g} tj|j} | j| d g| d d } g} x\| D]T} ytj|jj| Wn0tjk r}z| j |WYdd}~XnXd } PqW| s@|jtdjdj!| d |_"dS)a Set SSL parameters. Args: ssl_mode (str): SSL mode. ssl_ca (str): The certification authority certificate. ssl_crl (str): The certification revocation lists. ssl_cert (str): The certificate. ssl_key (str): The certificate key. Raises: :class:`mysqlx.RuntimeError`: If Python installation has no SSL support. :class:`mysqlx.InterfaceError`: If the parameters are invalid. z&Python installation has no SSL support PROTOCOL_TLSzInvalid CA Certificate: {}NzInvalid CRL: {}zInvalid Certificate/Key: {}nt localhostz 127.0.0.1rrFTz$Unable to verify server identity: {}z, )# SSL_AVAILABLErPrGhasattrsslrSPROTOCOL_TLSv1 SSLContextload_default_certsload_verify_locations CERT_REQUIRED verify_modeIOErrorSSLErrorrformatVERIFY_CRL_CHECK_LEAF verify_flagsload_cert_chain wrap_socketr6rZVERIFY_IDENTITYosnamer9r< gethostbyaddrextendmatch_hostname getpeercertCertificateErrorr+r.r7)r:Zssl_modeZssl_caZssl_crlZssl_certZssl_keyprotocolcontexterrZ hostnamesaliasesZ match_foundZerrshostnamer2r2r3set_sslsd             zSocketStream.set_sslcCs|jS)zpVerifies if SSL is being used. Returns: bool: Returns `True` if SSL is being used. )r7)r:r2r2r3is_sslszSocketStream.is_sslcCs|jS)zVerifies if socket connection is being used. Returns: bool: Returns `True` if socket connection is being used. )r8)r:r2r2r3 is_socketszSocketStream.is_socketcCs |jp |jS)zvVerifies if connection is secure. Returns: bool: Returns `True` if connection is secure. )r7r8)r:r2r2r3 is_secureszSocketStream.is_securecCs |jdk S)zrVerifies if connection is open. Returns: bool: Returns `True` if connection is open. N)r6)r:r2r2r3is_openszSocketStream.is_openN)__name__ __module__ __qualname____doc__r;_CONNECT_TIMEOUTrArLrMrPrRrrrsrtrurvr2r2r2r3r5os   Kr5cstfdd}|S)zDecorator used to catch socket.error or RuntimeError. Raises: :class:`mysqlx.InterfaceError`: If `socket.Error` or `RuntimeError` is raised. c s@y|f||Stjtfk r:|jtdYnXdS)zWrapper function.zCannot connect to hostN)r<rQrG disconnectr)r:argsr/)funcr2r3wrapper.s z(catch_network_exception..wrapper)r)r~rr2)r~r3catch_network_exception'src@s4eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZeddZed d!Zed"d#Zed$d%Zed&d'Zed(d)Zed*d+Zed,d-Zed.d/Zed0d1Zed2d3Zd4d5Zd6d7Zd8d9Z d:d;Z!dd?Z#d@S)A ConnectionzkConnection to a MySQL Server. Args: settings (dict): Dictionary with connection settings. cCs||_t|_d|_d|_d|_|jd|_|jd|_|jd|_ d|_ |jdg|_ d|kr|dr|j j |jd|jdddd|_ d |_|j|j jd d d d |jd t|_|jdkrd|_d|_g|_d |_dS)Nr'passwordschemaroutersr%r&)r%r&rTcSs|dS)Npriorityr2)xr2r2r3Tsz%Connection.__init__..)r1reversezconnect-timeoutr)settingsr5stream reader_writerrm keep_openget_user _passwordZ_schema_active_result_routersr+ _cur_router _can_failover_ensure_prioritiessortr{_connect_timeout _stmt_counter_prepared_stmt_ids_prepared_stmt_supported)r:rr2r2r3r;?s2    zConnection.__init__cCs|jdk r|jjd|_dS)zFetch active result.N)r fetch_all)r:r2r2r3fetch_active_result`s  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)r:resultr2r2r3set_active_resultfs zConnection.set_active_resultcCsd}d}xN|jD]D}|jdd}|dkr:|d7}||d<n|dkrLtdd|d8}qWd|kopt|jknrtdd dS) zuEnsure priorities. Raises: :class:`mysqlx.ProgrammingError`: If priorities are invalid. rdrNrz(The priorities must be between 0 and 100iz\You must either assign no priority to any of the routers or give a priority for every routeri)rrr rH)r:Zpriority_countrrouterprir2r2r3rqs    zConnection._ensure_prioritiescCs|jsFd|_d|jkr.|jd|jjddfSd|jkrB|jdSd S|j sX|jd kr|d |_d|_x|jD] }d|d <qlW|jd7_|j|jd}|j|jd}|jd krd|j|jdd <|jt|jdkrd|_||fS)zkReturns the connection parameters. Returns: tuple: The connection parameters. Fr%r&$r<rUrT availabler)rUrrr)rrrrrrH)r:rr%r&r2r2r3_get_connection_paramss(      z!Connection._get_connection_paramscCsd}xt|jrxy@|jj|j|jt|j|_t|j|_|j |j dSt j k rt}z |}WYdd}~XqXqW|dk rt |t js|jd krt rt|jdkrtdj|jtdj|jt|jdkrtdj|tdd dS) 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)rr)rrrArrrrrrm_handle_capabilities _authenticater<rQ isinstancetimeouterrnor rHrr rar)r:rQror2r2r3rAs,      zConnection.connectcCs^|jjdtjkrdS|jjr:|jjdr6tjddS|jj j }|r`t |ddj dkndst|j tdd}tjd krt\}}}yd |ko|jd dd k}Wntk rd}YnXtjdkr| r|j td|jjdd|jj|jjdtj|jjd|jjd|jjd|jjdd|jkrZ|jd}|jj|ddS)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.rrgtlsFzSSL not enabled at serverLinuxz Oracle Linux.7 zPLAIN authentication is not allowed via unencrypted connection)rN) rrurrrrrmrrrr)r:rr2r2r3r"s    zConnection._authenticate_plaincCsHt|j|j}|jj|j|jj}|jj|j||jj dS)zBAuthenticate with the MySQL server using `Sha256MemoryAuthPlugin`.N) rrrrmrrrrrr)r:rrr2r2r3r,s  z&Connection._authenticate_sha256_memorycCs,|jr(|jj|j|jj|jd|_dS)zrDeallocates statement. Args: statement (Statement): A `Statement` based type object. FN)preparedrmsend_prepare_deallocatestmt_idrremove)r: statementr2r2r3_deallocate_statement4sz Connection._deallocate_statementc CsPy|j|jj|||Wntk r6d|_dSX|jj|jd|_dS)zPrepares a statement. Args: msg_type (str): Message ID string. msg (mysqlx.protobuf.Message): MySQL X Protobuf Message. statement (Statement): A `Statement` based type object. FNT) rrmZsend_prepare_preparerrrr+rr)r:msg_typemsgrr2r2r3_prepare_statement?szConnection._prepare_statementcCsb|js|jj|||dS|jrr|j||j||||jsR|jj|||dS|jj|||d|_|jn|jr|j r|jj|||n|j r|j r|j||jj|||d|_ |jn|j o|j r|js|j||||js |jj|||dS|jj|||n8|j rV|j rV|j||jj|||d|_ |j|j dS)zExecutes the prepared statement pipeline. Args: msg_type (str): Message ID string. msg (mysqlx.protobuf.Message): MySQL X Protobuf Message. statement (Statement): A `Statement` based type object. NF) rrmsend_msg_without_psZdeallocate_prepare_executerrZsend_prepare_executeZreset_exec_counterrchangedZrepeatedZincrement_exec_counter)r:rrrr2r2r3_execute_prepared_pipelinePs@      z%Connection._execute_prepared_pipelinecGs|j}|jdkrtdt|ts*tdt rZt|trZ|jjdt t |d|\}}n|jjd||\}}|jj |||t |S)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. NzMySQLx Connection not availablez'The SQL statement is not a valid stringsqlzutf-8) rrmr rrr r rbuild_execute_statementbytes bytearrayrr)r:rr}rrrr2r2r3send_sqls   zConnection.send_sqlcCsN|jdkrtd|jj|\}}|jj||d}t|trD|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. NzMySQLx Connection not available)rmr Z build_insertsend_msgrridsr)r:rrrrr2r2r3 send_inserts  zConnection.send_insertcCs6|jj|\}}|j||||jr.t|St|S)aDSend an find statement. Args: statement (`Statement`): It can be :class:`mysqlx.SelectStatement` or :class:`mysqlx.FindStatement`. Returns: `Result`: It can be class:`mysqlx.DocResult` or :class:`mysqlx.RowResult`. )rmZ build_findrZ is_doc_basedrr)r:rrrr2r2r3 send_finds zConnection.send_findcCs&|jj|\}}|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. )rm build_deleterr)r:rrrr2r2r3 send_deletes zConnection.send_deletecCs&|jj|\}}|j|||t|S)aSend an delete statement. Args: statement (`Statement`): It can be :class:`mysqlx.ModifyStatement` or :class:`mysqlx.UpdateStatement`. Returns: :class:`mysqlx.Result`: The result object. )rmZ build_updaterr)r:rrrr2r2r3 send_updates zConnection.send_updatec GsJy*|jj|||\}}|jj||t|Stk rD|r@YnXdS)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)rmrrrr )r: namespacecmdZ raise_on_failr}rrr2r2r3execute_nonqueryszConnection.execute_nonquerycGsP|jjd||\}}|jj||t|}|j|jdkrDtd|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. rrz No data found)rmrrrrrIr)r:rr}rrrr2r2r3execute_sql_scalars zConnection.execute_sql_scalarcGs*|jjd||\}}|jj||t|S)zReturns the row result. Args: cmd (str): The command. *args: Arbitrary arguments. Returns: :class:`mysqlx.RowResult`: The result object. Zxplugin)rmrrr)r:rr}rrr2r2r3get_row_result s zConnection.get_row_resultcCs |jj|S)zdRead row. Args: result (:class:`mysqlx.RowResult`): The result object. )rmread_row)r:rr2r2r3rszConnection.read_rowcCs|jj|dS)zeClose result. Args: result (:class:`mysqlx.Result`): The result object. N)rm close_result)r:rr2r2r3r&szConnection.close_resultcCs |jj|S)zlGet column metadata. Args: result (:class:`mysqlx.Result`): The result object. )rmget_column_metadata)r:rr2r2r3r/szConnection.get_column_metadatacCs|jd7_|jS)z|Returns the next statement ID. Returns: int: A statement ID. .. versionadded:: 8.0.16 r)r)r:r2r2r3get_next_statement_id8sz Connection.get_next_statement_idcCs |jjS)zgCheck if connection is open. Returns: bool: `True` if connection is open. )rrv)r:r2r2r3rvCszConnection.is_opencCs|js dS|jjdS)zDisconnect from server.N)rvrrP)r:r2r2r3r|KszConnection.disconnectcCs|js dSzyH|j|jr@x|jD]}|jj|q&Wd|_|jj|jjWn6t t fk r}zt j dj |WYdd}~XnXWd|jjXdS)z*Close a sucessfully authenticated session.NrzGWarning: An error occurred while attempting to close the connection: {})rvrrrrmrrZ send_closeread_okrr rrrarrP)r:rror2r2r3 close_sessionQs  zConnection.close_sessioncCsp|js dS|jdk r |jjy|jj|j|_Wn6ttfk rj}ztj dj |WYdd}~XnXdS)z*Reset a sucessfully authenticated session.NzDWarning: An error occurred while attempting to reset the session: {}) rvrrrmZ send_resetrrr rrra)r:ror2r2r3 reset_sessionis  zConnection.reset_sessioncCsB|js dS|jdk r |jj|jj|jj|jjdS)zAnnounce to the server that the client wants to close the connection. Discards any session state of the server. N)rvrrrmZsend_connection_closerrrP)r:r2r2r3rus    zConnection.close_connectionN)$rwrxryrzr;rrrrrArrrrrrrrrrrrrrrrrrrrrrvr|rrrr2r2r2r3r9s@!  &.  2           rcsLeZdZdZfddZfddZddZdd 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|tstdtt|j|j||_|jd|_|jd|_ dS)Nz&pool should be a ConnectionPool objectr%r&) rConnectionPoolrBsuperrr; cnx_configpoolr%r&)r:r) __class__r2r3r;s   zPooledConnection.__init__cstt|jdS)zGCloses the connection. This method closes the socket. N)rrr)r:)rr2r3rsz!PooledConnection.close_connectioncCs|jj|dS)aDo 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)radd_connection)r:r2r2r3rs zPooledConnection.close_sessioncCs |jdk r|jj|jdS)z#Reconnect this connection. N)rrr)r:r2r2r3 reconnects  zPooledConnection.reconnectcCs |jdS)zQReset the connection. Resets the connection by re-authenticate. N)r)r:r2r2r3resetszPooledConnection.resetcCs t||S)aCreates a :class:`mysqlx.SqlStatement` object to allow running the SQL statement on the target MySQL Server. Args: sql (string): The SQL statement to be executed. Returns: mysqlx.SqlStatement: SqlStatement object. )r)r:rr2r2r3rs zPooledConnection.sql) rwrxryrzr;rrrrr __classcell__r2r2)rr3rs   rc@s`eZdZdZddZddZeddZdd d Zdd d Z d dZ ddZ ddZ ddZ dS)raThis 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|j|d|_g|_|jdd|_tjj||j|jdd|_|jdd|_ ||_ |jdd|_ |jdd|_ ||_ |d |_|d |_dS) Nrmax_sizerT max_idle_time queue_timeoutrr%r&)_set_pool_nameZ_open_sessions_connections_opennedr pool_max_sizerQueuer;rrrrZ _priorityrr%r&)r:rgr/r2r2r3r;s  zConnectionPool.__init__cCs<tj|rtdj|t|tkr2tdj|||_dS)aSet 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_REGEXsearchrBrarH_CNX_POOL_MAX_NAME_SIZErg)r: pool_namer2r2r3rs    zConnectionPool._set_pool_namecCs t|jS)zMReturns the number of open connections that can return to this pool. )rHr)r:r2r2r3open_connections szConnectionPool.open_connectionsNcCs|jj|dS)zuRemoves a connection to this pool. Args: cnx (PooledConnection): The connection object. N)rr)r:cnxr2r2r3remove_connectionsz ConnectionPool.remove_connectioncCs|jstd|jrtd|st|}|jdjjdd}tdd|jddjd Ddkr|jd j |j j|j j |nt |tstd |j|dS)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 availablez'Failed adding connection; queue is fullzshow variables like "version"rrcSsg|] }t|qSr2)int).0nr2r2r3 5sz1ConnectionPool.add_connection..-r zset mysqlx_wait_timeout = {}z2Connection instance not subclass of PooledSession.N)rrr)rr fullrrexecutertuplerrarrr+rqueue_connection)r:rverr2r2r3rs"   zConnectionPool.add_connectionc CsVt|tstd|jr |jy|j|ddWntjk rPtdYnXdS)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. z2Connection instance not subclass of PooledSession.F)blockz'Failed adding connection; queue is fullN)rrr rputrFull)r:rr2r2r3rAs zConnectionPool.queue_connectioncCs|jj|dS)zNTracks connection in order of close it when client.close() is invoke. N)rr+)r: connectionr2r2r3track_connectionZszConnectionPool.track_connectioncCs|jS)N)rg)r:r2r2r3__str___szConnectionPool.__str__cCsx|jD] }|jqWdS)z#Empty this ConnectionPool. N)rr)r:rr2r2r3rPbs zConnectionPool.close)N)N)rwrxryrzr;rpropertyrrrrrrrPr2r2r2r3rs   'rc@sPeZdZdZdZiZddZddZddZd d Z d d Z d dZ ddZ dS) PoolsManagerzManages a pool of connections for a host or hosts in routers. This class handles all the pools of Connections. .. versionadded:: 8.0.13 NcCs"tjdkrtj|t_it_tjS)N)r_PoolsManager__instanceobject__new___PoolsManager__pools)clsr2r2r3rss  zPoolsManager.__new__cCs.|jj|g}x|D]}|j|krdSqWdS)zVerifies 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)r rrg)r:r)rpoolsrr2r2r3 _pool_existsys   zPoolsManager._pool_existscCsfg}g}|j|}x|D]\}}|j|qWx2|jj|jddgD]}|j|krF|j|qFW|S)zRetrieves 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)zNo id)_get_connections_settingsr+r rrg)r:rZavailable_poolsZ pool_namesconnections_settings router_namer*rr2r2r3 _get_poolss   zPoolsManager._get_poolscCs|j}|jdg}g}d|kr*|jdd|krTd|krT|jd|d|dd|jdddd xL|D]D}|j}|d|d<|d|d<|d |d <|jtf||fqlW|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 rr%r&r)rr%r&cSs|dS)Nrr2)rr2r2r3rsz8PoolsManager._get_connections_settings..T)r1rr)copyrpopr+rr4)r:rZ pool_settingsrr%rZconnection_settingsr2r2r3r$s&        z&PoolsManager._get_connections_settingscCs|j|}|jdd|jkr.g|j|jdd<xP|D]H\}}|j|jdd|rTq4q4|jj|jddg}|jt|f|q4WdS)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)zNo idN)r$rr r#r+r)r: cnx_settingsr%r&rrr2r2r3 create_pools zPoolsManager.create_poolc/Cs|j|}t|}xt|D]}||}y|jdkrtj4y|jd|jd}Wntj k r|t dYnXy,|j s|j |j djjdd}Wnttjtfk r8z.y |jWnttjtfk rYnXWd|j|Xt|}|j||j|j djjdd}YnXtdd |jd djd Ddkrx|j dj|jj|SQRXn.|j|jkr t|}|j||j|j djjdd}tdd |jd djd Ddkr|j dj|jj|Stjyt|jd|jd}|j |j djjdd}tdd |jd djd Ddkr|j dj|jj|Stj k rt dYnXWdQRXWqttfk r||dkrނnwYqXqWdS)aGet 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)rrz)Failed getting connection; pool exhaustedzshow variables like "version"rNcSsg|] }t|qSr2)r )r r r2r2r3r  sz/PoolsManager.get_connection..r rrrzset mysqlx_wait_timeout = {}cSsg|] }t|qSr2)r )r r r2r2r3r scSsg|] }t|qSr2)r )r r r2r2r3r )szpool max size has been reached)rrr)rrr)rrr) r'rHrangeqsize threadingRLockrrrEmptyr rrrrrrGr<rQrrrrrrArrrarrrr )r:rr" num_poolsZ pool_numberrrrr2r2r3get_connections              zPoolsManager.get_connectioncCs\|j|}xH|D]@}|j|jdddk r|jj|jd}||kr|j|qWt|S)zjCloses the connections in the pools Returns: int: The number of closed pools r)N)r'rPrr rrH)r:r*r"rZ client_poolsr2r2r3 close_pool8s  zPoolsManager.close_pool) rwrxryrzrr rr#r'r$r+r2r3r2r2r2r3ris!^rc@seZdZdZddZddZeddZejddZd d Z d d Z d dZ ddZ ddZ ddZddZddZddZddZddZd*d d!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. cCs|jdtj|_||_d|jks.|jddkr@i|jd<|jd|kr|drtj|tj||_|jdkrt d|dnt |j|_|jj |jjd}|ry|j d j t|jWnJtk r}z,|jd kr|jnd j |}t||jWYdd}~XnXdS) Nzuse-purezconnection-attributesFrpoolingz.connection could not be retrieve from pool. %s)valuesrzUSE {}iz#Default schema '{}' does not exists)rruse_pure _settings_init_attributesrr+r2 _connectionr rrArrarrr rrr)r:rrroerrmsgr2r2r3r;Ws,         zSession.__init__c CstjdkrRdtjdkr d}ndtjdkr6d}ntj}djtjd}n>tj}tjd kr|d jd tjd}nd j t dd }t j d}|ddkrd}nd}t tj||tjddj ddtddD|d}|jdj|d|jkrx|jdD]}|jd|}t|ts.)Z_pid _platform_osZ _source_hostZ _client_nameZ_client_versionZ_client_licenserzconnection-attributesz)Attribute name '{}' must be a string type z4Attribute name '{}' exceeds 32 characters limit sizer*zJKey names in 'session-connect-attributes' cannot start with '_', found: {}z4Attribute name '{}' value '{}' must be a string typeizBAttribute name '{}' value: '{}' exceeds 1024 characters limit sizeN)rfrgr architecturera win32_vermachinermac_verr.r r"rr,getpidr< gethostnamer!r8updaterrrrH startswith)r:Z platform_archZos_verZlicense_chunksZclient_licenseZdefault_attributes attr_name attr_valuer2r2r3r9usT           zSession._init_attributescCstjS)zAbool: `True` to use pure Python Protobuf implementation. )rr7)r:r2r2r3r7szSession.use_purecCs t|tstdtj|dS)Nz)'use_pure' option should be True or False)rboolr rZ set_use_pure)r:valuer2r2r3r7s cCs |jjjS)zzReturns `True` if the session is open. Returns: bool: Returns `True` if the session is open. )r:rrv)r:r2r2r3rvszSession.is_opencCs t|j|S)aCreates a :class:`mysqlx.SqlStatement` object to allow running the SQL statement on the target MySQL Server. Args: sql (string): The SQL statement to be executed. Returns: mysqlx.SqlStatement: SqlStatement object. )rr:)r:rr2r2r3rs z Session.sqlcCs|jS)z~Returns the underlying connection. Returns: mysqlx.connection.Connection: The connection object. )r:)r:r2r2r3r2szSession.get_connectioncCs |jdj}dd|jDS)zReturns the list of schemas in the current session. Returns: `list`: The list of schemas in the current session. .. versionadded:: 8.0.12 zSHOW DATABASEScSsg|] }|dqS)rr2)r rowr2r2r3r sz'Session.get_schemas..)rrr)r:rr2r2r3 get_schemasszSession.get_schemascCs t||S)zRetrieves 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)r:rgr2r2r3 get_schemas zSession.get_schemac Cst|jjjd}|rp|jtjt|jj}y|dd|krJt ||SWn"t k rnt dj|YnXdS)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. rrz#Default schema '{}' does not existsN) r:rrr_SELECT_SCHEMA_NAME_QUERYrarrrrrr )r:rresr2r2r3get_default_schemaszSession.get_default_schemacCs|jjdtjt|ddS)zDrops the schema with the specified name. Args: name (string): The name of the Schema object to be retrieved. rTN)r:r_DROP_DATABASE_QUERYrar)r:rgr2r2r3 drop_schemaszSession.drop_schemacCs$|jjdtjt|dt||S)zCreates a schema on the database and returns the corresponding object. Args: name (string): A string value indicating the schema name. rT)r:r_CREATE_DATABASE_QUERYrarr)r:rgr2r2r3 create_schemaszSession.create_schemacCs|jjddddS)z+Starts a transaction context on the server.rzSTART TRANSACTIONTN)r:r)r:r2r2r3start_transaction*szSession.start_transactioncCs|jjddddS)zXCommits all the operations executed after a call to startTransaction(). rZCOMMITTN)r:r)r:r2r2r3commit.szSession.commitcCs|jjddddS)zYDiscards all the operations executed after a call to startTransaction(). rZROLLBACKTN)r:r)r:r2r2r3rollback4szSession.rollbackNcCsZ|dkrdjtj}n$t|t s4t|jdkrsn         9MIa+