U eT@sUdZddlmZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z mZmZmZddlmZzddlZddlZWnek rdZYnXdZzd d lmZWnek rdZYnXd d lmZd d lmZmZd d lm Z m!Z!m"Z"m#Z#m$Z$d dl%m&Z&e'Z(dZ)dZ*e+dZ,dZ-edkr:eneefZ.de/d<iZ0de/d<dddddZ1dddddZ2ddddd d!Z3dd"dd#d$Z4Gd%ddZ5Gd&d'd'Z6dS)(z5Implementing pooling of connections to MySQL servers.) annotationsN) TracebackType)AnyDictNoReturnOptionalTupleTypeUnion)uuid4FT)CMySQLConnection)MySQLConnection) CNX_POOL_ARGSDEFAULT_CONFIGURATION)ErrorInterfaceErrorNotSupportedError PoolErrorProgrammingError)read_option_files @z[^a-zA-Z0-9._:\-*$#]z0MySQL Connector/Python C Extension not availablezUnion[type, Tuple[type, ...]]MYSQL_CNX_CLASSzDict[str, MySQLConnectionPool]_CONNECTION_POOLSrPooledMySQLConnectionkwargsreturnc Ksd|kr|dntf|}|ddkr8tdkr8tttR|tkrVtf|t|<n4tt|trt|j }d|kr|d|krt dW5QRXzt| WSt k rt d|ddYnXdS) z!Return a pooled MySQL connection. pool_nameuse_pureFN pool_sizez)Size can not be changed for active pools.z%Failed getting connection from pool '')generate_pool_namegetr ImportError ERROR_NO_CEXTCONNECTION_POOL_LOCKrMySQLConnectionPool isinstancer!rget_connectionAttributeErrorr)rrZ check_sizer,X/var/www/html/assets/scripts/venv/lib/python3.8/site-packages/mysql/connector/pooling.py_get_pooled_connectionMs&  r.z?Union[PooledMySQLConnection, MySQLConnection, CMySQLConnection]c Ks<|}z |d}Wntk r2tddYnX|d=tdddddd d d d g }d }|D]}t||}|rt|dkrdnd}d|}td|d|t|d r|d7}|d d|d <|d d ks|d dkrt d|d t |d t s\t d|d q\d |kr0t|krz*_get_failover_connection..T)keyreversez,Unable to connect to any of the target hosts)copyKeyError ValueErrorsetkeyslenjoinhasattrr$rr)intrsortedappendrangerandomrandintpopupdateconnectr)rconfigr/Zsupport_cnx_argsZpriority_countserverdiffarglstZserver_directoryZserver_priority_listr6Z failover_list_lastindex new_configr,r,r-_get_failover_connectionjs~            rZ)argsrrcsdkrdnd}t|ts(td|rts:tddkrJtddkrZtdd krjtd d kr~td d <ztjd d }Wn.tj j k rtd d ddYnXg}|D](}| |j j dd|j|j|jdq|jddddd|Dd <dkr8dd<ddkrVtf}tf|Sd krjtfSz(tfddtDrtfWSWntk rYnXdd}dkrވd=|stdkrtttr|st|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 documentationr4z.)r>cSsg|]}|d|ddqS)r2r3)r2r3r,).0srvr,r,r- szconnect..Zread_default_fileZ option_filesc3s|]}|kVqdSNr,)r^r>rr,r- szconnect..r )rNr)boolrHAVE_DNSPYTHONrdnsresolverquery exceptionZ DNSExceptionrJtargetZto_textr3r6r]sortrrPrZanyrr. NameErrorr$r r%r&r)r[rr\Z srv_recordsr/r_rYr r,rbr-rPs                 rPstrc KsPg}dD]0}z|t||Wqtk r6YqXq|sFtdd|S)zGenerate a pool name This function takes keyword arguments, usually the connection arguments for MySQLConnection, and tries to generate a name for a pool. Raises PoolError when no name can be generated. Returns a string. )r2r3r0r5z.Failed generating pool name; specify pool_namerV)rJrnrArrF)rpartsr>r,r,r-r#)s r#c@seZdZdZddddddZddd d Zd d d ddddZdddddZddddZe dddddZ e ddddZ dS)raClass holding a MySQL Connection in a pool PooledMySQLConnection is used by MySQLConnectionPool to return an instance holding a MySQL connection. It works like a MySQLConnection except for methods like close() and config(). The close()-method will add the connection back to the pool rather than disconnecting from the MySQL server. Configuring the connection have to be done through the MySQLConnectionPool method set_config(). Using config() on pooled connection will raise a PoolError. r((Union[MySQLConnection, CMySQLConnection]None)poolcnxrcCs4t|tstdt|ts$td||_||_dS)zInitialize The pool argument must be an instance of MySQLConnectionPoll. cnx if an instance of MySQLConnection. z$pool should be a MySQLConnectionPoolzcnx should be a MySQLConnectionN)r)r(r+r _cnx_pool_cnx)selfrrrsr,r,r-__init__Ps   zPooledMySQLConnection.__init__rcCs|Srar,rvr,r,r- __enter___szPooledMySQLConnection.__enter__zType[BaseException] BaseExceptionr)exc_type exc_value tracebackrcCs |dSra)close)rvr|r}r~r,r,r-__exit__bszPooledMySQLConnection.__exit__r)attrrcCs t|j|S)z0Calls attributes of the MySQLConnection instance)getattrru)rvrr,r,r- __getattr__jsz!PooledMySQLConnection.__getattr__c Cs4z|j}|jjr|W5|j|d|_XdS)aYDo not close, but add connection back to pool The close() 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. N)rtadd_connectionru reset_sessionrvrsr,r,r-rns   zPooledMySQLConnection.closerrcKs tddS)z&Configuration is done through the poolzKConfiguration for pooled connections should be done through the pool itselfN)rrbr,r,r-rQszPooledMySQLConnection.configrncCs|jjSz&Return the name of the connection pool)rtrryr,r,r-rszPooledMySQLConnection.pool_nameN) __name__ __module__ __qualname____doc__rwrzrrr staticmethodrQpropertyrr,r,r,r-rAsc@seZdZdZd*ddddd d d d Zed dddZeddddZeddddZdd dddZ dd dddZ d d dddZ dd dd d!Z d+d"d dd#d$Z d%dd&d'Zddd(d)ZdS),r(z*Class defining a pool of MySQL connectionsNTrHz Optional[str]rdrrq)r!rpool_reset_sessionrrcKsd|_d|_||_||||p,tf|i|_t|j|_ t |_ |r||j f|d}||jkr|| |d7}q`dS)zInitialize Initialize a MySQL connection pool with a maximum number of connections set to pool_size. The rest of the keywords arguments, kwargs, are configuration arguments for MySQLConnection instances. Nrr ) _pool_size _pool_name_reset_session_set_pool_size_set_pool_namer# _cnx_configqueueQueue _cnx_queuer _config_version set_configr)rvr!rrrcntr,r,r-rws    zMySQLConnectionPool.__init__rnrxcCs|jSr)rryr,r,r-rszMySQLConnectionPool.pool_namecCs|jS)z0Return number of connections managed by the pool)rryr,r,r-r!szMySQLConnectionPool.pool_sizecCs|jS)zReturn whether to reset session)rryr,r,r-rsz!MySQLConnectionPool.reset_sessionrc Kst|sdSt^z$t}|jf|||_t|_Wn2tk rd}ztd||W5d}~XYnXW5QRXdS)aTSet the connection configuration for MySQLConnection instances This method sets the configuration used for creating MySQLConnection instances. See MySQLConnection for valid connection arguments. Raises PoolError when a connection argument is not valid, missing or not supported by MySQLConnection. Nz$Connection configuration not valid: )r'rPrQrr rr+r)rvrZtest_cnxerrr,r,r-rs   zMySQLConnectionPool.set_config)r!rcCs(|dks|tkrtdt||_dS)aSet the size of the pool This method sets the size of the pool but it will not resize the pool. Raises an AttributeError when the pool_size is not valid. Invalid size is 0, negative or higher than pooling.CNX_POOL_MAXSIZE. rz8Pool size should be higher than 0 and lower or equal to N)CNX_POOL_MAXSIZEr+r)rvr!r,r,r-rs z"MySQLConnectionPool._set_pool_size)rrcCs@t|rtd|dt|tkr6td|d||_dS)aSet the name of the pool. This method checks the validity and sets the name of the pool. Raises an AttributeError when pool_name contains illegal characters ([^a-zA-Z0-9._\-*$#]) or is longer than pooling.CNX_POOL_MAXNAMESIZE. z Pool name 'z' contains illegal charactersz ' is too longN)CNX_POOL_NAMEREGEXsearchr+rECNX_POOL_MAXNAMESIZEr)rvrr,r,r-rs   z"MySQLConnectionPool._set_pool_namerp)rsrc CsZt|tstdz|jj|ddWn.tjk rT}ztd|W5d}~XYnXdS)zPut connection back in the queue This method is putting a connection back in the queue. It will not acquire a lock as the methods using _queue_connection() will have it set. Raises PoolError on errors. 3Connection instance not subclass of MySQLConnectionFblock'Failed adding connection; queue is fullN)r)rrrputrFullrvrsrr,r,r-_queue_connections z%MySQLConnectionPool._queue_connectionz2Optional[Union[MySQLConnection, CMySQLConnection]]c Cst|jstd|jr&td|s~tf|j}z(|jr\|jdr\|dkr\tdWnt k rrYnX|j |_ nt |t std||W5QRXdS)aAdd a connection to the pool This method instantiates a MySQLConnection using the configuration passed when initializing the MySQLConnectionPool instance or using the set_config() method. If cnx is a MySQLConnection instance, it will be added to the queue. Raises PoolError when no configuration is set, when no more connection can be added (maximum reached) or when the connection can not be instantiated. z&Connection configuration not availablercompress)rz^Pool reset session is not supported with compression for MySQL server version 5.7.2 or earlierrN)r'rrrfullrPrZget_server_versionrrArpool_config_versionr)rrrr,r,r-rs2     z"MySQLConnectionPool.add_connectionrc Cstz|jjdd}Wn.tjk rF}ztd|W5d}~XYnX|r\|j|jkr|j f|j z | Wn t k r| |YnX|j|_t||W5QRSQRXdS)awGet a connection from the pool This method returns an PooledMySQLConnection instance which has a reference to the pool that created it, and the next available MySQL connection. When the MySQL connection is not connect, a reconnect is attempted. Raises PoolError on errors. Returns a PooledMySQLConnection instance. Frz)Failed getting connection; pool exhaustedN)r'rr$rEmptyrZ is_connectedrrrQrZ reconnectrrrrr,r,r-r*2s"    z"MySQLConnectionPool.get_connectionc Cstd}|j}|rz |jdd}||d7}Wqtjk r^|YW5QRStk rrYqtk rYqXq|W5QRSQRXdS)zClose all connections This method closes all connections. It returns the number of connections it closed. Used mostly for tests. Returns int. rFrr N) r'rqsizer$Z disconnectrrrr)rvrZcnxqrsr,r,r-_remove_connectionsTs   z'MySQLConnectionPool._remove_connections)rNT)N)rrrrrwrrr!rrrrrrr*rr,r,r,r-r(s& /"r()7r __future__rrrLre threadingtypesrtypingrrrrrr r uuidr Z dns.exceptionrfZ dns.resolverr%reZconnection_cextr connectionr constantsrrerrorsrrrrrZ optionfilesrRLockr'rrcompilerr&r__annotations__rr.rZrPr#rr(r,r,r,r-sH  $         ]bM