관리-도구
편집 파일: psycopg.cpython-38.pyc
U -?�f�Z � @ s& d Z ddlmZ ddlmZ ddlZddlZddlmZ ddlm Z ddl m Z dd l m Z dd l mZ dd lmZ dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddl mZ ddl mZ ddl m Z ddl m Z ddl m!Z! ddl m"Z" e �r.ddlm#Z# ddl$m%Z% e�&d �Z'G d d � d e j(�Z)G d d � d e�Z*G d!d"� d"e�Z+G d#d$� d$e�Z,G d%d&� d&e jj-�Z.G d'd(� d(e jj/�Z0G d)d*� d*e�Z1G d+d,� d,e�Z2G d-d.� d.e j3�Z4G d/d0� d0e j5�Z6G d1d2� d2e j7�Z8G d3d4� d4e j9�Z:G d5d6� d6e j;�Z<G d7d8� d8e j=�Z>G d9d:� d:e j?�Z@G d;d<� d<e jA�ZBG d=d>� d>e jC�ZDG d?d@� d@e jE�ZFG dAdB� dBe�ZGG dCdD� dDe�ZHG dEdF� dFe�ZIdGdH� ZJG dIdJ� dJe �ZKG dKdL� dL�ZLG dMdN� dNeL�ZMG dOdP� dPe �ZNG dQdR� dReN�ZOG dSdT� dT�ZPG dUdV� dVeK�ZQeKZReQZSdS )WaH .. dialect:: postgresql+psycopg :name: psycopg (a.k.a. psycopg 3) :dbapi: psycopg :connectstring: postgresql+psycopg://user:password@host:port/dbname[?key=value&key=value...] :url: https://pypi.org/project/psycopg/ ``psycopg`` is the package and module name for version 3 of the ``psycopg`` database driver, formerly known as ``psycopg2``. This driver is different enough from its ``psycopg2`` predecessor that SQLAlchemy supports it via a totally separate dialect; support for ``psycopg2`` is expected to remain for as long as that package continues to function for modern Python versions, and also remains the default dialect for the ``postgresql://`` dialect series. The SQLAlchemy ``psycopg`` dialect provides both a sync and an async implementation under the same dialect name. The proper version is selected depending on how the engine is created: * calling :func:`_sa.create_engine` with ``postgresql+psycopg://...`` will automatically select the sync version, e.g.:: from sqlalchemy import create_engine sync_engine = create_engine("postgresql+psycopg://scott:tiger@localhost/test") * calling :func:`_asyncio.create_async_engine` with ``postgresql+psycopg://...`` will automatically select the async version, e.g.:: from sqlalchemy.ext.asyncio import create_async_engine asyncio_engine = create_async_engine("postgresql+psycopg://scott:tiger@localhost/test") The asyncio version of the dialect may also be specified explicitly using the ``psycopg_async`` suffix, as:: from sqlalchemy.ext.asyncio import create_async_engine asyncio_engine = create_async_engine("postgresql+psycopg_async://scott:tiger@localhost/test") .. seealso:: :ref:`postgresql_psycopg2` - The SQLAlchemy ``psycopg`` dialect shares most of its behavior with the ``psycopg2`` dialect. Further documentation is available there. Using a different Cursor class ------------------------------ One of the differences between ``psycopg`` and the older ``psycopg2`` is how bound parameters are handled: ``psycopg2`` would bind them client side, while ``psycopg`` by default will bind them server side. It's possible to configure ``psycopg`` to do client side binding by specifying the ``cursor_factory`` to be ``ClientCursor`` when creating the engine:: from psycopg import ClientCursor client_side_engine = create_engine( "postgresql+psycopg://...", connect_args={"cursor_factory": ClientCursor}, ) Similarly when using an async engine the ``AsyncClientCursor`` can be specified:: from psycopg import AsyncClientCursor client_side_engine = create_async_engine( "postgresql+psycopg://...", connect_args={"cursor_factory": AsyncClientCursor}, ) .. seealso:: `Client-side-binding cursors <https://www.psycopg.org/psycopg3/docs/advanced/cursors.html#client-side-binding-cursors>`_ � )� annotations)�dequeN)�cast)� TYPE_CHECKING� )�ranges)�_PGDialect_common_psycopg)�"_PGExecutionContext_common_psycopg)�INTERVAL)� PGCompiler)�PGIdentifierPreparer)� REGCONFIG)�JSON)�JSONB)� JSONPathType)�CITEXT� )�pool)�util)�AdaptedConnection)�sqltypes)�await_fallback)� await_only)�Iterable)�AsyncConnectionz sqlalchemy.dialects.postgresqlc @ s e Zd ZdZdS )� _PGStringTN��__name__� __module__� __qualname__�render_bind_cast� r! r! �V/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/dialects/postgresql/psycopg.pyr x s r c @ s e Zd ZdZdS )� _PGREGCONFIGTNr r! r! r! r" r# | s r# c @ s e Zd ZdZdd� Zdd� ZdS )�_PGJSONTc C s | � d |j�S �N)�_make_bind_processor� _psycopg_Json��self�dialectr! r! r" �bind_processor� s z_PGJSON.bind_processorc C s d S r% r! �r) r* �coltyper! r! r" �result_processor� s z_PGJSON.result_processorN�r r r r r+ r. r! r! r! r" r$ � s r$ c @ s e Zd ZdZdd� Zdd� ZdS )�_PGJSONBTc C s | � d |j�S r% )r& �_psycopg_Jsonbr( r! r! r" r+ � s z_PGJSONB.bind_processorc C s d S r% r! r, r! r! r" r. � s z_PGJSONB.result_processorNr/ r! r! r! r" r0 � s r0 c @ s e Zd ZdZdZdS )�_PGJSONIntIndexTypeZjson_int_indexTN�r r r Z__visit_name__r r! r! r! r" r2 � s r2 c @ s e Zd ZdZdZdS )�_PGJSONStrIndexTypeZjson_str_indexTNr3 r! r! r! r" r4 � s r4 c @ s e Zd ZdS )�_PGJSONPathTypeN�r r r r! r! r! r" r5 � s r5 c @ s e Zd ZdZdS )� _PGIntervalTNr r! r! r! r" r7 � s r7 c @ s e Zd ZdZdS )� _PGTimeStampTNr r! r! r! r" r8 � s r8 c @ s e Zd ZdZdS )�_PGDateTNr r! r! r! r" r9 � s r9 c @ s e Zd ZdZdS )�_PGTimeTNr r! r! r! r" r: � s r: c @ s e Zd ZdZdS )� _PGIntegerTNr r! r! r! r" r; � s r; c @ s e Zd ZdZdS )�_PGSmallIntegerTNr r! r! r! r" r< � s r< c @ s e Zd ZdZdS )� _PGNullTypeTNr r! r! r! r" r= � s r= c @ s e Zd ZdZdS )� _PGBigIntegerTNr r! r! r! r" r> � s r> c @ s e Zd ZdZdS )� _PGBooleanTNr r! r! r! r" r? � s r? c @ s e Zd Zdd� Zdd� ZdS )� _PsycopgRangec s t t|�j� � fdd�}|S )Nc s& t | tj�r"� | j| j| j| j�} | S r% )� isinstancer �Range�lower�upper�bounds�empty��value�� psycopg_Ranger! r" �to_range� s �z._PsycopgRange.bind_processor.<locals>.to_range)r �PGDialect_psycopg�_psycopg_Range�r) r* rK r! rI r" r+ � s z _PsycopgRange.bind_processorc C s dd� }|S )Nc S s2 | d k r.t j| j| j| jr | jnd| j d�} | S )N�[)�rE rF �r rB Z_lowerZ_upperZ_boundsrG r! r! r" rK � s �z0_PsycopgRange.result_processor.<locals>.to_ranger! �r) r* r- rK r! r! r" r. � s z _PsycopgRange.result_processorN�r r r r+ r. r! r! r! r" r@ � s r@ c @ s e Zd Zdd� Zdd� ZdS )�_PsycopgMultiRangec s4 t t|�j�t t|�j�td �� � ��fdd�}|S )Nc s0 t | t� �f�r| S ��fdd�td| �D ��S )Nc s" g | ]}� |j |j|j|j��qS r! )rC rD rE rF )�.0�elementrI r! r" � <listcomp>� s ��zG_PsycopgMultiRange.bind_processor.<locals>.to_range.<locals>.<listcomp>zIterable[ranges.Range])rA �strr rG �ZNoneTypeZpsycopg_MultirangerJ r! r" rK � s ��z3_PsycopgMultiRange.bind_processor.<locals>.to_range)r rL rM �_psycopg_Multirange�typerN r! rY r" r+ � s �z!_PsycopgMultiRange.bind_processorc C s dd� }|S )Nc S s$ | d kr d S t �dd� | D ��S d S )Nc s s4 | ],}t j|j|j|jr |jnd |j d�V qdS )rO rP NrQ )rU �elemr! r! r" � <genexpr> s ��zH_PsycopgMultiRange.result_processor.<locals>.to_range.<locals>.<genexpr>)r Z MultiRangerG r! r! r" rK � s �z5_PsycopgMultiRange.result_processor.<locals>.to_ranger! rR r! r! r" r. � s z#_PsycopgMultiRange.result_processorNrS r! r! r! r" rT � s rT c @ s e Zd ZdS )�PGExecutionContext_psycopgNr6 r! r! r! r" r^ s r^ c @ s e Zd ZdS )�PGCompiler_psycopgNr6 r! r! r! r" r_ s r_ c @ s e Zd ZdS )� PGIdentifierPreparer_psycopgNr6 r! r! r! r" r` s r` c C s t �d| j| j� d S )Nz%s: %s)�logger�infoZseverityZmessage_primary)Z diagnosticr! r! r" � _log_notices s rc c ) s� e Zd ZdZdZdZdZdZeZ e Z e Z dZdZdZe�ejejeeeeee e ejee e ejj e ejj!e"ejj#e$ej%e&e'e&ej(e)ej*e+ej,e-ej.e/ej0e1ej2e3e4j5e6e4j7e8i�Z� fdd�Z9� fdd �Z:d d � Z;� fd d �Z<e=dd� �Z>e=dd� �Z?ej@dd� �ZAej@dd� �ZBej@dd� �ZCej@dd� �ZDej@dd� �ZEej@d d � �ZFd d � ZG� fd d!�ZHd"d#� ZId$d%� ZJd&d'� ZKd(d)� ZLd*d+� ZMd5d-d.�ZNd6d/d0�ZOd7d1d2�ZPej@d3d4� �ZQ� ZRS )8rL �psycopgTZpyformat)r r Nc s� t � jf |� | jr�t�d| jj�}|rFtdd� |�ddd�D ��| _| jdk rXt d��d d l m } || jj � | _ }| jd kr�d d l}|�d |jjj� |�d |jjj� | jr�d dlm} || j|� | jr�d dlm} || j|� d S )Nz(\d+)\.(\d+)(?:\.(\d+))?c s s | ]}|d k rt |�V qd S r% )�int)rU �xr! r! r" r] O s z-PGDialect_psycopg.__init__.<locals>.<genexpr>r � r )r r rg z,psycopg version 3.0.2 or higher is required.r )� AdaptersMapFZinetZcidr)�set_json_loads)�set_json_dumps)�super�__init__�dbapi�re�match� __version__�tuple�group�psycopg_version� ImportErrorZ psycopg.adaptrh �adapters�_psycopg_adapters_mapZ_native_inet_typesZpsycopg.types.stringZregister_loader�types�stringZ TextLoaderZ_json_deserializerZpsycopg.types.jsonri Z_json_serializerrj )r) �kwargs�mrh Z adapters_maprd ri rj �� __class__r! r" rl I s> � � � � � zPGDialect_psycopg.__init__c s< t � �|�\}}| jr | j|d< | jd k r4| j|d< ||fS )N�context�client_encoding)rk �create_connect_argsrv r~ )r) �urlZcargsZcparamsr{ r! r" r r s z%PGDialect_psycopg.create_connect_argsc C s ddl m} |�|jj|�S �Nr )�TypeInfo)� psycopg.typesr� �fetch� connection�driver_connection)r) r� �namer� r! r! r" �_type_info_fetch| s z"PGDialect_psycopg._type_info_fetchc sb t � �|� | jsd| _| jr^| �|d�}|d k | _| jr^ddlm} ||| j � |||j � d S )NFZhstorer )�register_hstore) rk � initializeZinsert_returningZ insert_executemany_returningZuse_native_hstorer� �_has_native_hstoreZpsycopg.types.hstorer� rv r� )r) r� rb r� r{ r! r" r� � s z PGDialect_psycopg.initializec C s dd l }|S )Nr )rd )�clsrd r! r! r" � import_dbapi� s z PGDialect_psycopg.import_dbapic C s t S r% )�PGDialectAsync_psycopg)r� r� r! r! r" �get_async_dialect_cls� s z'PGDialect_psycopg.get_async_dialect_clsc C s& | j jj| j jj| j jj| j jjd�S )N)zREAD COMMITTEDzREAD UNCOMMITTEDzREPEATABLE READ� SERIALIZABLE)rm ZIsolationLevelZREAD_COMMITTEDZREAD_UNCOMMITTEDZREPEATABLE_READr� �r) r! r! r" �_isolation_lookup� s �z#PGDialect_psycopg._isolation_lookupc C s ddl m} |jS �Nr )�json)r� r� ZJson�r) r� r! r! r" r' � s z PGDialect_psycopg._psycopg_Jsonc C s ddl m} |jS r� )r� r� ZJsonbr� r! r! r" r1 � s z PGDialect_psycopg._psycopg_Jsonbc C s ddl m} |S )Nr )�TransactionStatus)� psycopg.pqr� )r) r� r! r! r" �_psycopg_TransactionStatus� s z,PGDialect_psycopg._psycopg_TransactionStatusc C s ddl m} |S )Nr )rB )Zpsycopg.types.rangerB )r) rB r! r! r" rM � s z PGDialect_psycopg._psycopg_Rangec C s ddl m} |S )Nr )� Multirange)Zpsycopg.types.multiranger� )r) r� r! r! r" rZ � s z%PGDialect_psycopg._psycopg_Multirangec C s ||_ ||_d S r% �� autocommit�isolation_level�r) r� r� r� r! r! r" �_do_isolation_level� s z%PGDialect_psycopg._do_isolation_levelc s, |j j}t� �|�}|| jjkr(|�� |S r% )rb �transaction_statusrk �get_isolation_levelr� �IDLE�rollback)r) �dbapi_connectionZ status_beforerH r{ r! r" r� � s z%PGDialect_psycopg.get_isolation_levelc C s4 |dkr| j |dd d� n| j |d| j| d� d S )NZ AUTOCOMMITTr� F)r� r� )r) r� �levelr! r! r" �set_isolation_level� s ��z%PGDialect_psycopg.set_isolation_levelc C s ||_ d S r% �Z read_only�r) r� rH r! r! r" � set_readonly� s z PGDialect_psycopg.set_readonlyc C s |j S r% r� �r) r� r! r! r" � get_readonly� s z PGDialect_psycopg.get_readonlyc s> dd� }|g� �j d k r.�fdd�}� �|� � fdd�}|S )Nc S s | � t� d S r% )Zadd_notice_handlerrc ��connr! r! r" �notices� s z-PGDialect_psycopg.on_connect.<locals>.noticesc s � � | � j� d S r% )r� r� r� r� r! r" � on_connect� s z0PGDialect_psycopg.on_connect.<locals>.on_connectc s � D ] }|| � qd S r% r! )r� �fn)�fnsr! r" r� � s )r� �append)r) r� r� r! )r� r) r" r� � s z PGDialect_psycopg.on_connectc C s* t || jj�r&|d k r&|js"|jr&dS dS )NTF)rA rm �Error�closed�broken)r) �er� �cursorr! r! r" � is_disconnect� s z PGDialect_psycopg.is_disconnectFc C s` |j j}|s |jj| jjkr$|�� |j}z |s<| �|d� |� |� W 5 |sZ| �||� X d S )NT) r� r� rb r� r� r� r� r� �_do_autocommit�execute)r) r� �command�recoverZ dbapi_connZbefore_autocommitr! r! r" �_do_prepared_twophase s ��� z'PGDialect_psycopg._do_prepared_twophasec C s. |r | j |d|� d�|d� n | �|j� d S )NzROLLBACK PREPARED '�'�r� )r� Z do_rollbackr� �r) r� �xidZ is_preparedr� r! r! r" �do_rollback_twophase s �z&PGDialect_psycopg.do_rollback_twophasec C s. |r | j |d|� d�|d� n | �|j� d S )NzCOMMIT PREPARED 'r� r� )r� Z do_commitr� r� r! r! r" �do_commit_twophase! s �z$PGDialect_psycopg.do_commit_twophasec C s dS )N�;r! r� r! r! r" � _dialect_specific_select_one+ s z.PGDialect_psycopg._dialect_specific_select_one)F)TF)TF)Sr r r Zdriver�supports_statement_cacheZ supports_server_side_cursorsZdefault_paramstyleZ supports_sane_multi_rowcountr^ Zexecution_ctx_clsr_ Zstatement_compilerr` �preparerrs r� rv r Z update_copyr Zcolspecsr �Stringr r r# r r$ r r r0 r r5 �JSONIntIndexTyper2 �JSONStrIndexTyper4 ZIntervalr7 r �Dater9 �DateTimer8 �Timer: �Integerr; � SmallIntegerr< � BigIntegerr> r ZAbstractSingleRanger@ ZAbstractMultiRangerT rl r r� r� � classmethodr� r� Zmemoized_propertyr� r' r1 r� rM rZ r� r� r� r� r� r� r� r� r� r� r� � __classcell__r! r! r{ r" rL s� �� ) � � rL c @ s� e Zd ZdZdZdd�dd�Zdd� Zed d � �Zej d d � �Zd d � Z ddd�Z dd� Z dd� Z dd� Zddd�Zdd� ZdS ) �AsyncAdapt_psycopg_cursor)�_cursor�await_�_rowsN�None��returnc C s || _ || _t� | _d S r% )r� r� r r� )r) r� r� r! r! r" rl 5 s z"AsyncAdapt_psycopg_cursor.__init__c C s t | j|�S r% )�getattrr� �r) r� r! r! r" � __getattr__: s z%AsyncAdapt_psycopg_cursor.__getattr__c C s | j jS r% �r� � arraysizer� r! r! r" r� = s z#AsyncAdapt_psycopg_cursor.arraysizec C s || j _d S r% r� �r) rH r! r! r" r� A s c C s | j �� | j�� d S r% )r� �clearr� �_closer� r! r! r" �closeE s z AsyncAdapt_psycopg_cursor.closec K sP | � | jj||f|��}| jj}|rL|j| jjkrL| � | j�� �}t|�| _ |S r% ) r� r� r� Zpgresult�status�_psycopg_ExecStatusZ TUPLES_OK�fetchallr r� )r) �query�params�kw�result�res�rowsr! r! r" r� J s z!AsyncAdapt_psycopg_cursor.executec C s | � | j�||��S r% )r� r� � executemany)r) r� Z params_seqr! r! r" r� V s z%AsyncAdapt_psycopg_cursor.executemanyc c s | j r| j �� V q d S r% �r� �popleftr� r! r! r" �__iter__Y s z"AsyncAdapt_psycopg_cursor.__iter__c C s | j r| j �� S d S d S r% r� r� r! r! r" �fetchone] s z"AsyncAdapt_psycopg_cursor.fetchonec s6 |d kr| j j}| j� � fdd�tt|t� ���D �S )Nc s g | ] }� � � �qS r! )r� )rU �_�Zrrr! r" rW h s z7AsyncAdapt_psycopg_cursor.fetchmany.<locals>.<listcomp>)r� r� r� �range�min�len�r) �sizer! r� r" � fetchmanyc s z#AsyncAdapt_psycopg_cursor.fetchmanyc C s t | j�}| j�� |S r% )�listr� r� )r) �retvalr! r! r" r� j s z"AsyncAdapt_psycopg_cursor.fetchall)N)N)r r r � __slots__r� rl r� �propertyr� �setterr� r� r� r� r� r� r� r! r! r! r" r� 0 s r� c @ s@ e Zd Zddd�Zdd� Zdd� Zdd d �Zd d � Zd d� ZdS )� AsyncAdapt_psycopg_ss_cursorNc K s | � | jj||f|�� | S r% )r� r� r� )r) r� r� r� r! r! r" r� q s z$AsyncAdapt_psycopg_ss_cursor.executec C s | � | j�� � d S r% )r� r� r� r� r! r! r" r� u s z"AsyncAdapt_psycopg_ss_cursor.closec C s | � | j�� �S r% )r� r� r� r� r! r! r" r� x s z%AsyncAdapt_psycopg_ss_cursor.fetchoner c C s | � | j�|��S r% )r� r� r� r� r! r! r" r� { s z&AsyncAdapt_psycopg_ss_cursor.fetchmanyc C s | � | j�� �S r% )r� r� r� r� r! r! r" r� ~ s z%AsyncAdapt_psycopg_ss_cursor.fetchallc c s> | j �� }z| �|�� �V W q tk r6 Y q:Y q X q d S r% )r� � __aiter__r� � __anext__�StopAsyncIteration)r) �iteratorr! r! r" r� � s z%AsyncAdapt_psycopg_ss_cursor.__iter__)N)r ) r r r r� r� r� r� r� r� r! r! r! r" r� p s r� c @ s� e Zd ZU ded< dZee�Zdd�dd�Zdd � Z d d d �Z d d� Z dd� Z dd� Z dd� Zedd� �Zejdd� �Zdd� Zdd� Zd d � Zd d � Zd S )!� AsyncAdapt_psycopg_connectionr � _connectionr! r� r� c C s || _ d S r% �r r� r! r! r" rl � s z&AsyncAdapt_psycopg_connection.__init__c C s t | j|�S r% )r� r r� r! r! r" r� � s z)AsyncAdapt_psycopg_connection.__getattr__Nc K s$ | � | jj||f|��}t|| j �S r% )r� r r� r� )r) r� r� r� r� r! r! r" r� � s z%AsyncAdapt_psycopg_connection.executec O s4 | j j||�}t|d�r$t|| j�S t|| j�S d S )Nr� )r r� �hasattrr� r� r� )r) �argsr� r� r! r! r" r� � s z$AsyncAdapt_psycopg_connection.cursorc C s | � | j�� � d S r% )r� r �commitr� r! r! r" r � s z$AsyncAdapt_psycopg_connection.commitc C s | � | j�� � d S r% )r� r r� r� r! r! r" r� � s z&AsyncAdapt_psycopg_connection.rollbackc C s | � | j�� � d S r% )r� r r� r� r! r! r" r� � s z#AsyncAdapt_psycopg_connection.closec C s | j jS r% )r r� r� r! r! r" r� � s z(AsyncAdapt_psycopg_connection.autocommitc C s | � |� d S r% ��set_autocommitr� r! r! r" r� � s c C s | � | j�|�� d S r% )r� r r r� r! r! r" r � s z,AsyncAdapt_psycopg_connection.set_autocommitc C s | � | j�|�� d S r% )r� r r� r� r! r! r" r� � s z1AsyncAdapt_psycopg_connection.set_isolation_levelc C s | � | j�|�� d S r% )r� r � set_read_onlyr� r! r! r" r � s z+AsyncAdapt_psycopg_connection.set_read_onlyc C s | � | j�|�� d S r% )r� r �set_deferrabler� r! r! r" r � s z,AsyncAdapt_psycopg_connection.set_deferrable)N)r r r �__annotations__r� � staticmethodr r� rl r� r� r� r r� r� r� r� r� r r� r r r! r! r! r" r � s$ r c @ s e Zd ZdZee�ZdS )�%AsyncAdaptFallback_psycopg_connectionr! N)r r r r� r r r� r! r! r! r" r � s r c @ s" e Zd Zdd�dd�Zdd� ZdS )�PsycopgAdaptDBAPIr� r� c C s2 || _ | j j�� D ]\}}|dkr|| j|< qd S )N�connect)rd �__dict__�items)r) rd �k�vr! r! r" rl � s zPsycopgAdaptDBAPI.__init__c O sP |� dd�}|� d| jjj�}t�|�r:tt|||���S tt |||���S d S )N�async_fallbackFZasync_creator_fn) �poprd r r r �asboolr r r r )r) �argr� r Z creator_fnr! r! r" r � s � � �zPsycopgAdaptDBAPI.connectN)r r r rl r r! r! r! r" r � s r c @ s\ e Zd ZdZdZedd� �Zedd� �Zdd� Zdd � Z d d � Z d d � Z dd� Z dd� Z dS )r� Tc C s" dd l }ddlm} |t_t|�S )Nr )� ExecStatus)rd r� r r� r� r )r� rd r r! r! r" r� � s z#PGDialectAsync_psycopg.import_dbapic C s( |j �dd�}t�|�r tjS tjS d S )Nr F)r� �getr r r Z FallbackAsyncAdaptedQueuePoolZAsyncAdaptedQueuePool)r� r� r r! r! r" �get_pool_class� s z%PGDialectAsync_psycopg.get_pool_classc C s& ddl m} |j}|�|�|j|��S r� )r� r� r� r� r� r� )r) r� r� r� Zadaptedr! r! r" r� � s z'PGDialectAsync_psycopg._type_info_fetchc C s |� |� |�|� d S r% )r r� r� r! r! r" r� � s z*PGDialectAsync_psycopg._do_isolation_levelc C s |� |� d S r% r r� r! r! r" r� � s z%PGDialectAsync_psycopg._do_autocommitc C s |� |� d S r% )r r� r! r! r" r� � s z#PGDialectAsync_psycopg.set_readonlyc C s |� |� d S r% )r r� r! r! r" r s z%PGDialectAsync_psycopg.set_deferrablec C s |j S r% r r� r! r! r" �get_driver_connection s z,PGDialectAsync_psycopg.get_driver_connectionN)r r r Zis_asyncr� r� r� r r� r� r� r� r r r! r! r! r" r� � s r� )T�__doc__� __future__r � collectionsr �loggingrn �typingr r � r Z_psycopg_commonr r �baser r r r r� r r r rw r r r Zenginer �sqlr Zutil.concurrencyr r r rd r � getLoggerra r� r r# r$ r0 r� r2 r� r4 r5 r7 r� r8 r� r9 r� r: r� r; r� r< ZNullTyper= r� r> �Booleanr? ZAbstractSingleRangeImplr@ ZAbstractMultiRangeImplrT r^ r_ r` rc rL r� r� r r r r� r* Z dialect_asyncr! r! r! r" �<module> sx M - @4-