U
|
¸ý°d£' ã @ sÐ d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z dd l
|
m Z dd
|
l
|
m Z dd l
|
m Z G d d d ZG dd deZG dd de ZG dd deZdd ZG dd dZG dd deZeZdS )a\
|
.. dialect:: mysql+asyncmy
|
:name: asyncmy
|
:dbapi: asyncmy
|
:connectstring: mysql+asyncmy://user:password@host:port/dbname[?key=value&key=value...]
|
:url: https://github.com/long2ice/asyncmy
|
|
.. note:: The asyncmy dialect as of September, 2021 was added to provide
|
MySQL/MariaDB asyncio compatibility given that the :ref:`aiomysql` database
|
driver has become unmaintained, however asyncmy is itself very new.
|
|
Using a special asyncio mediation layer, the asyncmy 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+asyncmy://user:pass@hostname/dbname?charset=utf8mb4")
|
|
|
é )Úasynccontextmanageré )Ú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_asyncmy_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 úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/dialects/mysql/asyncmy.pyÚ__init__6 s
|
z"AsyncAdapt_asyncmy_cursor.__init__c C s | j jS r )r Ú description©r r r r r @ s z%AsyncAdapt_asyncmy_cursor.descriptionc C s | j jS r )r Úrowcountr r r r r D s z"AsyncAdapt_asyncmy_cursor.rowcountc C s | j jS r ©r Ú arraysizer r r r r H s z#AsyncAdapt_asyncmy_cursor.arraysizec C s || j _d S r r ©r Úvaluer r r r L s c C s | j jS r )r Ú lastrowidr r r r r! P s z#AsyncAdapt_asyncmy_cursor.lastrowidc C s g | j d d
< d S r ©r r r r r ÚcloseT s z AsyncAdapt_asyncmy_cursor.closeNc C s | | ||¡¡S r )r Ú_execute_async)r Ú operationÚ
|
parametersr r r Úexecute^ s z!AsyncAdapt_asyncmy_cursor.executec C s | | ||¡¡S r )r Ú_executemany_async©r r% Zseq_of_parametersr r r Ú executemanya s
|
ÿz%AsyncAdapt_asyncmy_cursor.executemanyc à s | j ¡ 4 I d H d |d kr0| 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 Ú_mutex_and_adapt_errorsr r' Ú server_sideÚlistÚfetchallr )r r% r& Úresultr r r r$ f s z(AsyncAdapt_asyncmy_cursor._execute_asyncc à sJ | j ¡ 4 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( u s z,AsyncAdapt_asyncmy_cursor._executemany_asyncc G s d S r r )r Z
|
inputsizesr r r Ú setinputsizesy s z'AsyncAdapt_asyncmy_cursor.setinputsizesc c s | j r| j d¡V q d S ©Nr ©r Úpopr r r r Ú__iter__| s z"AsyncAdapt_asyncmy_cursor.__iter__c C s | j r| j d¡S d S d S r1 r2 r r r r Úfetchone s z"AsyncAdapt_asyncmy_cursor.fetchonec C s8 |d kr| j }| jd|
}| j|d
| jd d
< |S r1 )r r )r ÚsizeÚretvalr r r Ú fetchmany s
|
z#AsyncAdapt_asyncmy_cursor.fetchmanyc C s | j d d
}g | j d d
< |S r r" )r r7 r r r r. s z"AsyncAdapt_asyncmy_cursor.fetchall)N)N)Ú__name__Ú
|
__module__Ú __qualname__r, Ú __slots__r Úpropertyr r r Úsetterr! r# r' r* r$ r( r0 r4 r5 r8 r. r r r r r , s.
|
|
|
|
|
|
|
|
|
r c @ s> e Zd ZdZdZdd Zdd Zdd Zdd
|
d Zd d Z d S )Ú AsyncAdapt_asyncmy_ss_cursorr Tc C s> || _ |j| _|j| _| j |jjjj¡}| | ¡ ¡| _ d S r )
|
r r r r ÚdbapiÚasyncmyZcursorsZSSCursorr r r r r r r s
|
ÿz%AsyncAdapt_asyncmy_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_asyncmy_ss_cursor.closec C s | | j ¡ ¡S r )r r r5 r r r r r5 ¨ s z%AsyncAdapt_asyncmy_ss_cursor.fetchoneNc C s | | jj|d¡S )N)r6 )r r r8 )r r6 r r r r8 « s z&AsyncAdapt_asyncmy_ss_cursor.fetchmanyc C s | | j ¡ ¡S r )r r r. r r r r r. ® s z%AsyncAdapt_asyncmy_ss_cursor.fetchall)N)
|
r9 r: r; r< r, r r# r5 r8 r. r r r r r? s
|
r? c @ sn e Zd ZeeZdZdd Zedd Z dd Z
|
dd Z d
|
d Z d d Z dddZdd Zdd Zdd ZdS )Ú AsyncAdapt_asyncmy_connection)r@ Ú_execute_mutexc C s || _ || _t ¡ | _d S r )r@ r r ÚLockrC )r r@ Ú
|
connectionr r r r ¶ s z&AsyncAdapt_asyncmy_connection.__init__c
|
C sP | j 4 I d H 2 z
|
d V W n tk
|
r: | j d¡Y nX W 5 Q I d H R X d S )Nz7network operation failed due to asyncmy attribute error)rC ÚAttributeErrorr@ Ú InternalErrorr r r r r+ » s
|
ÿz5AsyncAdapt_asyncmy_connection._mutex_and_adapt_errorsc C s |rt | | ¡ ¡S r )ÚAssertionErrorr Ú_do_ping)r Z reconnectr r r ÚpingÅ s z"AsyncAdapt_asyncmy_connection.pingc à sF | ¡ 4 I d H & | j d¡I d H W 5 Q I d H R £ S Q I d H R X d S )NF)r+ r rJ r r r r rI É s z&AsyncAdapt_asyncmy_connection._do_pingc C s
|
| j ¡ S r )r Úcharacter_set_namer r r r rK Í s z0AsyncAdapt_asyncmy_connection.character_set_namec C s | | j |¡¡ d S r )r r Ú
|
autocommitr r r r rL Ð s z(AsyncAdapt_asyncmy_connection.autocommitFc C s |r t | S t| S d S r )r? r )r r, r r r r Ó s z$AsyncAdapt_asyncmy_connection.cursorc C s | | j ¡ ¡ d S r )r r Úrollbackr r r r rM Ù s z&AsyncAdapt_asyncmy_connection.rollbackc C s | | j ¡ ¡ d S r )r r Úcommitr r r r rN Ü s z$AsyncAdapt_asyncmy_connection.commitc C s | j ¡ d S r )r r# r r r r r# ß s z#AsyncAdapt_asyncmy_connection.closeN)F)r9 r: r; Ú staticmethodr r r< r r r+ rJ rI rK rL r rM rN r# r r r r rB ² s
|
|
rB c @ s e Zd ZdZeeZdS )Ú%AsyncAdaptFallback_asyncmy_connectionr N)r9 r: r; r< rO r
|
r r r r r rP ä s rP c C s t | S )zReturn x as a binary type.)Úbytes)Úxr r r Ú_Binaryê s rS c @ s^ e Zd Zdd Zdd Ze d¡Ze d¡Ze d¡Z e d¡Z
|
e d ¡Z e e Zd
|
d Zd S ) ÚAsyncAdapt_asyncmy_dbapic C s || _ d| _| ¡ d S )NÚformat)rA Z
|
paramstyleÚ_init_dbapi_attributes)r rA r r r r ð s z!AsyncAdapt_asyncmy_dbapi.__init__c C s$ dD ]}t | |t| jj| qd S )N) ÚWarningÚErrorÚInterfaceErrorZ DataErrorZ DatabaseErrorZOperationalErrorrY ZIntegrityErrorZProgrammingErrorrG ZNotSupportedError)ÚsetattrÚgetattrrA Úerrors)r Únamer r r rV õ s z/AsyncAdapt_asyncmy_dbapi._init_dbapi_attributesÚSTRINGÚNUMBERÚBINARYÚDATETIMEÚ TIMESTAMPc O sJ | dd¡}t |¡r.t| t| jj||S t| t| jj||S d S ©NÚasync_fallbackF) r3 r ÚasboolrP r
|
rA ÚconnectrB r )r ÚargÚkwrd r r r rf s
|
þþz AsyncAdapt_asyncmy_dbapi.connectN)r9 r: r; r rV r Úsymbolr^ r_ r` ra rb rO rS ÚBinaryrf r r r r rT ï s
|
|
|
|
|
rT c sd e Zd ZdZdZdZeZdZe dd Z
|
e dd Z fddZ fd d
|
Z d d Zd d Z ZS )ÚMySQLDialect_asyncmyrA Tc C s t tdS )NrA )rT Ú
|
__import__)Úclsr r r Ú import_dbapi$ s z!MySQLDialect_asyncmy.import_dbapic C s( |j dd¡}t |¡r tjS tjS d S rc )ÚqueryÚgetr re r Z FallbackAsyncAdaptedQueuePoolZAsyncAdaptedQueuePool)rm Úurlrd r r r Úget_pool_class( s
|
z#MySQLDialect_asyncmy.get_pool_classc s t j|tddddS )NÚuserÚdb)ÚusernameZdatabase)Z_translate_args)ÚsuperÚcreate_connect_argsÚdict)r rq ©Ú __class__r r rw 2 s
|
ÿz(MySQLDialect_asyncmy.create_connect_argsc s4 t |||¡rdS t| ¡ }d|kp.d|kS d S )NTz not connectedznetwork operation failed)rv Ú is_disconnectÚstrÚlower)r ÚerE r Zstr_ery r r r{ 7 s
|
ÿz"MySQLDialect_asyncmy.is_disconnectc C s ddl m} |jS )Nr )ÚCLIENT)Zasyncmy.constantsr Z
|
FOUND_ROWS)r r r r r Ú_found_rows_client_flag@ s z,MySQLDialect_asyncmy._found_rows_client_flagc C s |j S r )r )r rE r r r Úget_driver_connectionE s z*MySQLDialect_asyncmy.get_driver_connection)r9 r: r; ZdriverZsupports_statement_cacheZ supports_server_side_cursorsr? Z _sscursorZis_asyncÚ classmethodrn rr rw r{ r r Ú __classcell__r r ry r rk s
|
|
rk N)Ú__doc__Ú
|
contextlibr Zpymysqlr Ú r r Zenginer Zutil.concurrencyr r
|
r r r? rB rP rS rT rk Údialectr r r r Ú<module> s h 2,.
|
