o
    i/                    @  s  U d dl mZ d dlZd dlm  mZ d dlZd dl	Z	d dl
Zd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl	mZmZmZmZmZmZmZ d dlmZ d dlmZmZ d dlm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z' d dl
m(Z( d dl)m*Z*m+Z+ d d	lm,Z,m-Z- d d
l.m/Z/m0Z0 d dl1m2Z2m3Z3 d dl4m5Z5m6Z6m7Z7m8Z8 d dl9m:Z: d dlm;Z; d dl<m=Z= d dl>m?Z? d dlm@Z@mAZA d dlmBZB d dlCmDZDmEZE d dlFmGZGmHZHmIZImJZJmKZKmLZL d dlmMZM ddlNmOZOmPZPmQZQmRZRmSZS ddlTmUZUmVZVmWZW ddlXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_ ddl`maZa ddlbmcZc ddldmeZf ddldmgZh ddldmiZj ddldmkZkmlZl dd ldmmZn dd!lompZq dd"lSmrZrmsZsmtZtmuZumvZv dd#lwmxZx dd$lymzZz dd%l{m|Z|m}Z} eHrnd d&l~mZ neZejd'kr}d d(lFmZ nd d(lmZ ejd)krd d*l	mZ d d+lFmZmZ n@d dlZd dlZd dl>Z>d d,l	mZmZmZmZ d d-lmZ d d+lmZmZ G d.d/ d/ejZG d0d1 d1Zdd6d7Zdd8d9ZeKd:ZeKd;d<d=Zed>Zed?Zezd@ZdAed@< ddCdDZddHdIZeM ZdJedK< ddNdOZddRdSZG dTdU dUeqZpG dVdW dWZeM ZdXedY< G dZd[ d[eSjZejd\kr;e	jjZd]ed^< ndZG d_d` d`eSjZeeJe eJe f ZG dadb dbeBZezdcZddedc< ezdeZdfede< e/dgdhG didj djeSjZe/dgdhG dkdl dleSjZe/dgdhG dmdn dneSjZddsdtZddvdwZG dxdy dye	jZG dzd{ d{e	jZG d|d} d}eSjZG d~d dZG dd deeSjZG dd deSjtZG dd deSjtZG dd deSjZG dd deSjZG dd deeSjZG dd deeSjZezdZded< ezdZded< G dd dehZgG dd dejZiG dd denZmG dd defZeezdZded< G dd dZG dd deRZG dd deSjÃZG dd derZeZdS )    )annotationsN)AbstractEventLoopCancelledError	all_taskscreate_taskcurrent_taskget_running_loopsleep)_run_until_complete_cb)OrderedDictdeque)AsyncGeneratorAsyncIterator	AwaitableCallable
Collection	CoroutineIterableSequence)Future)AbstractContextManagersuppress)Contextcopy_context)	dataclassfield)partialwraps)CORO_RUNNINGCORO_SUSPENDEDgetcoroutinestateiscoroutine)IOBase)PathLike)Queue)Signals)AddressFamily
SocketKind)Thread)CodeTypeTracebackType)IOTYPE_CHECKINGAnyOptionalTypeVarcast)WeakKeyDictionary   )CapacityLimiterStatisticsEventStatisticsLockStatisticsTaskInfoabc)claim_worker_threadset_current_async_librarythreadlocals)BrokenResourceErrorBusyResourceErrorClosedResourceErrorEndOfStreamRunFinishedError
WouldBlockiterate_exceptions)convert_ipv6_sockaddr)create_memory_object_streamCapacityLimiterEventLock)ResourceGuardSemaphoreStatistics	SemaphoreCancelScope)AsyncBackendIPSockAddrTypeSocketListenerUDPPacketTypeUNIXDatagramPacketType)StrOrBytesPath)RunVar)MemoryObjectReceiveStreamMemoryObjectSendStream)FileDescriptorLike)   
   )	ParamSpecrZ      )Runner)TypeVarTupleUnpack)
coroutinesevents
exceptionstasks)BaseExceptionGroupc                   @  s   e Zd ZdZdZdZdS )_StatecreatedinitializedclosedN)__name__
__module____qualname__CREATEDINITIALIZEDCLOSED rq   rq   V/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/anyio/_backends/_asyncio.pyrg      s    rg   c                   @  sj   e Zd Zdddd&ddZd'd
dZd(ddZd)ddZd*ddZddd+ddZd)d d!Z	d,d$d%Z
dS )-r_   Ndebugloop_factoryrt   bool | Noneru   &Callable[[], AbstractEventLoop] | Nonec                C  s0   t j| _|| _|| _d | _d | _d| _d| _d S )Nr   F)	rg   rn   _state_debug_loop_factory_loop_context_interrupt_count_set_event_loop)selfrt   ru   rq   rq   rr   __init__   s   
zRunner.__init__returnc                 C  s   |    | S N)
_lazy_initr   rq   rq   rr   	__enter__   s   zRunner.__enter__exc_typetype[BaseException] | Noneexc_valBaseException | Noneexc_tbTracebackType | NoneNonec                 C  s   |    d S r   )closer   r   r   r   rq   rq   rr   __exit__   s   zRunner.__exit__c              	   C  s   | j }| jtjus|du rdS z5t| ||  t|dr(||  n|t	| W | j
r8td |  d| _ tj| _dS | j
rMtd |  d| _ tj| _w )zShutdown and close event loop.Nshutdown_default_executor)r{   rx   rg   ro   _cancel_all_tasksrun_until_completeshutdown_asyncgenshasattrr   _shutdown_default_executorr~   rc   set_event_loopr   rp   r   looprq   rq   rr   r      s(   



zRunner.closer   c                 C  s   |    | jS )zReturn embedded event loop.)r   r{   r   rq   rq   rr   get_loop   s   zRunner.get_loopcontextcoroCoroutine[T_Retval]T_Retvalc             
   C  sb  t |std|t durtd|   |du r!| j}|| j	j
|}t t u rVttjtju rVt| j|d}z	ttj| W n tyU   d}Y nw d}d| _z>z| j	|W W |duryttj|u rzttjtj S S S  tjy   | jdkrt|dd}|dur| dkrt w |durttj|u rttjtj w w w )z/Run a coroutine inside the embedded event loop.za coroutine was expected, got Nz7Runner.run() cannot be called from a running event loop)	main_taskr   uncancel)rb   r!   
ValueErrorrc   _get_running_loopRuntimeErrorr   r|   runr{   r   	threadingcurrent_threadmain_threadsignal	getsignalSIGINTdefault_int_handlerr   
_on_sigintr}   r   rd   r   getattrKeyboardInterrupt)r   r   r   tasksigint_handlerr   rq   rq   rr   r      sN   
	
z
Runner.runc                 C  s   | j tju r
td| j tju rd S | jd u r)t | _| j	s(t
| j d| _	n|  | _| jd ur:| j| j t | _tj| _ d S )NzRunner is closedT)rx   rg   rp   r   ro   rz   rc   new_event_loopr{   r~   r   ry   	set_debugcontextvarsr   r|   r   rq   rq   rr   r      s   




zRunner._lazy_initr   asyncio.Taskc                 C  sB   |  j d7  _ | j dkr| s|  | jdd  d S t )N   c                   S  s   d S r   rq   rq   rq   rq   rr   <lambda>  s    z#Runner._on_sigint.<locals>.<lambda>)r}   donecancelr{   call_soon_threadsafer   )r   signumframer   rq   rq   rr   r      s   zRunner._on_sigint)rt   rv   ru   rw   )r   r_   r   r   r   r   r   r   r   r   r   r   r   r   )r   r   r   r   )r   r   r   r   )rk   rl   rm   r   r   r   r   r   r   r   r   rq   rq   rq   rr   r_      s    




0r_   r   r   r   r   c                 C  st   t | }|s	d S |D ]}|  q| t j|ddi |D ]}| r&q| d ur7| d| |d qd S )Nreturn_exceptionsTz1unhandled exception during asyncio.run() shutdown)message	exceptionr   )re   r   r   r   gather	cancelledr   call_exception_handler)r   	to_cancelr   rq   rq   rr   r     s$   

r   c                   sh   d
 fdd}d _  jdu rdS   }tj||fd	}|  z|I dH  W |  dS |  w )z.Schedule the shutdown of the default executor.futureasyncio.futures.Futurer   r   c              
     sX   z j jdd  | jd  W d S  ty+ } z | j| W Y d }~d S d }~ww )NT)wait)_default_executorshutdownr   
set_result	Exceptionset_exception)r   exr   rq   rr   _do_shutdown  s   z0_shutdown_default_executor.<locals>._do_shutdownTN)targetargs)r   r   r   r   )_executor_shutdown_calledr   create_futurer   r(   startjoin)r   r   r   threadrq   r   rr   r     s   
r   r   T_contraT)contravariantPosArgsTP
_root_taskzRunVar[asyncio.Task | None]r   c                  C  s   t d } | d ur|  s| S t D ]-}|jr?| s?dd |jD }|D ]}|tu s3t|dd dkr>t | |    S q%qtt	j
t }t|}|rm|j}|rb|jd urb|j}|rb|jd usX|d urmtt	j
|jS |S )Nc                 S  s   g | ]\}}|qS rq   rq   ).0cbr   rq   rq   rr   
<listcomp>B      z"find_root_task.<locals>.<listcomp>rl   zuvloop.loop)r   getr   r   
_callbacksr
   r   setr0   asyncioTaskr   _task_statescancel_scope_parent_scope
_host_task)	root_taskr   	callbacksr   stater   rq   rq   rr   find_root_task:  s.   



r   funcr   strc                 C  s0   t | dd }t | dd }ddd ||fD S )Nrl   rm   .c                 S  s   g | ]}|r|qS rq   rq   )r   xrq   rq   rr   r   \  r   z%get_callable_name.<locals>.<listcomp>)r   r   )r   modulequalnamerq   rq   rr   get_callable_nameY  s   r   z1WeakKeyDictionary[asyncio.AbstractEventLoop, Any]	_run_varsr   boolc                 C  s   |   }d}||u}|s<td|fd||fdt v s!t|r&t|ndt|d }dd|i }tt|d }}z	t	|t
tfv W S  tyY   td|  d	dw )
zBReturn ``True`` if the task has been started and has not finished.Nis notz%(py0)s is not %(py3)sr   py0py3zassert %(py5)spy5zCannot determine if task z has started or not)get_coro
@pytest_ar_call_reprcompare@py_builtinslocals_should_repr_global_name	_safereprAssertionError_format_explanationr    r   r   AttributeErrorr   )r   r   @py_assert2@py_assert1@py_format4@py_format6rq   rq   rr   _task_startedf  s   xr  excr   c                 C  sD   	 | j rt| j d tr| j d drdS t| jtr | j} q dS )NTr   Cancelled via cancel scope F)r   
isinstancer   
startswith__context__r   r  rq   rq   rr   is_anyio_cancellationx  s   r  c                   @  s   e Zd Zejddd2dd	Zejdfd3d
dZd4ddZd5ddZe	d6ddZ
e	d6ddZd7ddZd8ddZd7d d!Zd9d:d%d&Ze	d;d'd(Zejd<d*d(Ze	d6d+d,Ze	d6d-d.Ze	d6d/d0Zejd=d1d0Zd"S )>rO   Fdeadlineshieldr  floatr  r   r   c                C  
   t | S r   object__new__clsr  r  rq   rq   rr   r    s   
zCancelScope.__new__c                 C  sj   || _ || _d | _t | _d| _d | _d| _d| _d | _	d | _
t | _d | _tjdkr0d| _d S d | _d S )NFr]   r   )	_deadline_shieldr   r   _child_scopes_cancel_called_cancel_reason_cancelled_caught_active_timeout_handle_cancel_handle_tasksr   sysversion_info_pending_uncancellations)r   r  r  rq   rq   rr   r     s   


zCancelScope.__init__c                 C  s   | j rtdttjt  | _}| j| zt	| }W n t
y/   td | }|t	|< Y nw |j| _| |_| jd urJ| jj|  | jj| |   d| _ | jrY| |  | S )Nz;Each CancelScope may only be used for a single 'with' blockT)r$  r   r0   r   r   r   r   r'  addr   KeyError	TaskStater   r   r   discard_timeoutr!  _deliver_cancellation)r   	host_task
task_staterq   rq   rr   r     s,   


zCancelScope.__enter__r   r   r   r   r   r   c                 C  s"  ~| j stdt | jurtd| j}d }||u}|sQtd|fd||fdt v s2t| r7t	| ndt	|t	|d }dd|i }t
t|d  } }}t| j}	|	d u sf|	j| urjtd	z d
| _ | jrz| j  d | _| j| j | jd ur| jj|  | jj| j | j|	_|   | jr| js| jr| j  |  jd8  _| jsd
}
|d urt|D ]}t|trt|rd| _qd}
q| jo|
 W d | _~S | jr| j}d }||u}|s"td|fd||fdt v st| rt	| ndt	|t	|d }dd|i }t
t|d  } }}| j}|j}d }||u}|sqtd|fd||fdt v sNt| rSt	| ndt	|t	|t	|d }dd|i }t
t|d  } } }}| j j| j7  _d| _W d | _~d
S d | _~w )NzThis cancel scope is not activezIAttempted to exit cancel scope in a different task than it was entered inr   )z6%(py2)s
{%(py2)s = %(py0)s._host_task
} is not %(py5)sr   r   py2r   assert %(py7)spy7zTAttempted to exit a cancel scope that isn't the current tasks's current cancel scopeFr   T)z9%(py2)s
{%(py2)s = %(py0)s._parent_scope
} is not %(py5)s)zg%(py4)s
{%(py4)s = %(py2)s
{%(py2)s = %(py0)s._parent_scope
}._pending_uncancellations
} is not %(py7)s)r   r4  py4r6  zassert %(py9)spy9r   ) r$  r   r   r   r   r   r  r  r  r  r  r  r   r   r   r%  r   r'  remover   r   r+  _restart_cancellation_in_parentr!  %_parent_cancellation_is_visible_to_usr*  r   rA   r  r   r  r#  )r   r   r   r   r	  @py_assert4@py_assert3r  @py_format8host_task_statecannot_swallow_exc_valr  @py_assert6@py_assert5@py_format10rq   rq   rr   r     sh   


zCancelScope.__exit__c                 C  s2   | }|d ur|j rdS |jrdS |j}|d usdS )NTF)r!  r  r   )r   r   rq   rq   rr   _effectively_cancelled  s   	z"CancelScope._effectively_cancelledc                 C  s   | j d uo| j o| j jS r   )r   r  rD  r   rq   rq   rr   r;    s
   
z1CancelScope._parent_cancellation_is_visible_to_usr   c                 C  sH   | j tjkr"t }| | j kr| d d S || j | j| _d S d S )Nzdeadline exceeded)	r  mathinfr   timer   call_atr/  r%  r   rq   rq   rr   r/  $  s   zCancelScope._timeoutoriginc                 C  s   d}t  }| jD ]9}d}|jrq||urA|| ju st|rA|j}t|tjr*|	 sA|
|j ||ju rA|jdurA| jd7  _q| jD ]}|jsT|jsT||pS|}qE|| u ri|rft | j|| _|S d| _|S )aZ  
        Deliver cancellation to directly contained tasks and nested cancel scopes.

        Schedule another run at the end if we still have tasks eligible for
        cancellation.

        :param origin: the cancel scope that originated the cancellation
        :return: ``True`` if the delivery needs to be retried on the next cycle

        FTNr   )r   r'  _must_cancelr   r  _fut_waiterr  r   r   r   r   r"  r*  r   r  cancel_calledr0  r   	call_soonr&  )r   rI  should_retrycurrentr   waiterscoperq   rq   rr   r0  ,  s4   



z!CancelScope._deliver_cancellationc                 C  sL   | j }|dur$|jr|jdu r|| dS |jrdS |j }|dusdS dS )zb
        Restart the cancellation effort in the closest directly cancelled parent scope.

        N)r   r!  r&  r0  r  r   rQ  rq   rq   rr   r:  Z  s   

z+CancelScope._restart_cancellation_in_parentNreason
str | Nonec                 C  s   | j sA| jr| j  d | _d| _ dt| d| _t  }r)|  jd| 7  _|r5|  jd| 7  _| jd urC| |  d S d S d S )NTr  r   z by z
; reason: )r!  r%  r   idr"  r   r   r0  )r   rS  r   rq   rq   rr   r   m  s   


zCancelScope.cancelc                 C     | j S r   )r  r   rq   rq   rr   r  ~     zCancelScope.deadlinevaluec                 C  sD   t || _| jd ur| j  d | _| jr| js |   d S d S d S r   )r  r  r%  r   r$  r!  r/  r   rX  rq   rq   rr   r    s   


c                 C  rV  r   )r!  r   rq   rq   rr   rL    rW  zCancelScope.cancel_calledc                 C  rV  r   )r#  r   rq   rq   rr   cancelled_caught  rW  zCancelScope.cancelled_caughtc                 C  rV  r   )r  r   rq   rq   rr   r    rW  zCancelScope.shieldc                 C  s(   | j |kr|| _ |s|   d S d S d S r   )r  r:  rY  rq   rq   rr   r    s   
r  r  r  r   r   rO   )r  r  r  r   )r   rO   r   r   r   r   r   r   r   r   r   r   r   )rI  rO   r   r   r   )rS  rT  r   r   r   r  rX  r  r   r   )rX  r   r   r   )rk   rl   rm   rE  rF  r  r   r   r   propertyrD  r;  r/  r0  r:  r   r  setterrL  rZ  r  rq   rq   rq   rr   rO     s4    

L


.	rO   c                   @  s   e Zd ZdZdZd
ddZd	S )r-  z
    Encapsulates auxiliary task information that cannot be added to the Task instance
    itself because there are no guarantees about its implementation.
    )	parent_idr   __weakref__rb  
int | Noner   CancelScope | Nonec                 C     || _ || _d S r   rb  r   )r   rb  r   rq   rq   rr   r        
zTaskState.__init__N)rb  rd  r   re  )rk   rl   rm   __doc__	__slots__r   rq   rq   rq   rr   r-    s    r-  z*WeakKeyDictionary[asyncio.Task, TaskState]r   c                   @  s"   e Zd ZdddZddddZdS )_AsyncioTaskStatusr   asyncio.Futurerb  intc                 C  rf  r   )_future
_parent_id)r   r   rb  rq   rq   rr   r     rh  z_AsyncioTaskStatus.__init__NrX  T_contra | Noner   r   c                 C  sX   z| j | W n tjy   | j  stdd Y nw ttjt }| j	t
| _d S )Nz.called 'started' twice on the same task status)rn  r   r   InvalidStateErrorr   r   r0   r   r   ro  r   rb  )r   rX  r   rq   rq   rr   started  s   
z_AsyncioTaskStatus.started)r   rl  rb  rm  r   )rX  rp  r   r   )rk   rl   rm   r   rr  rq   rq   rq   rr   rk    s    
rk  rZ      zCodeType | None_eager_task_factory_codec                   @  sX   e Zd Zd$ddZd%ddZd&ddZ	d'd(ddZddd)ddZddd*d"d#ZdS )+	TaskGroupr   r   c                 C  s&   t  | _d| _g | _t | _d | _d S )NF)rO   r   r$  _exceptionsr   r'  _on_completed_futr   rq   rq   rr   r     s
   
zTaskGroup.__init__c                   s   | j   d| _| S NT)r   r   r$  r   rq   rq   rr   
__aenter__  s   
zTaskGroup.__aenter__r   r   r   r   r   r   r   c                   s  z|d ur| j   t|ts| j| t }zq| jrqt H}| jra|	 | _
z| j
I d H  W n) tyZ } zd|_| j   |d u sNt|trPt|sP|}W Y d }~nd }~ww d | _
| js$W d    n1 skw   Y  nt I d H  d| _| jrtd| jd |r|W n$ ty } z| j t|||jrW Y d }~W ~~| `dS  d }~ww | j |||W ~~| `S ~~| `w )NTFzunhandled errors in a TaskGroup)r   r   r  r   rw  appendr   r'  rO   r   rx  r  r  AsyncIOBackendcancel_shielded_checkpointr$  rf   BaseExceptionr   type__traceback__)r   r   r   r   r   
wait_scoper  rq   rq   rr   	__aexit__  sd   



zTaskGroup.__aexit__Nr   ,Callable[[Unpack[PosArgsT]], Awaitable[Any]]r   tuple[Unpack[PosArgsT]]namer  task_status_futureasyncio.Future | Noner   c                   st  d fdd} j stdi }r%tt }tt jj|d< nt jj}||i |}t|sQt|d	rA|j	 d
nd}	t
d|	 |j d|d|d u rYt|nt|}t }
|
  }rt|dd tu rt|dd  }r|d j}|||
|dnt||dt| jdt<  jj  j tjdkr jjd urt jj | S )N_taskr   r   r   c              
     s  t jdkr jjd urt|  jj t|  }|j}d }||u}|sVtd|fd||fdt	
 v s7t|r<t|ndt|t|d }dd|i }tt|d  } }}|j}|j}| |v }|std|fd	| |fd
t	
 v s}t| rt| nd
dt	
 v st|rt|ndt|t|d }dd|i }	tt|	d  } }}|jj|   j t| =  jd urވ jsz jd  W n
 tjy   Y nw z|  }
W n$ ty } zt|jtr|j}t|jts|}
W Y d }~nd }~ww |
d urEd ur rd S d u s$ r>t|
ts0 j|
  jjs< j  d S d S |
 d S d urX sZtd d S d S d S )NrZ      r   )z8%(py2)s
{%(py2)s = %(py0)s.cancel_scope
} is not %(py5)sr2  r3  r5  r6  )in)zP%(py0)s in %(py6)s
{%(py6)s = %(py4)s
{%(py4)s = %(py2)s.cancel_scope
}._tasks
}r  )r   r4  r7  py6zassert %(py8)spy8z2Child exited without calling task_status.started()) r(  r)  r   r   r   future_discard_from_awaited_byr   r   r   r  r  r  r  r  r  r'  r9  rx  r   rq  r   r   r  r  r   r   rw  r{  rD  r   r   r   )r  r2  r	  r<  r=  r  r>  rB  @py_format7@py_format9r  er   r   r  rq   rr   	task_done%  sP   

z#TaskGroup._spawn.<locals>.task_donez;This task group is not active; no new tasks can be started.task_statusrl   r    z	Expected z0() to return a coroutine, but the return value (z) is not a coroutine object__code____closure__r   )r   r  r  rg  r  )r  r   r   r   )r$  r   rU  r   rk  r   r   r!   r   rl   	TypeErrorrm   r   r   r   r   get_task_factoryr   ru  cell_contentsr   r-  r   r'  r+  r(  r)  future_add_to_awaited_byadd_done_callback)r   r   r   r  r  r  kwargsrb  r   prefixr   factoryclosurecustom_task_constructorrq   r  rr   _spawn  sN   /





zTaskGroup._spawnr  Unpack[PosArgsT]c                G  s   |  ||| d S r   )r  )r   r   r  r   rq   rq   rr   
start_soon  s   zTaskGroup.start_soonCallable[..., Awaitable[Any]]r-   c                  s   t  }| ||||}z|I d H W S  tyT   |  tdd( tt |I d H  W d    n1 s8w   Y  W d     W d     1 sNw   Y   w )NTr  )r   r   r  r   r   rO   r   )r   r   r  r   r   r   rq   rq   rr   r     s    (zTaskGroup.startr   )r   rv  r\  r   )
r   r  r   r  r  r  r  r  r   r   )r   r  r   r  r  r  r   r   )r   r  r   r  r  r  r   r-   )	rk   rl   rm   r   rz  r  r  r  r   rq   rq   rq   rr   rv    s    


Dh	rv  c                      sB   e Zd ZdZd fdd	ZdddZdddZddddZ  ZS ) WorkerThreadr[   r   r   workersset[WorkerThread]idle_workersdeque[WorkerThread]c                   sF   t  jdd || _|| _|| _|j| _td| _t	
 | _d| _d S )NzAnyIO worker threadr  r2   F)superr   r   r  r  r{   r   r$   queuer|  current_time
idle_sincestopping)r   r   r  r  	__class__rq   rr   r     s   

zWorkerThread.__init__r   rl  resultr-   r  r   r   r   c                 C  sh   t  | _| js| j|  | s2|d ur+t|tr$t	d}||_
|}|| d S || d S d S )Nzcoroutine raised StopIteration)r|  r  r  r  r  r{  r   r  StopIterationr   	__cause__r   r   )r   r   r  r  new_excrq   rq   rr   _report_result  s   

zWorkerThread._report_resultc           
      C  s   t t| jk 	 | j }|d u r	 W d    d S |\}}}}}| sed }d }|t_z$z|j|g|R  }W n t	yL }	 z|	}W Y d }	~	nd }	~	ww W t`nt`w | j
 sc| j| j||| ~~| j  ~~~~~~q1 stw   Y  d S r   )r8   r|  r   r  r   r   r:   current_cancel_scoper   r~  	is_closedr   r  r  )
r   itemr   r   r   r   r   r  r   r  rq   rq   rr   r     s8   



zWorkerThread.runNfasyncio.Task | Nonec                 C  sF   d| _ | jd  | j|  z	| j|  W d S  ty"   Y d S w ry  )r  r  
put_nowaitr  r.  r  r9  r   )r   r  rq   rq   rr   stop  s   zWorkerThread.stop)r   r   r  r  r  r  )r   rl  r  r-   r  r   r   r   r   r   )r  r  r   r   )	rk   rl   rm   MAX_IDLE_TIMEr   r  r   r  __classcell__rq   rq   r  rr   r    s    

r  _threadpool_idle_workerszRunVar[deque[WorkerThread]]_threadpool_workerszRunVar[set[WorkerThread]]F)eqc                   @  s,   e Zd ZU ded< dddd	ZdddZdS )StreamReaderWrapperzasyncio.StreamReader_stream   	max_bytesrm  r   bytesc                   s    | j |I d H }|r|S tr   )r  readr>   )r   r  datarq   rq   rr   receive  s
   zStreamReaderWrapper.receiver   c                   s"   | j t  t I d H  d S r   )r  r   r=   r|  
checkpointr   rq   rq   rr   aclose
  s   zStreamReaderWrapper.acloseNr  r  rm  r   r  r   )rk   rl   rm   __annotations__r  r  rq   rq   rq   rr   r    s   
 r  c                   @  s>   e Zd ZU ded< edddZded< dddZdddZdS )StreamWriterWrapperzasyncio.StreamWriterr  F)initdefaultr   _closedr  r  r   r   c              
     s   t  I d H  | jjj}z| j| | j I d H  W n ttt	fy; } z| j
r.t|| j r6t| d }~ww |sGt  I d H  d S d S r   )r|  checkpoint_if_cancelledr  	_protocol_pausedwritedrainConnectionResetErrorBrokenPipeErrorr   r  r=   
is_closingr;   r}  )r   r  stream_pausedr  rq   rq   rr   send  s"   

zStreamWriterWrapper.sendc                   s$   d| _ | j  t I d H  d S ry  )r  r  r   r|  r  r   rq   rq   rr   r  (  s   
zStreamWriterWrapper.acloseNr  r  r   r   r   )rk   rl   rm   r  r   r  r  r  rq   rq   rq   rr   r    s
   
 
r  c                   @  s   e Zd ZU ded< ded< ded< ded< d$d
dZd%ddZd$ddZd$ddZd&ddZe	d%ddZ
e	d'ddZe	d(ddZe	d)dd Ze	d)d!d"Zd#S )*Processzasyncio.subprocess.Process_processzStreamWriterWrapper | None_stdinzStreamReaderWrapper | None_stdout_stderrr   r   c              	     s   t ddM}| jr| j I d H  | jr| j I d H  | jr(| j I d H  d|_z	|  I d H  W n tyJ   d|_|   |  I d H   w W d    d S 1 sVw   Y  d S )NTr  F)	rO   r  r  r  r  r  r   r~  killrR  rq   rq   rr   r  5  s&   "zProcess.acloserm  c                   s   | j  I d H S r   )r  r   r   rq   rq   rr   r   G  s   zProcess.waitc                 C     | j   d S r   )r  	terminater   rq   rq   rr   r  J     zProcess.terminatec                 C  r  r   )r  r  r   rq   rq   rr   r  M  r  zProcess.killr   c                 C  s   | j | d S r   )r  send_signal)r   r   rq   rq   rr   r  P     zProcess.send_signalc                 C     | j jS r   )r  pidr   rq   rq   rr   r  S     zProcess.pidrd  c                 C  r  r   )r  
returncoder   rq   rq   rr   r  W  r  zProcess.returncodeabc.ByteSendStream | Nonec                 C  rV  r   )r  r   rq   rq   rr   stdin[  rW  zProcess.stdinabc.ByteReceiveStream | Nonec                 C  rV  r   )r  r   rq   rq   rr   stdout_  rW  zProcess.stdoutc                 C  rV  r   )r  r   rq   rq   rr   stderrc  rW  zProcess.stderrNr   r   rm  )r   rm  r   r   r   rd  )r   r  )r   r  )rk   rl   rm   r  r  r   r  r  r  r`  r  r  r  r  r  rq   rq   rq   rr   r  .  s(   
 




r  r  set[Process]r  r  c                 C  s   d}t jdk rzt  }W n	 ty   Y nw |  D ])}|jdu r%q|jj	j
  |jj	j
  |jj	j
  |  |rF||j qdS )zG
    Forcibly shuts down worker processes belonging to this event loop.Nrs  )r(  r)  r   get_event_loop_policyget_child_watcherNotImplementedErrorcopyr  r  r  
_transportr   r  r  r  remove_child_handlerr  )r  r  child_watcherprocessrq   rq   rr   '_forcibly_shutdown_process_pool_on_exith  s$   

r   set[abc.Process]c                   sl   zt tjI dH  W dS  tjy5   |  } | D ]}|jdu r%|  q| D ]	}| I dH  q(Y dS w )z
    Shuts down worker processes belonging to this event loop.

    NOTE: this only works when the event loop was started using asyncio.run() or
    anyio.run().

    N)	r	   rE  rF  r   r   r  r  r  r  )r  r  rq   rq   rr   _shutdown_process_pool_on_exit  s   	
r  c                   @  sz   e Zd ZU ded< ded< ded< dZded< 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S )%StreamProtocolzdeque[bytes]
read_queueasyncio.Event
read_eventwrite_eventNException | Noner   Fr   	is_at_eof	transportasyncio.BaseTransportr   r   c                 C  s<   t  | _t | _t | _| j  ttj|	d d S Nr   )
r   r  r   rG   r  r  r   r0   	Transportset_write_buffer_limitsr   r
  rq   rq   rr   connection_made  s
   


zStreamProtocol.connection_mader  c                 C  s,   |r
t  | _|| j_| j  | j  d S r   )r;   r   r  r  r   r  r   r  rq   rq   rr   connection_lost  s
   
zStreamProtocol.connection_lostr  r  c                 C  s   | j t| | j  d S r   )r  r{  r  r  r   )r   r  rq   rq   rr   data_received  s   zStreamProtocol.data_receivedrv   c                 C  s   d| _ | j  dS ry  )r	  r  r   r   rq   rq   rr   eof_received  s   
zStreamProtocol.eof_receivedc                 C     t  | _d S r   )r   rG   r  r   rq   rq   rr   pause_writing  r  zStreamProtocol.pause_writingc                 C  r  r   r  r   r   rq   rq   rr   resume_writing  r  zStreamProtocol.resume_writingr
  r  r   r   r  r  r   r   )r  r  r   r   )r   rv   r   )rk   rl   rm   r  r   r	  r  r  r  r  r  r  rq   rq   rq   rr   r    s   
 




r  c                   @  sn   e Zd ZU ded< ded< ded< 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S )$DatagramProtocolz#deque[tuple[bytes, IPSockAddrType]]r  r  r  r  Nr  r   r
  r  r   r   c                 C  s.   t dd| _t | _t | _| j  d S )Nd   )maxlen)r   r  r   rG   r  r  r   r  rq   rq   rr   r    s   

z DatagramProtocol.connection_mader  c                 C  s   | j   | j  d S r   )r  r   r  r  rq   rq   rr   r    s   
z DatagramProtocol.connection_lostr  r  addrrQ   c                 C  s&   t |}| j||f | j  d S r   )rB   r  r{  r  r   )r   r  r  rq   rq   rr   datagram_received  s   z"DatagramProtocol.datagram_receivedr   c                 C  s
   || _ d S r   )r   r  rq   rq   rr   error_received     
zDatagramProtocol.error_receivedc                 C  r  r   )r  clearr   rq   rq   rr   r    r  zDatagramProtocol.pause_writingc                 C  r  r   r  r   rq   rq   rr   r    r  zDatagramProtocol.resume_writingr  r  )r  r  r  rQ   r   r   )r  r   r   r   r   )rk   rl   rm   r  r   r  r  r  r   r  r  rq   rq   rq   rr   r    s   
 




r  c                   @  sN   e Zd ZdddZedd	d
ZddddZdddZdddZdddZ	dS ) SocketStreamr
  asyncio.Transportprotocolr  c                 C  *   || _ || _td| _td| _d| _d S Nreading from
writing toFr  r  rJ   _receive_guard_send_guardr  r   r
  r%  rq   rq   rr   r     s
   


zSocketStream.__init__r   socket.socketc                 C     | j dS Nsocketr  get_extra_infor   rq   rq   rr   _raw_socket     zSocketStream._raw_socketr  r  rm  r  c              	     s  | j  | jj s(| j s(| jjs(| j  | jj I d H  | j	  nt
 I d H  z| jj }W n tyP   | jrDtd | jjrM| jjd td w t||krk|d | ||d  }}| jj| | jjs}| jj  W d    |S W d    |S 1 sw   Y  |S r   )r+  r  r  is_setr  r  r	  resume_readingr   pause_readingr|  r  r  popleft
IndexErrorr  r=   r   r>   len
appendleftr"  )r   r  chunkleftoverrq   rq   rr   r    sB   



 
  zSocketStream.receiver  r   c                   s   | j F t I d H  | jrt| jjd ur| jjz| j| W n t	y8 } z	| j
 r3t| d }~ww | jj I d H  W d    d S 1 sMw   Y  d S r   )r,  r|  r  r  r=   r  r   r  r  r   r  r;   r  r   )r   r  r  rq   rq   rr   r    s"   
"zSocketStream.sendc                   s(   z| j   W d S  ty   Y d S w r   )r  	write_eofOSErrorr   rq   rq   rr   send_eof#  s   zSocketStream.send_eofc                   s^   d| _ | j s-z| j  W n	 ty   Y nw | j  tdI d H  | j  d S d S NTr   )r  r  r  r?  r@  r   r	   abortr   rq   rq   rr   r  )  s   

zSocketStream.acloseN)r
  r$  r%  r  r   r.  r  r  r  r   )
rk   rl   rm   r   r`  r4  r  r  rA  r  rq   rq   rq   rr   r#    s    

#
r#  c                   @  s`   e Zd ZU dZded< dZded< dZddd	ZedddZ	dddZ
dddZdddZdS )_RawSocketMixinNr  _receive_future_send_futureF
raw_socketr.  c                 C  s   || _ td| _td| _d S )Nr(  r)  )_RawSocketMixin__raw_socketrJ   r+  r,  r   rH  rq   rq   rr   r   ;  s   
z_RawSocketMixin.__init__r   c                 C  rV  r   )rI  r   rq   rq   rr   r4  @  rW  z_RawSocketMixin._raw_socketr   asyncio.AbstractEventLooprl  c                   >   d fdd}t   }_ j|jd  || |S )Nr  r  r   r   c                      `  j d S r   )rF  remove_readerrI  r  r   r   rq   rr   callbackE     z6_RawSocketMixin._wait_until_readable.<locals>.callbackr  r  r   r   )r   r   rF  
add_readerrI  r   r  r   r   rQ  r  rq   rP  rr   _wait_until_readableD  
   
z$_RawSocketMixin._wait_until_readablec                   rL  )Nr  r  r   r   c                   rM  r   )rG  remove_writerrI  rO  rP  rq   rr   rQ  O  rR  z6_RawSocketMixin._wait_until_writable.<locals>.callbackrS  )r   r   rG  
add_writerrI  r   r  rU  rq   rP  rr   _wait_until_writableN  rW  z$_RawSocketMixin._wait_until_writabler   c                   sV   | j s'd| _ | j dkr| j  | jr| jd  | jr)| jd  d S d S d S )NT)_closingrI  filenor   rF  r   rG  r   rq   rq   rr   r  X  s   
z_RawSocketMixin.acloserH  r.  rD  )r   rK  r   rl  r   )rk   rl   rm   rF  r  rG  r\  r   r`  r4  rV  rZ  r  rq   rq   rq   rr   rE  6  s   
 




rE  c                   @  s@   e Zd ZdddZddd	d
ZdddZdddZdddZdS )UNIXSocketStreamr   r   c                   s<   | j  | jtj W d    d S 1 sw   Y  d S r   )r,  r4  r   r1  SHUT_WRr   rq   rq   rr   rA  e  s   "zUNIXSocketStream.send_eofr  r  rm  r  c                   s   t  }t I d H  | j> 	 z| j|}W n% ty)   | |I d H  Y n" ty= } z	| j	r6t
d t|d }~ww |sBt|W  d    S q1 sOw   Y  d S r   )r   r|  r  r+  r4  recvBlockingIOErrorrV  r@  r\  r=   r;   r>   )r   r  r   r  r  rq   rq   rr   r  i  s*   zUNIXSocketStream.receiver  c                   s   t  }t I d H  | jM t|}|rSz| j|}W n% ty.   | |I d H  Y n t	yB } z	| j
r;td t|d }~ww ||d  }|sW d    d S W d    d S 1 s^w   Y  d S r   )r   r|  r  r,  
memoryviewr4  r  rb  rZ  r@  r\  r=   r;   )r   r  r   view
bytes_sentr  rq   rq   rr   r  }  s*   "zUNIXSocketStream.sendmsglenmaxfdstuple[bytes, list[int]]c                   sl  t |tr
|dk rtdt |tr|dk rtdt }td}t I d H  | jJ 	 z| j	|t
||j \}}}}W n% tyS   | |I d H  Y n tyg }	 z	| jr`td t|	d }	~	ww |sn|sntnq/W d    n1 szw   Y  |D ].\}
}}|
t
jks|t
jkrtd|d|
 d	| ||d t|t||j    q|t|fS )
Nr   z%msglen must be a non-negative integerr   z!maxfds must be a positive integeriTz.Received unexpected ancillary data; message = z, cmsg_level = z, cmsg_type = )r  rm  r   r   arrayr|  r  r+  r4  recvmsgr1  CMSG_LENitemsizerb  rV  r@  r\  r=   r;   r>   
SOL_SOCKET
SCM_RIGHTSr   	frombytesr;  list)r   rf  rg  r   fdsr   ancdataflagsr  r  
cmsg_level	cmsg_type	cmsg_datarq   rq   rr   receive_fds  sN   
&zUNIXSocketStream.receive_fdsr   rr  Collection[int | IOBase]c                   s  |st d|st dt }g }|D ]}t|tr!|| qt|tr-||  qtd|}t	 I d H  | j
A 	 z| j|gtjtj|fg W n& tya   | |I d H  Y n tyu } z	| jrntd t|d }~ww q@W d    d S 1 sw   Y  d S )Nzmessage must not be emptyzfds must not be emptyri  )r   r   r  rm  r{  r"   r]  rj  r|  r  r,  r4  sendmsgr1  rn  ro  rb  rZ  r@  r\  r=   r;   )r   r   rr  r   filenosfdfdarrayr  rq   rq   rr   send_fds  sB   

"zUNIXSocketStream.send_fdsNr   r  r  r  )rf  rm  rg  rm  r   rh  )r   r  rr  ry  r   r   )rk   rl   rm   rA  r  r  rx  r~  rq   rq   rq   rr   r_  d  s    


'r_  c                   @  sJ   e Zd ZU dZded< dZdddZedd
dZdddZ	dddZ
dS )TCPSocketListenerNre  _accept_scopeFrH  r.  c                 C  s$   || _ ttjt | _td| _d S )Naccepting connections from)_TCPSocketListener__raw_socketr0   r   BaseEventLoopr   r{   rJ   _accept_guardrJ  rq   rq   rr   r     s   zTCPSocketListener.__init__r   c                 C  rV  r   )r  r   rq   rq   rr   r4    rW  zTCPSocketListener._raw_socketabc.SocketStreamc                   s  | j rt| j` t I d H  t F| _z8z| j| j	I d H \}}W n$ t
jyI   z	| j| j	 W n ttfyA   Y nw | j rHtd  w W d | _nd | _w W d    n1 s]w   Y  W d    n1 slw   Y  |tjtjd | jt|I d H \}}t||S Nr   )r  r=   r  r|  r  rO   r  r{   sock_acceptr4  r   r   rN  r   r  
setsockoptr1  IPPROTO_TCPTCP_NODELAYconnect_accepted_socketr  r#  )r   client_sock_addrr
  r%  rq   rq   rr   accept  s8   

zTCPSocketListener.acceptr   c              	     sh   | j rd S d| _ | jr-z	| j| j W n ttfy    Y nw | j  tdI d H  | j	  d S rB  )
r  r  r{   rN  r4  r   r  r   r	   r   r   rq   rq   rr   r    s   
zTCPSocketListener.acloser^  rD  r   r  r   )rk   rl   rm   r  r  r  r   r`  r4  r  r  rq   rq   rq   rr   r    s   
 

r  c                   @  s8   e Zd ZdddZdddZdd
dZedddZdS )UNIXSocketListenerrH  r.  c                 C  s"   || _ t | _td| _d| _d S )Nr  F)_UNIXSocketListener__raw_socketr   r{   rJ   r  r  rJ  rq   rq   rr   r     s   

zUNIXSocketListener.__init__r   r  c                   s   t  I d H   jU 	 z j \}}|d t|W W  d    S  tyJ   t	 } j
 j|jd  | fdd |I d H  Y n ty^ } z	 jrWtd t|d }~ww q1 scw   Y  d S )NTFc                   s    j  jS r   )r{   rN  r  _r   rq   rr   r   '  s    z+UNIXSocketListener.accept.<locals>.<lambda>)r|  r  r  r  r  setblockingr_  rb  r   r   r{   rT  r   r  r@  r  r=   r;   )r   r  r  r  r  rq   r   rr   r    s0   

zUNIXSocketListener.acceptr   c                   s   d| _ | j  d S ry  )r  r  r   r   rq   rq   rr   r  0  s   zUNIXSocketListener.aclosec                 C  rV  r   )r  r   rq   rq   rr   r4  4  rW  zUNIXSocketListener._raw_socketNr^  r  r   rD  )rk   rl   rm   r   r  r  r`  r4  rq   rq   rq   rr   r    s    


r  c                   @  sB   e Zd ZdddZedd	d
ZdddZdddZdddZdS )	UDPSocketr
  asyncio.DatagramTransportr%  r  c                 C  r&  r'  r*  r-  rq   rq   rr   r   :  
   


zUDPSocket.__init__r   r.  c                 C  r/  r0  r2  r   rq   rq   rr   r4  C  r5  zUDPSocket._raw_socketr   c                   $   d| _ | j s| j  d S d S ry  r  r  r  r   r   rq   rq   rr   r  G  
   
zUDPSocket.aclosetuple[bytes, IPSockAddrType]c              	     s   | j ? t I d H  | jjs$| j s$| jj  | jj	 I d H  z| jj
 W W  d    S  tyB   | jr?td td w 1 sFw   Y  d S r   r+  r|  r  r  r  r  r  r  r"  r   r9  r:  r  r=   r;   r   rq   rq   rr   r  L  s   
zUDPSocket.receiver  rS   c                   sr   | j + t I d H  | jj I d H  | jrt| j	 r!t
| jj|  W d    d S 1 s2w   Y  d S r   r,  r|  r  r  r  r   r  r=   r  r  r;   sendtor   r  rq   rq   rr   r  ]     
"zUDPSocket.sendNr
  r  r%  r  rD  r   )r   r  )r  rS   r   r   	rk   rl   rm   r   r`  r4  r  r  r  rq   rq   rq   rr   r  9  s    
	

r  c                   @  sB   e Zd ZdddZedd	d
ZdddZdddZdddZdS )ConnectedUDPSocketr
  r  r%  r  c                 C  r&  r'  r*  r-  rq   rq   rr   r   j  r  zConnectedUDPSocket.__init__r   r.  c                 C  r/  r0  r2  r   rq   rq   rr   r4  s  r5  zConnectedUDPSocket._raw_socketr   c                   r  ry  r  r   rq   rq   rr   r  w  r  zConnectedUDPSocket.acloser  c              	     s   | j D t I d H  | jjs$| j s$| jj  | jj	 I d H  z| jj
 }W n ty<   | jr9td td w |d W  d    S 1 sKw   Y  d S r  r  )r   packetrq   rq   rr   r  |  s   $zConnectedUDPSocket.receiver  c                   sr   | j + t I d H  | jj I d H  | jrt| j	 r!t
| j| W d    d S 1 s2w   Y  d S r   r  r  rq   rq   rr   r    r  zConnectedUDPSocket.sendNr  rD  r   r   r  r  r  rq   rq   rq   rr   r  i  s    
	

r  c                   @      e Zd Zd
ddZdddZd	S )UNIXDatagramSocketr   rT   c                      t  }t I d H  | j: 	 z| jd}W n% ty)   | |I d H  Y n ty= } z	| j	r6t
d t|d }~ww |W  d    S q1 sKw   Y  d S NTr  )r   r|  r  r+  r4  recvfromrb  rV  r@  r\  r=   r;   r   r   r  r  rq   rq   rr   r    &   zUNIXDatagramSocket.receiver  r   c                   s   t  }t I d H  | j: 	 z| jj|  W n% ty)   | |I d H  Y n ty= } z	| j	r6t
d t|d }~ww 	 W d    d S q1 sKw   Y  d S r   )r   r|  r  r,  r4  r  rb  rZ  r@  r\  r=   r;   r   r  r   r  rq   rq   rr   r    r  zUNIXDatagramSocket.sendN)r   rT   )r  rT   r   r   rk   rl   rm   r  r  rq   rq   rq   rr   r        
r  c                   @  r  )ConnectedUNIXDatagramSocketr   r  c                   r  r  )r   r|  r  r+  r4  ra  rb  rV  r@  r\  r=   r;   r  rq   rq   rr   r    r  z#ConnectedUNIXDatagramSocket.receiver  r   c                   s   t  }t I d H  | j: 	 z| j| W n% ty)   | |I d H  Y n ty= } z	| j	r6t
d t|d }~ww 	 W d    d S q1 sKw   Y  d S r   )r   r|  r  r,  r4  r  rb  rZ  r@  r\  r=   r;   r  rq   rq   rr   r    r  z ConnectedUNIXDatagramSocket.sendNr  r  r  rq   rq   rq   rr   r    r  r  read_eventsz'RunVar[dict[int, asyncio.Future[bool]]]_read_eventswrite_events_write_eventsc                   @  sH   e Z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 )rG   r   c                 C  r  r   r  r  rq   rq   rr   r    r!  zEvent.__new__r   c                 C  r  r   )r   rG   _eventr   rq   rq   rr   r     r  zEvent.__init__c                 C  r  r   )r  r   r   rq   rq   rr   r     r  z	Event.setr   c                 C  
   | j  S r   )r  r6  r   rq   rq   rr   r6    r!  zEvent.is_setc                   s0   |   rt I d H  d S | j I d H  d S r   )r6  r|  r  r  r   r   rq   rq   rr   r     s   z
Event.waitr4   c                 C  s   t t| jjS r   )r4   r;  r  _waitersr   rq   rq   rr   
statistics  r  zEvent.statisticsN)r   rG   r   r]  )r   r4   )	rk   rl   rm   r  r   r   r6  r   r  rq   rq   rq   rr   rG     s    




rG   c                   @  s^   e Zd ZdddddZd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S )rI   Ffast_acquirer  r   r   c                C  r  r   r  r  r  rq   rq   rr   r    r!  zLock.__new__r   c                C  s   || _ d | _t | _d S r   )_fast_acquire_owner_taskr   r  )r   r  rq   rq   rr   r     s   zLock.__init__c                   s   t tjt }| jd u r6| js6t I d H  || _| js4z
t	 I d H  W d S  t
y3   |    w d S | j|kr?tdt }||f}| j| z|I d H  W n t
yk   | j| | j|u rj|    w | j| d S Nz)Attempted to acquire an already held Lock)r0   r   r   r   r  r  r|  r  r  r}  r   releaser   r   r{  r9  )r   r   futr  rq   rq   rr   acquire  s8   

zLock.acquirec                 C  s>   t tjt }| jd u r| js|| _d S | j|u rtdtr  )r0   r   r   r   r  r  r   r@   )r   r   rq   rq   rr   acquire_nowait.  s   
zLock.acquire_nowaitc                 C  s
   | j d uS r   )r  r   rq   rq   rr   locked9  r!  zLock.lockedc                 C  sL   | j t kr
td| jD ]\}}| s || _ |d   d S qd | _ d S )Nz)The current task is not holding this lock)r  r   r   r  r   r   )r   r   r  rq   rq   rr   r  <  s   

zLock.releaser5   c                 C  s*   | j rt| j nd }t|  |t| jS r   )r  AsyncIOTaskInfor5   r  r;  r  )r   	task_inforq   rq   rr   r  H  s   zLock.statisticsN)r  r   r   rI   )r  r   r   r   r   r]  )r   r5   )
rk   rl   rm   r  r   r  r  r  r  r  rq   rq   rq   rr   rI     s    

"

rI   c                      s|   e Zd ZddddddZdddd fddZdddZdddZdddZed ddZ	ed!ddZ
d"ddZ  ZS )#rM   NF	max_valuer  initial_valuerm  r  rd  r  r   r   c                C  r  r   r  r  r  r  r  rq   rq   rr   r  N  s   
zSemaphore.__new__c                  s.   t  j||d || _|| _|| _t | _d S )N)r  )r  r   _value
_max_valuer  r   r  )r   r  r  r  r  rq   rr   r   W  s
   zSemaphore.__init__r   c                   s   | j dkr3| js3t I d H  |  j d8  _ | js1z
t I d H  W d S  ty0   |    w d S t	 }| j
| z|I d H  W d S  tyb   z| j| W   tya   |   Y  w w Nr   r   )r  r  r|  r  r  r}  r   r  r   r   r{  r9  r   r   r  rq   rq   rr   r  d  s4   
zSemaphore.acquirec                 C  s    | j dkrt|  j d8  _ d S r  )r  r@   r   rq   rq   rr   r    s   
zSemaphore.acquire_nowaitc                 C  s`   | j d ur| j| j krtd| jD ]}| s&|d  | j|  d S q|  jd7  _d S )Nz!semaphore released too many timesr   )r  r  r   r  r   r   r9  r  rq   rq   rr   r    s   

zSemaphore.releasec                 C  rV  r   )r  r   rq   rq   rr   rX    rW  zSemaphore.valuec                 C  rV  r   )r  r   rq   rq   rr   r    rW  zSemaphore.max_valuerK   c                 C  s   t t| jS r   )rK   r;  r  r   rq   rq   rr   r    r  zSemaphore.statistics)r  rm  r  rd  r  r   r   rM   )r  rm  r  rd  r  r   r   r  r  )r   rK   )rk   rl   rm   r  r   r  r  r  r`  rX  r  r  r  rq   rq   r  rr   rM   M  s    


rM   c                   @  s   e Zd ZU dZded< d2ddZd3dd	Zd4ddZd5ddZe	d6ddZ
e
jd7ddZ
e	d8ddZe	d6ddZd4ddZd4d d!Zd9d$d%Zd4d&d'Zd9d(d)Zd4d*d+Zd9d,d-Zd:d/d0Zd1S );rE   r   r  _total_tokenstotal_tokensr   c                 C  r  r   r  r  r  rq   rq   rr   r    r!  zCapacityLimiter.__new__c                 C  s   t  | _t | _|| _d S r   )r   
_borrowersr   _wait_queuer  )r   r  rq   rq   rr   r     s   
zCapacityLimiter.__init__r   c                   s   |   I d H  d S r   )r  r   rq   rq   rr   rz       zCapacityLimiter.__aenter__r   r   r   r   r   r   c                   s   |    d S r   )r  r   rq   rq   rr   r    s   zCapacityLimiter.__aexit__c                 C  rV  r   )r  r   rq   rq   rr   r    rW  zCapacityLimiter.total_tokensrX  c                 C  s   t |tst|std|dk rtdt|| j d}|| _| jr>|rB| jj	ddd }|
  |d8 }| jr@|s&d S d S d S d S )Nz'total_tokens must be an int or math.infr   ztotal_tokens must be >= 0Flastr   )r  rm  rE  isinfr  r   maxr  r  popitemr   )r   rX  waiters_to_notifyeventrq   rq   rr   r    s   
rm  c                 C  s
   t | jS r   )r;  r  r   rq   rq   rr   borrowed_tokens     
zCapacityLimiter.borrowed_tokensc                 C  s   | j t| j S r   )r  r;  r  r   rq   rq   rr   available_tokens     z CapacityLimiter.available_tokensc                 C  s<   | j rt| j| jk r| j jddd }|  dS dS dS )zCNotify the next task in line if this limiter has free capacity now.Fr  r   N)r  r;  r  r  r  r   )r   r  rq   rq   rr   _notify_next_waiter  s   z#CapacityLimiter._notify_next_waiterc                 C     |  t  d S r   )acquire_on_behalf_of_nowaitr   r   rq   rq   rr   r    r  zCapacityLimiter.acquire_nowaitborrowerr  c                 C  s<   || j v r	td| jst| j | jkrt| j | d S )NzEthis borrower is already holding one of this CapacityLimiter's tokens)r  r   r  r;  r  r@   r+  r   r  rq   rq   rr   r    s   
z+CapacityLimiter.acquire_on_behalf_of_nowaitc                   s   |  t I d H S r   )acquire_on_behalf_ofr   r   rq   rq   rr   r    r  zCapacityLimiter.acquirec                   s   t  I d H  z| | W n: tyI   t }|| j|< z	| I d H  W n ty?   | j	|d  |
 r>|    w | j| Y d S w z
t  I d H  W d S  ty`   |    w r   )r|  r  r  r@   r   rG   r  r   r~  popr6  r  r  r+  r}  r  )r   r  r  rq   rq   rr   r    s.   
z$CapacityLimiter.acquire_on_behalf_ofc                 C  r  r   )release_on_behalf_ofr   r   rq   rq   rr   r     r  zCapacityLimiter.releasec                 C  s6   z| j | W n ty   tdd w |   d S )Nz@this borrower isn't holding any of this CapacityLimiter's tokens)r  r9  r,  r   r  r  rq   rq   rr   r    s   z$CapacityLimiter.release_on_behalf_ofr3   c                 C  s   t | j| jt| jt| jS r   )r3   r  r  tupler  r;  r  r   rq   rq   rr   r    s   zCapacityLimiter.statisticsN)r  r  r   rE   )r  r  r   r   r^  r_  r  )r  r  r   r   )r   r3   )rk   rl   rm   r  r  r  r   rz  r  r`  r  ra  r  r  r  r  r  r  r  r  r  r  rq   rq   rq   rr   rE     s,   
 











rE   _default_thread_limiterzRunVar[CapacityLimiter]c                   @  sH   e Z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 )_SignalReceiversignalstuple[Signals, ...]c                 C  s,   || _ t | _t | _t | _t | _	d S r   )
_signalsr   r{   r   _signal_queuer   r   rn  r   _handled_signals)r   r  rq   rq   rr   r     s
   
z_SignalReceiver.__init__r   r%   r   r   c                 C  s*   | j | | j s| jd  d S d S r   )r  r{  rn  r   r   )r   r   rq   rq   rr   _deliver&  s   
z_SignalReceiver._deliverc                 C  s2   t | jD ]}| j|| j| | j| q| S r   )r   r  r{   add_signal_handlerr  r  r+  )r   sigrq   rq   rr   r   +  s   z_SignalReceiver.__enter__r   r   r   r   r   r   c                 C  s   | j D ]}| j| qd S r   )r  r{   remove_signal_handler)r   r   r   r   r  rq   rq   rr   r   2  s   
z_SignalReceiver.__exit__c                 C  s   | S r   rq   r   rq   rq   rr   	__aiter__;  s   z_SignalReceiver.__aiter__c                   s6   t  I d H  | jst | _| jI d H  | j S r   )r|  r  r  r   r   rn  r9  r   rq   rq   rr   	__anext__>  s   

z_SignalReceiver.__anext__N)r  r  )r   r%   r   r   )r   r  r   )r   r%   )	rk   rl   rm   r   r  r   r   r  r  rq   rq   rq   rr   r    s    




	r  c                      s(   e Zd Zd	 fddZd
ddZ  ZS )r  r   r   c           	        s   t |}|d u rd }n|j}| }d }||u}|sPtd|fd||fdt v s0t|r5t	|ndt	|d }t
dd d|i }tt|d  }}t t||| | t|| _d S )Nr   r   r   r   z&created TaskInfo from a completed Taskz
>assert %(py5)sr   )r   r   rb  r   r   r   r  r  r  r  _format_assertmsgr  r  r  r   rU  get_nameweakrefrefr  )	r   r   r2  rb  r   r  r	  r
  r  r  rq   rr   r   M  s   
zAsyncIOTaskInfo.__init__r   r   c                 C  sX   |    }sdS |jrdS t|jtjr|j rdS t| }r*|j	 }r*|j
S dS )NFT)r  rJ  r  rK  r   r   r   r   r   r   rD  )r   r   r2  r   rq   rq   rr   has_pending_cancellationY  s   
z(AsyncIOTaskInfo.has_pending_cancellation)r   r   r]  )rk   rl   rm   r   r  r  rq   rq   r  rr   r  L  s    r  c                   @  s   e Zd ZU ded< ddddd>ddZd?ddZd@ddZdAddZdBd!d"ZdCd#d$Z	dDd'd(Z
dEd0d1ZdFd5d6ZdGd8d9ZdHd<d=ZdS )I
TestRunnerzBMemoryObjectSendStream[tuple[Awaitable[Any], asyncio.Future[Any]]]_send_streamNF)rt   
use_uvloopru   rt   rv   r  r   ru   rw   r   r   c                C  sR   |r|d u rt jdkrdd l}|j}ndd l}|j}t||d| _g | _d | _d S )Nwin32r   rs   )	r(  platformuvloopr   winloopr_   _runnerrw  _runner_task)r   rt   r  ru   r   r  rq   rq   rr   r   p  s   

zTestRunner.__init__c                 C  s   | j   |  | j | S r   )r  r   r   set_exception_handler_exception_handlerr   rq   rq   rr   r     s   
zTestRunner.__enter__r   r   r   r   r   r   c                 C  s   | j ||| d S r   )r  r   r   rq   rq   rr   r     s   zTestRunner.__exit__r   c                 C  r  r   )r  r   r   rq   rq   rr   r     r!  zTestRunner.get_loopr   rK  r   dict[str, Any]c                 C  s2   t |dtr| j|d  d S || d S )Nr   )r  r   r   rw  r{  default_exception_handler)r   r   r   rq   rq   rr   r    s   zTestRunner._exception_handlerc                 C  s>   | j r| j g }| _ t|dkr|d |rtd|d S d S )Nr   r   z6Multiple exceptions occurred in asynchronous callbacks)rw  r;  rf   )r   rd   rq   rq   rr   _raise_async_exceptions  s   z"TestRunner._raise_async_exceptionsreceive_streamOMemoryObjectReceiveStream[tuple[Awaitable[T_Retval], asyncio.Future[T_Retval]]]c                   s  ddl m} |z | j^ |2 zR3 d H W \}}z|I d H }W n9 ty6 } z| s1|j|j   d }~w tyX } z| sF|| t	|t
|fsN W Y d }~qd }~ww | sb|| q6 W d    n1 snw   Y  W d    d S W d    d S 1 sw   Y  d S )Nr   )OutcomeException)_pytest.outcomesr  r  r   r   r   r   r~  r   r  r   r   )r   r	  r  r   r   retvalr  rq   rq   rr   _run_tests_and_fixtures  s0   

Pz"TestRunner._run_tests_and_fixturesr    Callable[P, Awaitable[T_Retval]]r   P.argsr  P.kwargsr   c                   sr   | j stttt tjf  d\| _}|  	| 
|| _ ||i |}|   }| j||f |I d H S r  )r  rC   r  r   r-   r   r   r  r   r   r  r   send_nowait)r   r   r   r  r	  r   r   rq   rq   rr   _call_in_runner_task  s   

zTestRunner._call_in_runner_taskfixture_func,Callable[..., AsyncGenerator[T_Retval, Any]]Iterable[T_Retval]c                 c  s    |di |}|   | |jd }|   |V  z|   | |jd  W n ty7   |   Y d S w |   |  td)Nz$Async generator fixture did not stoprq   )r   r   r  asendr  StopAsyncIterationr  r   )r   r  r  asyncgenfixturevaluerq   rq   rr   run_asyncgen_fixture  s    zTestRunner.run_asyncgen_fixture,Callable[..., Coroutine[Any, Any, T_Retval]]c                 C  s(   |   | j|fi |}|   |S r   )r   r   r  r  )r   r  r  r  rq   rq   rr   run_fixture  s
   zTestRunner.run_fixture	test_func'Callable[..., Coroutine[Any, Any, Any]]c              
   C  s^   z|   | j|fi | W n ty( } z| j| W Y d }~nd }~ww |   d S r   )r   r   r  r   rw  r{  r  )r   r  r  r  rq   rq   rr   run_test  s   zTestRunner.run_test)rt   rv   r  r   ru   rw   r   r   )r   r  r   r   )r   rK  r   r  r   r   r   )r	  r
  r   r   )r   r  r   r  r  r  r   r   )r  r  r  r  r   r  )r  r  r  r  r   r   )r  r  r  r  r   r   )rk   rl   rm   r  r   r   r   r   r  r  r  r  r  r  r   rq   rq   rq   rr   r  m  s    
 








r  c                   @  s  e Zd Zedd
dZedddZedddZedddZedddZedddZ	edddZ
edddZeejdd dd%d&Zedd'd(Zedd*d+Zedd-d.Zedd1d2Zed3dd4dd:d;Zedd>d?Ze		3dddDdEZeddFdGZeddIdJZeddKdLZeddUdVZeddYdZZe	3dddadbZeddfdgZeddkdlZeddmdnZeddtduZeddzd{Z ed|d|d|d|d}dddZ!e	|ddddZ"edddZ#edddZ$edddZ%edddZ&edddZ'edddZ(edddZ)edddZ*edddZ+edddZ,edddZ-edddZ.edddZ/edddZ0edddZ1edddZ2d3S )r|  r   1Callable[[Unpack[PosArgsT]], Awaitable[T_Retval]]r   r  r  r  optionsr   r   c                   s   t d fdd}|dd }|dd }|d u r6|ddr6tjd	kr/d
d l}|j}nd
d l}	|	j}t||d}
|
| W  d    S 1 sMw   Y  d S )Nr   r   c                    sL   t tjt } | t td d t| < z  I d H W t| = S t| = w r   )r0   r   r   r   set_namer   r-  r   )r   r   r   rq   rr   wrapper	  s   z#AsyncIOBackend.run.<locals>.wrapperrt   ru   r  Fr  r   rs   r   r   )	r   r   r(  r  r   r   r  r_   r   )r  r   r   r  r"  r%  rt   ru   r   r  runnerrq   r$  rr   r   	  s   


$zAsyncIOBackend.runr  c                 C     t  S r   )r   r  rq   rq   rr   current_token(	  rW  zAsyncIOBackend.current_tokenr  c                 C  s
   t   S r   )r   rG  r  rq   rq   rr   r  ,	  r  zAsyncIOBackend.current_timetype[BaseException]c                 C  s   t S r   )r   r  rq   rq   rr   cancelled_exception_class0	  s   z(AsyncIOBackend.cancelled_exception_classr   c                   s   t dI d H  d S r  r	   r  rq   rq   rr   r  4	     zAsyncIOBackend.checkpointc                   sn   t  }|d u r
d S zt| j}W n
 ty   Y d S w |r5|jr)tdI d H  n|jr.d S |j}|sd S d S r  )r   r   r   r,  rL  r	   r  r   )r  r   r   rq   rq   rr   r  8	  s    z&AsyncIOBackend.checkpoint_if_cancelledc                   s@   t dd tdI d H  W d    d S 1 sw   Y  d S )NTr  r   )rO   r	   r  rq   rq   rr   r}  K	  s   "z)AsyncIOBackend.cancel_shielded_checkpointdelayc                   s   t |I d H  d S r   r,  )r  r.  rq   rq   rr   r	   P	  r-  zAsyncIOBackend.sleepFr  r  r  r   rO   c                C  s   t ||dS )Nr  rN   r  rq   rq   rr   create_cancel_scopeT	  s   z"AsyncIOBackend.create_cancel_scopec                 C  s   t   }d u r
tjS zt| j}W n ty   tj Y S w tj}|r>t||j}|jr3tj }	 |S |j	r9	 |S |j
}|s#|S r   )r   rE  rF  r   r   r,  minr  r!  r  r   )r  r   r   r  rq   rq   rr   current_effective_deadlineZ	  s(   

z)AsyncIOBackend.current_effective_deadlineabc.TaskGroupc                 C  r(  r   )rv  r  rq   rq   rr   create_task_groupq	  rW  z AsyncIOBackend.create_task_group	abc.Eventc                 C  r(  r   rF   r  rq   rq   rr   create_eventu	  rW  zAsyncIOBackend.create_eventr  abc.Lockc                C  s
   t |dS )Nr  rH   r  rq   rq   rr   create_locky	  r  zAsyncIOBackend.create_lockNr  r  rm  r  rd  abc.Semaphorec                C  s   t |||dS )Nr  rL   r  rq   rq   rr   create_semaphore}	  s   zAsyncIOBackend.create_semaphorer  abc.CapacityLimiterc                 C     t |S r   rD   r  rq   rq   rr   create_capacity_limiter	  r  z&AsyncIOBackend.create_capacity_limiter&Callable[[Unpack[PosArgsT]], T_Retval]abandon_on_cancellimiterabc.CapacityLimiter | Nonec              	     s  |   I d H  z
t }t }W n ty+   t }t }t| t| Y nw |p1|  4 I d H  t| d}t	j
t  }t }	|sct|	||}
|
  ||
 |	j|
jt d n&| }
|  }|r||d j tjk rxn| }|j|j |  |smt }|td  |s|jd u r|}n|j}|
j |||||f |I d H W  d    W  d   I d H  S 1 sw   Y  W d   I d H  d S 1 I d H sw   Y  d S )Nr  r   r   )!r  r  r   r  LookupErrorr   r   current_default_thread_limiterrO   r   r   r   r   r  r   r+  r  r  r   r   r  r  r  r  r9  r   remove_done_callbackr   r   r9   r   r  r  )r  r   r   r>  r?  r  r  rQ  r   r   workernowexpired_workerr   worker_scoperq   rq   rr   run_sync_in_worker_thread	  s\   


.z(AsyncIOBackend.run_sync_in_worker_threadc                 C  sH   t j}|d ur"|jrtdt|d|jrd S |j}|d usd S d S )NzCancelled by cancel scope r   )r:   r  rL  r   rU  r  r   )r  rQ  rq   rq   rr   check_cancelled	  s   zAsyncIOBackend.check_cancelledtokenc                   sj   d	 fdd}t d|ptjj}| rtt }|td t	tdd |jt
j| |d}| S )
Nr   r   c               
     s   d} d urt tjt }td t|< j| z*z  I d H W W d ur0j| S S  t	yE } z	t
j	t|d d }~ww d urQj| w w ry  )r0   r   r   r   r-  r   r'  r+  r.  r   
concurrentfuturesr   )__tracebackhide__r   r  r   r   rQ  rq   rr   task_wrapper	  s$   z:AsyncIOBackend.run_async_from_thread.<locals>.task_wrapperr   r   r  r   r&  )r0   r:   r)  native_tokenr  r?   r   r   r9   r   r   run_coroutine_threadsafer  )r  r   r   rJ  rO  r   r   r  rq   rN  rr   run_async_from_thread	  s   
z$AsyncIOBackend.run_async_from_threadc                   sP   t d fdd}td|ptjj}| rtt || 	 S )Nr   r   c               
     sZ   zt d    W d S  ty, }  z|  t| ts! W Y d } ~ d S d } ~ ww )Nr   )r9   r   r~  r   r  r   r  r   r  r   rq   rr   r%  	  s   

z4AsyncIOBackend.run_sync_from_thread.<locals>.wrapperr   r   )
r   r0   r:   r)  rP  r  r?   r   r   r  )r  r   r   rJ  r%  r   rq   rS  rr   run_sync_from_thread	  s   	
z#AsyncIOBackend.run_sync_from_threadcommand)StrOrBytesPath | Sequence[StrOrBytesPath]r  int | IO[Any] | Noner  r  r-   r  c          
        s   |   I d H  t|trt|}t|ttfr*tj|f|||d|I d H }ntj	||||d|I d H }|j
rAt|j
nd }|jrKt|jnd }|jrUt|jnd }	t||||	S )N)r  r  r  )r  r  r#   osfspathr   r  r   create_subprocess_shellcreate_subprocess_execr  r  r  r  r  r  )
r  rU  r  r  r  r  r  stdin_streamstdout_streamstderr_streamrq   rq   rr   open_process
  s4   


zAsyncIOBackend.open_processr  r  c                 C  s&   t t|dd t tt| d S )Nz AnyIO process pool shutdown taskr  )r   r  r   r  r   r   )r  r  rq   rq   rr   #setup_process_pool_exit_at_shutdown6
  s   z2AsyncIOBackend.setup_process_pool_exit_at_shutdownhostr   portlocal_addressIPSockAddrType | Noner  c                   sB   t ttjtf t jt|||dI d H \}}|  t||S )N)
local_addr)	r0   r  r   r  r  r   create_connectionr8  r#  )r  ra  rb  rc  r
  r%  rq   rq   rr   connect_tcp@
  s   

zAsyncIOBackend.connect_tcppathstr | bytesabc.UNIXSocketStreamc                   s   |   I d H  t  ttjd 	 z
| W tS  tyE   t } 	|j
d  | fdd |I d H  Y n tyP      w q)NFTc                   
     S r   rX  r  r   rH  rq   rr   r   Y
     
 z-AsyncIOBackend.connect_unix.<locals>.<lambda>)r  r   r1  AF_UNIXr  connectrb  r   r   rY  r   r  r~  r   r_  )r  rh  r  rq   rm  rr   connect_unixM
  s&   

zAsyncIOBackend.connect_unixsockr.  rR   c                 C  r;  r   r  r  rr  rq   rq   rr   create_tcp_listenera
  r  z"AsyncIOBackend.create_tcp_listenerc                 C  r;  r   )r  rt  rq   rq   rr   create_unix_listenere
  r  z#AsyncIOBackend.create_unix_listenerfamilyr&   remote_address
reuse_portUDPSocket | ConnectedUDPSocketc                   sN   t  jt||||dI d H \}}|jr|  |j|s"t||S t||S )N)re  remote_addrrw  ry  )r   create_datagram_endpointr  r   r   r  r  )r  rw  rc  rx  ry  r
  r%  rq   rq   rr   create_udp_socketi
  s   

z AsyncIOBackend.create_udp_socketrH  remote_pathstr | bytes | None8abc.UNIXDatagramSocket | abc.ConnectedUNIXDatagramSocketc                   s   |   I d H  t  |rI	 z
| W tS  ty<   t } |jd  | fdd |I d H  Y n t	yG   
   w qtS )NTc                   rk  r   rl  r  rm  rq   rr   r   
  rn  z<AsyncIOBackend.create_unix_datagram_socket.<locals>.<lambda>)r  r   rp  rb  r   r   rY  r   r  r~  r   r  r  )r  rH  r~  r  rq   rm  rr   create_unix_datagram_socket
  s&   
z*AsyncIOBackend.create_unix_datagram_socketr   rw  r  protort  bytes | str | Nonestr | int | Noneint | AddressFamilyr  int | SocketKindr  rt  uSequence[tuple[AddressFamily, SocketKind, int, str, tuple[str, int] | tuple[str, int, int, int] | tuple[int, bytes]]]c                  s    t  j||||||dI d H S )Nr  )r   getaddrinfo)r  ra  rb  rw  r  r  rt  rq   rq   rr   r  
  s   zAsyncIOBackend.getaddrinfosockaddrrQ   tuple[str, str]c                   s   t  ||I d H S r   )r   getnameinfo)r  r  rt  rq   rq   rr   r  
  s   zAsyncIOBackend.getnameinfoobjrY   c                   s:  zt  W n ty   i t  Y nw t|tr|n|   r,tdt }|	 d fdd}z|
 | W n ty`   ddlm} | }|
 | |jY nw |j < zI d H }W z = W n	 ty}   Y nw   nz = W n	 ty   Y w w   w |std S )	Nr(  r   r   c                     sN   z = W n	 t y   Y nw   zd W d S  tjy&   Y d S w ry  r,  r   r   rq  rq   r|  r  r  rN  rq   rr   r   
     
z(AsyncIOBackend.wait_readable.<locals>.cbr   get_selectorr   )r  r   rA  r   r  rm  r]  r<   r   r   rT  r  $anyio._core._asyncio_selector_threadr  rN  r,  r=   r  r  r   r   r  selectorsuccessrq   r  rr   wait_readable
  P   





zAsyncIOBackend.wait_readablec                   s:  zt  W n ty   i t  Y nw t|tr|n|   r,tdt }|	 d fdd}z|
 | W n ty`   ddlm} | }|
 | |jY nw |j < zI d H }W z = W n	 ty}   Y nw   nz = W n	 ty   Y w w   w |std S )	Nr)  r   r   c                     sN   z = W n	 t y   Y nw   zd W d S  tjy&   Y d S w ry  r  rq   r|  r  rX  r  rq   rr   r   
  r  z(AsyncIOBackend.wait_writable.<locals>.cbr   r  r   )r  r   rA  r   r  rm  r]  r<   r   r   rY  r  r  r  rX  r,  r=   r  rq   r  rr   wait_writable
  r  zAsyncIOBackend.wait_writablec                 C  sd  t |tr|n| }t }zt }W n	 ty   Y nAw z||}W n	 ty.   Y n0w z|	d W n
 t
jy@   Y nw z|| W n ty]   ddlm} | | Y nw zt }W n
 tyn   Y d S w z||}W n
 ty   Y d S w z|	d W n
 t
jy   Y nw z|| W d S  ty   ddlm} | | Y d S w )NFr   r  )r  rm  r]  r   r  r   rA  r  r,  r   r   rq  rX  r  r  r  r  rN  )r  r  r|  r   r  r  r  r  rq   rq   rr   notify_closing   sX   zAsyncIOBackend.notify_closingc                   
   t |S r   rs  rt  rq   rq   rr   wrap_listener_socketQ     z#AsyncIOBackend.wrap_listener_socketr#  c                   &   t  jt|dI d H \}}t||S N)rr  )r   rf  r  r#  r  rr  r
  r%  rq   rq   rr   wrap_stream_socketU  
   
z!AsyncIOBackend.wrap_stream_socketr_  c                   r  r   )r_  rt  rq   rq   rr   wrap_unix_stream_socket\  r  z&AsyncIOBackend.wrap_unix_stream_socketr  c                   r  r  )r   r|  r  r  r  rq   rq   rr   wrap_udp_socket`  r  zAsyncIOBackend.wrap_udp_socketr  c                   r  r  )r   r|  r  r  r  rq   rq   rr   wrap_connected_udp_socketg  r  z(AsyncIOBackend.wrap_connected_udp_socketr  c                   r  r   )r  rt  rq   rq   rr   wrap_unix_datagram_socketn  r  z(AsyncIOBackend.wrap_unix_datagram_socketr  c                   r  r   )r  rt  rq   rq   rr   #wrap_connected_unix_datagram_socketr  s   z2AsyncIOBackend.wrap_connected_unix_datagram_socketrE   c                 C  s4   zt  W S  ty   td}t | | Y S w )N(   )r  r   rA  rE   r   )r  r?  rq   rq   rr   rB  x  s   

z-AsyncIOBackend.current_default_thread_limiterr  r%   .AbstractContextManager[AsyncIterator[Signals]]c                 G  r;  r   )r  )r  r  rq   rq   rr   open_signal_receiver  s   z#AsyncIOBackend.open_signal_receiverr6   c                 C  s
   t t S r   )r  r   r  rq   rq   rr   get_current_task  r  zAsyncIOBackend.get_current_taskSequence[TaskInfo]c                 C  s   dd t  D S )Nc                 S  s   g | ]
}|  st|qS rq   )r   r  )r   r   rq   rq   rr   r     s    z4AsyncIOBackend.get_running_tasks.<locals>.<listcomp>)r   r  rq   rq   rr   get_running_tasks  r  z AsyncIOBackend.get_running_tasksc                   s\   |   I d H  t }	 t D ]}||u rq|j}|d u s!| r*tdI d H   nqd S q)NTg?)r  r   r   rK  r   r	   )r  	this_taskr   rP  rq   rq   rr   wait_all_tasks_blocked  s   
z%AsyncIOBackend.wait_all_tasks_blockedr  c                 C  s   t di |S )Nrq   )r  )r  r"  rq   rq   rr   create_test_runner  s   z!AsyncIOBackend.create_test_runner)
r   r!  r   r  r  r  r"  r  r   r   )r   r  r^  )r   r*  r   )r.  r  r   r   r[  )r   r2  )r   r4  )r  r   r   r6  )r  rm  r  rd  r  r   r   r8  )r  r  r   r:  )FN)
r   r=  r   r  r>  r   r?  r@  r   r   )r   r!  r   r  rJ  r  r   r   )r   r=  r   r  rJ  r  r   r   )rU  rV  r  rW  r  rW  r  rW  r  r-   r   r  r  r  r   r   r   )ra  r   rb  rm  rc  rd  r   r  )rh  ri  r   rj  )rr  r.  r   rR   )
rw  r&   rc  rd  rx  rd  ry  r   r   rz  )rH  r.  r~  r  r   r  )ra  r  rb  r  rw  r  r  r  r  rm  rt  rm  r   r  )r   )r  rQ   rt  rm  r   r  )r  rY   r   r   )rr  r.  r   r#  )rr  r.  r   r_  )rr  r.  r   r  )rr  r.  r   r  )rr  r.  r   r  )rr  r.  r   r  )r   rE   )r  r%   r   r  )r   r6   )r   r  )r"  r  r   r  )3rk   rl   rm   classmethodr   r)  r  r+  r  r  r}  r	   rE  rF  r/  r1  r3  r5  r7  r9  r<  rH  rI  rR  rT  r_  r`  rg  rq  ru  rv  r}  r  r  r  r  r  r  r  r  r  r  r  r  r  rB  r  r  r  r  r  rq   rq   rq   rr   r|  	  s    !	<"#	440r|  )r   r   r   r   )r   r   )r   r   r   r   )r   r   r   r   )r  r   r   r   )r  r  r  r  r   r   r  )
__future__r   builtinsr  _pytest.assertion.rewrite	assertionrewriter   rj  r   concurrent.futuresrK  r   rE  rX  r1  r(  r   r  r   r   r   r   r   r   r	   asyncio.base_eventsr
   collectionsr   r   collections.abcr   r   r   r   r   r   r   r   r   
contextlibr   r   r   r   dataclassesr   r   	functoolsr   r   inspectr   r   r    r!   ior"   r#   r  r$   r   r%   r&   r'   r(   typesr)   r*   typingr+   r,   r-   r.   r/   r0   r1   r  r3   r4   r5   r6   r7   _core._eventloopr8   r9   r:   _core._exceptionsr;   r<   r=   r>   r?   r@   rA   _core._socketsrB   _core._streamsrC   _core._synchronizationrE   BaseCapacityLimiterrG   	BaseEventrI   BaseLockrJ   rK   rM   BaseSemaphore_core._tasksrO   BaseCancelScoperP   rQ   rR   rS   rT   abc._eventlooprU   lowlevelrV   streams.memoryrW   rX   	_typeshedrY   r  r)  r\   typing_extensionsr_   r`   ra   enumrb   rc   rd   re   exceptiongrouprf   Enumrg   r   r   r   r   r   r   r   r  r   r   r   r  r  r-  r   
TaskStatusrk  eager_task_factoryr  ru  rv  r  r~  _Retval_Queue_Typer  r  r  ByteReceiveStreamr  ByteSendStreamr  r  r   r  Protocolr  r  r#  rE  r_  r  r  r  r  r  r  r  r  r  r  r  r  r|  backend_classrq   rq   rq   rr   <module>   s   "$	(
 $	 






   RN
9
'U.t<%02$$JQx.!      $