>ijg($dZgdZddlZddlmZddlmZddlmZm Z m Z ddl m Z d d l mZd d lmZ ddlZddlZd Zn #e$rd ZYnwxYw d dlmZn #e$rdZYnwxYwdededefdZdedefdZdS)z8MySQL Connector/Python - MySQL driver written in Python.)CMySQLConnectionMySQLConnectionconnectN)Any)DEFAULT_CONFIGURATION)ErrorInterfaceErrorProgrammingError) ERROR_NO_CEXT)MySQLConnectionAbstract)rTF)rargskwargsreturncKd|vr|dnd}t|tstd|r,tstdd|vrtdd|vrtdd |vrtd d |vrt d |d < t j|d d }n2#t j j $rtd |d ddwxYwg}|D]D}| |j d|j|j|jdE|dd|D|d <d |vrt%di|d{VS|dd}d|vr |d=|st(t+t,t(r|s t)|i|}n t/|i|}|d{V|S)aCreates or gets a MySQL connection object. In its simpliest form, `connect()` will open a connection to a MySQL server and return a `MySQLConnectionAbstract` subclass object such as `MySQLConnection` or `CMySQLConnection`. When any connection pooling arguments are given, for example `pool_name` or `pool_size`, a pool is created or a previously one is used to return a `PooledMySQLConnection`. Args: *args: N/A. **kwargs: For a complete list of possible arguments, see [1]. If no arguments are given, it uses the already configured or default values. Returns: A `MySQLConnectionAbstract` subclass instance (such as `MySQLConnection` or a `CMySQLConnection`) instance. Examples: A connection with the MySQL server can be established using either the `mysql.connector.connect()` method or a `MySQLConnectionAbstract` subclass: ``` >>> from mysql.connector.aio import MySQLConnection, HAVE_CEXT >>> >>> cnx1 = await mysql.connector.aio.connect(user='joe', database='test') >>> cnx2 = MySQLConnection(user='joe', database='test') >>> await cnx2.connect() >>> >>> cnx3 = None >>> if HAVE_CEXT: >>> from mysql.connector.aio import CMySQLConnection >>> cnx3 = CMySQLConnection(user='joe', database='test') ``` References: [1]: https://dev.mysql.com/doc/connector-python/en/connector-python-connectargs.html dns_srvFz(The value of 'dns-srv' must be a booleanztMySQL host configuration requested DNS SRV. This requires the Python dnspython module. Please refer to documentation unix_socketzzconnect..sQz]Qx[L$A)keyc0g|]}|d|ddS)rr)rrr).0srvs r! zconnect..s5   ;>S[#f+ 6 6   r#use_purer)pop isinstanceboolr HAVE_DNSPYTHONrdnsresolverquery exception DNSExceptionappendtargetto_textrrrsort_get_failover_connectiongetr ImportErrorr rr)rrr srv_recordsrr'r)cnxs r!rr:sP(1F':':fjj###G gt $ $IGHHH+   8  F " " N  V   M     S    26:F6N ,,,VF^UCCKK})    D6&>DDD     C OOJ..d.CCH # !j       AA BBB  BJ   z V-77777777777zz*e,,HV :  -,4m,, ,/////t.v.. ++-- Js )&C/C?cK|} |d}n#t$rtddwxYw|d=tgd}d}|D]}t||z }|rAt |dkrdnd}d |}td |d |t|d r|dz }|d d |d <|d dks |d d krtd|d t|d tstd|d d|cxkrt |krnntdi} g} t|ddD]U}|d | vr(|g| |d <| |d 4| |d |V| D]} | | } tt | D]} t | dz }t!jd|}| |}|}|||d d t)di|d{VccS#t*$rYwxYwtd)aReturn a MySQL connection and try to failover if needed. An InterfaceError is raise when no MySQL is available. ValueError is raised when the failover server configuration contains an illegal connection argument. Supported arguments are user, password, host, port, unix_socket and database. ValueError is also raised when the failover argument was not provided. Returns MySQLConnection instance. rzfailover argument not providedN) userpasswordrrrdatabase pool_name pool_sizerrr sz, z Unsupported connection argument z in failover: rdz9Priority value should be in the range of 0 to 100, got : zDPriority value should be an integer in the range of 0 to 100, got : z\You must either assign no priority to any of the routers or give a priority for every routerc|dS)Nrrrs r!r"z*_get_failover_connection..s :r#T)r$reversez,Unable to connect to any of the target hostsr)copyKeyError ValueErrorsetkeyslenjoinhasattrr8r r+intr sortedr3rangerandomrandintr*updaterr )rconfigrsupport_cnx_argspriority_countserverdiffarglstserver_directoryserver_priority_listr failover_list_lastindex new_configs r!r7r7s[[]]FE*% EEE9::DEz   N6;;==!!$44  TQ##BC))D//CK3KKcKK  6: & & a N#ZZ C88z *  ! !VJ%7#%=%= . +.. &,c22  3$Z033    >))))CMM)))))    '>'>MMM@@ * %5 5 54:8 VJ/ 0 ' 'z(: ; ; ; ; VJ/ 0 7 7 ? ? ? ?(  (2 s=))**  A}%%)DN1d++E"&&u--FJ   f % % % NN:t , , , $22z2222222222222      G H HHs!<J'' J43J4)__doc____all__rRtypingr constantsrerrorsr r r poolingr abstractsr connectionr dns.exceptionr. dns.resolverr-r9connection_cextrrr7rr#r!rns:?> < < < ------<<<<<<<<<<######......''''''NNNNN 1111111jjj0GjjjjZXISXI5LXIXIXIXIXIXIs!?A A  AAA