3 (h+5@sdZyddlZddlmZWnek r4dZYnXdZyddlZddlZWnek rfdZ YnXdZ ddl Z ddl m Z ddl mZdd lmZdd lmZmZmZmZmZmZmZmZmZmZmZmZdd lmZmZm Z m!Z!m"Z"dd l#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2dd l3m4Z4iZ5dZ6ddZ7ddZ8ddZ9e9Z:e j;Zdddddddddddd d!d"d#d$d%d&d'dd(d)d*d+d,d-d.d/d/d0d1d2d3d4d5d6d7g%Z?dS)8z9 MySQL Connector/Python - MySQL driver written in Python N)CMySQLConnectionFT)version)MySQLConnection)DEFAULT_CONFIGURATION) ErrorWarningInterfaceError DatabaseErrorNotSupportedError DataErrorIntegrityErrorProgrammingErrorOperationalError InternalErrorcustom_error_exception PoolError) FieldFlag FieldType CharacterSet RefreshOption ClientFlag)DateTime TimestampBinary DateFromTicksTimestampFromTicks TimeFromTicksSTRINGBINARYNUMBERDATETIMEROWIDapilevel threadsafety paramstyle)read_option_filesz0MySQL Connector/Python C Extension not availablecKsddlm}m}m}y |d}Wntk r>|f|}YnXd|kr`|d r`t r`tt|R|tkr~|f|t|<n4t t||rt|j }d|kr|d|krt dWdQRXy t|j St k rtdj|YnXdS) z Return a pooled MySQL connectionr)MySQLConnectionPoolgenerate_pool_nameCONNECTION_POOL_LOCK pool_nameuse_pure pool_sizez)Size can not be changed for active pools.Nz)Failed getting connection from pool '{0}')Zpoolingr(r)r*KeyError HAVE_CEXT ImportError ERROR_NO_CEXT_CONNECTION_POOLS isinstancer-rget_connectionAttributeErrorr format)kwargsr(r)r*r+Z check_sizer8L/tmp/pip-install-q3hcpn_q/mysql-connector-python/mysql/connector/__init__.py_get_pooled_connectionHs*    r:cKs8|j}y |d}Wntk r0tdYnX|d=tddddddd d d g }d }x|D]}t|j|}|rtd jt|dkrdnddj|t|d r|d7}|j d d|d <|d d ks|d dkrt dj|d t |d t s\t dj|d q\Wd |ko"t|knr2t d|jddddi}g}xL|D]D}|d |kr|g||d <|j|d n||d j|qRWx|D]} || } xxtt| D]h} t| d} tjd | } | j| }|j}|j||jd dy tf|Stk rYnXqWqWt ddS)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. failoverzfailover argument not provideduserpasswordhostport unix_socketZdatabaser+r-priorityrz4Unsupported connection argument {0} in failover: {1}rsz, dz;Priority value should be in the range of 0 to 100, got : {}zFPriority 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)NrAr8)xr8r8r9sz*_get_failover_connection..T)keyreverseNz,Unable to connect to any of the target hosts)copyr. ValueErrorsetkeysr6lenjoinhasattrgetr r3intrsortappendrangerandomrandintpopupdateconnectr)r7configr;Zsupport_cnx_argsZpriority_countserverZdiffZserver_directoryZserver_priority_listrAZ failover_list_lastindex new_configr8r8r9_get_failover_connectionlsh               r`c s dkrjdnd}t|ts(td|rts:tddkrJtddkrZtdd krjtd d kr~td d <ytjjd d }Wn*tj j k rtd j d YnXg}x0|D](}|j |j jdd|j|j|jdqW|jddddd|Dd <dkr8dd<jddkrVtf}tf|Sd krjtfSy2ddlm}tfdd|DrtfSWntk rYnXjdd}dkrd=| rt rtttr| rt|St |S)aCreate or get a MySQL connection object In its simpliest form, Connect() will open a connection to a MySQL server and return a MySQLConnection object. 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. Returns MySQLConnection or PooledMySQLConnection. 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 documentationr@zZSRVz$Unable to locate any hosts for '{0}'T)Zomit_final_dot)r>r?rAweightcSs|d|d fS)NrArbr8)rEr8r8r9rFszconnect..)rGcSsg|]}|d|ddqS)r>r?)r>r?r8).0srvr8r8r9 szconnect..Zread_default_fileZ option_filesr) CNX_POOL_ARGScsg|] }|kqSr8r8)rcrG)r7r8r9resr,)!rWr3boolr HAVE_DNSPYTHONrdnsresolverquery exceptionZ DNSExceptionr6rStargetZto_textr?rArbrRr'rYr` constantsrfanyr: NameErrorrPr/r0r1rr) argsr7raZ srv_recordsr;rdr_rfr,r8)r7r9rYsf                  rYrConnectrrrrrrr/rrr r r r r rrrr$r%r&rrrrrrrrr r!r"r#r)@__doc__Z_mysql_connectorZconnection_cextrr0r/Z dns.resolverriZ dns.exceptionrhrUrCr connectionrrnrerrorsrrr r r r r rrrrrrrrrrZdbapirrrrrrrrr r!r"r#r$r%r&Z optionfilesr'r2r1r:r`rYrrVERSION__version_info__Z VERSION_TEXT __version____all__r8r8r8r9sR      8D $NV