h dZddlmZddlmZ ddlZdZejejej dZ e edrej nej e d<e ed oejZn#e$r d Zd ZiZ YnwxYwddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZdd lmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z% ddl&Z'ddl(Z'dZ)n #e$rd Z)YnwxYwdd l*m*Z*m+Z+dd l,m-Z-ddl.m/Z/ddl0m1Z1m2Z2m3Z3ddl4m5Z5m6Z6m7Z7ddl8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@ddlAmBZBddlCmDZDmEZEmFZFmGZGmHZHmIZIddlJmKZKmLZLmMZMddlNmNZNddlOmPZPddlQmRZRmSZSmTZTmUZUmVZVddlWmXZXmYZYmZZZm[Z[m\Z\ddl]m^Z^m_Z_m`Z`maZambZbmcZcmdZdmeZemfZfmgZgmhZhddlmiZimjZjmkZkmlZlejmndddlompZpmqZqmrZrmsZsddltmuZuddlvmwZwmxZxd d!d"d#Zyd$Zzd%Z{d&Z|d'Z}d(Z~d)Zd*Zejd+Zd,Zd,Zd-Zd.Zd/Zd/Zd0Zd.Zeeeeeeeed1Zd2eiZejd3Zejd4ejZejd5ejZejd6Zgd7Zegd8zZdld=ZdmdAZGdBdCZdndFZGdGdHeZGdIdJZGdKdLZGdMdNeZGdOdPejZGdQdRZGdSdTZGdUdVZdodYZdpd\Zdqd^Z drdsdaZdqdbZdqdcZdqddZdtdfZdudgZdvdkZdS)wz4Implementation of communication for MySQL X servers.) annotations) TracebackTypeNT)TLSv1TLSv1.1TLSv1.2 PROTOCOL_TLSTLSv1.3 HAS_TLSv1_3F) AnyCallableDictListMappingOptionalTupleTypeUnion)datetime timedeltawraps)JSONDecodeError) parse_qslunquoteurlparse)MySQL41AuthPluginPlainAuthPluginSha256MemoryAuthPlugin)COMPRESSION_ALGORITHMSDEPRECATED_TLS_VERSIONSOPENSSL_CS_NAMESSUPPORTED_TLS_VERSIONSTLS_CIPHER_SUITESAuth CompressionSSLModeSchema)InterfaceErrorNotSupportedErrorOperationalError PoolErrorProgrammingError TimeoutError)escapeget_item_or_attriani_to_openssl_cs_name)logger)Protobuf)HAVE_LZ4 HAVE_ZSTD MessageReader MessageWriterProtocol) BaseResult DocResultResult RowResult SqlResult) AddStatementDeleteStatement FindStatementInsertStatementModifyStatement ReadStatementRemoveStatementSelectStatement SqlStatementUpdateStatementquote_identifier) ColumnType MessageTypeResultBaseType StatementType..)DUPLICATED_IN_LIST_ERRORTLS_VER_NO_SUPPORTEDTLS_VERSION_DEPRECATED_ERRORTLS_VERSION_ERROR)linux_distribution)LICENSEVERSIONzThis session was closed because the connection has been idle for too long. Use 'mysqlx.getSession()' or 'mysqlx.getClient()' to create a new one.z> penalty_dictMapping[str, Any]Nonecp|r1t|trt|dSdSdS)zUpdate the timeout penalties directory. Update the timeout penalties by error dictionary used to deactivate a pool. Args: penalty_dict (dict): The dictionary with the new timeouts. N) isinstancedict_TIMEOUT_PENALTIES_BY_ERR_NOupdate)r}s r{!update_timeout_penalties_by_errorrsK: <66:$++L99999::::cneZdZdZd"dZefd#d Zd$d Zd%dZd"dZ d"dZ d&dZ d'dZ d'dZ d'dZd'd Zd!S)( SocketStreamzImplements a socket stream.rprc>d|_d|_d|_d|_dS)NF)_socket_is_ssl _is_socket_hostselfs r{__init__zSocketStream.__init__s"04 " %$( rparamsrconnect_timeoutfloatc||dz } tj|||_|d|_n#t$r tjtj|_|j||j|d|_n#t$rtddwxYwYnwxYw|jddS)zConnects to a TCP service. Args: params (tuple): The connection parameters. Raises: :class:`mysqlx.InterfaceError`: If Unix socket is not supported. NrTzUnix socket unsupported) rfcreate_connectionrr ValueErrorAF_UNIX settimeoutconnectrAttributeErrorr*)rrrs r{rzSocketStream.connects  &-4O J!3DL DJJ J J J J%}V^<<  ''888 $$V,,,"&! J J J$%>??TI J   J %%%%%s''1 B<ABB<B66B<;B<countintbytesc |jtdg}|dkr\|j|}|dkrtd|||t |z}|dk\d|S)zReceive data from the socket. Args: count (int): Buffer size. Returns: bytes: The data received. NMySQLx Connection not availablerrzUnexpected connection close)rr,recv RuntimeErrorrvlenrx)rrbufdatas r{readzSocketStream.read"s < "#DEE Eaii<$$U++Ds{{"#@AAA JJt    SYY E aii xx}}rrc|jtd |j|dS#t$r}td||d}~wwxYw)z_Send data to the socket. Args: data (bytes): The data to be sent. NrzUnexpected socket error: )rr,sendallOSError)rrerrs r{rzSocketStream.sendall6sy < "#DEE E O L  & & & & & O O O"#Ds#D#DEE3 N Os4 AAAc|jsdS |jtj|jn#t $rYnwxYwd|_dS)zClose the socket.N)rshutdownrf SHUT_RDWRcloserrs r{rzSocketStream.closeCss|  F  L ! !&"2 3 3 3 L       D  s=A AAc.|dSNrrs r{__del__zSocketStream.__del__Os r ssl_protos List[str]ssl_moderqssl_cassl_crlssl_certssl_key ssl_cipherscts#|td||sLtj}|t jkr(d|_|t jkrtj |_ n| d|d} ts!| dkrt|dkr|d} t| } tj| }| dkrZd |vr|xjtjzc_d |vr|xjtjzc_d |vr|xjtjzc_|t jk|_|rm ||tj|_ nE#t,tjf$r,} |t1d | | d} ~ wwxYw|rm ||tj|_nE#t,tjf$r,} |t1d | | d} ~ wwxYw|r] |||nE#t,tjf$r,} |t1d| | d} ~ wwxYw|r<|dt=|d| ||j |j!|_ n(#tj"$r} t1| | d} ~ wwxYwd|_#|j $} | dvr"d| d} tKj&| tNdSdS)axSet SSL parameters. Args: ssl_protos (list): SSL protocol to use. 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. ssl_ciphers (list): SSL ciphersuites to use. 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 supportNFTreverserr rrrrzInvalid CA Certificate: z Invalid CRL: zInvalid Certificate/Key: :)server_hostname)rrzThis connection is using zZ which is now deprecated and will be removed in a future release of MySQL Connector/Python)( SSL_AVAILABLErrsslcreate_default_contextr'VERIFY_IDENTITYcheck_hostnameREQUIRED CERT_NONE verify_modesortTLS_V1_3_SUPPORTEDr TLS_VERSIONS SSLContextoptions OP_NO_TLSv1_2 OP_NO_TLSv1_1 OP_NO_TLSv1load_verify_locations CERT_REQUIREDIOErrorSSLErrorr*VERIFY_CRL_CHECK_LEAF verify_flagsload_cert_chain set_ciphersrxr2 wrap_socketrrCertificateErrorrversionwarningswarnDeprecationWarning) rrrrrrrrcontext tls_version ssl_protocolrwarn_msgs r{set_sslzSocketStream.set_sslRs4 I JJLLLGHH H  Z 022G7222).&w///*--G' OODO ) ) )$Q-K& ,9,, OOa''(m ' 4Ln\22Gi''J..OOs'88OOJ..OOs'88OO*,,OOs6OO%-1H%HG "  P P--f555&)&7##S\* P P P $%E%E%EFFCO P  E E--g666'*'@$$S\* E E E $%:S%:%:;;D E  Q Q'''::::S\* Q Q Q $%F%F%FGGSP Q     0A LLMM    4"..t|TZ.XXDLL# 4 4 4 C** 3 4 l**,, . . .)K)))  M($6 7 7 7 7 7 / .s`&FG'F>>G &G00H2'H--H28IJ%'J  J&K::L LLboolc|jS)zpVerifies if SSL is being used. Returns: bool: Returns `True` if SSL is being used. )rrs r{is_sslzSocketStream.is_ssls |rc|jS)zVerifies if socket connection is being used. Returns: bool: Returns `True` if socket connection is being used. )rrs r{ is_socketzSocketStream.is_sockets rc|jp|jS)zvVerifies if connection is secure. Returns: bool: Returns `True` if connection is secure. )rrrs r{ is_securezSocketStream.is_secures |.t.rc|jduS)zrVerifies if connection is open. Returns: bool: Returns `True` if connection is open. N)rrs r{is_openzSocketStream.is_opens |4''rNrpr)rrrrrpr)rrrpr)rrrpr)rrrrqrrqrrqrrqrrqrrrprrpr)__name__ __module__ __qualname____doc__r_CONNECT_TIMEOUTrrrrrrrrrrrr{rrs %%)))) ?O&&&&&4( O O O O    e8e8e8e8N////((((((rrfuncr c>tdfd }|S)zDecorator used to catch OSError or RuntimeError. Raises: :class:`mysqlx.InterfaceError`: If `OSError` or `RuntimeError` is raised. rr argsrorpc t|ttfr/|rt | |g|Ri|}t|t r|}|D]m}|dtvr\t|d}d|dd||df}t|ttfr| |nn|S#ttttf$rI} j dkr#|r t|dtr|d}|r|d}t|d}d|dd||df}t|trS|j|ddkr.t#|jt |t|ttfr| ||t |||d}~wwxYw) zWrapper function.codezConnection close: msgz: get_column_metadatarrVN)r ConnectionPooledConnectionis_server_disconnectedr*get_disconnected_reasonr: get_warningsCONNECTION_CLOSED_ERRORset_server_disconnectedrrr/rr>poolremove_connections PoolsManagerset_pool_unavailable disconnect) rrroresultwarnsr error_msgreasonrrs r{wrapperz(catch_network_exception..wrappers. 4*.>!?@@ F//11 F%d&B&B&D&DEET$000000F&*-- #0022!  DF|'>>>$;DL$I KeKK KK L"&dZ9I,JKKA 88@@@?M|D    !6667tAw 227Q,,..; 8D 7V EIGT%[GGIGGV F"$(899 44666<4//(NN?? $ >6+B"$5E(FGG=44V<<<OO%%%(&1s: OO    5 sC1C55IEII)rr rr ror rpr r)rr s` r{catch_network_exceptionr s; 4[[00000[0d Nrc<eZdZdZd fd Zd dZdd Zdd ZxZS)RouterzRepresents a set of connection parameters. Args: settings (dict): Dictionary with connection settings .. versionadded:: 8.0.20 connection_paramsr~rprct|||dd|d<dS)N availableT)superrrget)rr __class__s r{rzRouter.__init__ sJ  %&&& HH[$77[rrc|dS)zVerifies if the Router is available to open connections. Returns: bool: True if this Router is available else False. rrrs r{rzRouter.available%s K  rcd|d<dS)z1Sets this Router unavailable to open connections.FrNrrs r{set_unavailablezRouter.set_unavailable-s![r%Union[str, Tuple[str, Optional[int]]]c:d|vr|dS|d|dfS)zVerifies if the Router is available to open connections. Returns: tuple: host and port or socket information tuple. rfrcrdrrs r{get_connection_paramszRouter.get_connection_params1s, t  > !V d6l++r)rr~rprrr)rpr) rrrrrrrr __classcell__rs@r{rrs888888 !!!!"""",,,,,,,,rrcJeZdZdZddZdd Zdd ZddZddZddZ ddZ dS) RouterManagerzManages 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 re List[Router]settingsDict[str, Any]rprc||_||_d|_d|_i|_g|_|dS)NrT)_routers _settings_cur_priority_idx _can_failover_routers_directoryrouters_priority_list_ensure_priorities)rrer s r{rzRouterManager.__init__EsH !&'#';=02" !!!!!rc$d}|jD];}|dd}| |dz }d|d<%|dkrtddz2RouterManager._ensure_priorities..gs :rT)rzr) r#rr.rrr'rr(rv)rpriority_countrouterr+s r{r)z RouterManager._ensure_prioritiesOse m Y YFzz*d33H!#%(z""C&'QSWXXX  ~ 2 2 2 2DM 2 2 2 2 2 2 2"   66EEEm I IFj)Ht6665;F^^4D'1*11(;;;;'188HHHH  I Irr+rc8|j|}d|D}|S)zGet a list of the current available routers that shares the given priority. Returns: list: A list of the current available routers. c:g|]}||Sr)r).0r5s r{ z8RouterManager._get_available_routers..ys)NNN&6;K;K;M;MNvNNNrr')rr+ router_lists r{_get_available_routersz$RouterManager._get_available_routersrs* -h7 NNKNNN rrc||}|sdSt|dkr|dSt|dz }tjd|}||S)z{Get a random router from the group with the given priority. Returns: Router: A random router. Nrr)r<rrandomrandint)rr+r;lastindexs r{_get_random_connection_paramsz+RouterManager._get_random_connection_params|so 11(;;  4 {  q q> !;!#q$''5!!rrc|jS)zReturns the next connection parameters. Returns: bool: True if there is more server to failover to else False. )r&rs r{ can_failoverzRouterManager.can_failovers !!rcX|jskd|_|j}|jdd|d<|jdd|d<t |S|j|j}t|j}d}|r| |}| |j|kr<|j|dz kr-t| |d krd|_n/|xjdz c_|j|kr|j|j}||S) zvReturns the next connection parameters. Returns: Router: with the connection parameters. Frc localhostrd$TNr) r#r&r$copyrrr(r%rrBr<)rrouter_settings cur_priorityrouters_priority_lensearchr5s r{get_next_routerzRouterManager.get_next_routersL } +!&D "n1133O&*n&8&8&M&MOF #&*n&8&8&G&GOF #/** *1$2HI "4#=>> R77 EEF!T%;?S%S%S*.BQ.FFFD77 EEFFJJ).D&  " "a ' " "%(<<<#9$:PQ  R  rDict[int, List[Router]]c|jS)zReturns the directory containing all the routers managed. Returns: dict: Dictionary with priorities as connection settings. r:rs r{get_routers_directoryz#RouterManager.get_routers_directorys &&rN)rerr r!rprr)r+rrpr)r+rrprr)rpr)rprO) rrrrrr)r<rBrDrNrQrrr{rr<s""""!I!I!I!IF"""" """"!!!!F''''''rrceZdZdZdMdZdNdZdOd ZdNd ZdPd Z dQdRdZ dNdZ dNdZ dNdZ dNdZ dSdZdTd ZdTd!ZedUd$ZedVd'ZedWd*ZedXd,ZedYd.Ze dQdZd6Zed[d9Zed\d;Zed]d=Zed^d>Zed_d@Zd`dAZdadBZdbdEZdadFZ dcdHZ!dNdIZ"dNdJZ#dNdKZ$dNdLZ%dS)drzkConnection to a MySQL Server. Args: settings (dict): Dictionary with connection settings. r r!rprc||_t|_d|_d|_|d|_|d|_|d|_d|_ |dg|_ d|vrL|drD|j |d|dddt|j ||_ |dt|_|jd krd|_d |_g|_d |_d |_d|_dS) Nr`rarbrercrdrcrdrjrTF)r rstreamprotocol keep_openr_user _password_schema_active_resultr#rvrrouter_managerr_connect_timeout _stmt_counter_prepared_stmt_ids_prepared_stmt_supported_server_disconnected_server_disconnected_reason)rr s r{rzConnection.__init__sK(0 $0NN ,0 )-$,LL$8$8 (0 Z(@(@&.ll8&<&< 8<&.ll9b&A&A X  (6"2  M $LL00$LL66   .;4=(-S-S/7|| /0 0   A % %%)D !"#-/.2%*/!RV(((rcX|j"|jd|_dSdS)zFetch active result.N)r[ fetch_allrs r{fetch_active_resultzConnection.fetch_active_results6   *   ) ) + + +"&D    + *rrrLc||_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[rrs r{set_active_resultzConnection.set_active_results%rc<d}|jr |j}|j||jt|j}t|j}t|||_ |j j }|r d|Dni}| |d|jvr(|jd}|j ||jdt"j}|jd} |t"jkrdn|||| } ||j | dS#t.t0f$r } | }|Yd} ~ nd} ~ wwxYw|j|bt5|t6jrHt;|jdkrt?d|jd t?d |jd t;|jdkrtAd |tAd d)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. NcTi|]%}t|d|&Sname)r1lower)r8caps r{ z&Connection.connect..s1UUUC%c62288::CUUUr attributessession_connect_attrsrlrmrz9Connection attempt to the server was aborted. Timeout of z ms was exceededz8All server connection attempts were aborted. Timeout of z) ms was exceeded for each selected serverzCannot connect to host: ,Unable to connect to any of the target hostsi)!r\rDrNrUrrr]r7r8r9rVget_capabilites capabilities_set_tls_capabilitiesr set_capabilitiesrr& PREFERREDDISABLED_set_compression_capabilities _authenticateset_compressionrrrrrftimeoutrr#r/r*) rerrorr5readerwriter caps_datacaps conn_attrsrl algorithms algorithmrs r{rzConnection.connects!..00( )' ),<<>> ##0022D4I't{33&t{33 ( 8 8  M99;;H !UU9UUUU**4000 4=00!%|!!B!B 4=!!Q&&"J"&"7JJJ""3"""  t}   " " !CE!C!CDD DKTRRRsFF22G#GG#rc |jdtjkrdS|jr0|jdrt jddSd|vr#|tdd}tj dkrIt\}}} d|vo| d d d k}n#t$rd}YnwxYwtjd kr%|s#|t#d |jd|j|jdd|jdtj|jd|jd|jd|jd|jdd|jvr*|jd}|j|dSdS)aSet 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 rgNz(SSL not required when using Unix socket.tlszSSL not enabled at serverFLinuxz Oracle Linux.r7)rH z?? ? ?   ' '#5#7#7 Hgq '83T c8J8J18MQT8T      i ' ' '  ! ! # # #N  &&4&000  M  nd 3 3 M  j'*: ; ; M  h ' ' M  i ( ( M  j ) ) M  i ( ( M  0 1 1    4= ( (|4J M * * * L L L L L ) (s#C++ C:9C:NrlrqrOptional[List[str]] Optional[str]c |d}|7d}|tjkrt|t j|dSi}t |tr?|dddD])}d|dddD||d <*n7|jj j D]%}d |jj jD||j <&|d g d}|rXd |D} fd | D} | rtj| d}n!|tjkrtddS|trd vrd}nt rd vrd}nd}| vr7d}|tjkrt|t j|dS|jd |i|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 rlNz8Compression requested but the server does not support itvalueobjfldc\g|])}|dddd*S)scalarv_stringrutf-8)decoder8rs r{r9z.sD000(OJ/8??HH000rarrayrzcVg|]&}|jjjd'S)r)rrrrrs r{r9z.s>---L)/66w??---rrc:g|]}|tv t|Sr)r )r8items r{r9z.s3!!!111't,111rcg|]}|v| Srr)r8rserver_algorithmss r{r9z.s$WWWTEV=V=Vt=V=V=VrrzmThe connection compression is set as required, but none of the provided compression algorithms are supported. zstd_stream lz4_messagedeflate_streamzFCompression requested but the compression algorithm negotiation failed)rl)rr&rr+r3rrrrrrrrzr r*r6r5rVrw) rrrlrcompression_datarcompression_dictrrclient_algorithmsmatchedrs @r{rzz(Connection._set_compression_capabilitiesvsm< 88M22  #LCk222',,, N3   4 & - - '07>  00!$Wg!6w!?000 U,,  (-15  --!$!6--- )) -00bAA   !!&!!!  XWWW(9WWWG 26wqzBB  444$< t   -].???)  -m/@@@) , - - -% k222$S))) N3   4 &&K3K&LLLrcF|jd}|rt|tjkr|dS|tjkr|dS|tjkr|dSdS|j r|dS |dS#t$rYnwxYw |dS#t$r}td||d}~wwxYw)z#Authenticate with the MySQL server.rhNzrAuthentication failed using MYSQL41 and SHA256_MEMORY, check username and password or try a secure connection err:) r rr%PLAIN_authenticate_plain SHA256_MEMORY_authenticate_sha256_memoryMYSQL41_authenticate_mysql41rUrr*)rrhrs r{r{zConnection._authenticatesm}  ((  tz!!((*****+++0022222%%**,,,,,&% [ " " $ $   $ $ & & & & & **,,,"      0022222!   $E?BEE  s*C C$#C$(C>> D DD cPt|j|j}|j||j}|j|||j dS)z=Authenticate with the MySQL server using `MySQL41AuthPlugin`.N) rrXrYrVsend_auth_start auth_nameread_auth_continuesend_auth_continue auth_data read_auth_okrplugin extra_datas r{rz Connection._authenticate_mysql41s"4:t~>> %%f&6&6&8&8999]5577  (()9)9*)E)EFFF ""$$$$$rc<|jstdt|j|j}|j|| |j dS)z;Authenticate with the MySQL server using `PlainAuthPlugin`.z>PLAIN authentication is not allowed via unencrypted connection)rN) rUrr*rrXrYrVrrrr)rrs r{rzConnection._authenticate_plains{$$&&  P !T^<< %%f&6&6&8&8FDTDTDVDV%WWW ""$$$$$rcPt|j|j}|j||j}|j|||j dS)zBAuthenticate with the MySQL server using `Sha256MemoryAuthPlugin`.N) rrXrYrVrrrrrrrs r{rz&Connection._authenticate_sha256_memory s' DNCC %%f&6&6&8&8999]5577  (()9)9*)E)EFFF ""$$$$$r statementrMc|jrG|j|j|j|jd|_dSdS)zrDeallocates statement. Args: statement (Statement): A `Statement` based type object. FN)preparedrVsend_prepare_deallocatestmt_idr_remove)rrs r{_deallocate_statementz Connection._deallocate_statementsZ   ' M 1 1)2C D D D  # * *9+< = = =!&I    ' 'rmsg_typerrKgUnion[FindStatement, DeleteStatement, ModifyStatement, ReadStatement, RemoveStatement, UpdateStatement]c ||j|||n#t$r d|_YdSwxYw|j|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) rerVsend_prepare_preparer+r`r_rvrrrrrrs r{_prepare_statementzConnection._prepare_statements(   $ $ & & & M . .xi H H H H    ,1D ) FF  &&y'8999! s03AAc8|js|j|||dS|jr|||||||js|j|||dS|j|||d|_|nO|jr%|j s|j|||n#|j rT|j sM|||j|||d|_ |n|j sg|j s`|js|||||js|j|||dS|j|||nZ|j rS|j rL|||j|||d|_ || 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) r`rVsend_msg_without_psdeallocate_prepare_executerrsend_prepare_executereset_exec_counterrchangedrepeatedincrement_exec_counterrs r{_execute_prepared_pipelinez%Connection._execute_prepared_pipeline:sM*,  M - -hY G G G F  / +  & &y 1 1 1  # #Hc9 = = =0  11(CKKK M . .xi H H H38I 0  ( ( * * * *   + (9 + M . .xi H H H H   +y'9 +  & &y 1 1 1 M - -hY G G G %I   ( ( * * * *" +9+= +% B''#yAAA0  11(CKKK M . .xi H H H H   +9#5 +  & &y 1 1 1 M - -hY G G G %I   ( ( * * *((*****rrGr>c|j}|jtdt|tst d|jd|\}}|j|||t|S)zExecute a SQL statement. Args: sql (str): The SQL statement. Raises: :class:`mysqlx.ProgrammingError`: If the SQL statement is not a valid string. Nrz'The SQL statement is not a valid stringsql) rrVr,rrqr.build_execute_statementrr>)rrrrrs r{send_sqlzConnection.send_sqlxsm = "#DEE E#s## N"#LMM M ==eSII # ))(CCCCr$Union[AddStatement, InsertStatement]r<c|jtd|j|\}}|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. Nr)rVr, build_insertsend_msgrr?idsr<)rrrrrs r{ send_insertzConnection.send_insertsz = "#DEE E 229== # x--- i . . -CdC   r%Union[FindStatement, SelectStatement]Union[DocResult, RowResult]c|j|\}}|||||rt |nt |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`. )rV build_findr is_doc_basedr;r=rrrrs r{ send_findzConnection.send_finds\ 00;; # ''#yAAA"+"8"8":":Oy $Or'Union[DeleteStatement, RemoveStatement]c|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. )rV build_deleterr<rs r{ send_deletezConnection.send_deleteA 229== # ''#yAAAd||r'Union[ModifyStatement, UpdateStatement]c|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. )rV build_updaterr<rs r{ send_updatezConnection.send_updaterr namespacecmd raise_on_failrfieldsOptional[Dict[str, Any]]Optional[Result]c |j|||\}}|j||t|S#t$r|rYnwxYwdS)aExecute 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)rVrrr<r,)rrrrrrrs r{execute_nonqueryzConnection.execute_nonquerys,  MAA3MHc M " "8S 1 1 1$<<        tsAA AArrc |jd|\}}|j||t|}||jdkrt d|ddS)zExecute a SQL scalar. Args: sql (str): The SQL statement. Raises: :class:`mysqlx.InterfaceError`: If no data found. Returns: :class:`mysqlx.Result`: The result. rrz No data found)rVrrr=rdrr*)rrrrrs r{execute_sql_scalarzConnection.execute_sql_scalars} ==eSII # x---4 <1   11 1ay|rr=c|jd||\}}|j||t|S)zReturns the row result. Args: cmd (str): The command. fields (dict): The message fields. Returns: :class:`mysqlx.RowResult`: The result object. mysqlx)rVrrr=)rrrrrs r{get_row_resultzConnection.get_row_resultsE ==hVTT # x---rOptional[MessageType]c6|j|S)zdRead row. Args: result (:class:`mysqlx.RowResult`): The result object. )rVread_rowrgs r{rzConnection.read_rows}%%f---rc:|j|dS)zeClose result. Args: result (:class:`mysqlx.Result`): The result object. N)rV close_resultrgs r{rzConnection.close_results  ""6*****rList[ColumnType]c6|j|S)zlGet column metadata. Args: result (:class:`mysqlx.Result`): The result object. )rVrrgs r{rzConnection.get_column_metadata's}00888rc0|xjdz c_|jS)z|Returns the next statement ID. Returns: int: A statement ID. .. versionadded:: 8.0.16 r)r^rs r{get_next_statement_idz Connection.get_next_statement_id0s! a!!rc4|jS)zgCheck if connection is open. Returns: bool: `True` if connection is open. )rUrrs r{rzConnection.is_open;s {""$$$rr Union[str, Tuple[str, int]]c"d|_||_dS)zSet the disconnection message from the server. Args: reason (str): disconnection reason from the server. TN)rarb)rr s r{rz"Connection.set_server_disconnectedCs %)!+1(((rc|jS)zVerify if the session has been disconnect from the server. Returns: bool: `True` if the connection has been closed from the server otherwise `False`. )rars r{rz!Connection.is_server_disconnectedLs ((r%Optional[Union[str, Tuple[str, int]]]c|jS)zGet the disconnection message sent by the server. Returns: string: disconnection reason from the server. )rbrs r{rz"Connection.get_disconnected_reasonUs //rcd|sdS|jdS)zDisconnect from server.N)rrUrrs r{rzConnection.disconnect]s1||~~  F rc|sdS ||jr+|jD]}|j|d|_|j|jn9#tttf$r}tj d|Yd}~nd}~wwxYw|jdS#|jwxYw)z*Close a sucessfully authenticated session.NrzGWarning: An error occurred while attempting to close the connection: %s)rrer`r_rVrr^ send_closeread_okr*r,rr3rrUr)rrrs r{ close_sessionzConnection.close_sessioncs+||~~  F  $ $ & & &, '#6CCGM99'BBBB%&" M $ $ & & & M ! ! # # # # 0':    N!          K       DK      s0A8BC%C(C=C%CC%%Dc$|sdS|j|j |j|j|_dS#t tf$r }tj d|Yd}~dSd}~wwxYw)z*Reset a sucessfully authenticated session.NzDWarning: An error occurred while attempting to reset the session: %s) rr[rdrV send_resetrWr*r,r3r)rrs r{ reset_sessionzConnection.reset_session~s||~~  F   *   ) ) + + + !]55dnEEDNNN 01    N          s$AB/B  Bc|sdS|j|j|j|j|jdS)zAnnounce to the server that the client wants to close the connection. Discards any session state of the server. N)rr[rdrVsend_connection_closerrUrrs r{rzConnection.close_connectionsy||~~  F   *   ) ) + + + ++---  rr r!rprr)rrLrpr)rr!rprr)rr!rlrqrrrpr)rrMrpr)rrqrrKrrrpr)rrGrpr>)rrrpr<)rrrpr)rrrpr<)rrrpr<) rrqrrqrrrrrpr)rrMrpr)rrqrr!rpr=)rr=rpr)rr<rpr)rr<rprrprr)r r rpr)rpr )&rrrrrrerhrrvrzr{rrrrrrr rrrrrrrrrrrrrrrrrrrrrrr{rrst W W W WD'''' % % % %DSDSDSDSL4M4M4M4Mt+/ aaaaaF>%%%%%%%%%%%% ' ' ' '"""":<+<+<+<+|&!!!!&PPPP"         ,0 @(    ....++++9999 " " " "%%%%2222))))0000     6          rrcPeZdZdZdfd Zdfd ZddZdd Zdd ZddZ xZ S)raClass 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 rConnectionPoolrprct|tstdt|j||_|jd|_|jd|_dS)Nz&pool should be a ConnectionPool objectrcrd) rrrrr cnx_configrrcrd)rrrs r{rzPooledConnection.__init__sf$// K !IJJ J )))$( 0 0 rcHtdS)zGCloses the connection. This method closes the socket. N)rr)rrs r{rz!PooledConnection.close_connections! rc:|j|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_connectionrs r{rzPooledConnection.close_sessions    &&&&&rcn|j|j|dS)zReconnect this connection.N)r[rdr{rs r{ reconnectzPooledConnection.reconnects8   *   ) ) + + + rc.|dS)zQReset the connection. Resets the connection by re-authenticate. N)r#rs r{resetzPooledConnection.resets rrrqrGc"t||SaCreates 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. )rGrrs r{rzPooledConnection.sqlsD#&&&r)rrrprrrrqrprG) rrrrrrrr#r%rrrs@r{rrs$111111       ' ' ' '  ' ' ' ' ' ' ' 'rrceZdZdZd!dZd"d Zed#d Zd$d%dZd&dZ d$d%dZ d'dZ d(dZ d)dZ d*dZd+d,dZd&dZd-dZd&d Zd S).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 rlrqror rprc @||d|_g|_d|_d|_t j|_|dd|_ tj ||j |dd|_ |dd|_||_|dd|_|dd|_||_|d |_|d |_dS) NrTmax_sizer max_idle_time queue_timeoutr+rcrd)_set_pool_name_open_sessions_connections_openned _available_timeoutrnow_timeout_stampr pool_max_sizequeueQueuerrr.r r/r+rrcrd)rrlros r{rzConnectionPool.__init__s D!!!#$<>! $ (0 "(**Z"<"< T4#5666#)::ot#D#D"(**_b"A"A(. "(**_b"A"A#ZZ A66 *0  r pool_namect|rtd|dt|tkrtd|d||_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 'z' contains illegal charactersz ' is too longN)_CNX_POOL_NAME_REGEXrMrr_CNX_POOL_MAX_NAME_SIZErl)rr:s r{r0zConnectionPool._set_pool_namesl & &y 1 1 Y !Wy!W!W!WXX X y>>3 3 3 !Gy!G!G!GHH H rrc*t|jS)zDReturns the number of open connections that can return to this pool.)rr2rs r{open_connectionszConnectionPool.open_connections&s4,---rNcnxOptional[PooledConnection]c:|j|dS)zwRemoves a connection from this pool. Args: cnx (PooledConnection): The connection object. N)r2rrr@s r{remove_connectionz ConnectionPool.remove_connection+s! !((-----rc|dkr |d|j} |n#tt t f$rYnwxYw||n/#||wxYw#tj $rYnwxYw|dkdSdS)z*Removes all the connections from the pool.rTblockr}N) qsizerr/rrrr*rDr8EmptyrCs r{rz!ConnectionPool.remove_connections3sjjllQ 0hhT43EhFF0((****$g~>D**3////D**3////;    jjllQs:BA A? A&#A?%A&&A??BB+*B+c>|jstd|rtd|st|}|t dd}td| dd dDdkr/|d|j |j |n`t|tstd |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 fullrc34K|]}t|VdSrrr8ns r{ z0ConnectionPool.add_connection..\(BBSVVBBBBBBr-rr set mysqlx_wait_timeout = z1Connection instance not subclass of PooledSessionN)rr-fullrr_SELECT_VERSION_QUERYexecuterdtuplerr.r2rvrrrrqueue_connection)rr@vers r{r!zConnectionPool.add_connectionBs  FDEE E 99;; GEFF F "4((C''/0088::DDFFqI!LCBBSYYs^^A%6%<%)A99A> connectionc:|j|dS)zETracks connection in order of close it when client.close() is invoke.N)r2rv)rr`s r{track_connectionzConnectionPool.track_connections !((44444rc|jSrrkrs r{__str__zConnectionPool.__str__s yrrc|jS)zReturns if this pool is available for pool connections from it. Returns: bool: True if this pool is available else False. .. versionadded:: 8.0.20 )r3rs r{rzConnectionPool.availables rtime_outc|jr>tjd||d|_tj|_||_dSdS)zhSets this pool unavailable for a period of time (in seconds). .. versionadded:: 8.0.20 z4ConnectionPool.set_unavailable pool: %s time_out: %sFN)r3r3rrr5r6r4)rrgs r{rzConnectionPool.set_unavailablesW ? % NF    $DO"*,..D $DMMM % %rcDd|_tj|_dS)zaSets this pool available for pool connections from it. .. versionadded:: 8.0.20 TN)r3rr5r6rs r{ set_availablezConnectionPool.set_availables &lnnrTuple[int, datetime]c|j|jfS)zReturns the penalized time (timeout) and the time at the penalty. Returns: tuple: penalty seconds (int), timestamp at penalty (datetime object) .. versionadded:: 8.0.20 )r4r6rs r{get_timeout_stampz ConnectionPool.get_timeout_stamps t233rcB|jD]}|dS)zEmpty this ConnectionPool.N)r2rrCs r{rzConnectionPool.closes3, # #C  " " " " # #r)rlrqror rpr)r:rqrprrr)r@rArprrr@rrpr)r`rrpr)rprqr)rf)rgrrpr)rprk)rrrrrr0propertyr?rDrr!r[rbrdrrrjrmrrrr{rrsV8((((&&...X...... 0 0 0 0(#(#(#(#(#TPPPP05555 % % % % %----4444######rrceZdZUdZdZded<iZded<d+dZd,d Zd-dZ e d.dZ d/dZ e d0dZ e d1dZd2dZe d3d4d Z d3d5d#Zd6d%Zd7d&Ze d8d*ZdS)9rzManages a pool of connections for a host or hosts in routers. This class handles all the pools of Connections. .. versionadded:: 8.0.13 N_PoolsManager__instancer!_PoolsManager__poolsrpctj0t|t_it_tjSr)rrrobject__new__rs)clss r{rvzPoolsManager.__new__s0  " *&,nnS&9&9L ##%L &&rrtrqr:rcb|j|g}|D]}|j|krdSdS)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)rsrrl)rrtr:poolsrs r{ _pool_existszPoolsManager._pool_existssH   B//  DyI%%tt&urr rcg}g}||}|D]\}}|||j|ddgD] }|j|vr||!|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. rtNo id)_get_connections_settingsrvrsrrl)rr available_pools pool_namesconnections_settings router_namerurs r{ _get_poolszPoolsManager._get_poolss #==hGG3 + +NK   k * * * *L$$X\\+w%G%GLL - -DyJ&&&&t,,,r List[Tuple[str, Dict[str, Any]]]c|}|dg}g}d|vr|dd|vr*d|vr&|dd|d|dd|d|D]q}|}|d|d<|d|d<|d |d <|d d|d <|t d i||fr|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 rercrdr,r)r+weightrcrdc@|d|dd fS)Nr+rr)rr1s r{r3z8PoolsManager._get_connections_settings..s AjMAEE(A4F4F3F#Gr)rzr+rr)rIrpoprvrr|)r  pool_settingsrerr5connection_settingss r{r}z&PoolsManager._get_connections_settingssW!  ##Ir22!  % %   i ( ( ( ] " "v'>'> NN #)&1)&1      GG HHH  F"/"4"4"6"6 *0.  '*0.  '.4Z.@  +,2JJx,C,C  ) ' '&==)<=='     $#r cnx_settingsrc||}|dd|jvrg|j|dd<|D]\}}||dd|r0|j|ddg}|t |fi|dS)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. rtr|N)r}rrsrzrvr)rrrrr rs r{ create_poolzPoolsManager.create_pools $==lKK   K 1 1 E ECEDL))+w?? @&: A A !K  !1!1+w!G!GUU <##L$4$4['$J$JBOOD KK{??h?? @ @ @ @  A Ar pool_listList[ConnectionPool]rc|sdSt|dkr|dSt|dz }tjd|}||S)zGet a random router from the group with the given priority. Returns: Router: a random router. .. versionadded:: 8.0.20 Nrr)rr>r?)rr@rAs r{_get_random_poolzPoolsManager._get_random_pool5sX 4 y>>Q  Q< 9~~!q$''rryrArrKcg}d}|t|krh||j}||kr6||r|||n||krn|dz }|t|kh|S)Nr)rr+rrv)ryrArKsublist next_prioritys r{ _get_sublistzPoolsManager._get_sublistGs c%jj  !%L1M},,u1G1G1I1I,uU|,,,,.. QJE c%jj  rcd}|D](}|r ||jkrn|dz })g}|sD|t|kr1||||}|dz }|s|t|k1||S)Nrr)rr+rrr)rryrKrArsubpools r{_get_next_poolzPoolsManager._get_next_poolVs  D~~ LDM$A$A QJEE ec%jj00''ulCCG QJE ec%jj00$$W---r Optional[int]c||r |djS|D]!}|r |j}|cS"|djS)Nr)r+r)ryrKt_pools r{_get_next_priorityzPoolsManager._get_next_prioritydsf  E 8$ $ $ $F!! $% #### $Qx  rreviveOptional[bool]c||}|D]n}|r|\}}|r|}tj|t |zkr|odS)N)seconds)rrrmrr5rrj)rr rryrr} timeout_stamps r{_check_unavailable_poolsz%PoolsManager._check_unavailable_poolsqs)) % %D~~ %)%;%;%=%= "G] ! |~~71K1K1K!KLL""$$$ % %rrc ^ d fd }||}|dd}g}|||||}|t d||d<||| t j} ~ d kr|5 d j }n##tj $rt d dwxYw | r |js|||n#t t"t$f$r |n#t t"t$f$rYnwxYw |n# |wxYw d kr d j } |n#t t"t$f$rYnwxYw |n/# |wxYw#tj $rYnwxYw d k t+ } ||||n#t t"t$f$rYnwxYw d kr| d j }| |n#t t"t$f$rYnwxYw d k|n# d kr| d j }| |n#t t"t$f$rYnwxYw d k|wxYwYnwxYw|cdddS#1swxYwYnЉ j jkrEt+ } |||||S|5 d j }||||cdddS#tj $rt d dwxYw#1swxYwYn #t$t4tf$r}|d d|t9|tr dn| |||||| \|||}||d<||| )d|} t d| d|Yd}~vd}~wwxYw ~t d)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. r@rrprc|tdd}t d|dddDdkr1|djdSdS)Nrc34K|]}t|VdSrrMrNs r{rPzOPoolsManager.get_connection..set_mysqlx_wait_timeout..rQrrRrrSrV)rrXrYrdrZrr.)r@r\rs r{set_mysqlx_wait_timeoutz.set_mysqlx_wait_timeouts''/0088::DDFFqI!LCBBSYYs^^A%6%<%#J%%% $%!% 6 6s ; ; ; ; 6 6s ; ; ; ;#'**,,"2"2 !D*.((.2D(>(>(>,8'>+R%-%-%-(,%-)-(>(>s(C(C(C(C(>(>s(C(C(C(C(-{!)!)!)$(D!) #'**,,"2"2-&6t&<&< $ 5 5c : : : # 7 7 < < < <$0'>#J%%% $% '+jjllQ&6&6%-.2hh26484F/7/*/*),(<(<(>(>(>(,(>(>s(C(C(C(C,8'>+R%-%-%-(,%-'+jjllQ&6&6djjllQ&6&6%-.2hh26484F/7/*/*),(<(<(>(>(>(,(>(>s(C(C(C(C,8'>+R%-%-%-(,%-'+jjllQ&6&6&6&6&6&6G,-Z #{=#=#=#=#=#=#=#=#=#=#=#=#=#=#=#=#=#|*T-???*400C))#...KKMMM++C000JXXX"&((t?Q("R"RCIIKKK33C888#& XXXXXXXX %{XXX"+,L"M"MSWWX XXXXXXXX#L)<   !!">4">">">">???c9--9((++++--dC888--h777**5,??<#'#:#:5,#O#OL/;H^,..ulCCD|$kk*55'-$'--- ##/ iZFGGGs#T>P5CP5 C==P5AEP5P$*E?>F2?FF2FF2P$2G  P$%I$HI H1.I 0H11I 4P$ I!!P$$I63P$5I66P$AKNK0-N/K00N3P$ AMP$M,)P$+M,,P$P"AO('P(P ?PP PP$!P5#P$$P5( T5P99T<P9=ATTT#|// 0@0@0M0MNN <'' ''---5zzrrr#Union[InterfaceError, TimeoutError]cd} |j}t|}n#ttf$rYnwxYw|s,|j}t D] \}}||vr|} |r||dS|ddS)a\Sets a pool as unavailable. The time a pool is set unavailable depends on the given error message or the error number. Args: pool (ConnectionPool): The pool to set unavailable. err (Exception): The raised exception raised by a connection belonging to the pool. Ni)errnorrrwr_TIMEOUT_PENALTIESitemsr)rrpenaltyerr_noerr_msgrzrs r{rz!PoolsManager.set_pool_unavailable"s YF26:GG)    D  $gG06688 $ $ U'>>#G  )   ) ) ) ) )   ( ( ( ( (s --)rpr)rtrqr:rqrpr)r r!rpr)r r!rpr)rr!rpr)rrrpr)ryrrArrKrrpr)ryrrKrrprr)ryrrKrrpr)r r!rrrpr)r r!rpr)rr!rpr)rrrrrpr)rrrrrr__annotations__rsrvrzr staticmethodr}rrrrrrrrrrrr{rrs $J#### G    ''''  .*$*$*$\*$XAAAA,   \ "   \  . . . .CG ! ! ! !\ !BF % % % % %RHRHRHRHh )))\)))rrceZdZdZd.dZd/dZd0dZd1dZed2dZ e j d3dZ d2dZ d4dZ d5dZ d6dZd7dZd8d!Zd9d"Zd7d#Zd1d$Zd1d%Zd1d&Zd:d;d)Zd9d*Zd9d+Zd1d,Zd1d-Zd'S)<Sessiona*Enables interaction with a X Protocol enabled MySQL Product. The functionality includes: - Accessing available schemas. - Schema management operations. - Retrieval of connection information. Args: settings (dict): Connection data used to connect to the database. r r!rprc"|dtj|_||_|dr|drtst d t j|dd}n6#t j j $r}t d|dd|d}~wwxYwg|jd<|D]O}|jd |j d |j|j|jd Pd |jvs|jd d uri|jd<|d|vrf|dr^t%|t%||_|jt-dn2t/|j|_|j|jd}|rz |dt5|dS#t8$r3}|jdkr|jnd|d}t ||j|d}~wwxYwdS)NrircrnztMySQL host configuration requested DNS SRV. This requires the Python dnspython module. Please refer to documentationSRVz Unable to locate any hosts for ''reT)omit_final_dot)rcrdr+rrkFrppoolingz+Connection could not be retrieved from poolrbzUSE iDefault schema '' does not exists)rr4use_purer$HAVE_DNSPYTHONr*dnsresolverquery exception DNSExceptionrvtargetto_textrdr+r_init_attributesrrr _connectionr-rrrrIrYr,rr)rr  srv_recordsrsrvrberrmsgs r{rzSession.__init__Os&ll:x7HII )1 <<   HLL$;$; ! $<  !l00&1A5II =-   $Jx7GJJJ )+DN9 %"  y)00 # 2 2$ 2 G G #$'L"%*  $4> 9 9~56eCC+-DN< (  ! ! # # #  Xi%8 NN & &x 0 0 0+7>>+H+H+R+RD ' MNNN( *$.99D    $ $ & & &##H--  A A: 0 8 8::;;CCEEEEE# A A AyD((GGEFEEE %VSY77S@ A A As0.&BC)CC7I J .JJ c|Srrrs r{ __enter__zSession.__enter__ rexc_typeType[BaseException] exc_value BaseException tracebackrc.|dSrrrrrrs r{__exit__zSession.__exit__ rc <tjdkrldtjdvrd}n1dtjdvrd}ntj}dtjd}nptj}tjd krd tjd}n)d tdd }tj d }|ddkrd}nd}ttj ||tjdddt ddD|d}|jd|d|jvr|jdD]}|jd|}t'|tst)d|dt+|dkrt)d|d|drt)d|t'|tst)d|d|dt+|d krt)d|d!|d"||jd|<d#Sd#S)$z5Setup default and user defined connection-attributes.nt64rx86_6432i386zWindows-rDarwinzmacOS-rRrH GPLv2zGPL-2.0 Commercialzmysql-connector-pythonrc,g|]}t|Sr)rq)r8r2s r{r9z,Session._init_attributes..s(F(F(FAQ(F(F(Fr)_pid _platform_os _source_host _client_name_client_version_client_licenserprkAttribute name '' must be a string type "' exceeds 32 characters limit sizeruzHKey names in 'session-connect-attributes' cannot start with '_', found: z ' value 'z' must be a string type ' value: '$' exceeds 1024 characters limit sizeN)osrlr architecture win32_vermachinermac_verrxrSrTrrqgetpidrf gethostnamerUr$rrr*r startswith)r platform_archos_verlicense_chunksclient_licensedefault_attributes attr_name attr_values r{rzSession._init_attributess 7d??x,..q111=E .00333 & ( 5 7 7 9 2 4 4Q 799FF$,..M  H,,9("2"4"4Q"799"4"6"6qs";<< s++ !  ' '&NN)N $$&".004"xx(F(F1(F(F(FGG-   |$++,>??? "dn 4 4!^,CD! E! E !^,CDYO !)S11(M9MMMy>>B&&(%9%%% '',,(>2;>> "*c22(1911z111 z??T))(=9== === ;E|,Y77E 5 4! E! ErrctjS)z8bool: `True` to use pure Python Protobuf implementation.)r4rrs r{rzSession.use_pures   rrcvt|tstdtj|dS)Nz)'use_pure' option should be True or False)rrr.r4 set_use_pure)rrs r{rzSession.use_pures;%&& P"#NOO Oe$$$$$rc>|jjS)zzReturns `True` if the session is open. Returns: bool: Returns `True` if the session is open. )rrUrrs r{rzSession.is_opens &..000rrrqrGc,t|j|Sr')rGrr(s r{rz Session.sqlsD,c222rrc|jS)z~Returns the underlying connection. Returns: mysqlx.connection.Connection: The connection object. )rrs r{rzSession.get_connection s rrc|d}d|DS)zReturns the list of schemas in the current session. Returns: `list`: The list of schemas in the current session. .. versionadded:: 8.0.12 zSHOW DATABASEScg|] }|d S)rr)r8rows r{r9z'Session.get_schemas.. s5553A555r)rrYrdrgs r{ get_schemaszSession.get_schemas sB*++335555&"2"2"4"45555rrlr)c"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(rrls r{ get_schemazSession.get_schema sdD!!!rOptional[Schema]c|jjd}|r|tt |} |dd|krt||Sn"#t$rtd|ddwxYwdS)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. rbrrrN) rr rr_SELECT_SCHEMA_NAME_QUERYformatr0rYrdr)rr.)rrbress r{get_default_schemazSession.get_default_schema s!*..x88  299&..IIJJ   q6!9&&!$///'   &@v@@@ ts !B$$Cc|jdtt |ddS)zDrops the schema with the specified name. Args: name (string): The name of the Schema object to be retrieved. rTN)rr_DROP_DATABASE_QUERYrrIrs r{ drop_schemazSession.drop_schema= sH )) '../?/E/EFF     rc|jdtt |dt ||S)zCreates a schema on the database and returns the corresponding object. Args: name (string): A string value indicating the schema name. rT)rr_CREATE_DATABASE_QUERYrrIr)rs r{ create_schemazSession.create_schemaG sP )) )001A$1G1GHH$   dD!!!rc>|jddddS)z+Starts a transaction context on the server.rzSTART TRANSACTIONTNrrrs r{start_transactionzSession.start_transactionS s$ ))%1DdKKKKKrc>|jddddS)zXCommits all the operations executed after a call to startTransaction(). rCOMMITTNrrs r{commitzSession.commitW s% ))%4@@@@@rc>|jddddS)zYDiscards all the operations executed after a call to startTransaction(). rROLLBACKTNrrs r{rollbackzSession.rollback] s% ))%TBBBBBrNrc|tj}nIt|tr%t |dkrt d|jddt|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. NrInvalid SAVEPOINT namerz SAVEPOINT T) uuiduuid1rrqrstripr.rrrIrs r{ set_savepointzSession.set_savepointc s <jll$DDD#&& =#djjll*;*;q*@*@"#;<< < )) 8 0 6 688$    rct|tr%t|dkrt d|jddt|ddS)zRollback to a transaction savepoint with the given name. Args: name (string): The savepoint name. rr'rzROLLBACK TO SAVEPOINT TNrrqrr*r.rrrIrs r{ rollback_tozSession.rollback_tox s~ $$$ =DJJLL(9(9Q(>(>"#;<< < ))  =%5d%;%; = =      rct|tr%t|dkrt d|jddt|ddS)z{Release a transaction savepoint with the given name. Args: name (string): The savepoint name. rr'rzRELEASE SAVEPOINT TNr-rs r{release_savepointzSession.release_savepoint s~ $$$ =DJJLL(9(9Q(>(>"#;<< < ))  9!1$!7!7 9 9      rcj|jt|j|_dS)zCloses the session.N)rrrr$rs r{rz Session.close s/ &&(((%dn55rc8|jdS)z8Closes all underliying connections as pooled connectionsN)rrrs r{close_connectionszSession.close_connections s ))+++++rrrprrrrrrrrprrr)rrrprr))rpr)rpr)rlrqrpr))rpr)rlrqrprr)rlrrprq)rrrrrrrrrprsetterrrrr rrrrrr"r%r+r.r0rr3rrr{rrBs  9A9A9A9AvKEKEKEKEZ!!!X!_%%%_% 1111 3 3 3 3     6 6 6 6 " " " "<     " " " "LLLLAAAA CCCC *          6666 ,,,,,,rrc^eZdZdZ dd d Zd!d Zd"dZd#dZd$dZd%dZ d&dZ d'dZ d(dZ dS))Clienta(Class 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 Nconnection_dictr! options_dictrrprc||_|i}g|_tj|_||dd||dd||dd| |dd|j |jd<|j |jd<|j|jd <dS) Nr,r-r.rr/enabledTrrt) r sessionsr(uuid4rt_set_pool_sizer_set_max_idle_time_set_queue_timeout_set_pool_enabledpooling_enabledr,)rr9r:s r{rzClient.__init__ s )8  L') $(JLL L,,Z<<===  0 0! D DEEE  0 0! D DEEE |// 4@@AAA#'#7 i $(M j!%)^ k"""rc|Srrrs r{rzClient.__enter__ rrrrrrrrc.|dSrrrs r{rzClient.__exit__ rr pool_sizerct|tst|tr|dkstd|d|dkrtn||_dS)aSet 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. rzGPool max_size value must be an integer greater than 0, the given value is not validN)rrrr_CNX_POOL_MAXSIZEr,)rrFs r{r?zClient._set_pool_size sw y$ ' ' i-- q== 8(888  .7!^^)) rr.ct|tst|tr|dkstd|d||_|dkrt nt|dz |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. rfzYConnection max_idle_time value must be an integer greater or equal to 0, the given value rHrrr.N)rrrrr._CNX_POOL_MAX_IDLE_TIMEr )rr.s r{r@zClient._set_max_idle_time s }d + + mS11 !2%% L/<LLL  +'4'9'9 # #s=SWCW?X?X o&&&rr<rc\t|tstd||_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)rrrrC)rr<s r{rBzClient._set_pool_enabled s4'4(( O !MNN N&rr/ct|tst|tr|dkstd|d||_|dkrt nt|dz |jd<d|jvr|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. rfzYConnection queue_timeout value must be an integer greater or equal to 0, the given value rHrrr/rjN)rrrrr/_CNX_POOL_QUEUE_TIMEOUTr )rr/s r{rAzClient._set_queue_timeout s }d + + mS11 !2%% L/<LLL  +'4'9'9 # #s=SWCW?X?X o& DM 1 1/3/ADM+ , , , 2 1rrcbt|j}|j||S)z~Creates a Session instance using the provided connection data. Returns: Session: Session object. )rr r=rvrsessions r{ get_sessionzClient.get_session. s. $-(( W%%%rct|j|jD]}|dS)z*Closes the sessions opened by this client.N)rrr r=r3rPs r{rz Client.close8 sL!!$-000} ( (G  % % ' ' ' ' ( (rr)r9r!r:rrpr)rpr8r5)rFrrpr)r.rrpr)r<rrpr)r/rrprr4r) rrrrrrrr?r@rBrArRrrrr{r8r8 s2644444*KKKK2    8 ' ' ' 'BBBB@((((((rr8path"Union[Dict[str, List[Dict]], Dict]c|dd}d|vo1|ddko|ddk o)|do|d}g}t|r |ddn|}d }|D] }i}t |}|r@|d}t|d |d <|dz }n6t|}|r|d}d |d <td |}|j std| ||j |jn&#t $r}t#d|d|d}~wwxYw||"d |cxkrt'|krnnt#dd|rd|in|d S)zParses a list of host, port pairs. Args: path: String containing a list of routers or just router Returns: Returns a dict with parsed values of host, port and priority if specified. r,rr[]rfrrHr+r,z//zInvalid address: rTz Invalid URI: iNr.r/re)replacerrendswith _SPLIT_REr _PRIORITY_REmatchgroupr _ROUTER_RErhostnamer*rrdrr.rvr) rTrre address_listr4addressr5r_rs r{_parse_address_listre? sJ <<R D _ MC1!4 MCA9MN  OOC   MM#   G??#@4"::DAALN""7++  )kk!nnG!$U[[^^!4!4F:  a NN$$W--E )++a..%(z"g((~ @ !>W!>!>?? ? I MMu~EJM ? ? ? ? I I I"#83#8#8$??S H I v>----C ------- *    $) 8Iw  gaj8s'!F  F,F''F,urir!cddi}t|}|r|nd|f\}}|dvrtd|d|dkrd|d <|d d d d \}}|d d d d \}}|d}||d ddkr"|dkr|dd\}|d<|d}|r|rd|vrtd|d| dd\} } t| t| c|d<|d<| drt||d<nF| drtd| t|d} t|dD]\} } | dd}|| vrtd | d|t"vr&t| d||<s| }|d!vrd||<|d"vrd#||<|||<|S)$aqParses the connection string and returns a dictionary with the connection settings. Args: uri: mysqlx URI scheme to connect to a MySQL server/farm. Returns: Returns a dict with parsed values of credentials and address of the MySQL server/farm. Raises: :class:`mysqlx.InterfaceError`: If contains a invalid option. rbrWr)r mysqlx+srvzScheme 'z' is not validrhTrn@NrH?/)rfrrz()rzMalformed URI 'rr`ra)rkrNrrfz\.zWindows Pipe is not supported)r`rarnrurRzInvalid option: ')1true)0falseF)_URI_SCHEME_REr_groupsr* partitionrfindfindrsplitr*rrrrrerr[rm _SSL_OPTS)rfr r_schemeuserinfotmprc query_strposr`rainvalid_optionsrzvaloptval_strs r{_parse_connection_urirv s!)"~H   % %E$)>%,,...#KFC --->>>>??? "MM#&&sss+MHcmmC((1-OD) **S//C CDDzsr!!cAgg#';;sA#6#6 hx ::d  D 7x73h#6#65s555666^^C++ND(-4T]]GH>LJ"KK 55H4EHHHIIIHy) + +F FE * * * * + 8  !!!XJ' X#*,6x,E,ES  &&(((8$$HZ  + X X X !Mhz6J!M!M!MNNTW W X 8 # #(<  #  A ) ) !!BCC CH!9!9:;;;H8!;!;>???8 Z 8 8A!! CDDD  P#(24(=(=G ""$$$4  HV   + P P P !E(62B!E!E!EFFC O P  }-  &1k3//! !!##))+++ ''H] # #  +    A]+AAA    8++ h78# > > 1288::@@FF #& :5K5Q5QRU5V5V12259122H%=>u NN  R  << & &+*> > >15H- .(**'111H$$ (#45s;; O.1(;L2M.N.N*+x(9:C@@ !-.22  3   (   H(9-t44 M !KLL L << ! !  N  <<    M  << " "  S   8  HLL$8$8  !!x(((X%%"8,,,,,&%sX$A C22D"DD"A H,,I=II,A J::K* K%%K* A!P--Q default_portrcd|vr|dr t|d|d<|ddks |ddkrtddnB#t$rtdddt$rtd|dddwxYwd |vrA|d r9 t|d |d <dS#t$rt d dwxYwd |vr |r ||d <dSdSdS) zValidate hosts. Args: settings (dict): Settings dictionary. default_port (int): Default connection port. Raises: :class:`mysqlx.InterfaceError`: If priority or port are invalid. r+rr,z1Invalid priority value, must be between 0 and 100r-zInvalid priorityNzInvalid priority: rdz Invalid portrc)rr. NameErrorrr*)r rs r{rr: spX(:"6 #&x ';#<#+>s+C+C  BBB B4>BBB  ; ; ;///490114601 F$QrT*0055# - -Drzz JJsOOM%a(I+.}+=+=+A+A}Q''rHJ&&$PiPPP%-Jy ! ! - J % % # / /I#I.Jj#.. .!*-- $.Jy ! !  / J % % v)=)=  602H, - -05H, - J $ $ # ' 'I$&Jy ! ! ' J % %  - -Drzz JJsOOM%a(I+.}+=+=+A+A}Q''rHJ&&$PiPPP%-Jy ! ! - D ) )  ,( , , ,   %/%5%5%7%7   !Izi-- $IyIII9~~""$TyTTT##C(( $1$-111 j#.. $$)$$z$$$ :%%$1)11z111& )3H $%%%rcng}d|vrdS|d}t|tr|dr|dst d|d|ddd }|D]Y}|}|d kr||vr#t tjd | | |Znt|trS|st d |D]>}||vr#t tjd | | |?n^t|tr|D]}| |n.t dd td|d|st d g}g}g}|D]T}|tvr| ||tvr| |?| |U|rP|dgkr.ts't!t#j|t|||d<dS|r't!t'j|t|r't t)j|tdS)zValidate tls-versions. Args: settings (dict): Settings dictionary. Raises: :class:`mysqlx.InterfaceError`: If tls-versions name is not valid. r^NrYrZz%tls-versions must be a list, found: 'rrrfrXrW tls_versionsrrzKAt least one TLS protocol version must be specified in 'tls-versions' list.z>tls-versions should be a list with one or more of versions in z, z . found: 'r )rrqrr\r*rr*rOrrvrrrxr#r!rr+rPrrQrR) r rtls_versions_settingstls_verstls_verruse_tls_versionsdeprecated_tls_versionsnot_tls_versionss r{rr sLX%%$^4'--( ! , ,S 1 1 %..s33 !P8MPPP )2.44S99 - -G!--//Kb  l**$,3+;    , , , , - )4 0 0 $  4 - ) )G,&&$,3wWWW    ( ( ( (  ) )3 / / , ) )G    ( ( ( ( ) Lyy/00 L L.K sUUUvfUFUUUrzItls-ciphersuites should be a list with one or more ciphersuites. Found: 'r^TrrrRtls_ciphersuitesrz The value 'z.' in cipher suites is not a valid cipher suite)rrqrr\r*rr*upperrvrrrr#rrArr$extendr"rrOr) r rtls_ciphersuites_settingstls_cssrr newer_tls_vertranslated_namesiani_cipher_suites_namesossl_cipher_suites_namesrrltranslated_names r{rr) s?)) (); <+S11 % 0 0 5 5 )22377 !1-111 ,AbD177<<  L  0 0F\\^^))++F 0 ''/// 0 -c{ ; ; UU1JUUU 40 4 4 4    << - - 5 qqq!! n %aaa (d### OM!*,*9 &} 5 5 99CC !''(9'(BCCC ''(8(ABBBB  $;;4#;;;  # #D ) ) ) ) - - -6t??M nsDk 2 2LJKKK.#&& 8 :n55LL    S    (..00 8 8JC27LS#.. / / lD ) )  O   L  &**955/D22  N +:*=*=i*N*NY'+:+>+>z2+N+NZ(0?0C0COUV0W0W_-0?0C0COUV0W0W_-   ! # # !S/!S!STT T |  q  IL4E4E4G4GII  -!5 6 66s<B B0B++B0)ror rprq)r}r~rpr)rr rpr )rTrqrprU)rfrqrpr!rr)r r!rrrpr)rr ror rpr!)rr ror rpr)rrrrrpr8)r __future__rtypesrrrPROTOCOL_TLSv1PROTOCOL_TLSv1_1PROTOCOL_TLSv1_2rhasattrrPROTOCOL_SSLv23r r ImportErrorrrrr8r>rerfrrr(rtypingr r r rrrrrr dns.exceptionr dns.resolverrrr functoolsr json.decoderr urllib.parserrrauthenticationrrr constantsr r!r"r#r$r%r&r'crudr)errorsr*r+r,r-r.r/helpersr0r1r2r3protobufr4rVr5r6r7r8r9rr:r;r<r=r>rr?r@rArBrCrDrErFrGrHrIrJrKrLrMrTrvmysql.connector.abstractsrOrPrQrRmysql.connector.utilsrSmysql.connector.versionrTrUrrrrrrXrIr=compiler<rKrN_PENALTY_SERVER_OFFLINE_PENALTY_MAXED_OUT_PENALTY_NO_ADD_INFO_PENALTY_CONN_TIMEOUT_PENALTY_WRONG_PASSW_PENALTY_RESTARTINGrrr]VERBOSEr^rarqrwrr|rrr rrrrrr9rrrr8rerrrrrrrrRrrrr{rs >;:""""""JJJM#''L 73 ' ' !   !m44HMLLL    SSSSSSSSSSSSSSSSSSSSSSNNNNN )(((((((((((((5555555555VVVVVVVVVV                    GFFFFFFFFFQQQQQQQQQQQQQQGGGGGGGGGGGGGG                          JIIIIIIIIIII 54444444444444  I  3;R+!rz"9::!!"#+$/*-)&)!%&9: BJ) * * rz> KK RZ12: > > ABB     &8::::X(X(X(X(X(X(X(X(v;;;;|!,!,!,!,!,T!,!,!,HA'A'A'A'A'A'A'A'HW W W W W W W W tF'F'F'F'F'zF'F'F'RT#T#T#T#T#U[T#T#T#nE)E)E)E)E)E)E)E)P Z,Z,Z,Z,Z,Z,Z,Z,z ](](](](](](](](@49494949n<<<<~B-B-B-B-L=A!(!(!(!(!(Hm3m3m3m3`XXXXXXXXvV4V4V4V4r!!!!H"c7c7c7c7c7c7s$A A A*)A*0B;;CC