o Rh@sddlZddlZddlmZddlmZmZddlmZm Z ddl m Z m Z m Z mZmZzddlmZdd lmZWn eyEed wGd d d eZed dZee_d e_dddeddfddZdddeddfddZGdddZeZe ZGdddZdS)N) SAConnection)InvalidRequestError ArgumentError)_PoolContextManager_PoolAcquireContextManager)CursorDeserializationCursor DictCursorSSCursor SSDictCursor)MySQLDialect_pymysql)MySQLCompiler_mysqldbzaiomysql.sa requires sqlalchemycs&eZdZdfdd ZddZZS)MySQLCompiler_pymysqlNTcs2t|||}|jD] }||j||j<q |SN)superconstruct_paramsprefetch _exec_defaultdefaultkey)selfparams _group_number_checkpdcolumn __class__T/home/air/sanwanet/backup_V2/venv/lib/python3.10/site-packages/aiomysql/sa/engine.pyrs z&MySQLCompiler_pymysql.construct_paramscCs|jr ||jS|jSr) is_callableargdialect)rrr r r!rs z#MySQLCompiler_pymysql._exec_default)NNT)__name__ __module__ __qualname__rr __classcell__r r rr!rsrpyformat) paramstyle c sdttttg}|dtttrtfdd|Dr tdt d||||||d|}t |S)zA coroutine for Engine creation. Returns Engine instance with embedded connection pool. The pool has *minsize* opened connections to MySQL server. cursorclassc3s|]}t|VqdSr) issubclass).0 cursor_classr-r r! 7s  z create_engine..z4SQLAlchemy engine does not support this cursor class)minsizemaxsizeloopr$ pool_recyclecompiled_cacheNr ) r r r r getr r.anyr_create_engine_EngineContextManager) r3r4r5r$r6r7kwargsdeprecated_cursor_classescoror r1r! create_engine)s   r?c sp|dur t}tjd||||d|IdH}|IdH}zt||fd|i|W||S||w)N)r3r4r5r6r7r )asyncioget_event_loopaiomysql create_poolacquireEnginerelease) r3r4r5r$r6r7r<poolconnr r r!r:Ds  r:c@seZdZdZd)ddZeddZeddZed d Zed d Z ed dZ eddZ eddZ ddZ ddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(ZdS)*rEzConnects a aiomysql.Pool and sqlalchemy.engine.interfaces.Dialect together to provide a source of database connectivity and behavior. An Engine object is instantiated publicly using the create_engine coroutine. NcKs||_||_||_||_dSr)_dialect_pool_compiled_cache_conn_kw)rr$rGr7r<r r r!__init__]s zEngine.__init__cCs|jS)zAn dialect for engine.)rIrr r r!r$cszEngine.dialectcC|jjS)zA name of the dialect.)rInamerNr r r!rPhz Engine.namecCrO)zA driver of the dialect.)rIdriverrNr r r!rRmrQz Engine.drivercCrOr)rJr3rNr r r!r3rzEngine.minsizecCrOr)rJr4rNr r r!r4vrSzEngine.maxsizecCrOr)rJsizerNr r r!rTzrSz Engine.sizecCrOr)rJfreesizerNr r r!rU~rSzEngine.freesizecC|jdS)zClose engine. Mark all engine connections to be closed on getting back to pool. Closed engine doesn't allow to acquire new connections. N)rJcloserNr r r!rWz Engine.closecCrV)zwTerminate engine. Terminate engine pool with instantly closing all acquired connections also. N)rJ terminaterNr r r!rYrXzEngine.terminatecs|jIdHdS)z*Wait for closing all engine's connections.N)rJ wait_closedrNr r r!rZszEngine.wait_closedcCs|}t||S)zGet a connection from pool.)_acquire_EngineAcquireContextManager)rr>r r r!rDs zEngine.acquirecs&|jIdH}t|||jd}|S)N)r7)rJrDrrK)rrawrHr r r!r[szEngine._acquirecCs |jrtd|j}|j|S)zRevert back connection to pool.z9Cannot release a connection with not finished transaction)in_transactionr connectionrJrF)rrHr]r r r!rFs zEngine.releasecCstd)Nz9"yield from" should be used as context manager expression) RuntimeErrorrNr r r! __enter__szEngine.__enter__cGsdSrr rargsr r r!__exit__szEngine.__exit__ccs|EdH}t||Sr)rD_ConnectionContextManager)rrHr r r!__iter__s zEngine.__iter__cs|Srr rNr r r! __aenter__szEngine.__aenter__cs||IdHdSr)rWrZ)rexc_typeexc_valexc_tbr r r! __aexit__szEngine.__aexit__r)r%r&r'__doc__rMpropertyr$rPrRr3r4rTrUrWrYrZrDr[rFrardrfrgrkr r r r!rETs8         rEc@s,eZdZdZdZddZddZddZd S) rea+Context manager. This enables the following idiom for acquiring and releasing a connection around a block: with (yield from engine) as conn: cur = yield from conn.cursor() while failing loudly when accidentally using: with engine: _engine_conncCs||_||_dSrrn)renginerHr r r!rMs z"_ConnectionContextManager.__init__cCs|jdusJ|jSr)rprNr r r!rasz#_ConnectionContextManager.__enter__cGs0z|j|jWd|_d|_dSd|_d|_wr)rorFrprbr r r!rds  z"_ConnectionContextManager.__exit__N)r%r&r'rl __slots__rMrardr r r r!res  re)r@rBr_rexcrrutilsrrcursorsr r r r r !sqlalchemy.dialects.mysql.pymysqlr!sqlalchemy.dialects.mysql.mysqldbr ImportErrorrrIstatement_compilerdefault_paramstyler?r:rEr;r\rer r r r!s6      u