관리-도구
편집 파일: aiomysql.cpython-38.pyc
U -?�f' � @ 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� de�ZG dd� de �ZG dd� de�ZG dd� d�ZG dd� de�ZeZdS )a� .. 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 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") � )�deque� )�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_aiomysql_cursorF)�_adapt_connection� _connection�await_�_cursor�_rowsc C sB || _ |j| _|j| _| j�|jj�}| �|�� �| _t� | _ d S �N) r r r �cursor�dbapi�Cursor� __aenter__r r r ��selfZadapt_connectionr � r �R/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/dialects/mysql/aiomysql.py�__init__5 s z#AsyncAdapt_aiomysql_cursor.__init__c C s | j jS r )r � description�r r r r r @ s z&AsyncAdapt_aiomysql_cursor.descriptionc C s | j jS r )r �rowcountr r r r r D s z#AsyncAdapt_aiomysql_cursor.rowcountc C s | j jS r �r � arraysizer r r r r H s z$AsyncAdapt_aiomysql_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_aiomysql_cursor.lastrowidc C s | j �� d S r )r �clearr r r r �closeT s z AsyncAdapt_aiomysql_cursor.closeNc C s | � | �||��S r )r �_execute_async)r � operation� parametersr r r �execute^ s z"AsyncAdapt_aiomysql_cursor.executec C s | � | �||��S r )r �_executemany_async�r r' Zseq_of_parametersr r r � executemanya s �z&AsyncAdapt_aiomysql_cursor.executemanyc � sh | j j4 I d H �H | j�||�I d H }| jsBt| 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_sider �fetchallr )r r' r( �resultr r r r& f 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* r s z-AsyncAdapt_aiomysql_cursor._executemany_asyncc G s d S r r )r Z inputsizesr r r � setinputsizesv s z(AsyncAdapt_aiomysql_cursor.setinputsizesc c s | j r| j �� V q d S r �r �popleftr r r r �__iter__y s z#AsyncAdapt_aiomysql_cursor.__iter__c C s | j r| j �� S d S d S r r2 r r r r �fetchone} s z#AsyncAdapt_aiomysql_cursor.fetchonec s4 |d kr| j }| j� � fdd�tt|t� ���D �S )Nc s g | ] }� � � �qS r )r3 )�.0�_�Zrrr r � <listcomp>� s z8AsyncAdapt_aiomysql_cursor.fetchmany.<locals>.<listcomp>)r r �range�min�len�r �sizer r8 r � fetchmany� s z$AsyncAdapt_aiomysql_cursor.fetchmanyc C s t | j�}| j�� |S r )�listr r$ )r �retvalr 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* r1 r4 r5 r? 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�|jj�}| �|�� �| _d S r )r r r r r �SSCursorr 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 r5 r r r r r5 � s z&AsyncAdapt_aiomysql_ss_cursor.fetchoneNc C s | � | jj|d��S )N)r> )r r r? r= r r r r? � 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) rB rC rD rE r. r r% r5 r? r/ r r r r rH � s rH c @ sh e Zd Zee�ZdZdd� Zdd� Zdd� Z dd � Z dd d �Z d d� Z dd� Z dd� Zdd�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 rM � s z#AsyncAdapt_aiomysql_connection.pingc C s | j �� S r )r �character_set_namer r r r rN � s z1AsyncAdapt_aiomysql_connection.character_set_namec C s | � | j�|�� d S r )r r � autocommitr! r r r rO � s z)AsyncAdapt_aiomysql_connection.autocommitFc C s |r t | �S t| �S d S r )rH 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 rP � s z'AsyncAdapt_aiomysql_connection.rollbackc C s | � | j�� � d S r )r r �commitr r r r rQ � s z%AsyncAdapt_aiomysql_connection.commitc C s | j �� d S r )r r% r r r r � terminate� s z(AsyncAdapt_aiomysql_connection.terminateN��returnc C s | � | j�� � d S r )r r Z ensure_closedr r r r r% � s z$AsyncAdapt_aiomysql_connection.close)F)rB rC rD � staticmethodr r rE r rM rN rO r rP rQ rR r% r r r r rJ � s rJ c @ s e Zd ZdZee�ZdS )�&AsyncAdaptFallback_aiomysql_connectionr N)rB rC rD rE rU r r r r r r rV � s rV c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �AsyncAdapt_aiomysql_dbapic C s. || _ || _d| _| �� | �� \| _| _d S )N�format)�aiomysql�pymysqlZ paramstyle�_init_dbapi_attributes�_init_cursors_subclassesr rI )r rY rZ 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 DatabaseErrorZOperationalErrorr_ ZIntegrityErrorZProgrammingErrorZ InternalErrorZNotSupportedError)�NUMBER�STRINGZDATETIME�BINARY� TIMESTAMP�Binary)�setattr�getattrrY rZ )r �namer r r r[ � s z0AsyncAdapt_aiomysql_dbapi._init_dbapi_attributesc O sR |� dd�}|� d| jj�}t�|�r:t| t|||���S t| t|||���S d S )N�async_fallbackFZasync_creator_fn) �poprY �connectr �asboolrV r rJ r )r �arg�kwrh Z creator_fnr r r rj s � �z!AsyncAdapt_aiomysql_dbapi.connectc C s0 G dd� d| j j�}G dd� d| j j�}||fS )Nc @ s e Zd Zdd� ZdS )zBAsyncAdapt_aiomysql_dbapi._init_cursors_subclasses.<locals>.Cursorc � s d S r r �r �connr r r �_show_warnings s zQAsyncAdapt_aiomysql_dbapi._init_cursors_subclasses.<locals>.Cursor._show_warningsN�rB rC rD rp r r r r r s r c @ s e Zd Zdd� ZdS )zDAsyncAdapt_aiomysql_dbapi._init_cursors_subclasses.<locals>.SSCursorc � s d S r r rn r r r rp s zSAsyncAdapt_aiomysql_dbapi._init_cursors_subclasses.<locals>.SSCursor._show_warningsNrq r r r r rI s rI )rY r rI )r r rI r r r r\ s z2AsyncAdapt_aiomysql_dbapi._init_cursors_subclassesN)rB rC rD r r[ rj r\ r r r r rW � s rW 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_aiomysqlrY Tc C s t td�td��S )NrY rZ )rW � __import__)�clsr r r � import_dbapi& s �z"MySQLDialect_aiomysql.import_dbapic C s( |j �dd�}t�|�r tjS tjS d S )Nrh F)�query�getr rk r Z FallbackAsyncAdaptedQueuePoolZAsyncAdaptedQueuePool)rt �urlrh r r r �get_pool_class, s z$MySQLDialect_aiomysql.get_pool_classNrS c C s |� � d S r )rR )r Zdbapi_connectionr r r � do_terminate5 s z"MySQLDialect_aiomysql.do_terminatec s t � j|tddd�d�S )N�user�db)�usernameZdatabase)Z_translate_args)�super�create_connect_args�dict)r rx �� __class__r r r 8 s �z)MySQLDialect_aiomysql.create_connect_argsc s, t � �|||�rdS t|��� }d|kS d S )NTz not connected)r~ � is_disconnect�str�lower)r �erL r Zstr_er� r r r� = s z#MySQLDialect_aiomysql.is_disconnectc C s ddl m} |jS )Nr )�CLIENT)Zpymysql.constantsr� Z FOUND_ROWS)r r� r r r �_found_rows_client_flagD s z-MySQLDialect_aiomysql._found_rows_client_flagc C s |j S r )r )r rL r r r �get_driver_connectionI s z+MySQLDialect_aiomysql.get_driver_connection)rB rC rD ZdriverZsupports_statement_cacheZ supports_server_side_cursorsrH Z _sscursorZis_asyncZ has_terminate� classmethodru ry rz r r� r� r� � __classcell__r r r� r rr s rr N)�__doc__� collectionsr rZ r � r r Zenginer Zutil.concurrencyr r r r rH rJ rV rW rr �dialectr r r r �<module> s g (>1