o
    h#                     @  s  d dl mZ d dlZd dlmZ d dlZd dlZd dlmZ d dlmZ d dlm	Z	 d dlm
Z
 d dlmZ d d	lmZ d d
lmZ d dlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ edZejrG dd deZdIddZnd dlmZ d dlmZ ee dZdJd!d"ZG d#d$ d$eZ ed%d&d'Z!erdKd+d,Z"nej"Z"dLd/d0Z#dMd1d2Z$dNd4d5Z%dNd6d7Z&d8d9dOd@dAZ'G dBdC dCZ(dPdEdFZ)eserej*Z+dS G dGdH dHZ+dS )Q    )annotationsN)Context)Any)	Awaitable)Callable)	Coroutine)Optional)TYPE_CHECKING)TypeVar)Union   )memoized_property   )exc)py311)Literal)Protocol)Self)	TypeGuard_Tc                   @  s<   e Zd ZU ded< ded< ddd	ZdddZdddZdS )greenletbooldeadzOptional[Context]
gr_contextfnCallable[..., Any]driverc                 C     d S N selfr   r   r   r   u/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/sqlalchemy/util/_concurrency_py3k.py__init__(       zgreenlet.__init__argr   returnc                 G  r   r   r   )r!   r%   r   r   r"   throw*      zgreenlet.throwvaluec                 C  r   r   r   )r!   r)   r   r   r"   switch-   r(   zgreenlet.switchNr   r   r   r   )r%   r   r&   r   )r)   r   r&   r   )__name__
__module____qualname____annotations__r#   r'   r*   r   r   r   r"   r   $   s   
 

r   r&   c                   C  r   r   r   r   r   r   r"   
getcurrent0   r$   r0   )r0   )r   r   eBaseExceptionr   c                 C  s   t | t pt | tjtjfS r   )
isinstance	ExceptionasyncioTimeoutErrorCancelledError)r1   r   r   r"   is_exit_exception=   s   r8   c                   @  s$   e Zd ZU ded< dZddd	Zd
S )_AsyncIoGreenletr   r   Tr   r   r   r   c                 C  s"   t | || tr|j| _d S d S r   )r   r#   _has_gr_contextr   r    r   r   r"   r#   O   s   z_AsyncIoGreenlet.__init__Nr+   )r,   r-   r.   r/    __sqlalchemy_greenlet_provider__r#   r   r   r   r"   r9   J   s   
 r9   _T_coT)	covariant	awaitableAwaitable[_T_co]%TypeGuard[Coroutine[Any, Any, _T_co]]c                 C  r   r   r   r>   r   r   r"   iscoroutineY   s   rB   Awaitable[Any]Nonec                 C  s   t | r
|   d S d S r   )rB   closerA   r   r   r"   _safe_cancel_awaitablea   s   rF   c                  C  s   t  } t| ddS )Nr;   F)r0   getattr)currentr   r   r"   in_greenleth   s   rI   Awaitable[_T]c                 C  s0   t  }t|ddst|  td|j| S )zAwaits 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.

    r;   Fzjgreenlet_spawn has not been called; can't call await_only() here. Was IO attempted in an unexpected place?)r0   rG   rF   r   MissingGreenletparentr*   )r>   rH   r   r   r"   
await_onlym   s   
	rM   c                 C  sH   t  }t|ddst }| rt|  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`.

    r;   Fzgreenlet_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?)
r0   rG   get_event_loop
is_runningrF   r   rK   run_until_completerL   r*   )r>   rH   loopr   r   r"   await_fallback   s   
rR   F)_require_awaitr   Callable[..., _T]argsr   rS   kwargsc                  s   t | t }d}|j|i |}|js6d}z|I dH }W n ty-   |jt  }Y nw ||}|jr|r?|s?t	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.
    FTNzThe 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.)
r9   r0   r*   r   r2   r'   sysexc_infor   AwaitRequired)r   rS   rU   rV   contextswitch_occurredresultr)   r   r   r"   greenlet_spawn   s$   
r]   c                   @  s.   e Zd ZedddZdddZdddZdS )AsyncAdaptedLockr&   asyncio.Lockc                 C  s   t  S r   )r5   Lockr!   r   r   r"   mutex   s   zAsyncAdaptedLock.mutexr   c                 C  s   t | j S r   )rR   rb   acquirera   r   r   r"   	__enter__   s   zAsyncAdaptedLock.__enter__r%   r   kwrD   c                 O  s   | j   d S r   )rb   release)r!   r%   re   r   r   r"   __exit__   s   zAsyncAdaptedLock.__exit__N)r&   r_   r&   r   )r%   r   re   r   r&   rD   )r,   r-   r.   r   rb   rd   rg   r   r   r   r"   r^      s
    
r^   asyncio.AbstractEventLoopc                   C  s*   zt  W S  ty   Y nw t   S )z}vendor asyncio.get_event_loop() for python 3.7 and above.

    Python 3.10 deprecates get_event_loop() as a standalone.

    )r5   get_running_loopRuntimeErrorget_event_loop_policyrN   r   r   r   r"   rN      s   
rN   c                   @  s`   e Zd ZU dZded< dddZdd	d
Zd ddZdddZd!ddZ	d"ddZ
dddZdS )#_Runnerz#Runner implementation for test onlyz6Union[None, asyncio.AbstractEventLoop, Literal[False]]_loopr&   rD   c                 C  s
   d | _ d S r   )rn   ra   r   r   r"   r#      s   
z_Runner.__init__r   c                 C  s   |    | S r   )
_lazy_initra   r   r   r"   rd      s   z_Runner.__enter__exc_typer   exc_valexc_tbc                 C  s   |    d S r   )rE   )r!   rp   rq   rr   r   r   r"   rg     s   z_Runner.__exit__c                 C  sF   | j r!z| j | j   W | j   d| _ d S | j   d| _ w d S )NF)rn   rP   shutdown_asyncgensrE   ra   r   r   r"   rE     s   


z_Runner.closeri   c                 C  s   |    | js	J | jS )zReturn embedded event loop.)ro   rn   ra   r   r   r"   get_loop  s   
z_Runner.get_loopcoroCoroutine[Any, Any, _T]r   c                 C  s   |    | js	J | j|S r   )ro   rn   rP   )r!   ru   r   r   r"   run  s   
z_Runner.runc                 C  s.   | j du r	td| j d u rt | _ d S d S )NFzRunner is closed)rn   rk   r5   new_event_loopra   r   r   r"   ro     s
   

z_Runner._lazy_initN)r&   rD   )r&   r   )rp   r   rq   r   rr   r   r&   rD   r&   ri   )ru   rv   r&   r   )r,   r-   r.   __doc__r/   r#   rd   rg   rE   rt   rw   ro   r   r   r   r"   rm      s   
 






rm   )r&   r   )r1   r2   r&   r   )r>   r?   r&   r@   )r>   rC   r&   rD   rh   )r>   rJ   r&   r   )
r   rT   rU   r   rS   r   rV   r   r&   r   ry   ),
__future__r   r5   contextvarsr   rW   typingr   r   r   r   r   r	   r
   r   langhelpersr    r   utilr   util.typingr   r   r   r   r   r   r0   hasattrr:   r8   r9   r<   rB   rF   rI   rM   rR   r]   r^   rN   Runnerrm   r   r   r   r"   <module>   sX   




#/

