3 S^Q-@s6dZddlZddlZddlZddlmZddlmZmZm Z m Z m Z m Z ddl mZmZddlmZmZmZddlmZmZmZmZdd lmZdd lmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%dd l&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.dd l/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>dd l?m@Z?ejAdZBejAdejCZDddddgZEeEddddddddddddg ZFejGeHjIejJd d!ZKd"d#ZLd$d%ZMd&d'ZNd(d)ZOd*d+ZPd,d-ZQd.d/ZRd0d1d/d-d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZg-ZSdS)[z$MySQL X DevAPI Python implementationN) constants) INT_TYPES STRING_TYPESJSONDecodeErrorurlparseunquote parse_qsl)ClientSession)AuthLockContentionSSLMode)Schema CollectionTableView)DbDoc) ErrorInterfaceError DatabaseErrorNotSupportedError DataErrorIntegrityErrorProgrammingErrorOperationalError InternalError PoolError TimeoutError)ColumnRowResultBufferingResult RowResult SqlResult DocResult ColumnType) StatementFilterableStatement SqlStatement FindStatement AddStatementRemoveStatementModifyStatementSelectStatementInsertStatementDeleteStatementUpdateStatementCreateCollectionIndexStatementExpr ReadStatementWriteStatement) ExprParserz,(?![^\(\)]*\))z!^\(address=(.+),priority=(\d+)\)$zssl-certzssl-cazssl-keyzssl-crluserpasswordschemahostportrouterssocketzssl-modeauthzuse-purezconnect-timeoutzconnection-attributescCs|jdd}d|ko.|jddko.|jddk oD|jdoD|jd}g}tj|r^|ddn|}xx|D]p}i}tj|}|r|jd}t |jd|d <t d j |}|j st d j ||j|j |jd |j|qjW|rd |iS|dS)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[]priorityz//{0}zInvalid address: {0})r:r;r<r)replacecount startswithendswith_SPLITsplit _PRIORITYmatchgroupintrformathostnamerupdater;append)patharrayr<Z address_listaddressrouterrOrZA/tmp/pip-build-8app2_gc/mysql-connector-python/mysqlx/__init__.py_parse_address_listCs$     r\cCsddi}dj|jdrdnd|}|jdd\}}|jdddd \}}|jd ddd \}}|jd }||djd d!kr|d kr|jd d\}|d<|jd}| s| sd|krtdj||jdd\}} t |t | |d<|d<|jd"rt ||d<n$|jdr$tdn|j t |xt |dD]\} } | j ddj} | |krntdj| | tkrt | jd|| <n8| j} | d#krd|| <n| d$krd || <n| || <q>W|S)%aParses 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. r9r@z{0}{1}z mysqlx://z://r@NrE?/)rz()rBzMalformed URI '{0}'r7r8...r=z\.zWindows Pipe is not supported.T_-zDuplicate option '{0}'.1true0falseFrG)r_rarb)rerf)rgrh)rRrJrM partitionrfindfindrsplitstriprrrTr\r rHlower _SSL_OPTS)urisettingsrctempuserinfor:Z query_strposr7r8keyvaloptZval_strrZrZr[_parse_connection_urifs@            rxcstjjt}|r*tdjdj|dkrNx*dD] }t|qjdt jt jgkr>t dd krhjdt jt jgkrht ddkry"djd<tj dWn,t t fk rt djdYnXdkrtdS)aValidates the settings to be passed to a Session object the port values are converted to int if specified or set to 33060 otherwise. The priority values for each router is converted to int if specified. Args: settings: dict containing connection settings. zInvalid options: {0}.z, r<r:zssl-modezInvalid SSL Mode '{0}'.c3s|]}|kVqdS)NrZ).0ru)rqrZr[ sz%_validate_settings..z*SSL options used with ssl-mode 'disabled'.zssl-crlzssl-cazCA Certificate not provided.zssl-keyzssl-certz Client Certificate not provided.z Cannot verify Server without CA.z%Must verify Server if CA is provided.r>zInvalid Auth '{0}'zconnection-attributesN)setkeys difference _SESS_OPTSrrRjoin_validate_hostsrnrindexAttributeError ValueErrorrZDISABLEDanyrogetZVERIFY_IDENTITYZ VERIFY_CAr validate_connection_attributes)rqZ invalid_optsrYrZ)rqr[_validate_settingssJ   rcCsd|krB|drByt|d|d<Wntk r@tdYnXd|kr|dryt|d|d<Wqtk rtdYqXnd|krd|d<dS)zValidate hosts. Args: settings (dict): Settings dictionary. Raises: :class:`mysqlx.InterfaceError`: If priority or port are invalid. rFzInvalid priorityr;z Invalid portr:i$N)rQ NameErrorr)rqrZrZr[rs rc Csi}d|krdS|d}t|tr|dkr8i|d<dS|jdoJ|jd rf|dkrftd j|n|dkr|dkrd |d<ni|d<dS|d djd }xb|D]Z}|dkrq|jd}|d}t|d kr|d nd}||krtdj|q|||<qWn.t|trHx2|D]*}||}t|ts8t |}|||<qWnt|t s^|dkrz|rni|d<nd |d<dSt|t rx|D]}d||<qWnt|t rx|D]b}|dkrĐq|jd}|d}t|d kr|d nd}||kr tdj|n|||<qWnt|t s4td j||rx|D]}||}t|tsftdj|t|dkrtdj||jdrtdj|t|tstdj||t|dkr@tdj||q@W||d<dS)zValidate connection-attributes. Args: settings (dict): Settings dictionary. Raises: :class:`mysqlx.InterfaceError`: If attribute name or value exceeds size. zconnection-attributesNr@rCrDFalserhTruerfzOconnection-attributes must be Boolean or a list of key-value pairs, found: '{}'FrrA=rz0Duplicate key '{}' used in connection-attributesz(Attribute name '{}' must be a stringtype z5Attribute name '{}' exceeds 32 characters limit size.rczEKey names in connection-attributes cannot start with '_', found: '{}'z1Attribute '{}' value: '{}' must be a string type.iz=Attribute '{}' value: '{}' exceeds 1024 characters limit size)rrhrrf)rrhrrf)rrhrG)rr) isinstancerrJrKrrRrMlendictreprboolr{list) rq attributesZ conn_attrsZconn_attributesattrZ attr_name_val attr_nameattr_val attr_valuerZrZr[rs                          rc Osi}|r^t|dtr$t|d}qt|dtrxT|djD]\}}|||jdd<q@Wn*|rx$|jD]\}}|||jdd<qlW|stdd|kryBt|dtrt|d|d<t|dt s|ddkrt Wnt k rt dYnXt ||S)aParses the connection string and returns a dictionary with the connection settings. Args: *args: Variable length argument list with the connection data used to connect to the database. It can be a dictionary or a connection string. **kwargs: Arbitrary keyword arguments with connection data used to connect to the database. Returns: mysqlx.Session: Session object. Raises: TypeError: If connection timeout is not a positive integer. rrcrdzSettings not providedzconnect-timeoutzEThe connection timeout value must be a positive integer (including 0)) rrrxritemsrHrrQrr TypeErrorr)argskwargsrqrurvrZrZr[_get_connection_settingsUs. rcOst||}t|S)aCreates a Session instance using the provided connection data. Args: *args: Variable length argument list with the connection data used to connect to a MySQL server. It can be a dictionary or a connection string. **kwargs: Arbitrary keyword arguments with connection data used to connect to the database. Returns: mysqlx.Session: Session object. )rr )rrrqrZrZr[ get_sessions rc CsXt|ttfstdt|}t|ttfs4tdt|trlytj|}Wqtk rhtdYqXn*i}x$|jD]\}}|||j dd<qzWt|tstdi}d|krN|j d}t|tstd|j dd |d<|j d d |d <|j d d |d <|j dd |d<t |d kr.tdj |t |d krNtdj |j t||S)a8Creates a Client instance with the provided connection data and settings. Args: connection_string: A string or a dict type object to indicate the connection data used to connect to a MySQL server. The string must have the following uri format:: cnx_str = 'mysqlx://{user}:{pwd}@{host}:{port}' cnx_str = ('mysqlx://{user}:{pwd}@[' ' (address={host}:{port}, priority=n),' ' (address={host}:{port}, priority=n), ...]' ' ?[option=value]') And the dictionary:: cnx_dict = { 'host': 'The host where the MySQL product is running', 'port': '(int) the port number configured for X protocol', 'user': 'The user name account', 'password': 'The password for the given user account', 'ssl-mode': 'The flags for ssl mode in mysqlx.SSLMode.FLAG', 'ssl-ca': 'The path to the ca.cert' "connect-timeout": '(int) milliseconds to wait on timeout' } options_string: A string in the form of a document or a dictionary type with configuration for the client. Current options include:: options = { 'pooling': { 'enabled': (bool), # [True | False], True by default 'max_size': (int), # Maximum connections per pool "max_idle_time": (int), # milliseconds that a # connection will remain active while not in use. # By default 0, means infinite. "queue_timeout": (int), # milliseconds a request will # wait for a connection to become available. # By default 0, means infinite. } } Returns: mysqlx.Client: Client object. .. versionadded:: 8.0.13 z(connection_data must be a string or dictz+connection_options must be a string or dictzA'pooling' options must be given in the form of a document or dictrdrcZpoolingz<'pooling' options must be given in the form document or dictenabledTmax_sizeZ max_idle_timerZ queue_timeoutz Unrecognized pooling options: {}z#Unrecognized connection options: {})rrrrrjsonloadsrrrHpoprrRr|r )Zconnection_stringZoptions_stringZ settings_dictZ options_dictruvalueZpooling_options_dictZpooling_optionsrZrZr[ get_clients>2      rr r exprr r rrrrrrrrrrrrrrrrrr r!r"r#r$r%r&rr'r(r)r*r+r,r-r.r/r0r1r2r3)T__doc__rerloggingr@rcompatrrrrrr connectionr r r r rZcrudrrrrZdbdocrerrorsrrrrrrrrrrrresultrr r!r"r#r$r%r&Z statementr'r(r)r*r+r,r-r.r/r0r1r2r3r4r5rr6compilerLVERBOSErNror~ getLogger__name__ addHandler NullHandlerr\rxrrrrrr__all__rZrZrZr[sP   4(D      #46m.d