o
    i                    @  st  U d dl 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)m*Z* d dl+m,Z,m-Z- d dl.m/Z/m0Z0m1Z1m2Z2 d dl3m4Z4 d dlm5Z5 d dl6m7Z7 d dl8m9Z9 d dl	m:Z:m;Z; d dlm<Z< d dl=m>Z>m?Z? d dl@mAZAmBZBmCZCmDZDmEZEmFZF d dlmGZG ddlHmIZImJZJmKZKmLZLmMZM ddlNmOZOmPZPmQZQ ddlRmSZSmTZTmUZUmVZVmWZWmXZX ddlYmZZZ ddl[m\Z\ ddl]m^Z_ ddl]m`Za ddl]mbZc ddl]mdZdmeZe dd l]mfZg dd!lhmiZj dd"lMmkZkmlZlmmZmmnZnmoZo dd#lpmqZq dd$lrmsZs dd%ltmuZumvZv eBr_d d&lwmxZx neyZxe
jzd'krvd d(lm{Z{ d d)l@m|Z|m}Z} n@d dlZd dl~Z~d dl8Z8d d*lmZmZmZmZ d d+lmZ d d)lm|Z|m}Z} G d,d- d-e~jZG d.d/ d/Z{dd4d5Zdd6d7ZeEd8ZeEd9d:d;Ze|d<ZeDd=Zesd>Zd?ed>< ddAdBZddFdGZeG ZdHedI< ddLdMZddPdQZG dRdS dSejZiG dTdU dUZeG ZdVedW< G dXdY dYeMjZe
jzdZkrejjZd[ed\< ndZG d]d^ d^eMjZeedB edB f ZG d_d` d`e<ZesdaZdbeda< esdcZddedc< e)dedfG dgdh dheMjZe)dedfG didj djeMjZe)dedfG dkdl dleMjZddqdrZddtduZG dvdw dwejZG dxdy dyejZG dzd{ d{eMjZG d|d} d}ZG d~d deeMjZG dd deMjmZG dd deMjmZG dd deMjZG dd deMjZG dd deeMjZG dd deeMjZesdZded< esdZded< G dd deaZ`G dd decZbG dd degZfG dd de_Z^esdZded< G dd dZG dd deLZG dd deMjZG dd dek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	ParamSpecTypeVarcast)WeakKeyDictionary   )CapacityLimiterStatisticsEventStatisticsLockStatisticsTaskInfoabc)claim_worker_threadset_current_async_librarythreadlocals)BrokenResourceErrorBusyResourceErrorClosedResourceErrorEndOfStreamRunFinishedError
WouldBlock)convert_ipv6_sockaddr)create_memory_object_streamCapacityLimiterEventLock)ResourceGuardSemaphoreStatistics	SemaphoreCancelScope)AsyncBackendIPSockAddrTypeSocketListenerUDPPacketTypeUNIXDatagramPacketType)StrOrBytesPath)RunVar)MemoryObjectReceiveStreamMemoryObjectSendStream)FileDescriptorLike      )Runner)TypeVarTupleUnpack)
coroutinesevents
exceptionstasks)BaseExceptionGroupc                   @  s   e Zd ZdZdZdZdS )_StatecreatedZinitializedclosedN)__name__
__module____qualname__CREATEDINITIALIZEDCLOSED rm   rm   R/home/kim/smarthome/.venv/lib/python3.10/site-packages/anyio/_backends/_asyncio.pyrd   {   s    rd   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_factoryrp   bool | Nonerq   &Callable[[], AbstractEventLoop] | Nonec                C  s0   t j| _|| _|| _d | _d | _d| _d| _d S )Nr   F)	rd   rj   _state_debug_loop_factory_loop_context_interrupt_count_set_event_loop)selfrp   rq   rm   rm   rn   __init__   s   
zRunner.__init__returnc                 C  s   |    | S N)
_lazy_initr{   rm   rm   rn   	__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   rm   rm   rn   __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)rw   rt   rd   rk   _cancel_all_tasksrun_until_completeZshutdown_asyncgenshasattrr   _shutdown_default_executorrz   r`   set_event_loopr   rl   r{   looprm   rm   rn   r      s(   



zRunner.closer   c                 C  s   |    | jS )zReturn embedded event loop.)r   rw   r   rm   rm   rn   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)r_   r!   
ValueErrorr`   Z_get_running_loopRuntimeErrorr   rx   runrw   r   	threadingcurrent_threadmain_threadsignal	getsignalSIGINTdefault_int_handlerr   
_on_sigintry   r   ra   r   getattrKeyboardInterrupt)r{   r   r   taskZsigint_handlerr   rm   rm   rn   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)rt   rd   rl   r   rk   rv   r`   new_event_looprw   rz   r   ru   Z	set_debugcontextvarsr   rx   r   rm   rm   rn   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~   rm   rm   rm   rm   rn   <lambda>   s    z#Runner._on_sigint.<locals>.<lambda>)ry   donecancelrw   call_soon_threadsafer   )r{   signumframer   rm   rm   rn   r      s   zRunner._on_sigint)rp   rr   rq   rs   )r}   r\   r   r   r   r   r   r   r}   r   r}   r   r}   r   )r   r   r}   r   )r   r   r}   r   )rg   rh   ri   r|   r   r   r   r   r   r   r   rm   rm   rm   rn   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 )NZreturn_exceptionsTz1unhandled exception during asyncio.run() shutdown)message	exceptionr   )rb   r   r   r   Zgather	cancelledr   Zcall_exception_handler)r   Z	to_cancelr   rm   rm   rn   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   rm   rn   _do_shutdown  s   z0_shutdown_default_executor.<locals>._do_shutdownTN)targetargs)r   r   r}   r   )Z_executor_shutdown_calledr   create_futurer   r(   startjoin)r   r   r   threadrm   r   rn   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 rm   rm   ).0cbr   rm   rm   rn   
<listcomp><      z"find_root_task.<locals>.<listcomp>rh   zuvloop.loop)r   getr   r   Z
_callbacksr
   r   setr0   asyncioTaskr   _task_statescancel_scope_parent_scope
_host_task)	root_taskr   	callbacksr   stater   rm   rm   rn   find_root_task4  s.   



r   funcr   strc                 C  s0   t | dd }t | dd }ddd ||fD S )Nrh   ri   .c                 S  s   g | ]}|r|qS rm   rm   )r   xrm   rm   rn   r   V  r   z%get_callable_name.<locals>.<listcomp>)r   r   )r   modulequalnamerm   rm   rn   get_callable_nameS  s   r   z1WeakKeyDictionary[asyncio.AbstractEventLoop, Any]	_run_varsr   boolc                 C  sH   |   }|dus
J z	t|ttfv W S  ty#   td|  ddw )zBReturn ``True`` if the task has been started and has not finished.NzCannot determine if task z has started or not)get_coror    r   r   AttributeErrorr   )r   r   rm   rm   rn   _task_started`  s   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   rm   rm   rn   is_anyio_cancellationr  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 )>rN   Fdeadlineshieldr   floatr   r   r}   c                C  
   t | S r~   object__new__clsr   r   rm   rm   rn   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 )NFrY   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   rm   rm   rn   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{   Z	host_task
task_staterm   rm   rn   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sJ t| j}|d u s(|j| ur,tdzd| _ | jr;| j  d | _| j	
| j | jd urV| jj
|  | jj	| j | j|_|   | jr| js| jrv| j  |  jd8  _| jsgt|tr|dd \}}|d u rW d | _~dS d| _|d u rW d | _~dS |j}z|||_~w t|trt|rd| _W d | _~dS W d | _~dS | jr| jd usJ | jjd usJ | 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 inzTAttempted to exit a cancel scope that isn't the current tasks's current cancel scopeFr   c                 S  s   t | tot| S r~   )r   r   r   r   rm   rm   rn   r     s   
z&CancelScope.__exit__.<locals>.<lambda>Tr   )r  r   r   r   r   r   r   r  r   r  remover   r   r
  _restart_cancellation_in_parentr   %_parent_cancellation_is_visible_to_usr	  r   r   rc   splitr  r   r   r   )r{   r   r   r   Zhost_task_stateZcancelleds_caught	remainingr   rm   rm   rn   r     s   



#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   rm   rm   rn   _effectively_cancelled  s   	z"CancelScope._effectively_cancelledc                 C  s   | j d uo| j o| j jS r~   )r   r   r  r   rm   rm   rn   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   Zcall_atr  r  r   rm   rm   rn   r  4  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_calledr  r   Z	call_soonr  )r{   r  Zshould_retrycurrentr   waiterscoperm   rm   rn   r  <  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  r  r   r{   r   rm   rm   rn   r  j  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   r  )r{   r"  r   rm   rm   rn   r   }  s   


zCancelScope.cancelc                 C     | j S r~   )r   r   rm   rm   rn   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{   r'  rm   rm   rn   r     s   


c                 C  r%  r~   )r   r   rm   rm   rn   r    r&  zCancelScope.cancel_calledc                 C  r%  r~   )r  r   rm   rm   rn   cancelled_caught  r&  zCancelScope.cancelled_caughtc                 C  r%  r~   )r   r   rm   rm   rn   r     r&  zCancelScope.shieldc                 C  s(   | j |kr|| _ |s|   d S d S d S r~   )r   r  r(  rm   rm   rn   r     s   
r   r   r   r   r}   rN   )r   r   r   r   )r}   rN   r   r   r   r   r   r   r}   r   r}   r   r   )r  rN   r}   r   r~   )r"  r#  r}   r   r}   r   r'  r   r}   r   )r'  r   r}   r   )rg   rh   ri   r  r  r   r|   r   r   propertyr  r  r  r  r  r   r   setterr  r)  r   rm   rm   rm   rn   rN     s4    

b


.	rN   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__r1  
int | Noner   CancelScope | Nonec                 C     || _ || _d S r~   r1  r   )r{   r1  r   rm   rm   rn   r|        
zTaskState.__init__N)r1  r3  r   r4  )rg   rh   ri   __doc__	__slots__r|   rm   rm   rm   rn   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.Futurer1  intc                 C  r5  r~   )_future
_parent_id)r{   r   r1  rm   rm   rn   r|     r7  z_AsyncioTaskStatus.__init__Nr'  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)r=  r   r   InvalidStateErrorr   r   r0   r   r   r>  r   r1  )r{   r'  r   rm   rm   rn   started  s   
z_AsyncioTaskStatus.started)r   r;  r1  r<  r~   )r'  r?  r}   r   )rg   rh   ri   r|   rA  rm   rm   rm   rn   r:    s    
r:  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)rN   r   r  _exceptionsr   r  _on_completed_futr   rm   rm   rn   r|     s
   
zTaskGroup.__init__c                   s   | j   d| _| S NT)r   r   r  r   rm   rm   rn   
__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   rF  appendr   r  rN   r   rG  r   r   AsyncIOBackendcancel_shielded_checkpointr  rc   BaseExceptionr   type__traceback__)r{   r   r   r   r   Z
wait_scoper   rm   rm   rn   	__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sJ | |jjv s&J |jj|   j t| =  j	d urQ jsQz j	
d  W n
 tjyP   Y nw z|  }W n# tyz } zt|jtrn|j}t|jtse|}W Y d }~nd }~ww |d urd ur rd S d u s rt|ts j|  jjs j  d S d S | d S d ur sÈtd d S d S d S )NrZ      z2Child exited without calling task_status.started())r  r  r   r   r   Zfuture_discard_from_awaited_byr   r  r  rG  r   r@  r   r   r   r   r   r   rF  rJ  r  r   r   r   )rV  r  r   er{   r   rT  rm   rn   	task_done5  sP   
z#TaskGroup._spawn.<locals>.task_donez;This task group is not active; no new tasks can be started.Ztask_statusrh   r    z	Expected z0() to return a coroutine, but the return value (z) is not a coroutine object__code____closure__r   )r   rS  rS  r6  rW  )rV  r   r}   r   )r  r   r$  r   r:  r   r   r!   r   rh   	TypeErrorri   r   r   r   r   Zget_task_factoryr   rD  cell_contentsr   r  r   r  r
  r  r  Zfuture_add_to_awaited_byadd_done_callback)r{   r   r   rS  rT  r[  kwargsr1  r   prefixr   factoryclosureZcustom_task_constructorrm   rZ  rn   _spawn.  sN   /





zTaskGroup._spawnr_  Unpack[PosArgsT]c                G  s   |  ||| d S r~   )rg  )r{   r   rS  r   rm   rm   rn   
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   rg  r   r   rN   r   )r{   r   rS  r   r   r   rm   rm   rn   r     s    (zTaskGroup.startr   )r}   rE  r+  r~   )
r   rQ  r   rR  rS  r   rT  rU  r}   r   )r   rQ  r   rh  rS  r   r}   r   )r   rj  r   r   rS  r   r}   r-   )	rg   rh   ri   r|   rI  rP  rg  ri  r   rm   rm   rm   rn   rE    s    


Dh	rE  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 ) WorkerThread
   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   rn  rp  rw   r   r$   queuerK  current_time
idle_sincestopping)r{   r   rn  rp  	__class__rm   rn   r|     s   

zWorkerThread.__init__r   r;  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)rK  rt  ru  rv  rp  rJ  r   r   StopIterationr   	__cause__r   r   )r{   r   ry  r   new_excrm   rm   rn   _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   rK  r   rs  r   r   r:   current_cancel_scoper   rM  	is_closedr   r}  r[  )
r{   itemr   r   r   r   r   ry  r   r   rm   rm   rn   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 rH  )rv  rs  
put_nowaitrn  r  rp  r  r   )r{   r  rm   rm   rn   stop  s   zWorkerThread.stop)r   r   rn  ro  rp  rq  )r   r;  ry  r-   r   r   r}   r   r   r~   )r  r  r}   r   )	rg   rh   ri   MAX_IDLE_TIMEr|   r}  r   r  __classcell__rm   rm   rw  rn   rl    s    

rl  _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_bytesr<  r}   bytesc                   s    | j |I d H }|r|S tr~   )r  readr>   )r{   r  datarm   rm   rn   receive  s
   zStreamReaderWrapper.receiver   c                   s"   | j t  t I d H  d S r~   )r  r   r=   rK  
checkpointr   rm   rm   rn   aclose  s   zStreamReaderWrapper.acloseNr  r  r<  r}   r  r   )rg   rh   ri   __annotations__r  r  rm   rm   rm   rn   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~   )rK  checkpoint_if_cancelledr  	_protocolZ_pausedwriteZdrainConnectionResetErrorBrokenPipeErrorr   r  r=   
is_closingr;   rL  )r{   r  Zstream_pausedr   rm   rm   rn   send$  s"   

zStreamWriterWrapper.sendc                   s$   d| _ | j  t I d H  d S rH  )r  r  r   rK  r  r   rm   rm   rn   r  8  s   
zStreamWriterWrapper.acloseNr  r  r}   r   r   )rg   rh   ri   r  r   r  r  r  rm   rm   rm   rn   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 )NTrk  F)	rN   r  r  r  r  r   r   rM  killr!  rm   rm   rn   r  E  s&   "zProcess.acloser<  c                   s   | j  I d H S r~   )r  r   r   rm   rm   rn   r   W  s   zProcess.waitc                 C     | j   d S r~   )r  	terminater   rm   rm   rn   r  Z     zProcess.terminatec                 C  r  r~   )r  r  r   rm   rm   rn   r  ]  r  zProcess.killr   c                 C  s   | j | d S r~   )r  send_signal)r{   r   rm   rm   rn   r  `     zProcess.send_signalc                 C     | j jS r~   )r  pidr   rm   rm   rn   r  c     zProcess.pidr3  c                 C  r  r~   )r  
returncoder   rm   rm   rn   r  g  r  zProcess.returncodeabc.ByteSendStream | Nonec                 C  r%  r~   )r  r   rm   rm   rn   stdink  r&  zProcess.stdinabc.ByteReceiveStream | Nonec                 C  r%  r~   )r  r   rm   rm   rn   stdouto  r&  zProcess.stdoutc                 C  r%  r~   )r  r   rm   rm   rn   stderrs  r&  zProcess.stderrNr   r}   r<  )r   r<  r}   r   r}   r3  )r}   r  )r}   r  )rg   rh   ri   r  r  r   r  r  r  r/  r  r  r  r  r  rm   rm   rm   rn   r  >  s(   
 




r  rn  set[Process]rV  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.NrB  )r  r  r   Zget_event_loop_policyZget_child_watcherNotImplementedErrorcopyr  r  r  
_transportr   r  r  r  Zremove_child_handlerr  )rn  rV  Zchild_watcherprocessrm   rm   rn   '_forcibly_shutdown_process_pool_on_exitx  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	   r  r  r   r   r  r  r  r  )rn  r  rm   rm   rn   _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   rF   r  r  r   r0   	TransportZset_write_buffer_limitsr{   r  rm   rm   rn   connection_made  s
   


zStreamProtocol.connection_mader   c                 C  s"   |r|| _ | j  | j  d S r~   )r   r  r   r  r{   r   rm   rm   rn   connection_lost  s   
zStreamProtocol.connection_lostr  r  c                 C  s   | j t| | j  d S r~   )r  rJ  r  r  r   )r{   r  rm   rm   rn   data_received  s   zStreamProtocol.data_receivedrr   c                 C  s   d| _ | j  dS rH  )r  r  r   r   rm   rm   rn   eof_received  s   
zStreamProtocol.eof_receivedc                 C     t  | _d S r~   )r   rF   r  r   rm   rm   rn   pause_writing  r  zStreamProtocol.pause_writingc                 C  r  r~   r  r   r   rm   rm   rn   resume_writing  r  zStreamProtocol.resume_writingr  r  r}   r   r   r  r}   r   )r  r  r}   r   )r}   rr   r   )rg   rh   ri   r  r   r  r  r  r  r  r  r  rm   rm   rm   rn   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   rF   r  r  r   r  rm   rm   rn   r    s   

z DatagramProtocol.connection_mader   c                 C  s   | j   | j  d S r~   )r  r   r  r  rm   rm   rn   r    s   
z DatagramProtocol.connection_lostr  r  addrrP   c                 C  s&   t |}| j||f | j  d S r~   )rA   r  rJ  r  r   )r{   r  r  rm   rm   rn   datagram_received  s   z"DatagramProtocol.datagram_receivedr   c                 C  s
   || _ d S r~   )r   r  rm   rm   rn   error_received     
zDatagramProtocol.error_receivedc                 C  r  r~   )r  clearr   rm   rm   rn   r    r  zDatagramProtocol.pause_writingc                 C  r  r~   r  r   rm   rm   rn   r    r  zDatagramProtocol.resume_writingr  r  )r  r  r  rP   r}   r   )r   r   r}   r   r   )rg   rh   ri   r  r   r  r  r  r  r  r  rm   rm   rm   rn   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  rI   _receive_guard_send_guardr  r{   r  r  rm   rm   rn   r|     s
   


zSocketStream.__init__r}   socket.socketc                 C     | j dS Nsocketr  Zget_extra_infor   rm   rm   rn   _raw_socket     zSocketStream._raw_socketr  r  r<  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rMt| jj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  Zresume_readingr   pause_readingrK  r  r  popleft
IndexErrorr  r=   r   r;   r>   len
appendleftr  )r{   r  chunkleftoverrm   rm   rn   r    sB   



 
  zSocketStream.receiver  r   c                   s   | j G t I d H  | jrt| jjd urt| jjz| j	| W n t
y9 } z	| j r4t| d }~ww | jj I d H  W d    d S 1 sNw   Y  d S r~   )r  rK  r  r  r=   r  r   r;   r  r  r   r  r  r   )r{   r  r   rm   rm   rn   r    s"   

"zSocketStream.sendc                   s(   z| j   W d S  ty   Y d S w r~   )r  	write_eofOSErrorr   rm   rm   rn   send_eof2  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   rm   rm   rn   r  8  s   

zSocketStream.acloseN)r  r  r  r  r}   r  r  r  r  r   )
rg   rh   ri   r|   r/  r  r  r  r  r  rm   rm   rm   rn   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 )_RawSocketMixinNrU  _receive_future_send_futureF
raw_socketr  c                 C  s   || _ td| _td| _d S )Nr  r  )_RawSocketMixin__raw_socketrI   r  r  r{   r  rm   rm   rn   r|   J  s   
z_RawSocketMixin.__init__r}   c                 C  r%  r~   )r  r   rm   rm   rn   r  O  r&  z_RawSocketMixin._raw_socketr   asyncio.AbstractEventLoopr;  c                   >   d fdd}t   }_ j|jd  || |S )Nr  r   r}   r   c                      `  j d S r~   )r  remove_readerr  r  r   r{   rm   rn   callbackT     z6_RawSocketMixin._wait_until_readable.<locals>.callbackr  r   r}   r   )r   r   r  
add_readerr  r   rb  r{   r   r  r  rm   r  rn   _wait_until_readableS  
   
z$_RawSocketMixin._wait_until_readablec                   r  )Nr  r   r}   r   c                   r  r~   )r  remove_writerr  r
  r  rm   rn   r  ^  r  z6_RawSocketMixin._wait_until_writable.<locals>.callbackr  )r   r   r  
add_writerr  r   rb  r  rm   r  rn   _wait_until_writable]  r  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)_closingr  filenor   r  r   r  r   rm   rm   rn   r  g  s   
z_RawSocketMixin.acloser  r  r  )r   r  r}   r;  r   )rg   rh   ri   r  r  r  r  r|   r/  r  r  r  r  rm   rm   rm   rn   r   E  s   
 




r   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  r  r   r  SHUT_WRr   rm   rm   rn   r  t  s   "zUNIXSocketStream.send_eofr  r  r<  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   rK  r  r  r  recvBlockingIOErrorr  r  r  r=   r;   r>   )r{   r  r   r  r   rm   rm   rn   r  x  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   rK  r  r  
memoryviewr  r  r  r  r  r  r=   r;   )r{   r  r   viewZ
bytes_sentr   rm   rm   rn   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   r<  r   r   arrayrK  r  r  r  recvmsgr  CMSG_LENitemsizer  r  r  r  r=   r;   r>   
SOL_SOCKET
SCM_RIGHTSr   	frombytesr  list)r{   r   r!  r   fdsr   ancdataflagsr  r   
cmsg_level	cmsg_type	cmsg_datarm   rm   rn   receive_fds  sN   
&zUNIXSocketStream.receive_fdsr   r,  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 emptyr#  )r   r   r   r<  rJ  r"   r  r$  rK  r  r  r  sendmsgr  r(  r)  r  r  r  r  r=   r;   )r{   r   r,  r   ZfilenosfdZfdarrayr   rm   rm   rn   send_fds  sB   

"zUNIXSocketStream.send_fdsNr   r  r  r  )r   r<  r!  r<  r}   r"  )r   r  r,  r3  r}   r   )rg   rh   ri   r  r  r  r2  r6  rm   rm   rm   rn   r  s  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 )TCPSocketListenerNr4  _accept_scopeFr  r  c                 C  s$   || _ ttjt | _td| _d S )Naccepting connections from)_TCPSocketListener__raw_socketr0   r   ZBaseEventLoopr   rw   rI   _accept_guardr  rm   rm   rn   r|     s   zTCPSocketListener.__init__r}   c                 C  r%  r~   )r:  r   rm   rm   rn   r    r&  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;  rK  r  rN   r8  rw   Zsock_acceptr  r   r   r	  r   r  
setsockoptr  IPPROTO_TCPTCP_NODELAYZconnect_accepted_socketr  r  )r{   client_sockZ_addrr  r  rm   rm   rn   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 r  )
r  r8  rw   r	  r  r   r  r   r	   r   r   rm   rm   rn   r    s   
zTCPSocketListener.acloser  r  r}   r<  r   )rg   rh   ri   r8  r  r  r|   r/  r  rB  r  rm   rm   rm   rn   r7    s   
 

r7  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 )UNIXSocketListenerr  r  c                 C  s"   || _ t | _td| _d| _d S )Nr9  F)_UNIXSocketListener__raw_socketr   rw   rI   r;  r  r  rm   rm   rn   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~   )rw   r	  rE  _r   rm   rn   r   6  s    z+UNIXSocketListener.accept.<locals>.<lambda>)rK  r  r;  rE  rB  setblockingr  r  r   r   rw   r  r   rb  r  r  r=   r;   )r{   rA  rG  r  r   rm   r   rn   rB  *  s0   

zUNIXSocketListener.acceptr   c                   s   d| _ | j  d S rH  )r  rE  r   r   rm   rm   rn   r  ?  s   zUNIXSocketListener.aclosec                 C  r%  r~   )rE  r   rm   rm   rn   r  C  r&  zUNIXSocketListener._raw_socketNr  rC  r   r  )rg   rh   ri   r|   rB  r  r/  r  rm   rm   rm   rn   rD  #  s    


rD  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  rm   rm   rn   r|   I  
   


zUDPSocket.__init__r}   r  c                 C  r  r  r  r   rm   rm   rn   r  R  r  zUDPSocket._raw_socketr   c                   $   d| _ | j s| j  d S d S rH  r  r  r  r   r   rm   rm   rn   r  V  
   
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  rK  r  r  r  r  r  r  r  r   r  r  r  r=   r;   r   rm   rm   rn   r  [  s   
zUDPSocket.receiver  rR   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  rK  r  r  r  r   r  r=   r  r  r;   sendtor{   r  rm   rm   rn   r  l     
"zUDPSocket.sendNr  rJ  r  r  r  r   )r}   rO  )r  rR   r}   r   	rg   rh   ri   r|   r/  r  r  r  r  rm   rm   rm   rn   rI  H  s    
	

rI  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  rJ  r  r  c                 C  r  r  r  r  rm   rm   rn   r|   y  rK  zConnectedUDPSocket.__init__r}   r  c                 C  r  r  r  r   rm   rm   rn   r    r  zConnectedUDPSocket._raw_socketr   c                   rL  rH  rM  r   rm   rm   rn   r    rN  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  rP  )r{   packetrm   rm   rn   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~   rQ  rS  rm   rm   rn   r    rT  zConnectedUDPSocket.sendNrU  r  r   r}   r  r  rV  rm   rm   rm   rn   rW  x  s    
	

rW  c                   @      e Zd Zd
ddZdddZd	S )UNIXDatagramSocketr}   rS   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   rK  r  r  r  recvfromr  r  r  r  r=   r;   r{   r   r  r   rm   rm   rn   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   rK  r  r  r  rR  r  r  r  r  r=   r;   r{   r  r   r   rm   rm   rn   r    r`  zUNIXDatagramSocket.sendN)r}   rS   )r  rS   r}   r   rg   rh   ri   r  r  rm   rm   rm   rn   r[        
r[  c                   @  rZ  )ConnectedUNIXDatagramSocketr}   r  c                   r\  r]  )r   rK  r  r  r  r  r  r  r  r  r=   r;   r_  rm   rm   rn   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   rK  r  r  r  r  r  r  r  r  r=   r;   ra  rm   rm   rn   r    r`  z ConnectedUNIXDatagramSocket.sendNrY  r  rb  rm   rm   rm   rn   rd    rc  rd  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 )rF   r}   c                 C  r   r~   r   r   rm   rm   rn   r     r  zEvent.__new__r   c                 C  r  r~   )r   rF   _eventr   rm   rm   rn   r|     r  zEvent.__init__c                 C  r  r~   )rj  r   r   rm   rm   rn   r     r  z	Event.setr   c                 C  
   | j  S r~   )rj  r  r   rm   rm   rn   r    r  zEvent.is_setc                   s0   |   rt I d H  d S | j I d H  d S r~   )r  rK  r  rj  r   r   rm   rm   rn   r     s   z
Event.waitr4   c                 C  s   t t| jjS r~   )r4   r  rj  _waitersr   rm   rm   rn   
statistics  r  zEvent.statisticsN)r}   rF   r   r,  )r}   r4   )	rg   rh   ri   r   r|   r   r  r   rm  rm   rm   rm   rn   rF     s    




rF   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 )rH   Ffast_acquirero  r   r}   c                C  r   r~   r   r   ro  rm   rm   rn   r     r  zLock.__new__r   c                C  s   || _ d | _t | _d S r~   )_fast_acquire_owner_taskr   rl  )r{   ro  rm   rm   rn   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   rr  rl  rK  r  rq  rL  r   releaser   r   rJ  r  )r{   r   futr  rm   rm   rn   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rs  )r0   r   r   r   rr  rl  r   r@   )r{   r   rm   rm   rn   acquire_nowait=  s   
zLock.acquire_nowaitc                 C  s
   | j d uS r~   )rr  r   rm   rm   rn   lockedH  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)rr  r   r   rl  r   r   )r{   r   ru  rm   rm   rn   rt  K  s   

zLock.releaser5   c                 C  s*   | j rt| j nd }t|  |t| jS r~   )rr  AsyncIOTaskInfor5   rx  r  rl  )r{   Z	task_inform   rm   rn   rm  W  s   zLock.statisticsN)ro  r   r}   rH   )ro  r   r}   r   r   r,  )r}   r5   )
rg   rh   ri   r   r|   rv  rw  rx  rt  rm  rm   rm   rm   rn   rH     s    

"

rH   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 )#rL   NF	max_valuero  initial_valuer<  r{  r3  ro  r   r}   c                C  r   r~   r   r   r|  r{  ro  rm   rm   rn   r   ]  s   
zSemaphore.__new__c                  s.   t  j||d || _|| _|| _t | _d S )N)r{  )rr  r|   _value
_max_valuerq  r   rl  )r{   r|  r{  ro  rw  rm   rn   r|   f  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~  rl  rK  r  rq  rL  r   rt  r   r   rJ  r  r   r{   ru  rm   rm   rn   rv  s  s4   
zSemaphore.acquirec                 C  s    | j dkrt|  j d8  _ d S r  )r~  r@   r   rm   rm   rn   rw    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   rl  r   r   r  r  rm   rm   rn   rt    s   

zSemaphore.releasec                 C  r%  r~   )r~  r   rm   rm   rn   r'    r&  zSemaphore.valuec                 C  r%  r~   )r  r   rm   rm   rn   r{    r&  zSemaphore.max_valuerJ   c                 C  s   t t| jS r~   )rJ   r  rl  r   rm   rm   rn   rm    r  zSemaphore.statistics)r|  r<  r{  r3  ro  r   r}   rL   )r|  r<  r{  r3  ro  r   r   r  r  )r}   rJ   )rg   rh   ri   r   r|   rv  rw  rt  r/  r'  r{  rm  r  rm   rm   rw  rn   rL   \  s    


rL   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 );rD   r   r   _total_tokenstotal_tokensr}   c                 C  r   r~   r   r   r  rm   rm   rn   r     r  zCapacityLimiter.__new__c                 C  s   t  | _t | _|| _d S r~   )r   
_borrowersr   _wait_queuer  )r{   r  rm   rm   rn   r|     s   
zCapacityLimiter.__init__r   c                   s   |   I d H  d S r~   )rv  r   rm   rm   rn   rI       zCapacityLimiter.__aenter__r   r   r   r   r   r   c                   s   |    d S r~   )rt  r   rm   rm   rn   rP    s   zCapacityLimiter.__aexit__c                 C  r%  r~   )r  r   rm   rm   rn   r    r&  zCapacityLimiter.total_tokensr'  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   r<  r  isinfr`  r   maxr  r  popitemr   )r{   r'  Zwaiters_to_notifyeventrm   rm   rn   r    s   
r<  c                 C  s
   t | jS r~   )r  r  r   rm   rm   rn   borrowed_tokens     
zCapacityLimiter.borrowed_tokensc                 C  s   | j t| j S r~   )r  r  r  r   rm   rm   rn   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  rm   rm   rn   _notify_next_waiter  s   z#CapacityLimiter._notify_next_waiterc                 C     |  t  d S r~   )acquire_on_behalf_of_nowaitr   r   rm   rm   rn   rw    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  rm   rm   rn   r    s   
z+CapacityLimiter.acquire_on_behalf_of_nowaitc                   s   |  t I d H S r~   )acquire_on_behalf_ofr   r   rm   rm   rn   rv    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~   )rK  r  r  r@   r   rF   r  r   rM  popr  r  r  r
  rL  rt  )r{   r  r  rm   rm   rn   r    s.   
z$CapacityLimiter.acquire_on_behalf_ofc                 C  r  r~   )release_on_behalf_ofr   r   rm   rm   rn   rt    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  r  r  r   r  r  rm   rm   rn   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   rm   rm   rn   rm    s   zCapacityLimiter.statisticsN)r  r   r}   rD   )r  r   r   r   r-  r.  r  )r  r   r}   r   )r}   r3   )rg   rh   ri   r  r  r   r|   rI  rP  r/  r  r0  r  r  r  rw  r  rv  r  rt  r  rm  rm   rm   rm   rn   rD     s,   
 











rD   _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   rw   r   _signal_queuer   r   r=  r   _handled_signals)r{   r  rm   rm   rn   r|   .  s
   
z_SignalReceiver.__init__r   r%   r}   r   c                 C  s*   | j | | j s| jd  d S d S r~   )r  rJ  r=  r   r   )r{   r   rm   rm   rn   _deliver5  s   
z_SignalReceiver._deliverc                 C  s2   t | jD ]}| j|| j| | j| q| S r~   )r   r  rw   Zadd_signal_handlerr  r  r
  )r{   sigrm   rm   rn   r   :  s   z_SignalReceiver.__enter__r   r   r   r   r   r   c                 C  s   | j D ]}| j| qd S r~   )r  rw   Zremove_signal_handler)r{   r   r   r   r  rm   rm   rn   r   A  s   
z_SignalReceiver.__exit__c                 C  s   | S r~   rm   r   rm   rm   rn   	__aiter__J  s   z_SignalReceiver.__aiter__c                   s6   t  I d H  | jst | _| jI d H  | j S r~   )rK  r  r  r   r   r=  r  r   rm   rm   rn   	__anext__M  s   

z_SignalReceiver.__anext__N)r  r  )r   r%   r}   r   )r}   r  r   )r}   r%   )	rg   rh   ri   r|   r  r   r   r  r  rm   rm   rm   rn   r  -  s    




	r  c                      s(   e Zd Zd	 fddZd
ddZ  ZS )ry  r   r   c                   s`   t |}|d u rd }n|j}| }|d usJ dt t||| | t	|| _
d S )Nz&created TaskInfo from a completed Task)r   r   r1  r   rr  r|   r$  get_nameweakrefrefrV  )r{   r   r  r1  r   rw  rm   rn   r|   \  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)rV  r  r   r  r   r   r   r   r   r   r  )r{   r   r  r   rm   rm   rn   has_pending_cancellationh  s   
z(AsyncIOTaskInfo.has_pending_cancellation)r   r   r,  )rg   rh   ri   r|   r  r  rm   rm   rw  rn   ry  [  s    ry  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)rp   
use_uvlooprq   rp   rr   r  r   rq   rs   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   ro   )	r  platformuvloopr   winloopr\   _runnerrF  _runner_task)r{   rp   r  rq   r  r  rm   rm   rn   r|     s   

zTestRunner.__init__c                 C  s   | j   |  | j | S r~   )r  r   r   Zset_exception_handler_exception_handlerr   rm   rm   rn   r     s   
zTestRunner.__enter__r   r   r   r   r   r   c                 C  s   | j ||| d S r~   )r  r   r   rm   rm   rn   r     s   zTestRunner.__exit__r   c                 C  rk  r~   )r  r   r   rm   rm   rn   r     r  zTestRunner.get_loopr   r  r   dict[str, Any]c                 C  s2   t |dtr| j|d  d S || d S )Nr   )r   r   r   rF  rJ  Zdefault_exception_handler)r{   r   r   rm   rm   rn   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)rF  r  rc   )r{   ra   rm   rm   rn   _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)Z_pytest.outcomesr  r  r   r   r   r   rM  r   r   r   r   )r{   r  r  r   r   retvalr   rm   rm   rn   _run_tests_and_fixtures  s0   

Pz"TestRunner._run_tests_and_fixturesr    Callable[P, Awaitable[T_Retval]]r   P.argsrc  P.kwargsr   c                  sr   | j stttt tjf  d\| _}|  	| 
|| _ ||i |}|   }| j||f |I d H S r=  )r  rB   r  r   r-   r   r   r  r   r   r  r   Zsend_nowait)r{   r   r   rc  r  r   r   rm   rm   rn   _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 stoprm   )r   r   r  asendr  StopAsyncIterationr  r   )r{   r  rc  ZasyncgenZfixturevaluerm   rm   rn   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  rc  r  rm   rm   rn   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   rF  rJ  r  )r{   r  rc  r   rm   rm   rn   run_test	  s   zTestRunner.run_test)rp   rr   r  r   rq   rs   r}   r   )r}   r  r   r   )r   r  r   r  r}   r   r   )r  r  r}   r   )r   r  r   r  rc  r  r}   r   )r  r  rc  r  r}   r  )r  r  rc  r  r}   r   )r  r  rc  r  r}   r   )rg   rh   ri   r  r|   r   r   r   r  r  r  r  r  r  r  rm   rm   rm   rn   r  |  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 )rK  r   1Callable[[Unpack[PosArgsT]], Awaitable[T_Retval]]r   rR  rc  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   rm   rn   wrapper	  s   z#AsyncIOBackend.run.<locals>.wrapperrp   rq   r  Fr  r   ro   r}   r   )	r   r   r  r  r  r   r  r\   r   )r   r   r   rc  r  r  rp   rq   r  r  runnerrm   r  rn   r   	  s   


$zAsyncIOBackend.runr   c                 C     t  S r~   )r   ri  rm   rm   rn   current_token8	  r&  zAsyncIOBackend.current_tokenr   c                 C  s
   t   S r~   )r   r  ri  rm   rm   rn   rt  <	  r  zAsyncIOBackend.current_timetype[BaseException]c                 C  s   t S r~   )r   ri  rm   rm   rn   cancelled_exception_class@	  s   z(AsyncIOBackend.cancelled_exception_classr   c                   s   t dI d H  d S r  r	   ri  rm   rm   rn   r  D	     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  r  r	   r   r   )r   r   r   rm   rm   rn   r  H	  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 )NTrk  r   )rN   r	   ri  rm   rm   rn   rL  [	  s   "z)AsyncIOBackend.cancel_shielded_checkpointdelayc                   s   t |I d H  d S r~   r  )r   r  rm   rm   rn   r	   `	  r  zAsyncIOBackend.sleepFr   r   r   r   rN   c                C  s   t ||dS )Nr   rM   r   rm   rm   rn   create_cancel_scoped	  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   r  r  r   r   r  minr   r   r   r   )r   r   r   r   rm   rm   rn   current_effective_deadlinej	  s(   

z)AsyncIOBackend.current_effective_deadlineabc.TaskGroupc                 C  r  r~   )rE  ri  rm   rm   rn   create_task_group	  r&  z AsyncIOBackend.create_task_group	abc.Eventc                 C  r  r~   rE   ri  rm   rm   rn   create_event	  r&  zAsyncIOBackend.create_eventro  abc.Lockc                C  s
   t |dS )Nrn  rG   rp  rm   rm   rn   create_lock	  r  zAsyncIOBackend.create_lockNrz  r|  r<  r{  r3  abc.Semaphorec                C  s   t |||dS )Nrz  rK   r}  rm   rm   rn   create_semaphore	  s   zAsyncIOBackend.create_semaphorer  abc.CapacityLimiterc                 C     t |S r~   rC   r  rm   rm   rn   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 )Nrk  r   r   )!r  r  r   r  LookupErrorr   r   current_default_thread_limiterrN   r   r   r   r   rl  r   r
  rb  r  r   r   r  rt  ru  r  r  r   Zremove_done_callbackr   r   r9   r   rs  r  )r   r   r   r  r  rp  rn  r   r   r   ZworkernowZexpired_workerr   Zworker_scoperm   rm   rn   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~  r  r   r$  r   r   )r   r   rm   rm   rn   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 rH  )r0   r   r   r   r  r   r  r
  r  r   
concurrentZfuturesr   )Z__tracebackhide__r   r   r   r   r   rm   rn   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   Zrun_coroutine_threadsafery  )r   r   r   r  r  r   r   r  rm   r  rn   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   rM  r   r   r   r   r   r  r   rm   rn   r  
  s   

z4AsyncIOBackend.run_sync_from_thread.<locals>.wrapperr   r   )
r   r0   r:   r  r  r  r?   r   r   ry  )r   r   r   r  r  r   rm   r  rn   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   Zcreate_subprocess_shellZcreate_subprocess_execr  r  r  r  r  r  )
r   r  r  r  r  rc  r  Zstdin_streamZstdout_streamZstderr_streamrm   rm   rn   open_process"
  s4   


zAsyncIOBackend.open_processrn  r  c                 C  s&   t t|dd t tt| d S )Nz AnyIO process pool shutdown taskr_  )r   r  r   rb  r   r  )r   rn  rm   rm   rn   #setup_process_pool_exit_at_shutdownF
  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_connectionr  r  )r   r  r  r   r  r  rm   rm   rn   connect_tcpP
  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~   r  rF  r   r  rm   rn   r   i
     
 z-AsyncIOBackend.connect_unix.<locals>.<lambda>)r  r   r  AF_UNIXrH  connectr  r   r   r  r   rb  rM  r   r  )r   r  r  rm   r
  rn   connect_unix]
  s&   

zAsyncIOBackend.connect_unixsockr  rQ   c                 C  r  r~   r7  r   r  rm   rm   rn   create_tcp_listenerq
  r  z"AsyncIOBackend.create_tcp_listenerc                 C  r  r~   )rD  r  rm   rm   rn   create_unix_listeneru
  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)r  Zremote_addrr  r  )r   create_datagram_endpointr  r   r   rI  rW  )r   r  r   r  r  r  r  rm   rm   rn   create_udp_sockety
  s   

z AsyncIOBackend.create_udp_socketr  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                   r  r~   r	  rF  r
  rm   rn   r   
  r  z<AsyncIOBackend.create_unix_datagram_socket.<locals>.<lambda>)r  r   r  r  r   r   r  r   rb  rM  r   rd  r[  )r   r  r  r  rm   r
  rn   create_unix_datagram_socket
  s&   
z*AsyncIOBackend.create_unix_datagram_socketr   r  rN  protor.  bytes | str | Nonestr | int | Noneint | AddressFamilyrN  int | SocketKindr  r.  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   r  r  r  rN  r  r.  rm   rm   rn   r%  
  s   zAsyncIOBackend.getaddrinfosockaddrrP   tuple[str, str]c                   s   t  ||I d H S r~   )r   getnameinfo)r   r&  r.  rm   rm   rn   r(  
  s   zAsyncIOBackend.getnameinfoobjrX   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 rH  r  r   r   r@  rm   r5  ru  re  r	  rm   rn   r   
     
z(AsyncIOBackend.wait_readable.<locals>.cbr   get_selectorr   )rf  r   r  r   r   r<  r  r<   r   r   r  r  $anyio._core._asyncio_selector_threadr.  r	  r  r=   r   r)  r   r   r.  selectorsuccessrm   r+  rn   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 rH  r*  rm   r5  ru  r  rg  rm   rn   r   
  r,  z(AsyncIOBackend.wait_writable.<locals>.cbr   r-  r   )rh  r   r  r   r   r<  r  r<   r   r   r  r  r/  r.  r  r  r=   r0  rm   r5  rn   wait_writable
  r4  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   r<  r  r   rh  r   r  r  r  r   r   r@  r  r  r/  r.  rf  r	  )r   r)  r5  r   rg  ru  r.  re  rm   rm   rn   notify_closing0  sX   zAsyncIOBackend.notify_closingc                   
   t |S r~   r  r  rm   rm   rn   wrap_listener_socketa     z#AsyncIOBackend.wrap_listener_socketr  c                   &   t  jt|dI d H \}}t||S N)r  )r   r  r  r  r   r  r  r  rm   rm   rn   wrap_stream_sockete  
   
z!AsyncIOBackend.wrap_stream_socketr  c                   r8  r~   )r  r  rm   rm   rn   wrap_unix_stream_socketl  r:  z&AsyncIOBackend.wrap_unix_stream_socketrI  c                   r;  r<  )r   r  r  rI  r=  rm   rm   rn   wrap_udp_socketp  r?  zAsyncIOBackend.wrap_udp_socketrW  c                   r;  r<  )r   r  r  rW  r=  rm   rm   rn   wrap_connected_udp_socketw  r?  z(AsyncIOBackend.wrap_connected_udp_socketr[  c                   r8  r~   )r[  r  rm   rm   rn   wrap_unix_datagram_socket~  r:  z(AsyncIOBackend.wrap_unix_datagram_socketrd  c                   r8  r~   )rd  r  rm   rm   rn   #wrap_connected_unix_datagram_socket  s   z2AsyncIOBackend.wrap_connected_unix_datagram_socketrD   c                 C  s4   zt  W S  ty   td}t | | Y S w )N(   )r  r   r  rD   r   )r   r  rm   rm   rn   r    s   

z-AsyncIOBackend.current_default_thread_limiterr  r%   .AbstractContextManager[AsyncIterator[Signals]]c                 G  r  r~   )r  )r   r  rm   rm   rn   open_signal_receiver  s   z#AsyncIOBackend.open_signal_receiverr6   c                 C  s
   t t S r~   )ry  r   ri  rm   rm   rn   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 rm   )r   ry  )r   r   rm   rm   rn   r     s    z4AsyncIOBackend.get_running_tasks.<locals>.<listcomp>)r   ri  rm   rm   rn   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   r  r   r	   )r   Z	this_taskr   r  rm   rm   rn   wait_all_tasks_blocked  s   
z%AsyncIOBackend.wait_all_tasks_blockedr  c                 C  s   t di |S )Nrm   )r  )r   r  rm   rm   rn   create_test_runner  s   z!AsyncIOBackend.create_test_runner)
r   r  r   rR  rc  r  r  r  r}   r   )r}   r   r-  )r}   r  r   )r  r   r}   r   r*  )r}   r  )r}   r  )ro  r   r}   r  )r|  r<  r{  r3  ro  r   r}   r  )r  r   r}   r  )FN)
r   r  r   rR  r  r   r  r  r}   r   )r   r  r   rR  r  r   r}   r   )r   r  r   rR  r  r   r}   r   )r  r  r  r  r  r  r  r  rc  r-   r}   r  rn  r  r}   r   r~   )r  r   r  r<  r   r  r}   r<  )r  r  r}   r  )r  r  r}   rQ   )
r  r&   r   r  r  r  r  r   r}   r  )r  r  r  r  r}   r  )r  r   r  r!  r  r"  rN  r#  r  r<  r.  r<  r}   r$  )r   )r&  rP   r.  r<  r}   r'  )r)  rX   r}   r   )r  r  r}   r  )r  r  r}   r  )r  r  r}   rI  )r  r  r}   rW  )r  r  r}   r[  )r  r  r}   rd  )r}   rD   )r  r%   r}   rF  )r}   r6   )r}   rI  )r  r  r}   r  )3rg   rh   ri   classmethodr   r  rt  r  r  r  rL  r	   r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r%  r(  r3  r6  r7  r9  r>  r@  rA  rB  rC  rD  r  rG  rH  rJ  rK  rL  rm   rm   rm   rn   rK  	  s    !	<"#	440rK  )r   r   r}   r   )r}   r   )r   r   r}   r   )r   r   r}   r   )r   r   r}   r   )rn  r  rV  r   r}   r   rM  )
__future__r   r$  r   concurrent.futuresr  r   r  r  r  r  r   r  r   r   r   r   r   r   r	   Z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#   rs  r$   r   r%   r&   r'   r(   typesr)   r*   typingr+   r,   r-   r.   r/   r0   r1   r\  r3   r4   r5   r6   r7   Z_core._eventloopr8   r9   r:   Z_core._exceptionsr;   r<   r=   r>   r?   r@   Z_core._socketsrA   Z_core._streamsrB   Z_core._synchronizationrD   ZBaseCapacityLimiterrF   Z	BaseEventrH   ZBaseLockrI   rJ   rL   ZBaseSemaphoreZ_core._tasksrN   ZBaseCancelScoperO   rP   rQ   rR   rS   Zabc._eventlooprT   ZlowlevelrU   Zstreams.memoryrV   rW   Z	_typeshedrX   r   r  r\   r]   r^   enumr_   r`   ra   rb   Zexceptiongrouprc   Ztyping_extensionsEnumrd   r   r   r   r   r   r   r   r  r   r   r   r   r   r  r   Z
TaskStatusr:  Zeager_task_factoryr]  rD  rE  r  rM  Z_Retval_Queue_Typerl  r  r  ZByteReceiveStreamr  ZByteSendStreamr  r  r  r  Protocolr  r  r  r   r  r7  rD  rI  rW  r[  rd  rf  rh  r  r  ry  r  rK  Zbackend_classrm   rm   rm   rn   <module>   s    $	(
   





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