U
|
¸ý°dc+ ã @ sà d Z ddlZddlmZ ddlmZ ddlmZ ddlm Z dd lm
|
Z
|
dd
|
l m Z dd l mZ dd l mZ G d d dZG dd deZG dd de ZG dd deZG dd dZG dd deZG dd deZeZdS )aZ
|
|
.. dialect:: sqlite+aiosqlite
|
:name: aiosqlite
|
:dbapi: aiosqlite
|
:connectstring: sqlite+aiosqlite:///file_path
|
:url: https://pypi.org/project/aiosqlite/
|
|
The aiosqlite dialect provides support for the SQLAlchemy asyncio interface
|
running on top of pysqlite.
|
|
aiosqlite is a wrapper around pysqlite that uses a background thread for
|
each connection. It does not actually use non-blocking IO, as SQLite
|
databases are not socket-based. However it does provide a working asyncio
|
interface that's useful for testing and prototyping purposes.
|
|
Using a special asyncio mediation layer, the aiosqlite 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("sqlite+aiosqlite:///filename")
|
|
The URL passes through all arguments to the ``pysqlite`` driver, so all
|
connection arguments are the same as they are for that of :ref:`pysqlite`.
|
|
.. _aiosqlite_udfs:
|
|
User-Defined Functions
|
----------------------
|
|
aiosqlite extends pysqlite to support async, so we can create our own user-defined functions (UDFs)
|
in Python and use them directly in SQLite queries as described here: :ref:`pysqlite_udfs`.
|
|
|
é N)Úpartialé )ÚSQLiteExecutionContext)ÚSQLiteDialect_pysqliteé )Úpool)Úutil)ÚAdaptedConnection)Úawait_fallback)Ú
|
await_onlyc @ s` e Zd ZdZdZdd Zdd Zddd Zd
|
d Zd d Z dd Z
|
dd Z dddZ dd Z dS )ÚAsyncAdapt_aiosqlite_cursor)Ú_adapt_connectionÚ _connectionÚ descriptionÚawait_Ú_rowsÚ arraysizeÚrowcountÚ lastrowidFc C s2 || _ |j| _|j| _d| _d| _d | _g | _d S )Nr éÿÿÿÿ)r r r r r r r )ÚselfZadapt_connection© r ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/dialects/sqlite/aiosqlite.pyÚ__init__L s z$AsyncAdapt_aiosqlite_cursor.__init__c C s g | j d d
< d S ©N©r ©r r r r ÚcloseU s z!AsyncAdapt_aiosqlite_cursor.closeNc
|
C sØ z¤| | j ¡ ¡}|d kr,| | |¡¡ n| | ||¡¡ |jrp|j| _d | _| _| js| | ¡ ¡| _ nd | _|j| _|j| _| js| |
|
¡ ¡ n|| _ W n. t k
|
rÒ } z| j |¡ W 5 d }~X Y nX d S )Nr )r r ÚcursorÚexecuter r r Ú server_sideÚfetchallr r Ú_cursorÚ Exceptionr Ú_handle_exception)r Ú operationÚ
|
parametersr" Úerrorr r r r X s$
|
z#AsyncAdapt_aiosqlite_cursor.executec
|
C s~ zJ| | j ¡ ¡}| | ||¡¡ d | _|j| _|j| _| | ¡ ¡ W n. tk
|
rx } z| j
|
|¡ W 5 d }~X Y nX d S r ) r r r Ú executemanyr r r r r# r r$ )r r% Zseq_of_parametersr" r' r r r r( s s z'AsyncAdapt_aiosqlite_cursor.executemanyc G s d S r r )r Z
|
inputsizesr r r Ú setinputsizes~ s z)AsyncAdapt_aiosqlite_cursor.setinputsizesc c s | j r| j d¡V q d S ©Nr ©r Úpopr r r r Ú__iter__ s z$AsyncAdapt_aiosqlite_cursor.__iter__c C s | j r| j d¡S d S d S r* r+ r r r r Úfetchone
s z$AsyncAdapt_aiosqlite_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_aiosqlite_cursor.fetchmanyc C s | j d d
}g | j d d
< |S r r )r r0 r r r r! s z$AsyncAdapt_aiosqlite_cursor.fetchall)N)N)Ú__name__Ú
|
__module__Ú __qualname__Ú __slots__r r r r r( r) r- r. r1 r! r r r r r > s
|
|
r c sF e Zd ZdZdZ fddZdd Zdd Zdd
|
d Zd d Z Z
|
S )Ú AsyncAdapt_aiosqlite_ss_cursorr" Tc s t j|| d | _d S r )Úsuperr r" )r ÚargÚkw©Ú __class__r r r s z'AsyncAdapt_aiosqlite_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_aiosqlite_ss_cursor.closec C s | | j ¡ ¡S r )r r" r. r r r r r. § s z'AsyncAdapt_aiosqlite_ss_cursor.fetchoneNc C s" |d kr| j }| | jj|d¡S )N)r/ )r r r" r1 )r r/ r r r r1 ª s z(AsyncAdapt_aiosqlite_ss_cursor.fetchmanyc C s | | j ¡ ¡S r )r r" r! r r r r r! ¯ s z'AsyncAdapt_aiosqlite_ss_cursor.fetchall)N) r2 r3 r4 r5 r r r r. r1 r! Ú __classcell__r r r: r r6 s
|
r6 c @ st e Zd ZeeZdZdd Zedd Z e j
|
dd Z dd Z dd
|
d Z d d Z dd Zdd Zdd Zdd ZdS )Ú AsyncAdapt_aiosqlite_connection)Údbapic C s || _ || _d S r )r> r )r r> Ú
|
connectionr r r r · s z(AsyncAdapt_aiosqlite_connection.__init__c C s | j jS r )r Úisolation_levelr r r r r@ » s z/AsyncAdapt_aiosqlite_connection.isolation_levelc
|
C st dd }t || jj|}t ¡ ¡ }| jj ||f¡ z | |¡W S t k
|
rn } z|
|
|¡ W 5 d }~X Y nX d S )Nc S s
|
|| _ d S r )r@ )r? Úvaluer r r Úset_isoÇ s z@AsyncAdapt_aiosqlite_connection.isolation_level.<locals>.set_iso) r r Z_connÚasyncioZget_event_loopZ create_futureZ_txÚ
|
put_nowaitr r# r$ )r rA rB ÚfunctionÚfuturer' r r r r@ ¿ s c
|
O sJ z| | jj||¡ W n, tk
|
rD } z| |¡ W 5 d }~X Y nX d S r )r r Úcreate_functionr# r$ )r Úargsr9 r' r r r rG Ô s z/AsyncAdapt_aiosqlite_connection.create_functionFc C s |r t | S t| S d S r )r6 r )r r r r r r Ú s z&AsyncAdapt_aiosqlite_connection.cursorc O s | | jj||¡S r )r r r )r rH r9 r r r r à s z'AsyncAdapt_aiosqlite_connection.executec
|
C sF z| | j ¡ ¡ W n, tk
|
r@ } z| |¡ W 5 d }~X Y nX d S r )r r Úrollbackr# r$ ©r r' r r r rI ã s z(AsyncAdapt_aiosqlite_connection.rollbackc
|
C sF z| | j ¡ ¡ W n, tk
|
r@ } z| |¡ W 5 d }~X Y nX d S r )r r Úcommitr# r$ rJ r r r rK é s z&AsyncAdapt_aiosqlite_connection.commitc
|
C sX z| | j ¡ ¡ W n> tk
|
r( Y n, tk
|
rR } z| |¡ W 5 d }~X Y nX d S r )r r r Ú
|
ValueErrorr# r$ rJ r r r r ï s z%AsyncAdapt_aiosqlite_connection.closec C s2 t |tr*|jd dkr*| jj d¡|n|d S )Nr úno active connection)Ú
|
isinstancerL rH r> ÚsqliteÚOperationalErrorrJ r r r r$ ÿ s ÿ þÿþz1AsyncAdapt_aiosqlite_connection._handle_exceptionN)F)r2 r3 r4 Ú staticmethodr r r5 r Úpropertyr@ ÚsetterrG r r rI rK r r$ r r r r r= ³ s
|
|
|
r= c @ s e Zd ZdZeeZdS )Ú'AsyncAdaptFallback_aiosqlite_connectionr N)r2 r3 r4 r5 rQ r
|
r r r r r rT s rT c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚAsyncAdapt_aiosqlite_dbapic C s || _ || _d| _| ¡ d S )NZqmark)Ú aiosqliterO Z
|
paramstyleÚ_init_dbapi_attributes)r rV rO r r r r s z#AsyncAdapt_aiosqlite_dbapi.__init__c C s^ dD ]}t | |t| j| qdD ]}t | |t| j| q"dD ]}t | |t| j| q@d S )N)Z DatabaseErrorÚErrorZIntegrityErrorZNotSupportedErrorrP ZProgrammingErrorZsqlite_versionZsqlite_version_info)ZPARSE_COLNAMESZPARSE_DECLTYPES)ÚBinary)ÚsetattrÚgetattrrV rO )r Únamer r r rW s
|
z1AsyncAdapt_aiosqlite_dbapi._init_dbapi_attributesc O sJ | dd¡}| jj||}d|_t |¡r8t| t|S t| t |S d S )NÚasync_fallbackFT)
|
r, rV ÚconnectÚdaemonr ZasboolrT r
|
r= r )r r8 r9 r] r? r r r r^ + s
|
þþz"AsyncAdapt_aiosqlite_dbapi.connectN)r2 r3 r4 r rW r^ r r r r rU s rU c @ s e Zd Zdd ZdS )Ú SQLiteExecutionContext_aiosqlitec C s | j jddS )NT)r )Z_dbapi_connectionr r r r r Úcreate_server_side_cursor@ s z:SQLiteExecutionContext_aiosqlite.create_server_side_cursorN)r2 r3 r4 ra r r r r r` ? s r` c sP e Zd ZdZdZdZdZeZe dd Z
|
e dd Z fddZ d d
|
Z ZS ) ÚSQLiteDialect_aiosqliterV Tc C s t tdtdS )NrV Zsqlite3)rU Ú
|
__import__)Úclsr r r Ú import_dbapiN s ÿz$SQLiteDialect_aiosqlite.import_dbapic C s | |¡rtjS tjS d S r )Z_is_url_file_dbr ZNullPoolZ
|
StaticPool)rd Úurlr r r Úget_pool_classT s
|
z&SQLiteDialect_aiosqlite.get_pool_classc s. t || jjr dt|kr dS t |||¡S )NrM T)rN r> rP Ústrr7 Ú is_disconnect)r Úer? r r: r r ri [ s ÿ
|
þz%SQLiteDialect_aiosqlite.is_disconnectc C s |j S r )r )r r? r r r Úget_driver_connectionc s z-SQLiteDialect_aiosqlite.get_driver_connection)r2 r3 r4 ZdriverZsupports_statement_cacheZis_asyncZ supports_server_side_cursorsr` Zexecution_ctx_clsÚ classmethodre rg ri rk r< r r r: r rb D s
|
|
rb )Ú__doc__rC Ú functoolsr Úbaser Zpysqliter Ú r r Zenginer Zutil.concurrencyr
|
r r r6 r= rT rU r` rb Údialectr r r r Ú<module>
|
s" ( [X.#
|
