관리-도구
편집 파일: asyncmy.cpython-38.pyc
U -?�fS' � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z dd l m Z dd l m Z dd l mZ dd l mZ G d d� d�ZG dd� de�ZG dd� de �ZG dd� de�Zdd� ZG dd� d�ZG dd� de�ZeZdS ) 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 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") � )�deque)�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#dd�Z 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�� }| �|�� �| _t� | _d S �N)r r r �cursor� __aenter__r r r ��selfZadapt_connectionr � r �Q/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/dialects/mysql/asyncmy.py�__init__4 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 B s z"AsyncAdapt_asyncmy_cursor.rowcountc C s | j jS r �r � arraysizer r r r r F s z#AsyncAdapt_asyncmy_cursor.arraysizec C s || j _d S r r �r �valuer r r r J s c C s | j jS r )r � lastrowidr r r r r" N s z#AsyncAdapt_asyncmy_cursor.lastrowidc C s | j �� d S r )r �clearr r r r �closeR 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 � executemany_ 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_sider �fetchallr )r r& r' �resultr r r r% d 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) s s z,AsyncAdapt_asyncmy_cursor._executemany_asyncc G s d S r r )r Z inputsizesr r r � setinputsizesw s z'AsyncAdapt_asyncmy_cursor.setinputsizesc c s | j r| j �� V q d S r �r �popleftr r r r �__iter__z s z"AsyncAdapt_asyncmy_cursor.__iter__c C s | j r| j �� S d S d S r r1 r r r r �fetchone~ s z"AsyncAdapt_asyncmy_cursor.fetchonec s4 |d kr| j }| j� � fdd�tt|t� ���D �S )Nc s g | ] }� � � �qS r )r2 )�.0�_�Zrrr r � <listcomp>� s z7AsyncAdapt_asyncmy_cursor.fetchmany.<locals>.<listcomp>)r r �range�min�len�r �sizer r7 r � fetchmany� s z#AsyncAdapt_asyncmy_cursor.fetchmanyc C s t | j�}| j�� |S r )�listr r# )r �retvalr 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 r3 r4 r> 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 r4 r r r r r4 � s z%AsyncAdapt_asyncmy_ss_cursor.fetchoneNc C s | � | jj|d��S )N)r= )r r r> r< r r r r> � 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) rA rB rC rD r- r r$ r4 r> r. r r r r rG � s rG c @ s| e Zd Zee�ZdZdd� Zedd� �Z dd� Z dd � Z d d � Z d d � Z ddd�Zdd� Zdd� Zdd� Zdd�dd�ZdS ) � AsyncAdapt_asyncmy_connection)rH �_execute_mutexc C s || _ || _t�� | _d S r )rH r r �LockrK )r rH � 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)rK �AttributeErrorrH � 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 rR r r r r rQ � s z&AsyncAdapt_asyncmy_connection._do_pingc C s | j �� S r )r �character_set_namer r r r rS � s z0AsyncAdapt_asyncmy_connection.character_set_namec C s | � | j�|�� d S r )r r � autocommitr r r r rT � s z(AsyncAdapt_asyncmy_connection.autocommitFc C s |r t | �S t| �S d S r )rG 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 rU � s z&AsyncAdapt_asyncmy_connection.rollbackc C s | � | j�� � d S r )r r �commitr r r r rV � s z$AsyncAdapt_asyncmy_connection.commitc C s | j �� d S r )r r$ r r r r � terminate� s z'AsyncAdapt_asyncmy_connection.terminateN��returnc C s | � | j�� � d S r )r r Z ensure_closedr r r r r$ � s z#AsyncAdapt_asyncmy_connection.close)F)rA rB rC � staticmethodr r rD r r r, rR rQ rS rT r rU rV rW r$ r r r r rJ � s rJ c @ s e Zd ZdZee�ZdS )�%AsyncAdaptFallback_asyncmy_connectionr N)rA rB rC rD rZ r r r r r r r[ � s r[ c C s t | �S )zReturn x as a binary type.)�bytes)�xr r r �_Binary� s r^ 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)rI Z paramstyle�_init_dbapi_attributes)r rI r r r r � s z!AsyncAdapt_asyncmy_dbapi.__init__c C s$ dD ]}t | |t| jj|�� qd S )N) �Warning�Error�InterfaceErrorZ DataErrorZ DatabaseErrorZOperationalErrorrd ZIntegrityErrorZProgrammingErrorrO ZNotSupportedError)�setattr�getattrrI �errors)r �namer r r ra � s z/AsyncAdapt_asyncmy_dbapi._init_dbapi_attributes�STRING�NUMBER�BINARY�DATETIME� TIMESTAMPc O sR |� dd�}|� d| jj�}t�|�r:t| t|||���S t| t|||���S d S )N�async_fallbackFZasync_creator_fn) �poprI �connectr �asboolr[ r rJ r )r �arg�kwrn Z creator_fnr r r rp s � �z AsyncAdapt_asyncmy_dbapi.connectN)rA rB rC r ra r �symbolri rj rk rl rm rZ r^ �Binaryrp r r r r r_ � s r_ c sv e Zd ZdZdZdZeZdZdZ e dd� �Z e dd� �Z dd�d d �Z � fd d �Z� fd d�Zdd� Zdd� Z� ZS )�MySQLDialect_asyncmyrI Tc C s t td��S )NrI )r_ � __import__)�clsr r r � import_dbapi* s z!MySQLDialect_asyncmy.import_dbapic C s( |j �dd�}t�|�r tjS tjS d S )Nrn F)�query�getr rq r Z FallbackAsyncAdaptedQueuePoolZAsyncAdaptedQueuePool)rx �urlrn r r r �get_pool_class. s z#MySQLDialect_asyncmy.get_pool_classNrX c C s |� � d S r )rW )r Zdbapi_connectionr r r � do_terminate7 s z!MySQLDialect_asyncmy.do_terminatec s t � j|tddd�d�S )N�user�db)�usernameZdatabase)Z_translate_args)�super�create_connect_args�dict)r r| �� __class__r r r� : s �z(MySQLDialect_asyncmy.create_connect_argsc s4 t � �|||�rdS t|��� }d|kp.d|kS d S )NTz not connectedznetwork operation failed)r� � is_disconnect�str�lower)r �erM r Zstr_er� r r r� ? 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_flagH s z,MySQLDialect_asyncmy._found_rows_client_flagc C s |j S r )r )r rM r r r �get_driver_connectionM s z*MySQLDialect_asyncmy.get_driver_connection)rA rB rC ZdriverZsupports_statement_cacheZ supports_server_side_cursorsrG Z _sscursorZis_asyncZ has_terminate� classmethodry r} r~ r� r� r� r� � __classcell__r r r� r rv s rv N)�__doc__� collectionsr � contextlibr Zpymysqlr � r r Zenginer Zutil.concurrencyr r r r rG rJ r[ r^ r_ rv �dialectr r r r �<module> s" i 7-1