o g($@sdZgdZddlZddlmZddlmZddlmZm Z m Z ddl m Z d d l mZd d lmZz ddlZddlZWn eyGd ZYnwd Zzd dlmZWn ey]dZYnwdededefddZdedefddZdS)z8MySQL Connector/Python - MySQL driver written in Python.)CMySQLConnectionMySQLConnectionconnectN)Any)DEFAULT_CONFIGURATION)ErrorInterfaceErrorProgrammingError) ERROR_NO_CEXT)MySQLConnectionAbstract)rFT)rargskwargsreturncsd|vr |dnd}t|tstd|rtstdd|vr%tdd|vr-tdd |vr5td d |vr?td |d <z tj|d d }Wntj j y^td |d ddwg}|D]}| |j j dd|j|j|jdqc|jddddd|D|d <d |vrtdi|IdHS|dd}d|vr|d=|stdurtttr|st|i|}nt|i|}|IdH|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_socketzszconnect..)keycSsg|] }|d|ddqS)rr)rrr).0srvrrr szconnect..use_purer)pop isinstanceboolr HAVE_DNSPYTHONrdnsresolverquery exception DNSExceptionappendtargetto_textrrrsort_get_failover_connectiongetr ImportErrorr rr)rrr srv_recordsrr"r$cnxrrrr:sr(       rc s|}z|d}Wn tytddw|d=tgd}d}|D][}t||}|rJt|dkr9dnd}d |}td |d |t|d rS|d7}|d d |d <|d dksg|d d krpt d|d t |d t st d|d q%d|krt|krt di} g} t |ddddD] }|d | vr|g| |d <| |d q| |d |q| D]H} | | } tt| D];} t| d}td|}| |}|}|||d dztdi|IdHWStyYqwqt d)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 routercSs|dS)Nrrrrrrrsz*_get_failover_connection..T)r reversez,Unable to connect to any of the target hostsr)copyKeyError ValueErrorsetkeyslenjoinhasattrr3r r&intr sortedr.rangerandomrandintr%updaterr)rconfigrsupport_cnx_argspriority_countserverdiffarglstserver_directoryserver_priority_listr failover_list_lastindex new_configrrrr2s             r2)__doc____all__rKtypingr constantsrerrorsrr r poolingr abstractsr connectionr dns.exceptionr) dns.resolverr4r(connection_cextrrr2rrrrs.        m