관리-도구
편집 파일: aiosqlite.cpython-38.pyc
U -?�f90 � @ s� d Z ddlZddlmZ ddlmZ ddlmZ ddlm Z dd l m Z dd l m Z dd l mZ dd lmZ dd lmZ G dd� d�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� d�ZG dd� de�ZG dd� de �ZeZdS ) 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`. .. _aiosqlite_serializable: Serializable isolation / Savepoints / Transactional DDL (asyncio version) ------------------------------------------------------------------------- Similarly to pysqlite, aiosqlite does not support SAVEPOINT feature. The solution is similar to :ref:`pysqlite_serializable`. This is achieved by the event listeners in async:: from sqlalchemy import create_engine, event from sqlalchemy.ext.asyncio import create_async_engine engine = create_async_engine("sqlite+aiosqlite:///myfile.db") @event.listens_for(engine.sync_engine, "connect") def do_connect(dbapi_connection, connection_record): # disable aiosqlite's emitting of the BEGIN statement entirely. # also stops it from emitting COMMIT before any DDL. dbapi_connection.isolation_level = None @event.listens_for(engine.sync_engine, "begin") def do_begin(conn): # emit our own BEGIN conn.exec_driver_sql("BEGIN") .. warning:: When using the above recipe, it is advised to not use the :paramref:`.Connection.execution_options.isolation_level` setting on :class:`_engine.Connection` and :func:`_sa.create_engine` with the SQLite driver, as this function necessarily will also alter the ".isolation_level" setting. � N)�deque)�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 ddd�Z dd� Z dS )�AsyncAdapt_aiosqlite_cursor)�_adapt_connection� _connection� description�await_�_rows� arraysize�rowcount� lastrowidFc C s4 || _ |j| _|j| _d| _d| _d | _t� | _d S )Nr ���)r r r r r r r r )�selfZadapt_connection� r �T/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/dialects/sqlite/aiosqlite.py�__init__n s z$AsyncAdapt_aiosqlite_cursor.__init__c C s | j �� d S �N)r �clear�r r r r �closew s z!AsyncAdapt_aiosqlite_cursor.closeNc C s� z�| � | j�� �}|d kr,| � |�|�� n| � |�||�� |jrt|j| _d | _| _| js�t| � |� � ��| _ 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_sider �fetchallr r �_cursor� Exceptionr �_handle_exception)r � operation� parametersr# �errorr r r r z 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 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 �� V q d S r �r �popleftr r r r �__iter__� s z$AsyncAdapt_aiosqlite_cursor.__iter__c C s | j r| j �� S d S d S r r+ r r r r �fetchone� s z$AsyncAdapt_aiosqlite_cursor.fetchonec s4 |d kr| j }| j� � fdd�tt|t� ���D �S )Nc s g | ] }� � � �qS r )r, )�.0�_�Zrrr r � <listcomp>� s z9AsyncAdapt_aiosqlite_cursor.fetchmany.<locals>.<listcomp>)r r �range�min�len�r �sizer r1 r � fetchmany� s z%AsyncAdapt_aiosqlite_cursor.fetchmanyc C s t | j�}| j�� |S r )�listr r )r �retvalr r r r" � s z$AsyncAdapt_aiosqlite_cursor.fetchall)N)N)�__name__� __module__� __qualname__� __slots__r! r r r r) r* r- r. r8 r" r r r r r ] s r c sF e Zd ZdZdZ� fdd�Zdd� 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)r7 )r r r# r8 r6 r r r r8 � 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) r; r< r= r> r! r r r. r8 r" � __classcell__r r rC r r? � s r? c @ st e Zd Zee�ZdZdd� 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 )rG r )r rG � connectionr r r r � s z(AsyncAdapt_aiosqlite_connection.__init__c C s | j jS r )r �isolation_levelr r r r rI � 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 )rI )rH �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 rJ rK �function�futurer( r r r rI � 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 �argsrB r( r r r rP � s z/AsyncAdapt_aiosqlite_connection.create_functionFc C s |r t | �S t| �S d S r )r? r )r r! r r r r � s z&AsyncAdapt_aiosqlite_connection.cursorc O s | � | jj||��S r )r r r )r rQ rB 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 rR 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% rS r r r rT 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% rS r r r r s z%AsyncAdapt_aiosqlite_connection.closec C s2 t |t�r*|jd dkr*| jj�d�|�n|�d S )Nr �no active connection)� isinstancerU rQ rG �sqlite�OperationalErrorrS r r r r% ! s � ���z1AsyncAdapt_aiosqlite_connection._handle_exceptionN)F)r; r< r= � staticmethodr r r> r �propertyrI �setterrP r r rR rT r r% r r r r rF � s rF c @ s e Zd ZdZee�ZdS )�'AsyncAdaptFallback_aiosqlite_connectionr N)r; r< r= r> rZ r r r r r r r] - s r] c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�AsyncAdapt_aiosqlite_dbapic C s || _ || _d| _| �� d S )NZqmark)� aiosqliterX Z paramstyle�_init_dbapi_attributes)r r_ rX r r r r 4 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�ErrorZIntegrityErrorZNotSupportedErrorrY ZProgrammingErrorZsqlite_versionZsqlite_version_info)ZPARSE_COLNAMESZPARSE_DECLTYPES)�Binary)�setattr�getattrr_ rX )r �namer r r r` : s z1AsyncAdapt_aiosqlite_dbapi._init_dbapi_attributesc O sf |� dd�}|� dd �}|r(|||�}n| jj||�}d|_t�|�rTt| t|��S t| t |��S d S )N�async_fallbackFZasync_creator_fnT) �popr_ �connect�daemonr Zasboolr] r rF r )r rA rB rf Z creator_fnrH r r r rh M s ��z"AsyncAdapt_aiosqlite_dbapi.connectN)r; r< r= r r` rh r r r r r^ 3 s r^ c @ s e Zd Zdd� ZdS )� SQLiteExecutionContext_aiosqlitec C s | j jdd�S )NT)r! )Z_dbapi_connectionr r r r r �create_server_side_cursore s z:SQLiteExecutionContext_aiosqlite.create_server_side_cursorN)r; r< r= rk r r r r rj d s rj c sP e Zd ZdZdZdZdZeZe dd� �Z e dd� �Z � fdd�Z d d � Z � ZS ) �SQLiteDialect_aiosqliter_ Tc C s t td�td��S )Nr_ �sqlite3)r^ � __import__)�clsr r r � import_dbapis s �z$SQLiteDialect_aiosqlite.import_dbapic C s | � |�rtjS tjS d S r )Z_is_url_file_dbr ZNullPoolZ StaticPool)ro �urlr r r �get_pool_classy s z&SQLiteDialect_aiosqlite.get_pool_classc s. t || jj�r dt|�kr dS t� �|||�S )NrV T)rW rG rY �strr@ � is_disconnect)r �erH r rC r r rt � s � �z%SQLiteDialect_aiosqlite.is_disconnectc C s |j S r )r )r rH r r r �get_driver_connection� s z-SQLiteDialect_aiosqlite.get_driver_connection)r; r< r= ZdriverZsupports_statement_cacheZis_asyncZ supports_server_side_cursorsrj Zexecution_ctx_cls� classmethodrp rr rt rv rE r r rC r rl i s rl )�__doc__rL � collectionsr � functoolsr �baser Zpysqliter � r r Zenginer Zutil.concurrencyr r r r? rF r] r^ rj rl �dialectr r r r �<module> s$ F ] W1#