관리-도구
편집 파일: _concurrency_py3k.cpython-38.pyc
U -?�f�# � @ s d dl mZ d dlZd dlmZ d dlZd dlZd dlmZ d dlmZ d dlm Z d dlm Z d dlm Z d d lm Z d d lm Z d d lmZ d d lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ e d�Zej �rG dd� de�Zdd�dd�Z nd dlm Z d d lmZ e e � d �Z d d d �d!d"�Z G d#d$� d$e�Z e d%d&d'�Z!e �r|d(d)d*�d+d,�Z"nej"Z"d-d.d*�d/d0�Z#d d�d1d2�Z$d3dd*�d4d5�Z%d3dd*�d6d7�Z&d8d9�d:d;d d;dd<�d=d>�Z'G d?d@� d@�Z(dAd�dBdC�Z)e �s e�r ej*Z+nG dDdE� dE�Z+dS )F� )� annotationsN)�Context)�Any)� Awaitable)�Callable)� Coroutine)�Optional)� TYPE_CHECKING)�TypeVar)�Union� )�memoized_property� )�exc)�py311)�Literal)�Protocol)�Self)� TypeGuard�_Tc @ sN e Zd ZU ded< ded< dd d�dd�Zd d d �d d �Zd d d �dd�ZdS )�greenlet�bool�deadzOptional[Context]� gr_context�Callable[..., Any]��fn�driverc C s d S �N� ��selfr r r r �Q/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/util/_concurrency_py3k.py�__init__( � zgreenlet.__init__r )�arg�returnc G s d S r r )r! r% r r r" �throw* s zgreenlet.throw)�valuer& c C s d S r r )r! r( r r r" �switch- s zgreenlet.switchN)�__name__� __module__� __qualname__�__annotations__r# r' r) r r r r" r $ s r �r& c C s d S r r r r r r" � getcurrent0 r$ r/ )r/ )r r � BaseExceptionr )�er& c C s t | t� p t | tjtjf�S r )� isinstance� Exception�asyncio� TimeoutErrorZCancelledError)r1 r r r" �is_exit_exception= s �r6 c @ s* e Zd ZU ded< dZddd�dd�Zd S ) �_AsyncIoGreenletr r Tr r r c C s t �| ||� tr|j| _d S r )r r# �_has_gr_contextr r r r r" r# O s z_AsyncIoGreenlet.__init__N)r* r+ r, r- � __sqlalchemy_greenlet_provider__r# r r r r" r7 J s r7 �_T_coT)� covariantzAwaitable[_T_co]z%TypeGuard[Coroutine[Any, Any, _T_co]])� awaitabler& c C s d S r r �r< r r r" � iscoroutineY s r> zAwaitable[Any]�Nonec C s t | �r| �� d S r )r> �closer= r r r" �_safe_cancel_awaitablea s rA c C s t � } t| dd�S )Nr9 F)r/ �getattr)�currentr r r" � in_greenleth s rD z Awaitable[_T]c C s0 t � }t|dd�s$t| � t�d��|j�| �S )z�Awaits an async function in a sync method. The sync method must be inside a :func:`greenlet_spawn` context. :func:`await_only` calls cannot be nested. :param awaitable: The coroutine to call. r9 Fzjgreenlet_spawn has not been called; can't call await_only() here. Was IO attempted in an unexpected place?)r/ rB rA r �MissingGreenlet�parentr) )r< rC r r r" � await_onlym s � rG c C sH t � }t|dd�s<t� }|�� r2t| � t�d��|�| �S |j� | �S )a� Awaits an async function in a sync method. The sync method must be inside a :func:`greenlet_spawn` context. :func:`await_fallback` calls cannot be nested. :param awaitable: The coroutine to call. .. deprecated:: 2.0.24 The ``await_fallback()`` function will be removed in SQLAlchemy 2.1. Use :func:`_util.await_only` instead, running the function / program / etc. within a top-level greenlet that is set up using :func:`_util.greenlet_spawn`. r9 Fz�greenlet_spawn has not been called and asyncio event loop is already running; can't call await_fallback() here. Was IO attempted in an unexpected place?) r/ rB �get_event_loopZ is_runningrA r rE �run_until_completerF r) )r< rC �loopr r r" �await_fallback� s � rK F)�_require_awaitzCallable[..., _T]r )r �argsrL �kwargsr& c � sz t | t� �}d}|j||�}|jsdd}z|I dH }W n" tk rV |jt�� � }Y q X |�|�}q |rv|svt� d��|S )aG Runs a sync function ``fn`` in a new greenlet. The sync function can then use :func:`await_only` to wait for async functions. :param fn: The sync callable to call. :param \*args: Positional arguments to pass to the ``fn`` callable. :param \*\*kwargs: Keyword arguments to pass to the ``fn`` callable. FTNz�The current operation required an async execution but none was detected. This will usually happen when using a non compatible DBAPI driver. Please ensure that an async DBAPI is used.) r7 r/ r) r r0 r' �sys�exc_infor Z AwaitRequired)r rL rM rN �contextZswitch_occurred�resultr( r r r" �greenlet_spawn� s �rS c @ s> e Zd Zedd�dd��Zdd�dd�Zddd d �d d �Zd S )�AsyncAdaptedLockz asyncio.Lockr. c C s t �� S r )r4 �Lock�r! r r r" �mutex� s zAsyncAdaptedLock.mutexr c C s t | j�� �S r )rK rW �acquirerV r r r" � __enter__� s zAsyncAdaptedLock.__enter__r r? )r% �kwr& c O s | j �� d S r )rW �release)r! r% rZ r r r" �__exit__� s zAsyncAdaptedLock.__exit__N)r* r+ r, r rW rY r\ r r r r" rT � s rT �asyncio.AbstractEventLoopc C s, z t �� W S tk r Y nX t �� �� S )z}vendor asyncio.get_event_loop() for python 3.7 and above. Python 3.10 deprecates get_event_loop() as a standalone. )r4 Zget_running_loop� RuntimeErrorZget_event_loop_policyrH r r r r" rH � s rH c @ s� e Zd ZU dZded< dd�dd�Zdd�d d �Zd d d dd �d d�Zdd�dd�Zdd�dd�Z ddd�dd�Z dd�dd�Z dS ) �_Runnerz#Runner implementation for test onlyz6Union[None, asyncio.AbstractEventLoop, Literal[False]]�_loopr? r. c C s d | _ d S r )r` rV r r r" r# � s z_Runner.__init__r c C s | � � | S r )� _lazy_initrV r r r" rY s z_Runner.__enter__r )�exc_type�exc_val�exc_tbr& c C s | � � d S r )r@ )r! rb rc rd r r r" r\ s z_Runner.__exit__c C s4 | j r0z| j �| j �� � W 5 | j �� d| _ X d S )NF)r` r@ rI Zshutdown_asyncgensrV r r r" r@ s � z _Runner.closer] c C s | � � | jst�| jS )zReturn embedded event loop.)ra r` �AssertionErrorrV r r r" �get_loop s z_Runner.get_loopzCoroutine[Any, Any, _T]r )�coror& c C s | � � | jst�| j�|�S r )ra r` re rI )r! rg r r r" �run s z _Runner.runc C s* | j dkrtd��| j d kr&t�� | _ d S )NFzRunner is closed)r` r^ r4 Znew_event_looprV r r r" ra s z_Runner._lazy_initN) r* r+ r, �__doc__r- r# rY r\ r@ rf rh ra r r r r" r_ � s r_ ),� __future__r r4 Z contextvarsr rO �typingr r r r r r r r Z langhelpersr � r �utilr Z util.typingr r r r r r r/ �hasattrr8 r6 r7 r: r> rA rD rG rK rS rT rH ZRunnerr_ r r r r" �<module> sV #�/