a còf¿bã @s(dZz\ddlZdZejejejdœZeedƒr:ejed<n ej ed<eedƒrZej rZdZ ndZ WndZdZ Yn0ddl Z ddl Z ddlZddlZddlZddlZddlZddlZddlZddlZddlZzddlZddlZWneyødZYn0dZdd lmZmZdd lmZd d l m!Z!m"Z"m#Z#d d l$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*d dl+m,Z,d dl-m.Z.m/Z/m0Z0d dl1m2Z2m3Z3m4Z4d dl5m6Z6m7Z7m8Z8m9Z9m:Z:d dl;mZ>m?Z?m@Z@d dlAmBZBmCZCmDZDd dlEmFZFe jG Hd¡ddlImJZJddlKmLZLmMZMdZNdZOdZPdZQdZRdZSdZTe Ud¡ZVd ZWd ZXd!ZYd"ZZd#Z[d#Z\d$Z]d"Z^eZeZeYeYe\e[e[e]d%œZ_d&e^iZ`d'd(d)d*œZae bd+¡Zcd,d-„Zdd.d/„ZeGd0d1„d1efƒZgd2d3„ZhGd4d5„d5eiƒZjGd6d7„d7ƒZkGd8d9„d9efƒZlGd:d;„d;elƒZmGdd?„d?efƒZpGd@dA„dAefƒZqGdBdC„dCefƒZrdS)Dz4Implementation of communication for MySQL X servers.éNT)ÚTLSv1úTLSv1.1úTLSv1.2Ú PROTOCOL_TLSúTLSv1.3Ú HAS_TLSv1_3F)ÚdatetimeÚ timedelta©Úwrapsé)ÚMySQL41AuthPluginÚPlainAuthPluginÚSha256MemoryAuthPlugin)ÚInterfaceErrorÚNotSupportedErrorÚOperationalErrorÚ PoolErrorÚProgrammingErrorÚ TimeoutError©ÚSchema)ÚSSLModeÚAuthÚCOMPRESSION_ALGORITHMS)ÚescapeÚget_item_or_attrÚiani_to_openssl_cs_name)ÚProtocolÚ MessageReaderÚ MessageWriterÚHAVE_LZ4Ú HAVE_ZSTD)Ú BaseResultÚResultÚ RowResultÚ SqlResultÚ DocResult)Ú SqlStatementÚ AddStatementÚquote_identifier)ÚProtobufz..)Úlinux_distribution)ÚVERSIONÚLICENSEi'zDROP DATABASE IF EXISTS {0}z!CREATE DATABASE IF NOT EXISTS {0}zLSELECT SCHEMA_NAME FROM INFORMATION_SCHEMA.SCHEMATA WHERE SCHEMA_NAME = '{}'zSHOW VARIABLES LIKE "VERSION"écéxz[^a-zA-Z0-9._:\-*$#]i›Ä i@Bé<ii€Q)z[WinError 10053]z [Errno 32]z[WinError 10061]z [Errno 111]z[WinError 10060]z [Errno 11001]z [Errno -2]z Access deniedéz‘This session was closed because the connection has been idle for too long. Use "mysqlx.getSession()" or "mysqlx.getClient()" to create a new one.z.wrapperr )r¹rºr@r¸rAÚcatch_network_exception—s+r»c@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S) ÚRouterz“Represents a set of connection parameters. Args: settings (dict): Dictionary with connection settings .. versionadded:: 8.0.20 cCs| |¡| dd¡|d<dS)NÚ availableT)rFÚget)rNZconnection_paramsr@r@rArOÔs zRouter.__init__cCs|dS)z’Verifies if the Router is available to open connections. Returns: bool: True if this Router is available else False. r½r@rMr@r@rAr½ØszRouter.availablecCs d|d<dS)z:Sets this Router unavailable to open connections. Fr½Nr@rMr@r@rAÚset_unavailableàszRouter.set_unavailablecCs d|vr|dS|d|dfS)z‘Verifies if the Router is available to open connections. Returns: tuple: host and port or socket information tuple. rQr4r5r@rMr@r@rAÚget_connection_paramsåszRouter.get_connection_paramsN)ržrŸr r¡rOr½r¿rÀr@r@r@rAr¼Ís r¼c@sHeZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dS)Ú RouterManagerzÒManages the connection parameters of all the routers. Args: Routers (list): A list of Router objects. settings (dict): Dictionary with connection settings. .. versionadded:: 8.0.20 cCs0||_||_d|_d|_i|_g|_| ¡dS)NrT)Ú_routersÚ _settingsÚ_cur_priority_idxÚ _can_failoverÚ_routers_directoryÚrouters_priority_listÚ_ensure_priorities)rNÚroutersÚsettingsr@r@rArOøszRouterManager.__init__cCsÜd}|jD]<}| dd¡}|dur4|d7}d|d<q |dkr tddƒ‚q d|krbt|jƒkrpnn tdd ƒ‚|jjd d „d d |jD]L}|d}||j ¡vrÂt|ƒg|j|<|j  |¡qŠ|j|  t|ƒ¡qŠdS)zuEnsure priorities. Raises: :class:`mysqlx.ProgrammingError`: If priorities are invalid. rÚpriorityNr édz(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 cSs|dS)NrËr@©Úxr@r@rAÚr[z2RouterManager._ensure_priorities..T)r?rm) rÂr¾rr^ryrÆÚkeysr¼rÇr9)rNZpriority_countÚrouterrËr@r@rArÈs*   ÿþ z RouterManager._ensure_prioritiescCs|j|}dd„|Dƒ}|S)z¡Get a list of the current available routers that shares the given priority. Returns: list: A list of the current available routers. cSsg|]}| ¡r|‘qSr@)r½)Ú.0rÑr@r@rAÚ *r[z8RouterManager._get_available_routers..©rÆ)rNrËÚ router_listr@r@rAÚ_get_available_routers#s z$RouterManager._get_available_routerscCsF| |¡}|sdSt|ƒdkr&|dSt|ƒd}t d|¡}||S)z{Get a random router from the group with the given priority. Returns: Router: A random router. Nr r)rÖr^ÚrandomÚrandint)rNrËrÕÚlastÚindexr@r@rAÚ_get_random_connection_params-s    z+RouterManager._get_random_connection_paramscCs|jS)zŒReturns the next connection parameters. Returns: bool: True if there is more server to failover to else False. )rÅrMr@r@rAÚ can_failover=szRouterManager.can_failovercCsÎ|jsBd|_|j ¡}|j dd¡|d<|j dd¡|d<t|ƒS|j|j}t|jƒ}d}|rÊ|  |¡}|dus||j|kr¤|j|dkrÊt|  |¡ƒd krÊd|_qÊ|jd7_|j|kr\|j|j}q\|S) zvReturns the next connection parameters. Returns: Router: with the connection parameters. Fr4rqr5i$TNr é) rÂrÅrÃÚcopyr¾r¼rÇrÄr^rÛrÖ)rNZrouter_settingsÚ cur_priorityZrouters_priority_lenÚsearchrÑr@r@rAÚget_next_routerEs.    ÿÿ zRouterManager.get_next_routercCs|jS)z™Returns the directory containing all the routers managed. Returns: dict: Dictionary with priorities as connection settings. rÔrMr@r@rAÚget_routers_directorygsz#RouterManager.get_routers_directoryN) ržrŸr r¡rOrÈrÖrÛrÜrárâr@r@r@rArÁðs ! "rÁc@sHeZdZdZdd„Zdd„Zdd„Zdd „Zd d „ZdEd d„Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zedd„ƒZedd „ƒZed!d"„ƒZed#d$„ƒZed%d&„ƒZedFd'd(„ƒZed)d*„ƒZed+d,„ƒZed-d.„ƒZed/d0„ƒZed1d2„ƒZd3d4„Zd5d6„Zd7d8„Zd9d:„Z d;d<„Z!d=d>„Z"d?d@„Z#dAdB„Z$dCdD„Z%d S)Gr¦zkConnection to a MySQL Server. Args: settings (dict): Dictionary with connection settings. cCsÐ||_tƒ|_d|_d|_| d¡|_| d¡|_| d¡|_d|_ | dg¡|_ d|vr‚|dr‚|j   | d¡| dd¡dœ¡t |j |ƒ|_ | dt¡|_|jd kr®d|_d |_g|_d |_d |_d|_dS) Nr6ÚpasswordÚschemarÉr4r5)r4r5úconnect-timeoutrTF)rÊrHÚstreamÚprotocolÚ keep_openr¾Ú_userÚ _passwordZ_schemaÚ_active_resultrÂr9rÁÚrouter_managerr¢Ú_connect_timeoutÚ _stmt_counterÚ_prepared_stmt_idsÚ_prepared_stmt_supportedÚ_server_disconnectedÚ_server_disconnected_reason)rNrÊr@r@rArOvs2    þÿ zConnection.__init__cCs|jdur|j ¡d|_dS)zFetch active result.N)rëÚ fetch_allrMr@r@rAÚfetch_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ë©rNr´r@r@rAÚset_active_result›s zConnection.set_active_resultc Csšd}|j ¡r&zÚ|j ¡}|j | ¡|j¡t|jƒ}t|jƒ}t ||ƒ|_ |j   ¡j }|rndd„|Dƒni}|  |¡d|jvrž|jd}|j j|d|j dd¡}|j d¡} |d krÄdn | ||| ¡} | ¡|j  | ¡WdStjtfy"} z| }| ¡WYd} ~ qd} ~ 00q|durnt|tjƒrnt|jƒd kr^td  |j¡ƒ‚td  |j¡ƒ‚t|jƒd krŒtd  |¡ƒ‚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. NcSsi|]}t|dƒ ¡|“qS©rŒ)rÚlower)rÒÚcapr@r@rAÚ ºsÿ z&Connection.connect..Ú attributes©Zsession_connect_attrsÚ compressionÚ preferredzcompression-algorithmsÚdisabledr zLConnection 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}ú,Unable to connect to any of the target hostsi¡)rìrÜrárærVrÀrírr rrçZget_capabilitesZ capabilitiesÚ_set_tls_capabilitiesrÊÚset_capabilitiesr¾Ú_set_compression_capabilitiesÚ _authenticateZset_compressionrQrdr]r¿rCÚtimeoutr^rÂrrer) rNrdrÑÚreaderÚwriterZ caps_dataÚcapsÚ conn_attrsrýÚ algorithmsÚ algorithmrfr@r@rArV¦sd    ÿ    ýþý   ÿ  þÿ  þþÿzConnection.connectc CsN|j d¡tjkrdS|j ¡r:|j d¡r6t d¡dSd|vrR| ¡t dƒ‚d}t   ¡dkr¤t ƒ\}}}zd|voˆ|  d ¡d d k}Wnty¢d}Yn0tjd krÂ|sÂ| ¡td ƒ‚|jjdd|j |j dd¡|j dtj¡|j d¡|j d¡|j d¡|j d¡|j d¡¡d|jvrJ|jd}|jj|ddS)a©Set the TLS capabilities. Args: caps (dict): Dictionary with the server 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. .. versionadded:: 8.0.21 zssl-modeNz(SSL not required when using Unix socket.ÚtlszSSL not enabled at serverFÚLinuxz Oracle LinuxÚ.rÚ7)rÝéé zdˆvr>d}nd}|ˆvrpd}|dkrbt |ƒ‚t |¡dS|jjd |id|S)aSet the compression capabilities. If compression is available, negociates client and server algorithms. By trying to find an algorithm from the requested compression algorithms list, which is supported by the server. If no compression algorithms list is provided, the following priority is used: 1) zstd_stream 2) lz4_message 3) deflate_stream Args: caps (dict): Dictionary with the server capabilities. compression (str): The compression connection setting. algorithms (list): List of requested compression algorithms. Returns: str: The compression algorithm. .. versionadded:: 8.0.21 .. versionchanged:: 8.0.22 rýNz8Compression requested but the server does not support itÚrequiredÚvalueÚobjÚfldcSs"g|]}|ddd d¡‘qS)ÚscalarÚv_stringrúutf-8)Údecode©rÒrr@r@rArÓ?sÿz.Úarrayr?cSsg|]}|jjj d¡‘qS)r!)rr rr"r#r@r@rArÓEsÿr cSsg|]}|tvrt|‘qSr@)r©rÒÚitemr@r@rArÓQsÿcsg|]}|ˆvr|‘qSr@r@r%©Zserver_algorithmsr@rArÓUsÿrzmThe connection compression is set as required, but none of the provided compression algorithms are supported.Z zstd_streamZ lz4_messageZdeflate_streamzFCompression requested but the compression algorithm negotiation failed)rý)r¾rrrµrCrDrrrr$r?rrr"r!rçr) rNrrýr Zcompression_datar¤Zcompression_dictrr Zclient_algorithmsÚmatchedr@r'rArsZ   þþ ÿ ÿ      z(Connection._set_compression_capabilitiescCs°|j d¡}|rL|tjkr$| ¡q¬|tjkr8| ¡q¬|tjkr¬| ¡n`|j   ¡r`| ¡nLz | ¡Wnt y~Yn0dSz | ¡Wnt yªt dƒ‚Yn0dS)z#Authenticate with the MySQL server.ÚauthNzmAuthentication failed using MYSQL41 and SHA256_MEMORY, check username and password or try a secure connection) rÊr¾rZPLAINÚ_authenticate_plainZ SHA256_MEMORYÚ_authenticate_sha256_memoryZMYSQL41Ú_authenticate_mysql41rærœr)rNr)r@r@rArws&             zConnection._authenticatecCsHt|j|jƒ}|j | ¡¡|j ¡}|j | |¡¡|j  ¡dS)z=Authenticate with the MySQL server using `MySQL41AuthPlugin`.N) r rérêrçÚsend_auth_startÚ auth_nameÚread_auth_continueÚsend_auth_continueÚ auth_dataÚ read_auth_ok©rNÚpluginÚ extra_datar@r@rAr,”s  z Connection._authenticate_mysql41cCsF|j ¡stdƒ‚t|j|jƒ}|jj| ¡|  ¡d|j  ¡dS)z;Authenticate with the MySQL server using `PlainAuthPlugin`.z>PLAIN authentication is not allowed via unencrypted connection)r1N) rærœrrrérêrçr-r.r1r2)rNr4r@r@rAr*œs  ÿzConnection._authenticate_plaincCsHt|j|jƒ}|j | ¡¡|j ¡}|j | |¡¡|j  ¡dS)zBAuthenticate with the MySQL server using `Sha256MemoryAuthPlugin`.N) rrérêrçr-r.r/r0r1r2r3r@r@rAr+¦s  z&Connection._authenticate_sha256_memorycCs,|jr(|j |j¡|j |j¡d|_dS)zrDeallocates statement. Args: statement (Statement): A `Statement` based type object. FN)ÚpreparedrçÚsend_prepare_deallocateÚstmt_idrïÚremove)rNÚ statementr@r@rAÚ_deallocate_statement®sz Connection._deallocate_statementcCsPz| ¡|j |||¡Wnty6d|_YdS0|j |j¡d|_dS)zåPrepares 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) rôrçZsend_prepare_preparerrðrïr9r8r6©rNÚmsg_typer¤r:r@r@rAÚ_prepare_statement¹s zConnection._prepare_statementcCs\|js|j |||¡dS|jrr| |¡| |||¡|jsR|j |||¡dS|j |||¡d|_| ¡nÞ|jr|j s|j |||¡nÀ|j rÆ|j sÆ| |¡|j |||¡d|_ | ¡nŠ|j s|j s|jsê| |||¡|js|j |||¡dS|j |||¡n8|j rP|j rP| |¡|j |||¡d|_ | ¡|  ¡dS)zùExecutes 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) rðrçÚsend_msg_without_psZdeallocate_prepare_executer;r>Zsend_prepare_executeZreset_exec_counterr6ÚchangedZrepeatedZincrement_exec_counterr<r@r@rAÚ_execute_prepared_pipelineÊs@        z%Connection._execute_prepared_pipelinecCsV|j}|jdurtdƒ‚t|tƒs,tdƒ‚n|j d|¡\}}|j |||¡t|ƒS)zóExecute a SQL statement. Args: sql (str): The SQL statement. Raises: :class:`mysqlx.ProgrammingError`: If the SQL statement is not a valid string. NrZz'The SQL statement is not a valid stringÚsql) rBrçrrCr:rÚbuild_execute_statementr?r&)rNr:rBr=r¤r@r@rAÚsend_sqlüs    ÿzConnection.send_sqlcCsN|jdurtdƒ‚|j |¡\}}|j ||¡d}t|tƒrD|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. NrZ)rçrZ build_insertÚsend_msgrCr)Úidsr$)rNr:r=r¤rFr@r@rAÚ send_inserts  zConnection.send_insertcCs6|j |¡\}}| |||¡| ¡r.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`. )rçZ build_findrAZ is_doc_basedr'r%©rNr:r=r¤r@r@rAÚ send_find&s zConnection.send_findcCs&|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çZ build_deleterAr$rHr@r@rAÚ send_delete6s zConnection.send_deletecCs&|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. )rçZ build_updaterAr$rHr@r@rAÚ send_updateEs zConnection.send_updatecCsJz,|j |||¡\}}|j ||¡t|ƒWStyD|r@‚Yn0dS)a‹Execute a non query command. Args: namespace (str): The namespace. cmd (str): The command. raise_on_fail (bool): `True` to raise on fail. fields (Optional[dict]): The message fields. Raises: :class:`mysqlx.OperationalError`: On errors. Returns: :class:`mysqlx.Result`: The result object. N)rçrCrEr$r)rNÚ namespaceÚcmdZ raise_on_failÚfieldsr=r¤r@r@rAÚexecute_nonqueryTsÿ  zConnection.execute_nonquerycCsN|j d|¡\}}|j ||¡t|ƒ}| ¡|jdkrBtdƒ‚|ddS)zèExecute a SQL scalar. Args: sql (str): The SQL statement. Raises: :class:`mysqlx.InterfaceError`: If no data found. Returns: :class:`mysqlx.Result`: The result. rBrz No data found)rçrCrEr%rór_r)rNrBr=r¤r´r@r@rAÚexecute_sql_scalarms  zConnection.execute_sql_scalarcCs*|j d||¡\}}|j ||¡t|ƒS)zÎReturns the row result. Args: cmd (str): The command. fields (dict): The message fields. Returns: :class:`mysqlx.RowResult`: The result object. r3)rçrCrEr%)rNrMrNr=r¤r@r@rAÚget_row_result‚s ÿzConnection.get_row_resultcCs |j |¡S)zdRead row. Args: result (:class:`mysqlx.RowResult`): The result object. )rçÚread_rowrõr@r@rArR’szConnection.read_rowcCs|j |¡dS)zeClose result. Args: result (:class:`mysqlx.Result`): The result object. N)rçÚ close_resultrõr@r@rArS›szConnection.close_resultcCs |j |¡S)zlGet column metadata. Args: result (:class:`mysqlx.Result`): The result object. )rçr¥rõr@r@rAr¥¤szConnection.get_column_metadatacCs|jd7_|jS)z|Returns the next statement ID. Returns: int: A statement ID. .. versionadded:: 8.0.16 r )rîrMr@r@rAÚget_next_statement_id­sz Connection.get_next_statement_idcCs |j ¡S)zgCheck if connection is open. Returns: bool: `True` if connection is open. )rærrMr@r@rAr¸szConnection.is_opencCsd|_||_dS)z†Set the disconnection message from the server. Args: reason (str): disconnection reason from the server. TN)rñrò)rNr·r@r@rAr«Àsz"Connection.set_server_disconnectedcCs|jS)zÅVerify if the session has been disconnect from the server. Returns: bool: `True` if the connection has been closed from the server otherwise `False`. )rñrMr@r@rAr¨Ész!Connection.is_server_disconnectedcCs|jS)z†Get the disconnection message sent by the server. Returns: string: disconnection reason from the server. )ròrMr@r@rAr©Òsz"Connection.get_disconnected_reasoncCs| ¡s dS|j ¡dS)zDisconnect from server.N)rrærirMr@r@rAr²ÚszConnection.disconnectc Cs¬| ¡s dSzŽzD| ¡|jr<|jD]}|j |¡q$d|_|j ¡|j ¡Wn:t t t fyŒ}zt   d |¡¡WYd}~n d}~00W|j ¡n |j ¡0dS)z*Close a sucessfully authenticated session.NrzGWarning: An error occurred while attempting to close the connection: {})rrôrðrïrçr7rîZ send_closeÚread_okrrÚOSErrorrrµreræri)rNr8rfr@r@rAÚ close_sessionàs  ÿzConnection.close_sessionc Csr| ¡s dS|jdur |j ¡z|j |j¡|_Wn8ttfyl}zt  d  |¡¡WYd}~n d}~00dS)z*Reset a sucessfully authenticated session.NzDWarning: An error occurred while attempting to reset the session: {}) rrërórçZ send_resetrèrrrrµre)rNrfr@r@rAÚ reset_sessionøs  ÿzConnection.reset_sessioncCsB| ¡s dS|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çZsend_connection_closerUrærirMr@r@rArs    zConnection.close_connection)N)N)&ržrŸr r¡rOrôrörVrrrr,r*r+r;r>rAr»rDrGrIrJrKrOrPrQrRrSr¥rTrr«r¨r©r²rWrXrr@r@r@rAr¦ps\ @5ÿ ]  2               r¦csLeZdZdZ‡fdd„Z‡fdd„Zdd„Zdd „Zd d „Zd d „Z ‡Z S)r§aClass 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ƒstdƒ‚tt|ƒ |j¡||_|jd|_|jd|_ dS)Nz&pool should be a ConnectionPool objectr4r5) rCÚConnectionPoolrWÚsuperr§rOÚ cnx_configr®r4r5)rNr®©Ú __class__r@rArO#s   zPooledConnection.__init__cstt|ƒ ¡dS)zGCloses the connection. This method closes the socket. N)rZr§rWrMr\r@rAr+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_connectionrMr@r@rArW2s zPooledConnection.close_sessioncCs |jdur|j ¡| ¡dS)z#Reconnect this connection. N)rërórrMr@r@rAÚ reconnect?s  zPooledConnection.reconnectcCs | ¡dS)zQReset the connection. Resets the connection by re-authenticate. N)r_rMr@r@rAÚresetFszPooledConnection.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(©rNrBr@r@rArBMs zPooledConnection.sql) ržrŸr r¡rOrrWr_r`rBÚ __classcell__r@r@r\rAr§s   r§c@sŠeZdZdZdd„Zdd„Zedd„ƒZd d d „Zd d „Z d!d d„Z dd„Z dd„Z dd„Z dd„Zd"dd„Zdd„Zdd„Zdd„ZdS)#rYaµ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|_d|_d|_t ¡|_| dd¡|_ t j   ||j ¡| dd¡|_ | dd¡|_||_| dd¡|_| dd¡|_||_|d |_|d |_dS) NrTÚmax_sizeérXÚ max_idle_timeÚ queue_timeoutrËr4r5)Ú_set_pool_nameZ_open_sessionsÚ_connections_opennedÚ _availableÚ_timeoutrÚnowÚ_timeout_stampr¾Ú pool_max_sizeÚqueueÚQueuerOrXrfrÊrgrËr[r4r5)rNrŒr=r@r@rArOvs    zConnectionPool.__init__cCs<t |¡rtd |¡ƒ‚t|ƒtkr2td |¡ƒ‚||_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_REGEXràrWrer^Ú_CNX_POOL_MAX_NAME_SIZErŒ)rNÚ pool_namer@r@rArh‰s ÿ ÿzConnectionPool._set_pool_namecCs t|jƒS)zMReturns the number of open connections that can return to this pool. )r^rirMr@r@rAÚopen_connectionsžszConnectionPool.open_connectionsNcCs|j |¡dS)zwRemoves a connection from this pool. Args: cnx (PooledConnection): The connection object. N)rir9©rNÚcnxr@r@rAÚremove_connection¤sz ConnectionPool.remove_connectionc Cs€| ¡dkr|z|jd|jd}Wntjy4Yq0z6z | ¡Wnttjt fy^Yn0W|  |¡q|  |¡0qdS)z*Removes all the connections from the pool.rT©ÚblockrN) Úqsizer¾rgroÚEmptyrr]rQrdrrwrur@r@rAr¯¬s ÿ  z!ConnectionPool.remove_connectionscCsÆ|jstdƒ‚| ¡rtdƒ‚|sŽt|ƒ}| d¡ ¡ ¡dd}tdd„| d¡d d ¡Dƒƒd kr€| d   |j ¡¡ ¡|j   |¡n*t |tƒs td ƒ‚| ¡r¸| ¡| ¡| |¡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 fullzshow variables like "version"rr cSsg|] }t|ƒ‘qSr@©Úint©rÒÚnr@r@rArÓ×r[z1ConnectionPool.add_connection..ú-r©éré úset mysqlx_wait_timeout = {}ú2Connection instance not subclass of PooledSession.N)r[rÚfullr§rBÚexecuteróÚtuplerrerfrir9rCr¨r¯riÚqueue_connection)rNrvÚverr@r@rAr^¼s4ÿÿ ÿÿ  ÿzConnectionPool.add_connectioncCsTt|tƒstdƒ‚|jr | ¡z|j|ddWntjyNtdƒYn0dS)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)ryr|N)rCr§rrXÚputroÚFullrur@r@rArŠæs ÿzConnectionPool.queue_connectioncCs|j |¡dS)zNTracks connection in order of close it when client.close() is invoke. N)rir9)rNÚ connectionr@r@rAÚtrack_connectionÿszConnectionPool.track_connectioncCs|jSrjr÷rMr@r@rAÚ__str__szConnectionPool.__str__cCs|jS)z¸Returns if this pool is available for pool connections from it. Returns: bool: True if this pool is available else False. .. versionadded:: 8.0.20 )rjrMr@r@rAr½szConnectionPool.availableéÿÿÿÿcCs.|jr*t d||¡d|_t ¡|_||_dS)zhSets this pool unavailable for a period of time (in seconds). .. versionadded:: 8.0.20 z4ConnectionPool.set_unavailable pool: %s time_out: %sFN)rjrrµrrlrmrk)rNZtime_outr@r@rAr¿sÿ zConnectionPool.set_unavailablecCsd|_t ¡|_dS)zaSets this pool available for pool connections from it. .. versionadded:: 8.0.20 TN)rjrrlrmrMr@r@rAÚ set_availableszConnectionPool.set_availablecCs |j|jfS)zÎReturns the penalized time (timeout) and the time at the penalty. Returns: tuple: penalty seconds (int), timestamp at penalty (datetime object) .. versionadded:: 8.0.20 )rkrmrMr@r@rAÚget_timeout_stamp$sz ConnectionPool.get_timeout_stampcCs|jD] }| ¡qdS)z#Empty this ConnectionPool. N)rirrur@r@rAri-s zConnectionPool.close)N)N)r‘)ržrŸr r¡rOrhÚpropertyrtrwr¯r^rŠrrr½r¿r’r“rir@r@r@rArYZs    *  rYc@s„eZdZdZdZiZdd„Zdd„Zdd„Zd d „Z d d „Z d d„Z dd„Z dd„Z ddd„Zddd„Zdd„Zdd„Zdd„ZdS)r°z–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_tjSrj)r°Ú_PoolsManager__instanceÚobjectÚ__new__Ú_PoolsManager__pools)Úclsr@r@rAr—>s  zPoolsManager.__new__cCs,|j |g¡}|D]}|j|krdSqdS)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)r˜r¾rŒ)rNr7rsÚpoolsr®r@r@rAÚ _pool_existsDs  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. r7úNo id)Ú_get_connections_settingsr9r˜r¾rŒ)rNrÊZavailable_poolsZ pool_namesÚconnections_settingsÚ router_namer8r®r@r@rAÚ _get_poolsTs      zPoolsManager._get_poolscCsÈ| ¡}| dg¡}g}d|vr*| d¡d|vrVd|vrV| dd|d|ddœ¡|jdd„d |D]X}| ¡}|d|d<|d|d<|d |d <| d d¡|d <| tfi|¤Ž|f¡qj|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Ér4r5rÌr)rËÚweightr4r5cSs|d| dd¡ fS)NrËr¡r)r¾rÍr@r@rArÏ‚r[z8PoolsManager._get_connections_settings..)r?rËr¡)rÞr¾Úpopr9ryrB)rNrÊZ pool_settingsrÉržrÑZconnection_settingsr@r@rArks0   ý    ÿÿz&PoolsManager._get_connections_settingscCs„| |¡}| dd¡|jvr.g|j| dd¡<|D]L\}}| | dd¡|¡rRq2q2|j | dd¡g¡}| t|fi|¤Ž¡q2dS)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. r7rœN)rr¾r˜r›r9rY)rNÚ cnx_settingsržrŸrÊr®r@r@rAÚ create_poolŽs  ÿÿzPoolsManager.create_poolcCs<|sdSt|ƒdkr|dSt|ƒd}t d|¡}||S)zGet a random router from the group with the given priority. Returns: Router: a random router. .. versionadded:: 8.0.20 Nr r)r^r×rØ)rNZ pool_listrÙrÚr@r@rAÚ_get_random_pool§s   zPoolsManager._get_random_poolcCsZg}d}|t|ƒkrV||j}||krB|| ¡rB| ||¡n ||krLqV|d7}q|S)Nr )r^rËr½r9)rNršrÚrßZsublistZ next_priorityr@r@rAÚ get_sublist¸s   zPoolsManager.get_sublistcCsbd}|D]"}| ¡r"||jkr"q,|d7}qg}|sX|t|ƒkrX| |||¡}|d7}q0| |¡S)Nrr )r½rËr^r¦r¥)rNršrßrÚr®Zsubpoolr@r@rAÚ_get_next_poolÄs  zPoolsManager._get_next_poolcCs@|dur|r|djS|D]}| ¡r|j}|Sq|djS)Nr)rËr½)rNršrßZt_poolr@r@rAÚ_get_next_priorityÐs   zPoolsManager._get_next_prioritycCsT| |¡}|D]@}| ¡rq| ¡\}}|r0|}t ¡|t|dkr| ¡qdS)N)Úseconds)r r½r“rrlr r’)rNrÊZreviveršr®rZ timeout_stampr@r@rAÚ_check_unavailable_poolsÛs  z%PoolsManager._check_unavailable_poolsc s†‡fdd„}| |¡}| dd¡}g}| |¡| ||¡}|durLtdƒ‚||d<| ||¡‰ˆdurzz:ˆ ¡dkrît ¡Tzˆjdˆj d}Wnt j y¸td ƒ‚Yn0z,|  ¡r̈  ¡|jsÚ| ¡||ƒWnÞttjtfyÄz8z | ¡Wnttjtfy(Yn0Wˆ |¡n ˆ |¡0ˆ ¡dkrÈzˆjdˆj d}Wnt j y|YnH0z8z | ¡Wnttjtfy¨Yn0Wˆ |¡n ˆ |¡0qDzžz&tˆƒ}ˆ |¡| ¡||ƒWnttjtfy Yn0Wˆ ¡dkrÀz&ˆjdˆj d}| ¡ˆ |¡Wnttjtfy`Yn0qnXˆ ¡dkr¾z&ˆjdˆj d}| ¡ˆ |¡Wnttjtfy¸Yn0qh0Yn0|WdƒWS1sâ0Yn¶ˆjˆjkr$tˆƒ}ˆ |¡| ¡||ƒ|WSt ¡fz6ˆjdˆj d}| ¡||ƒ|WWdƒWSt j y‚td ƒ‚Yn0Wdƒn1sš0YWq`tttfyv}z°| d  ˆ|¡¡t|tƒr舠d ¡n |  ˆ|¡| |¡| ||¡‰ˆdurV| ||¡}||d<| ||¡‰ˆdurVd  !|¡}td |¡ƒ‚WYd}~q`WYd}~q`d}~00q`tdƒ‚dS)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. csZ| t¡ ¡ ¡dd}tdd„| d¡d d¡DƒƒdkrV| d ˆj¡¡ ¡dS) Nrr cSsg|] }t|ƒ‘qSr@r}rr@r@rArÓ÷r[zPPoolsManager.get_connection..set_mysqlx_wait_timeout..rrr‚r…)rBÚ_SHOW_VERSION_QUERYrˆrór‰rrerf)rvr‹©r®r@rAÚset_mysqlx_wait_timeoutôsÿÿÿz.set_mysqlx_wait_timeoutrßNzCUnable to connect to any of the target hosts. No pool is available.rTrxz)Failed getting connection; pool exhaustedzpool max size has been reachedzpool: {} error: {}rÝz z6Unable to connect to any of the target hosts: [ {} ]r)"r r¾rªr¨rr§rzÚ threadingÚRLockrgror{r¨r¯rèr`r]rQrdrrrwr§rrVrtrnrr9rerCr¿r±r<) rNrÊr­ršrßZ error_listrvrfr¤r@r¬rAÚget_connectionæsÔ        ÿ ÿ  ÿ    ÿ ùÿ(  ÿ2          ÿ$zPoolsManager.get_connectioncCsX| |¡}|D]@}| ¡| dd¡dur|j | d¡¡}||vr| |¡qt|ƒS)zjCloses the connections in the pools Returns: int: The number of closed pools r7N)r rir¾r˜r9r^)rNr£ršr®Z client_poolsr@r@rAÚ close_poolqs  zPoolsManager.close_poolc Cspd}z|j}t|}Wnttfy,Yn0|sR|j}tD]}||vrt ¡}t ¡d kr|d  d t ¡d¡}nd   t ƒdd …¡}t   d¡}|ddkr¬d}nd}t t ¡ƒ||t ¡dd  dd„tdd…Dƒ¡|dœ}|jd |¡d|jvr¼|jdD]¬}|jd|}t|t ƒs:td |¡ƒ‚t|ƒdkrVtd |¡ƒ‚| d¡rptd |¡ƒ‚t|t ƒsŒtd ||¡ƒ‚t|ƒd krªtd! ||¡ƒ‚||jd|<qd"S)#z5Setup default and user defined connection-attributes.rpZ64rÚx86_64Z32Úi386z Windows-{}r ÚDarwinz{}-{}ÚmacOSrrÝú ZGPLv2zGPL-2.0Z Commercialzmysql-connector-pythonrcSsg|] }t|ƒ‘qSr@)r:)rÒrÎr@r@rArÓr[z,Session._init_attributes..é)Ú_pidÚ _platformÚ_osZ _source_hostZ _client_nameZ_client_versionZ_client_licenserûr¶z)Attribute name '{}' must be a string typeé z4Attribute name '{}' exceeds 32 characters limit sizer8zJKey names in 'session-connect-attributes' cannot start with '_', found: {}z4Attribute name '{}' value '{}' must be a string typeizBAttribute name '{}' value: '{}' exceeds 1024 characters limit sizeN)r‹rŒrÚ architecturereÚ win32_verÚmachinerÚmac_verr<r,r.rr:ÚgetpidrQÚ gethostnamer-rÃrFrCrr^Ú startswith)rNZ platform_archZos_verZlicense_chunksZclient_licenseZdefault_attributesÚ attr_nameÚ attr_valuer@r@rArÀâs`     ò  ÿþ þ þþzSession._init_attributescCstjS)zAbool: `True` to use pure Python Protobuf implementation. )r+r¹rMr@r@rAr¹*szSession.use_purecCs t|tƒstdƒ‚t |¡dS)Nz)'use_pure' option should be True or False)rCÚboolrr+Z set_use_pure)rNrr@r@rAr¹0s cCs |jj ¡S)zzReturns `True` if the session is open. Returns: bool: Returns `True` if the session is open. )rÁrærrMr@r@rAr6szSession.is_opencCs t|j|ƒSra)r(rÁrbr@r@rArB>s z Session.sqlcCs|jS)z~Returns the underlying connection. Returns: mysqlx.connection.Connection: The connection object. )rÁrMr@r@rAr°JszSession.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)rr@)rÒÚrowr@r@rArÓ[r[z'Session.get_schemas..)rBrˆrórõr@r@rAÚ get_schemasRszSession.get_schemascCs t||ƒS)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©rNrŒr@r@rAÚ get_schema]s zSession.get_schemacCst|jj d¡}|rp| t t|ƒ¡¡ ¡ ¡}z |dd|krLt ||ƒWSWn t ynt d |¡ƒ‚Yn0dS)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. rärr¸N) rÁrÊr¾rBÚ_SELECT_SCHEMA_NAME_QUERYrerrˆrórrr)rNräÚresr@r@rAÚget_default_schemahs ÿ  ÿ 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. rBTN)rÁrOÚ_DROP_DATABASE_QUERYrer*rßr@r@rAÚ 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. rBT)rÁrOÚ_CREATE_DATABASE_QUERYrer*rrßr@r@rAÚ create_schemaŒsÿzSession.create_schemacCs|j ddd¡dS)z+Starts a transaction context on the server.rBzSTART TRANSACTIONTN©rÁrOrMr@r@rAÚstart_transaction—szSession.start_transactioncCs|j ddd¡dS)zXCommits all the operations executed after a call to startTransaction(). rBZCOMMITTNrèrMr@r@rAÚcommit›szSession.commitcCs|j ddd¡dS)zYDiscards all the operations executed after a call to startTransaction(). rBZROLLBACKTNrèrMr@r@rAÚrollback¡szSession.rollbackNcCsX|durd t ¡¡}n"t|tƒr2t| ¡ƒdkr:tdƒ‚|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 namerBz SAVEPOINT {0}T) reÚuuidÚuuid1rCr:r^ÚstriprrÁrOr*rßr@r@rAÚ set_savepoint§s  ÿþzSession.set_savepointcCs@t|tƒrt| ¡ƒdkr"tdƒ‚|j dd t|ƒ¡d¡dS)zRollback to a transaction savepoint with the given name. Args: name (string): The savepoint name. rrìrBzROLLBACK TO SAVEPOINT {0}TN© rCr:r^rïrrÁrOrer*rßr@r@rAÚ rollback_to¼s ÿþzSession.rollback_tocCs@t|tƒrt| ¡ƒdkr"tdƒ‚|j dd t|ƒ¡d¡dS)z{Release a transaction savepoint with the given name. Args: name (string): The savepoint name. rrìrBzRELEASE SAVEPOINT {0}TNrñrßr@r@rAÚrelease_savepointÈs ÿþzSession.release_savepointcCs|j ¡t|jƒ|_dS)zCloses the session.N)rÁrWr¦rÃrMr@r@rAriÔs z Session.closecCs|j ¡dS)z8Closes all underliying connections as pooled connectionsN)rÁrrMr@r@rAÚclose_connectionsÚszSession.close_connections)N)ržrŸr r¡rOrÃrÈrÀr”r¹ÚsetterrrBr°rÞràrãrårçrérêrërðròrórirôr@r@r@rArµs2 1H          rµc@sZeZdZdZddd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dd„Z dd„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’||_|duri}g|_t ¡|_| | dd¡¡| | dd¡¡| | dd¡¡|  | dd¡¡|j |jd<|j |jd<|j|jd <dS) NrdrerfrrgÚenabledTr·r7) rÊÚsessionsríÚuuid4r7Ú_set_pool_sizer¾Ú_set_max_idle_timeÚ_set_queue_timeoutÚ_set_pool_enabledÚpooling_enabledrd)rNZconnection_dictZ options_dictr@r@rArOés   zClient.__init__cCs|Srjr@rMr@r@rArÃúszClient.__enter__cCs | ¡dSrjrkrÄr@r@rArÈýszClient.__exit__cCs@t|tƒst|tƒr|dks*td |¡ƒ‚|dkr6tn||_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)rCrÜr~rWreÚ_CNX_POOL_MAXSIZErd)rNZ pool_sizer@r@rArú s ÿþzClient._set_pool_sizecCsRt|tƒst|tƒr|dks*td |¡ƒ‚||_|dkrsÎý              óÿû  b6#&I[kD