U
|
¸ý°d& ã @ s¼ d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm
|
Z
|
dd lm Z G d
|
d d Z G d d d e Z G dd deZG dd deZG dd dZG dd deZeZdS )uV
|
.. dialect:: mysql+aiomysql
|
:name: aiomysql
|
:dbapi: aiomysql
|
:connectstring: mysql+aiomysql://user:password@host:port/dbname[?key=value&key=value...]
|
:url: https://github.com/aio-libs/aiomysql
|
|
.. warning:: The aiomysql dialect is not currently tested as part of
|
SQLAlchemyâs continuous integration. As of September, 2021 the driver
|
appears to be unmaintained and no longer functions for Python version 3.10,
|
and additionally depends on a significantly outdated version of PyMySQL.
|
Please refer to the :ref:`asyncmy` dialect for current MySQL/MariaDB asyncio
|
functionality.
|
|
The aiomysql dialect is SQLAlchemy's second Python asyncio dialect.
|
|
Using a special asyncio mediation layer, the aiomysql dialect is usable
|
as the backend for the :ref:`SQLAlchemy asyncio <asyncio_toplevel>`
|
extension package.
|
|
This dialect should normally be used only with the
|
:func:`_asyncio.create_async_engine` engine creation function::
|
|
from sqlalchemy.ext.asyncio import create_async_engine
|
engine = create_async_engine("mysql+aiomysql://user:pass@hostname/dbname?charset=utf8mb4")
|
|
|
é )ÚMySQLDialect_pymysqlé )Úpool)Úutil)ÚAdaptedConnection)Úasyncio)Úawait_fallback)Ú
|
await_onlyc @ s® e Zd ZdZdZdd Zedd Zedd Zed d
|
Z e j
|
d d
|
Z ed d Z dd Z d#ddZ dd Zdd Zdd Zdd Zdd Zd d Zd$d d Zd!d" ZdS )%ÚAsyncAdapt_aiomysql_cursorF)Ú_adapt_connectionÚ _connectionÚawait_Ú_cursorÚ_rowsc C s: || _ |j| _|j| _| j ¡ }| | ¡ ¡| _g | _d S ©N)r r r ÚcursorÚ
|
__aenter__r r ©ÚselfZadapt_connectionr © r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/dialects/mysql/aiomysql.pyÚ__init__9 s
|
z#AsyncAdapt_aiomysql_cursor.__init__c C s | j jS r )r Ú description©r r r r r D s z&AsyncAdapt_aiomysql_cursor.descriptionc C s | j jS r )r Úrowcountr r r r r H s z#AsyncAdapt_aiomysql_cursor.rowcountc C s | j jS r ©r Ú arraysizer r r r r L s z$AsyncAdapt_aiomysql_cursor.arraysizec C s || j _d S r r ©r Úvaluer r r r P s c C s | j jS r )r Ú lastrowidr r r r r T s z$AsyncAdapt_aiomysql_cursor.lastrowidc C s g | j d d
< d S r ©r r r r r ÚcloseX s z AsyncAdapt_aiomysql_cursor.closeNc C s | | ||¡¡S r )r Ú_execute_async)r Ú operationÚ
|
parametersr r r Úexecuteb s z"AsyncAdapt_aiomysql_cursor.executec C s | | ||¡¡S r )r Ú_executemany_async©r r# Zseq_of_parametersr r r Ú executemanye s
|
ÿz&AsyncAdapt_aiomysql_cursor.executemanyc à s | j j4 I d H d |d kr.| j |¡I d H }n| j ||¡I d H }| js^t| j ¡ I d H | _|W 5 Q I d H R £ S Q I d H R X d S r )r Ú_execute_mutexr r% Ú server_sideÚlistÚfetchallr )r r# r$ Úresultr r r r" j s z)AsyncAdapt_aiomysql_cursor._execute_asyncc à sH | j j4 I d H ( | j ||¡I d H W 5 Q I d H R £ S Q I d H R X d S r )r r) r r( r' r r r r& y s z-AsyncAdapt_aiomysql_cursor._executemany_asyncc G s d S r r )r Z
|
inputsizesr r r Ú setinputsizes} s z(AsyncAdapt_aiomysql_cursor.setinputsizesc c s | j r| j d¡V q d S ©Né ©r Úpopr r r r Ú__iter__ s z#AsyncAdapt_aiomysql_cursor.__iter__c C s | j r| j d¡S d S d S r/ r1 r r r r Úfetchone s z#AsyncAdapt_aiomysql_cursor.fetchonec C s8 |d kr| j }| jd|
}| j|d
| jd d
< |S r/ )r r )r ÚsizeÚretvalr r r Ú fetchmany s
|
z$AsyncAdapt_aiomysql_cursor.fetchmanyc C s | j d d
}g | j d d
< |S r r )r r6 r r r r, s z#AsyncAdapt_aiomysql_cursor.fetchall)N)N)Ú__name__Ú
|
__module__Ú __qualname__r* Ú __slots__r Úpropertyr r r Úsetterr r! r% r( r" r& r. r3 r4 r7 r, r r r r r
|
/ s.
|
|
|
|
|
|
|
|
r
|
c @ s> e Zd ZdZdZdd Zdd Zdd Zdd
|
d Zd d Z d S )Ú AsyncAdapt_aiomysql_ss_cursorr Tc C s< || _ |j| _|j| _| j |jjj¡}| | ¡ ¡| _d S r ) r r r r ÚdbapiÚaiomysqlZSSCursorr r r r r r r s ÿz&AsyncAdapt_aiomysql_ss_cursor.__init__c C s$ | j d k r | | j ¡ ¡ d | _ d S r )r r r! r r r r r! § s
|
z#AsyncAdapt_aiomysql_ss_cursor.closec C s | | j ¡ ¡S r )r r r4 r r r r r4 ¬ s z&AsyncAdapt_aiomysql_ss_cursor.fetchoneNc C s | | jj|d¡S )N)r5 )r r r7 )r r5 r r r r7 ¯ s z'AsyncAdapt_aiomysql_ss_cursor.fetchmanyc C s | | j ¡ ¡S r )r r r, r r r r r, ² s z&AsyncAdapt_aiomysql_ss_cursor.fetchall)N)
|
r8 r9 r: r; r* r r! r4 r7 r, r r r r r> s
|
r> c @ sZ e Zd ZeeZdZdd Zdd Zdd Z dd Z
|
dd d Z d d Z dd Z dd ZdS )Ú AsyncAdapt_aiomysql_connection)r? r) c C s || _ || _t ¡ | _d S r )r? r r ÚLockr) )r r? Ú
|
connectionr r r r º s z'AsyncAdapt_aiomysql_connection.__init__c C s | | j |¡¡S r )r r Úping)r Z reconnectr r r rD ¿ s z#AsyncAdapt_aiomysql_connection.pingc C s
|
| j ¡ S r )r Úcharacter_set_namer r r r rE Â s z1AsyncAdapt_aiomysql_connection.character_set_namec C s | | j |¡¡ d S r )r r Ú
|
autocommitr r r r rF Å s z)AsyncAdapt_aiomysql_connection.autocommitFc C s |r t | S t| S d S r )r> r
|
)r r* r r r r È s z%AsyncAdapt_aiomysql_connection.cursorc C s | | j ¡ ¡ d S r )r r Úrollbackr r r r rG Î s z'AsyncAdapt_aiomysql_connection.rollbackc C s | | j ¡ ¡ d S r )r r Úcommitr r r r rH Ñ s z%AsyncAdapt_aiomysql_connection.commitc C s | j ¡ d S r )r r! r r r r r! Ô s z$AsyncAdapt_aiomysql_connection.closeN)F)r8 r9 r: Ú staticmethodr r r; r rD rE rF r rG rH r! r r r r rA ¶ s
|
rA c @ s e Zd ZdZeeZdS )Ú&AsyncAdaptFallback_aiomysql_connectionr N)r8 r9 r: r; rI r r r r r r rJ Ù s rJ c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚAsyncAdapt_aiomysql_dbapic C s || _ || _d| _| ¡ d S )NÚformat)r@ ÚpymysqlZ
|
paramstyleÚ_init_dbapi_attributes)r r@ rM r r r r à s z"AsyncAdapt_aiomysql_dbapi.__init__c C s@ dD ]}t | |t| j| qdD ]}t | |t| j| q"d S )N) ÚWarningÚErrorÚInterfaceErrorZ DataErrorZ DatabaseErrorZOperationalErrorrQ ZIntegrityErrorZProgrammingErrorZ InternalErrorZNotSupportedError)ÚNUMBERÚSTRINGZDATETIMEÚBINARYÚ TIMESTAMPÚBinary)ÚsetattrÚgetattrr@ rM )r Únamer r r rN æ s z0AsyncAdapt_aiomysql_dbapi._init_dbapi_attributesc O sJ | dd¡}t |¡r.t| t| jj||S t| t| jj||S d S ©NÚasync_fallbackF) r2 r ÚasboolrJ r r@ ÚconnectrA r )r ÚargÚkwr[ r r r r] s
|
þþz!AsyncAdapt_aiomysql_dbapi.connectN)r8 r9 r: r rN r] r r r r rK ß s rK c sd e Zd ZdZdZdZeZdZe dd Z
|
e dd Z fddZ fd d
|
Z d d Zd d Z ZS )ÚMySQLDialect_aiomysqlr@ Tc C s t tdtdS )Nr@ rM )rK Ú
|
__import__)Úclsr r r Ú import_dbapi s ÿz"MySQLDialect_aiomysql.import_dbapic C s( |j dd¡}t |¡r tjS tjS d S rZ )ÚqueryÚgetr r\ r Z FallbackAsyncAdaptedQueuePoolZAsyncAdaptedQueuePool)rb Úurlr[ r r r Úget_pool_class s
|
z$MySQLDialect_aiomysql.get_pool_classc s t j|tddddS )NÚuserÚdb)ÚusernameZdatabase)Z_translate_args)ÚsuperÚcreate_connect_argsÚdict)r rf ©Ú __class__r r rl ( s
|
ÿz)MySQLDialect_aiomysql.create_connect_argsc s, t |||¡rdS t| ¡ }d|kS d S )NTz not connected)rk Ú is_disconnectÚstrÚlower)r ÚerC r Zstr_ern r r rp - s z#MySQLDialect_aiomysql.is_disconnectc C s ddl m} |jS )Nr0 )ÚCLIENT)Zpymysql.constantsrt Z
|
FOUND_ROWS)r rt r r r Ú_found_rows_client_flag4 s z-MySQLDialect_aiomysql._found_rows_client_flagc C s |j S r )r )r rC r r r Úget_driver_connection9 s z+MySQLDialect_aiomysql.get_driver_connection)r8 r9 r: ZdriverZsupports_statement_cacheZ supports_server_side_cursorsr> Z _sscursorZis_asyncÚ classmethodrc rg rl rp ru rv Ú __classcell__r r rn r r` s
|
|
r` N)Ú__doc__rM r Ú r r Zenginer Zutil.concurrencyr r r r
|
r> rA rJ rK r` Údialectr r r r Ú<module> s i #0.
|
