Format: application/x-python-code
1 o
2
3 ;'h� � @ s� U d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
4 Z
5 m Z m Z m
6 Z
7 mZmZmZmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d d lm Z d d
8 l m Z d d lm Z m Z d d lm!Z! d d
9 l"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z* d dl+Z,d dl-Z,d dl.m/Z/m0Z0m1Z1 d dl-m2Z2m3Z3m4Z4m5Z5 d dl6m7Z8 d dl9m:Z: ddl;m<Z<m=Z=m>Z>m?Z?m@Z@mAZA ddlBmCZC ddlDmEZEmFZFmGZGmHZH ddlImJZJ ddlKmLZL ddlMmNZO ddlMmPZQ ddlMmRZS ddlMmTZTmUZU dd lMmVZW dd lXmYZZ dd lAm[Z[m\Z\m]Z] dd l^m_Z_m`Z` dd lambZb e$�r"d d!lcmdZd ejed"k�r/d d#l"mfZf nd d#lgmfZf ejed$k�rDd d%l"mhZhmiZi nd d&ljmkZk d d%lgmhZhmiZi e(d'�Zle(d(�Zme(d)ene[�Zoehd*�Zpefd+�Zqe,jrjsZsG d,d-� d-eZ�ZYG d.d/� d/eAjt�ZtG d0d1� d1eAju�Zued2d3�G d4d5� d5eAjv��Zwed2d3�G d6d7� d7eAjx��Zyed2d3�G d8d9� d9eAjz��ZzG d:d;� d;e,jAj{�Z|esd<�Z}d=e~d<< dgdBdC�ZG dDdE� dEe&eo �Z�G dFdG� dGe�eAj��Z�G dHdI� dIe�eAj��Z�G dJdK� dKe�eAj��Z�G dLdM� dMe�eAj��Z�G dNdO� dOe�e[ eAj��Z�G dPdQ� dQe�e[ eAj��Z�G dRdS� dSe�en eAj��Z�G dTdU� dUe�en eAj��Z�G dVdW� dWeQ�ZPG dXdY� dYeS�ZRG dZd[� d[eW�ZVG d\d]� d]eO�ZNesd^�Z�d=e~d^< G d_d`� d`�Z�G dadb� dbeAj��Z�G dcdd� dde?�Z�G dedf� dfe_�Z�e�Z�dS )h� )� annotationsN)�AsyncGenerator�
10 AsyncIterator� Awaitable�Callable�
11 Collection� Coroutine�Iterable�Sequence)�Future)�AbstractContextManager)� dataclass)�partial)�IOBase)�PathLike)�Signals)�
12 AddressFamily�
13 SocketKind)�
14 TracebackType)�IO�
15 TYPE_CHECKING�Any�Generic�NoReturn�TypeVar�cast�overload)�Error�Outcome�Value)�current_root_task� current_task�
16 wait_readable�
17 wait_writable)�
18 SocketType)�run_sync� )�CapacityLimiterStatistics�EventStatistics�LockStatistics�TaskInfo�
19 WouldBlock�abc)�claim_worker_thread)�BrokenResourceError�BusyResourceError�ClosedResourceError� EndOfStream)�convert_ipv6_sockaddr)�create_memory_object_stream��CapacityLimiter��Event��Lock)�
20 ResourceGuard�SemaphoreStatistics�� Semaphore�� CancelScope)�IPSockAddrType�
21 UDPPacketType�UNIXDatagramPacketType)� AsyncBackend�StrOrBytesPath)�MemoryObjectSendStream)� HasFileno)� �
22 )� ParamSpec)rG � )� TypeVarTuple�Unpack)�BaseExceptionGroup�T�T_Retval�
23 T_SockAddr�PosArgsT�Pc @ s� e Zd Z d&d'dd�Zd&d(d d �Zd)d
24 d�Zd*dd�Zd+dd�Zed,dd ��Z e j
25 d-d d ��Z ed.d d ��Z ed.d!d"��Z ed.d#d$��Z
26 e
27 j
28 d/d%d$��Z
29 dS )0r? N�original�trio.CancelScope | None�kwargs�object�returnc K �
30 t �| �S �N�rV �__new__)�clsrS rU � r] �KC:\Users\shubh\Videos\agent\nova\Lib\site-packages\anyio/_backends/_trio.pyr[ v s
31 zCancelScope.__new__r �Nonec K s |p t jdi |��| _d S �Nr] )�trior? �_CancelScope__original)�selfrS rU r] r] r^ �__init__{ s zCancelScope.__init__c C s | j �� | S rY )rb � __enter__�rc r] r] r^ re ~ s
32 zCancelScope.__enter__�exc_type�type[BaseException] | None�exc_val�BaseException | None�exc_tb�TracebackType | None�boolc C � | j �|||�S rY )rb �__exit__�rc rg ri rk r] r] r^ ro � � zCancelScope.__exit__c C � | j �� d S rY )rb �cancelrf r] r] r^ rs � � zCancelScope.cancel�floatc C � | j jS rY �rb �deadlinerf r] r] r^ rx � � zCancelScope.deadline�valuec C � || j _d S rY rw �rc rz r] r] r^ rx � � c C rv rY )rb �
33 cancel_calledrf r] r] r^ r~ � ry zCancelScope.cancel_calledc C rv rY )rb �cancelled_caughtrf r] r] r^ r � ry z CancelScope.cancelled_caughtc C rv rY �rb �shieldrf r] r] r^ r� � ry zCancelScope.shieldc C r{ rY r� r| r] r] r^ r� � r} rY )rS rT rU rV rW r? )rS rT rU r rW r_ )rW r? �rg rh ri rj rk rl rW rm �rW r_ �rW ru �rz ru rW r_ �rW rm )rz rm rW r_ )�__name__�
34 __module__� __qualname__r[ rd re ro rs �propertyrx �setterr~ r r� r] r] r] r^ r? u s& �
35
36
37 r? c @ sJ e Zd Zd dd�Zd dd�Zd dd�Zdd�d!dd�Zdd�d"d d �ZdS )#� TaskGrouprW r_ c C s d| _ tjdd�| _d | _d S )NFT)Zstrict_exception_groups)�_activera Z open_nursery�_nursery_manager� cancel_scoperf r] r] r^ rd � s
38 zTaskGroup.__init__c � s, �d| _ | j�� I d H | _t| jj�| _| S )NT)r� r� �
39 __aenter__�_nurseryr? r� rf r] r] r^ r� � s
40 �zTaskGroup.__aenter__rg rh ri rj rk rl rm c
41 � sl �z.z| j �|||�I d H W W ~~d| _S ty/ } z|�tj�d s*tj�� |�� d }~ww ~~d| _w )NF� )r� � __aexit__r� rM �splitra � Cancelled�_create)rc rg ri rk �excr] r] r^ r� � s �� ��zTaskGroup.__aexit__N��name�func�,Callable[[Unpack[PosArgsT]], Awaitable[Any]]�args�Unpack[PosArgsT]r� rV c G s, | j std��| jj|g|�R d|i� d S �Nz;This task group is not active; no new tasks can be started.r� )r� � RuntimeErrorr� �
42 start_soon�rc r� r� r� r] r] r^ r� � s
43 � zTaskGroup.start_soon� Callable[..., Awaitable[Any]]r c � s0 �| j std��| jj|g|�R d|i�I d H S r� )r� r� r� �startr� r] r] r^ r� � s �� zTaskGroup.startr� )rW r� r� )r� r� r� r� r� rV rW r_ )r� r� r� rV r� rV rW r )r� r� r� rd r� r� r� r� r] r] r] r^ r� � s
44
45
46 ��r� c s2 e Zd Zddd�Zd� fdd� Zddd�Z� ZS )�BlockingPortalrW c C rX rY rZ �r\ r] r] r^ r[ � �
47 zBlockingPortal.__new__r_ c s t � �� tj�� | _d S rY )�superrd ra �lowlevel�current_trio_token�_tokenrf �� __class__r] r^ rd � s
48 zBlockingPortal.__init__r� �<Callable[[Unpack[PosArgsT]], Awaitable[T_Retval] | T_Retval]r� �tuple[Unpack[PosArgsT]]rU �dict[str, Any]r� rV �future�Future[T_Retval]c C s. t jjt| jj|d�| j||||| jd� d S )Nr� )Z
49 trio_token)ra � from_threadr% r Z _task_groupr� Z
50 _call_funcr� )rc r� r� rU r� r� r] r] r^ �_spawn_task_from_thread� s
51 �z&BlockingPortal._spawn_task_from_thread)rW r� r� ) r� r� r� r� rU r� r� rV r� r� rW r_ )r� r� r� r[ rd r� �
52 __classcell__r] r] r� r^ r� � s
53 r� F)�eqc @ s, e Zd ZU ded< d
54 ddd �Zdd d �ZdS )�ReceiveStreamWrapperztrio.abc.ReceiveStream�_streamN� max_bytes�
55 int | NonerW �bytesc
56 � sh �z | j �|�I d H }W n! tjy } zt|j�d }~w tjy- } zt|j�d }~ww |r2|S t�rY )r� Z receive_somera r0 � __cause__r. r1 �rc r� �datar� r] r] r^ �receive
57 s ����z ReceiveStreamWrapper.receiver_ c � � �| j �� I d H d S rY �r� �acloserf r] r] r^ r� � �zReceiveStreamWrapper.acloserY )r� r� rW r� r� )r� r� r� �__annotations__r� r� r] r] r] r^ r� s
58
59 r� c @ s* e Zd ZU ded< d dd�Zd
60 d d
61 �Zd S )�SendStreamWrapperztrio.abc.SendStreamr� �itemr� rW r_ c
62 � s^ �z | j �|�I d H W d S tjy } zt|j�d }~w tjy. } zt|j�d }~ww rY )r� Zsend_allra r0 r� r. �rc r� r� r] r] r^ �send s ����zSendStreamWrapper.sendc � r� rY r� rf r] r] r^ r� ' r� zSendStreamWrapper.acloseN�r� r� rW r_ r� )r� r� r� r� r� r� r] r] r] r^ r� s
63
64 r� c @ s� e Zd ZU ded< ded< ded< ded< d#d
65 d �Zd$d
66 d�Zd#dd�Zd#dd�Zd%dd�Ze d$dd��Z
67 e d&dd��Z e d'd d ��Z e d(d d ��Z
68 e d(d d!��Zd"S ))�Processz trio.Process�_process�abc.ByteSendStream | None�_stdin� abc.ByteReceiveStream | None�_stdout�_stderrrW r_ c � s� �t dd��) | jr| j�� I d H | jr | j�� I d H | jr(| j�� I d H W d � n1 s2w Y z
69 | �� I d H W d S tyi | �� t dd�� | �� I d H W d � � 1 scw Y � w �NT)r� )r? r� r� r� r� �wait�
70 BaseException�killrf r] r] r^ r� 2 s* � � �
71 ���zProcess.aclose�intc � � �| j �� I d H S rY )r� r� rf r] r] r^ r� C � �z Process.waitc C rr rY )r� � terminaterf r] r] r^ r� F rt zProcess.terminatec C rr rY )r� r� rf r] r] r^ r� I rt z Process.kill�signalr c C � | j �|� d S rY )r� � send_signal)rc r� r] r] r^ r� L � zProcess.send_signalc C rv rY )r� �pidrf r] r] r^ r� O ry z Process.pidr� c C rv rY )r� �
72 returncoderf r] r] r^ r� S ry zProcess.returncodec C � | j S rY )r� rf r] r] r^ �stdinW � z
73 Process.stdinc C r� rY )r� rf r] r] r^ �stdout[ r� zProcess.stdoutc C r� rY )r� rf r] r] r^ �stderr_ r� zProcess.stderrNr� �rW r� )r� r rW r_ �rW r� )rW r� )rW r� )r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r] r] r] r^ r� + s(
74
75
76
77
78
79 r� c s e Zd Zd� fdd� Z� ZS )� _ProcessPoolShutdownInstrumentrW r_ c s t � �� d S rY )r� � after_runrf r� r] r^ r� e rt z(_ProcessPoolShutdownInstrument.after_runr� )r� r� r� r� r� r] r] r� r^ r� d s r� �¤t_default_worker_process_limiterztrio.lowlevel.RunVar�workers�set[abc.Process]rW r_ c � s� �z t �tj�I d H W d S t jyI | D ] }|jd u r"|�� qtdd�� | D ] }|�� I d H q+W d � Y d S 1 sAw Y Y d S w r� ) ra �sleep�math�infr� r� r� r? r� )r� �processr] r] r^ �_shutdown_process_pooln s �
80 � �&��r� c @ sB e Zd Zddd�Zddd�Zedd
81 d ��Zdd d
82 �Zddd�ZdS )�_TrioSocketMixin� trio_socket�TrioSocketTyperW r_ c C s || _ d| _d S �NF)� _trio_socket�_closed�rc r� r] r] r^ rd � s
83 z_TrioSocketMixin.__init__c C s | j rt�| j�� dk rt�d S �Nr )r� r0 r� �filenor. rf r] r] r^ �
84 _check_closed� s
85 �z _TrioSocketMixin._check_closed�
86 socket.socketc C rv rY )r� �_sockrf r] r] r^ � _raw_socket� ry z _TrioSocketMixin._raw_socketc � s( �| j �� dkrd| _| j �� d S d S )Nr T)r� r� r� �closerf r] r] r^ r� � s
87 ��z_TrioSocketMixin.acloser� r� r c C s@ t |tj�r t|�| j�� dk r| jrtd �t |t�r t|�|�r� )�
88 isinstancera r0 r� r� r� �OSErrorr. �rc r� r] r] r^ �_convert_socket_error� s
89 z&_TrioSocketMixin._convert_socket_errorN�r� r� rW r_ r� )rW r� )r� r� rW r ) r� r� r� rd r� r� r� r� r r] r] r] r^ r� � s
90
91
92 r� c s> e Zd Zd� fdd� Zddd d �Zddd�Zddd�Z� ZS )� SocketStreamr� r� rW r_ c �$ t � �|� td�| _td�| _d S �N� reading from�
93 writing to�r� rd r: �_receive_guard� _send_guardr� r� r] r^ rd � �
94 zSocketStream.__init__� r� r� r� c � s~ �| j �1 z | j�|�I d H }W n ty' } z | �|� W Y d }~nd }~ww |r3|W d � S t�1 s8w Y d S rY )r r� �recvr� r r1 r� r] r] r^ r� � s ���� �zSocketStream.receiver� c � s� �| j �B t|�}|r>z | j�|�I d H }W n ty- } z | �|� W Y d }~nd }~ww ||d � }|s W d � d S W d � d S 1 sIw Y d S rY )r �
95 memoryviewr� r� r� r )rc r� �viewZ
96 bytes_sentr� r] r] r^ r� � s ��� ��"�zSocketStream.sendc � s �| j �tj� d S rY )r� �shutdown�socket�SHUT_WRrf r] r] r^ �send_eof� s �zSocketStream.send_eofr )r )r� r� rW r� r� r� )r� r� r� rd r� r� r r� r] r] r� r^ r � s
97
98 r c @ s e Zd Zddd�Zdd
99 d�ZdS )�UNIXSocketStream�msglenr� �maxfdsrW �tuple[bytes, list[int]]c � sR �t |t�r
100 |dk rtd��t |t�r|dk rtd��t�d�}tj�� I d H | j�? z| j� |t
101 � ||j ��I d H \}}}}W n t
102 yZ } z | �|� W Y d }~n d }~ww |sa|sat�nq-W d � n1 smw Y |D ].\} }
103 } | t
104 jks�|
105 t
106 jkr�td|�d| � d |
107 � ���|�| d t| �t| �|j � � qt|t|�fS )
108 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� �
109 ValueError�arrayra r� �
110 checkpointr r� �recvmsgr ZCMSG_LEN�itemsizer� r r1 �
111 SOL_SOCKET�
112 SCM_RIGHTSr� � frombytes�len�list) rc r r �fds�messageZancdata�flags�addrr� Z
113 cmsg_levelZ cmsg_typeZ cmsg_datar] r] r^ � receive_fds� sD �
114 ���� ����& z UNIXSocketStream.receive_fdsr# r� r" �Collection[int | IOBase]r_ c � s� �|st d��|s
115 t d��g }|D ]}t|t�r |�|� qt|t�r*|�|�� � qt�d|�}tj� � I d H | j
116 �6 z| j � |gt
117 jt
118 j|fg�I d H W n tyh } z | �|� W Y d }~nd }~ww q>W d � d S 1 suw Y d S )Nzmessage must not be emptyzfds must not be emptyr )r r� r� �appendr r� r ra r� r r r� �sendmsgr r r r� r )rc r# r" Zfilenos�fdZfdarrayr� r] r] r^ �send_fds� sB �
119
120 � ��
121 �
122 ��� "�zUNIXSocketStream.send_fdsN)r r� r r� rW r )r# r� r" r' rW r_ )r� r� r� r& r+ r] r] r] r^ r � s
123 !r c �( e Zd Zd � fdd� Zd
124 dd�Z� ZS ) �TCPSocketListener�
125 raw_socketr� c �" t � �tj�|�� td�| _d S �Nzaccepting connections from�r� rd ra r �from_stdlib_socketr: �
126 _accept_guard�rc r. r� r] r^ rd � zTCPSocketListener.__init__rW r c � s� �| j �, z | j�� I d H \}}W n ty( } z | �|� W Y d }~nd }~ww W d � n1 s3w Y |�tjtjd� t |�S �Nr� )
127 r3 r� �acceptr� r �
128 setsockoptr � IPPROTO_TCP� TCP_NODELAYr �rc r� Z_addrr� r] r] r^ r7 s ���� �zTCPSocketListener.accept�r. r� )rW r �r� r� r� rd r7 r� r] r] r� r^ r- � r- c r, ) �UNIXSocketListenerr. r� c r/ r0 r1 r4 r� r] r^ rd r5 zUNIXSocketListener.__init__rW r c � s� �| j �9 z | j�� I d H \}}W n ty( } z | �|� W Y d }~nd }~ww W d � t|�S W d � t|�S 1 s@w Y t|�S rY )r3 r� r7 r� r r r; r] r] r^ r7 s ����
129 ��
130 ��zUNIXSocketListener.acceptr< )rW r r= r] r] r� r^ r? r> r? c s2 e Zd Zd� fdd� Zddd �Zdd d
131 �Z� ZS )� UDPSocketr� r� rW r_ c r r r r� r� r] r^ rd # r
132 zUDPSocket.__init__� tuple[bytes, IPSockAddrType]c � s� �| j �: z| j�d�I d H \}}|t|�fW W d � S ty5 } z | �|� W Y d }~nd }~ww W d � d S 1 sAw Y d S �Nr )r r� �recvfromr2 r� r �rc r� r% r� r] r] r^ r� ( s � ���"�zUDPSocket.receiver� rA c � � �| j �4 z | jj|� I d H W n ty' } z | �|� W Y d }~n
133 d }~ww W d � d S W d � d S 1 s;w Y d S rY �r r� �sendtor� r r� r] r] r^ r� 0 � �����"�zUDPSocket.sendr )rW rA )r� rA rW r_ �r� r� r� rd r� r� r� r] r] r� r^ r@ " �
134 r@ c �2 e Zd Zd
135 � fdd� Zddd �Zdd d �Z� ZS )�ConnectedUDPSocketr� r� rW r_ c r r r r� r� r] r^ rd 9 r
136 zConnectedUDPSocket.__init__r� c � � �| j �2 z| j�d�I d H W W d � S ty- } z | �|� W Y d }~nd }~ww W d � d S 1 s9w Y d S rB �r r� r r� r r� r] r] r^ r� > � ����"�zConnectedUDPSocket.receiver� c � � �| j �4 z | j�|�I d H W n ty' } z | �|� W Y d }~n
137 d }~ww W d � d S W d � d S 1 s;w Y d S rY �r r� r� r� r r� r] r] r^ r� E rH zConnectedUDPSocket.sendr �rW r� r� rI r] r] r� r^ rL 8 s
138 rL c rK )�UNIXDatagramSocketr� r� rW r_ c r r r r� r� r] r^ rd N r
139 zUNIXDatagramSocket.__init__rB c � s� �| j �8 z| j�d�I d H \}}||fW W d � S ty3 } z | �|� W Y d }~nd }~ww W d � d S 1 s?w Y d S rB )r r� rC r� r rD r] r] r^ r� S s ����"�zUNIXDatagramSocket.receiver� c � rE rY rF r� r] r] r^ r� [ rH zUNIXDatagramSocket.sendr )rW rB )r� rB rW r_ rI r] r] r� r^ rS M rJ rS c rK )�ConnectedUNIXDatagramSocketr� r� rW r_ c r r r r� r� r] r^ rd f r
140 z$ConnectedUNIXDatagramSocket.__init__r� c � rM rB rN r� r] r] r^ r� k rO z#ConnectedUNIXDatagramSocket.receiver� c � rP rY rQ r� r] r] r^ r� r rH z ConnectedUNIXDatagramSocket.sendr rR r� rI r] r] r� r^ rT c s
141 rT c @ sH e Zd Zddd�Zddd�Zddd �Zdd
142 d �Zdd
143 d�Zddd�ZdS )r7 rW c C rX rY rZ r� r] r] r^ r[ � r� z
144 Event.__new__r_ c C s t �� | _d S rY )ra r7 �_Event__originalrf r] r] r^ rd � rt zEvent.__init__rm c C �
145 | j �� S rY )rU �is_setrf r] r] r^ rW � r� z Event.is_setc � r� rY )rU r� rf r] r] r^ r� � r� z
146 Event.waitr( c C s | j �� }t|jd�S )N)�
147 tasks_waiting)rU �
148 statisticsr( rX �rc �orig_statisticsr] r] r^ rY � s
149 zEvent.statisticsc C rr rY )rU �setrf r] r] r^ r\ � rt z Event.setN)rW r7 r� r� )rW r( ) r� r� r� r[ rd rW r� rY r\ r] r] r] r^ r7 s
150
151
152
153
154 r7 c @ sl e Zd Zdd�ddd�Zdd�d d d
155 �Zed d
156 d��Zd dd�Zd dd�Zd dd�Z d dd�Z
157 d dd�Z dS )!r9 F�� fast_acquirer^ rm rW c C rX rY rZ �r\ r^ r] r] r^ r[ � r� z Lock.__new__r_ c C s || _ t�� | _d S rY )�
158 _fast_acquirera r9 �_Lock__original)rc r^ r] r] r^ rd � s z
159 Lock.__init__r� r� c C s | j dkr
160 d| _ d S d S )N)z*attempt to re-acquire an already held Lock)z)Attempted to acquire an already held Lock)r� )r� r] r] r^ �_convert_runtime_error_msg� s
161
162 �z Lock._convert_runtime_error_msgc
163 � s� �| j s"z | j�� I d H W d S ty! } z| �|� � d }~ww tj�� I d H z| j�� W d S tj yF | jj
164 � � I d H Y d S tyW } z| �|� � d }~ww rY ) r` ra �acquirer� rb ra r� �checkpoint_if_cancelled�acquire_nowaitr+ �_lot�parkr� r] r] r^ rc � s( ��
165 ��
166 ��z Lock.acquirec
167 C sJ z| j �� W d S tjy td � ty$ } z| �|� � d }~ww rY )ra re ra r+ r� rb r� r] r] r^ re � s
168 ��zLock.acquire_nowaitc C rV rY )ra �lockedrf r] r] r^ rh � r� z Lock.lockedc C rr rY )ra �releaserf r] r] r^ ri � rt z Lock.releaser) c C s. | j �� }|jr
169 t|j�nd }t|j||j�S rY )ra rY �owner� TrioTaskInfor) rh rX )rc r[ rj r] r] r^ rY � s
170
171
172 �zLock.statisticsN�r^ rm rW r9 )r^ rm rW r_ )r� r� rW r_ r� r� )rW r) ) r� r� r� r[ rd � staticmethodrb rc re rh ri rY r] r] r] r^ r9 � s
173
174
175
176 r9 c s| e Zd Zddd�d d d �Zddd�d � fdd�Zd dd�Zd dd�Zed dd��Zed!dd��Z d dd�Z
177 d"dd �Z � Z S )#r= NF�� max_valuer^ �
178 initial_valuer� ro r� r^ rm rW c C rX rY rZ �r\ rp ro r^ r] r] r^ r[ � s
179 zSemaphore.__new__r_ c s& t � j|||d� tj||d�| _d S )Nrn )ro )r� rd ra r= �_Semaphore__original)rc rp ro r^ r� r] r^ rd � s zSemaphore.__init__c � sf �| j s| j�� I d H d S tj�� I d H z| j�� W d S tjy2 | jj� � I d H Y d S w rY )
180 r` rr rc ra r� rd re r+ rf rg rf r] r] r^ rc � s ��zSemaphore.acquirec C s( z| j �� W d S tjy td �w rY )rr re ra r+ rf r] r] r^ re � s
181 �zSemaphore.acquire_nowaitc C rv rY )rr ro rf r] r] r^ ro � ry zSemaphore.max_valuec C rv rY )rr rz rf r] r] r^ rz � ry zSemaphore.valuec C rr rY )rr ri rf r] r] r^ ri � rt zSemaphore.releaser; c C s | j �� }t|j�S rY )rr rY r; rX rZ r] r] r^ rY � s
182
183 zSemaphore.statistics)rp r� ro r� r^ rm rW r= )rp r� ro r� r^ rm rW r_ r� r� r� )rW r; )
184 r� r� r� r[ rd rc re r� ro rz ri rY r� r] r] r� r^ r= � s �
185 �
186
187
188
189 r= c @ s� e Zd Z d2dd�d3dd �Z d2dd�d4d d �Zd5d
190 d�Zd6dd�Zed7dd��Zej d8dd��Zed9d d ��Z
191 ed7d d ��Z d5d!d"�Z d:d%d&�Z
192 d5d'd(�Zd:d)d*�Zd5d+d,�Zd:d-d.�Zd;d0d1�ZdS )<r5 N�rS � total_tokens� float | NonerS �trio.CapacityLimiter | NonerW c C rX rY rZ )r\ rt rS r] r] r^ r[ s
193 zCapacityLimiter.__new__r_ c C s. |d ur || _ d S |d usJ �t�|�| _ d S rY )�_CapacityLimiter__originalra r5 )rc rt rS r] r] r^ rd s
194 zCapacityLimiter.__init__c � r� rY )rw r� rf r] r] r^ r� r� zCapacityLimiter.__aenter__rg rh ri rj rk rl c � s �| j �|||�I d H d S rY )rw r� rp r] r] r^ r� s �zCapacityLimiter.__aexit__ru c C rv rY �rw rt rf r] r] r^ rt " ry z CapacityLimiter.total_tokensrz c C r{ rY rx r| r] r] r^ rt & r} r� c C rv rY )rw �borrowed_tokensrf r] r] r^ ry * ry z CapacityLimiter.borrowed_tokensc C rv rY )rw �available_tokensrf r] r] r^ rz . ry z CapacityLimiter.available_tokensc C rr rY )rw re rf r] r] r^ re 2 rt z CapacityLimiter.acquire_nowait�borrowerrV c C r� rY )rw �acquire_on_behalf_of_nowait�rc r{ r] r] r^ r| 5 r� z+CapacityLimiter.acquire_on_behalf_of_nowaitc � r� rY )rw rc rf r] r] r^ rc 8 r� zCapacityLimiter.acquirec � s �| j �|�I d H d S rY )rw �acquire_on_behalf_ofr} r] r] r^ r~ ; s �z$CapacityLimiter.acquire_on_behalf_ofc C rV rY )rw ri rf r] r] r^ ri > r� zCapacityLimiter.releasec C s | j �|�S rY )rw �release_on_behalf_ofr} r] r] r^ r A s z$CapacityLimiter.release_on_behalf_ofr' c C s& | j �� }t|j|jt|j�|jd�S )N)ry rt � borrowersrX )rw rY r' ry rt �tupler� rX )rc �origr] r] r^ rY D s
195 �zCapacityLimiter.statisticsrY )rt ru rS rv rW r5 )rt ru rS rv rW r_ r� )rg rh ri rj rk rl rW r_ r� r� r� )r{ rV rW r_ )rW r' )r� r� r� r[ rd r� r� r� rt r� ry rz re r| rc r~ ri r rY r] r] r] r^ r5 s4 ��
196 ��
197
198
199
200
201
202
203
204 r5 �_capacity_limiter_wrapperc @ sH e Zd ZU ded< ddd�Zddd �Zddd�Zddd�Zd dd�ZdS ) �_SignalReceiverzAsyncIterator[int]� _iterator�signals�tuple[Signals, ...]c C s
205 || _ d S rY )�_signals)rc r� r] r] r^ rd Y r� z_SignalReceiver.__init__rW c C s t j| j� | _| j�� | _| S rY )ra �open_signal_receiverr� �_cmre r� rf r] r] r^ re \ s z_SignalReceiver.__enter__rg rh ri rj rk rl � bool | Nonec C rn rY )r� ro rp r] r] r^ ro a rq z_SignalReceiver.__exit__c C s | S rY r] rf r] r] r^ � __aiter__i s z_SignalReceiver.__aiter__r c � s �| j �� I d H }t|�S rY )r� � __anext__r )rc Zsignumr] r] r^ r� l s �z_SignalReceiver.__anext__N)r� r� )rW r� )rg rh ri rj rk rl rW r� )rW r ) r� r� r� r� rd re ro r� r� r] r] r] r^ r� V s
206
207
208
209
210 r� c @ s\ e Zd Zd,dd�Zd-d
211 d�Zd.dd�Zd/dd�Zd0d d �Zd1d"d#�Zd2d%d&�Z d3d)d*�Z
212 d+S )4�
213 TestRunner�optionsr rW r_ c K s$ ddl m} |� | _d | _|| _d S )Nr )�Queue)�queuer� � _call_queue� _send_stream�_options)rc r� r� r] r] r^ rd w s
214 zTestRunner.__init__rg rh ri rj rk �types.TracebackType | Nonec C s< | j r| j �� | j d ur | j�� � | j d us
215 d S d S d S rY )r� r� r� �getrp r] r] r^ ro ~ s
216
217 ��zTestRunner.__exit__c � s� �t d�\| _}|�= |2 z03 d H W \}}z|I d H }W n ty5 } z
218 |�t|�� W Y d }~q
219 d }~ww |�t|�� q
220 6 W d � d S 1 sJw Y d S r6 )r3 r� r� r( r r )rc Zreceive_stream�coro�outcome_holder�retvalr� r] r] r^ �_run_tests_and_fixtures� s ����"�z"TestRunner._run_tests_and_fixtures�outcomerV c C s
221 d | _ d S rY )r� )rc r� r] r] r^ �_main_task_finished� r� z TestRunner._main_task_finishedr� � Callable[P, Awaitable[T_Retval]]r� �P.argsrU �P.kwargsrO c O s� | j d u r'tjj| jf| jj| jd�| j�� | j d u r'| j� � � | j d u s g }| j �
222 ||i |��|f� |s@| j� � � |r8|d � � S )N)Zrun_sync_soon_threadsafe�
223 done_callbackr ) r� ra r� Zstart_guest_runr� r� �putr� r� r� Z send_nowait�unwrap)rc r� r� rU r� r] r] r^ �_call_in_runner_task� s$
224 ���
225
226 � � z TestRunner._call_in_runner_task� fixture_func�,Callable[..., AsyncGenerator[T_Retval, Any]]r� �Iterable[T_Retval]c c s` � |di |��}| � |jd �}|V z | � |jd � W n
227 ty% Y d S w | � |j� td��)Nz$Async generator fixture did not stopr] )r� �asend�StopAsyncIterationr� r� )rc r� rU �asyncgenZ fixturevaluer] r] r^ �run_asyncgen_fixture� s � � z TestRunner.run_asyncgen_fixture�,Callable[..., Coroutine[Any, Any, T_Retval]]c C s | j |fi |��S rY �r� )rc r� rU r] r] r^ � run_fixture� s zTestRunner.run_fixture� test_func�'Callable[..., Coroutine[Any, Any, Any]]c C s | j |fi |�� d S rY r� )rc r� rU r] r] r^ �run_test� s zTestRunner.run_testN)r� r rW r_ )rg rh ri rj rk r� rW r_ r� )r� rV rW r_ )r� r� r� r� rU r� rW rO )r� r� rU r� rW r� )r� r� rU r� rW rO )r� r� rU r� rW r_ ) r� r� r� rd ro r� r� r� r� r� r� r] r] r] r^ r� v s
228
229
230
231
232
233
234 r� c r, ) rk �task�trio.lowlevel.Taskc sH d }|j r|j jrt|j j�}t� �t|�||j|j� t�|�| _ d S rY )
235 Zparent_nurseryZ parent_task�idr� rd r� r� �weakref�proxy�_task)rc r� Z parent_idr� r] r^ rd � s
236 zTrioTaskInfo.__init__rW rm c C s" z| j jjW S ty Y dS w r� )r� Z_cancel_statusZeffectively_cancelled�ReferenceErrorrf r] r] r^ �has_pending_cancellation� s
237 �z%TrioTaskInfo.has_pending_cancellation)r� r� r� )r� r� r� rd r� r� r] r] r� r^ rk � s rk c @ sp e Zd Zed�d
238 d ��Zed�d
239 d��Zed�dd��Zed�dd��Zed�dd��Zed�dd��Z ed�dd��Z
240 ed�d d ��Z ee j
241 d d �d�d%d&��Zed�d'd(��Zed�d*d+��Zed�d-d.��Zed�d1d2��Zed3d d4�d�d:d;��Zed�d>d?��Ze 3d�d�dDdE��Zed�dFdG��Zed�dIdJ��Zed�dKdL��Zed�dNdO��Zed�dXdY��Zed�d\d]��Ze 3d�d�ddde��Z ed�didj��Z ed�dndo��Z ed�dpdq��Z ed�dwdx��Z ee!d�d|d}���Z"ee!d�dd}���Z"ed�d�d}��Z"ed�d�d�d�d��d�d�d���Z#e �d�d�d�d���Z$ed�d�d���Z%ed�d�d���Z&ed�d�d���Z'ed�d�d���Z(ed�d�d���Z)ed�d�d���Z*ed�d�d���Z+ed�d�d���Z,d3S )�� TrioBackendr� �1Callable[[Unpack[PosArgsT]], Awaitable[T_Retval]]r� r� rU r� r� rW rO c C s t j|g|�R � S rY )ra �run)r\ r� r� rU r� r] r] r^ r� � s zTrioBackend.runrV c C s
242 t j�� S rY )ra r� r� r� r] r] r^ �
243 current_token� �
244 zTrioBackend.current_tokenru c C � t �� S rY )ra � current_timer� r] r] r^ r� � ry zTrioBackend.current_time�type[BaseException]c C s t jS rY )ra r� r� r] r] r^ �cancelled_exception_class� r� z%TrioBackend.cancelled_exception_classr_ c � � �t j�� I d H d S rY )ra r� r r� r] r] r^ r � � �zTrioBackend.checkpointc � r� rY )ra r� rd r� r] r] r^ rd � r� z#TrioBackend.checkpoint_if_cancelledc � r� rY )ra r� �cancel_shielded_checkpointr� r] r] r^ r� � r� z&TrioBackend.cancel_shielded_checkpoint�delayc � s �t �|�I d H d S rY )ra r� )r\ r� r] r] r^ r� r� zTrioBackend.sleepF�rx r� rx r� rm �abc.CancelScopec C s t ||d�S )Nr� r> )r\ rx r� r] r] r^ �create_cancel_scope s z TrioBackend.create_cancel_scopec C r� rY )ra �current_effective_deadliner� r] r] r^ r� ry z&TrioBackend.current_effective_deadline�
245 abc.TaskGroupc C � t � S rY )r� r� r] r] r^ �create_task_group r� z TrioBackend.create_task_group� abc.Eventc C r� rY r6 r� r] r] r^ � create_event r� zTrioBackend.create_eventr^ r9 c C s
246 t |d�S )Nr] r8 r_ r] r] r^ � create_lock r� zTrioBackend.create_lockNrn rp r� ro r� �
247 abc.Semaphorec C s t |||d�S )Nrn r< rq r] r] r^ �create_semaphore s z TrioBackend.create_semaphorert r5 c C � t |�S rY r4 )r\ rt r] r] r^ �create_capacity_limiter& ry z#TrioBackend.create_capacity_limiter�&Callable[[Unpack[PosArgsT]], T_Retval]�abandon_on_cancel�limiter�abc.CapacityLimiter | Nonec � s8 �d� ��fdd� }t �� �t||ttj|�d�I d H S )NrW rO c s6 t t��� �� � W d � S 1 sw Y d S rY )r- r� r] �r� r� �tokenr] r^ �wrapper2 s $�z6TrioBackend.run_sync_in_worker_thread.<locals>.wrapper)r� r� )rW rO )r� r� r% r ra r5 )r\ r� r� r� r� r� r] r� r^ �run_sync_in_worker_thread* s �
248 �z%TrioBackend.run_sync_in_worker_threadc C s t j�� d S rY )ra r� �check_cancelledr� r] r] r^ r� = � zTrioBackend.check_cancelledr� c C � t jj|g|�R � S rY )ra r� r� �r\ r� r� r� r] r] r^ �run_async_from_threadA � z!TrioBackend.run_async_from_threadc C r� rY )ra r� r% r� r] r] r^ �run_sync_from_threadJ r� z TrioBackend.run_sync_from_thread�abc.BlockingPortalc C r� rY )r� r� r] r] r^ �create_blocking_portalS r� z"TrioBackend.create_blocking_portal�command�)StrOrBytesPath | Sequence[StrOrBytesPath]r� �int | IO[Any] | Noner� r� r r� c
249 � s� �d dd�� t |tttf�r#tjj� |�f|||dd�|��I d H }ntjj� fd d
250 �|D �f|||d d�|��I d H }|jrDt|j�nd }|j rNt
251 |j �nd }|j rXt
252 |j �nd } t |||| �S )
253 Nr� rD rW �strc S s" t �| �}t|t�r |S t �|�S rY )�os�fspathr� r� �fsdecode)r� Z str_or_bytesr] r] r^ � convert_itema s
254
255
256 z.TrioBackend.open_process.<locals>.convert_itemT)r� r� r� �shellc s g | ]}� |��qS r] r] )�.0r� �r� r] r^ �
257 <listcomp>s s z,TrioBackend.open_process.<locals>.<listcomp>F)r� rD rW r� )
258 r� r� r� r ra r� � open_processr� r� r� r� r� r� )
259 r\ r� r� r� r� rU r� Z stdin_streamZ
260 stdout_streamZ
261 stderr_streamr] r� r^ r� W s6 �
262
263 ��� �� � zTrioBackend.open_processr� r� c C s t j�t|� d S rY )ra r� Zspawn_system_taskr� )r\ r� r] r] r^ �#setup_process_pool_exit_at_shutdown� s z/TrioBackend.setup_process_pool_exit_at_shutdown�hostr� �port�
264 local_address�IPSockAddrType | Noner c � s� �d|v rt jnt j}tj � |�}|�t jt jd� |r$|�|�I d H z|�||f�I d H W t |�S t y? |�
265 � � w )N�:r� ) r �AF_INET6�AF_INETra r8 r9 r: �bind�connectr� r� r )r\ r� r� r� �familyr� r] r] r^ � connect_tcp� s � ��zTrioBackend.connect_tcp�path� str | bytes�abc.UNIXSocketStreamc � sD �t j�tj�}z
266 |�|�I d H W t|�S ty! |�� � w rY )ra r �AF_UNIXr� r� r� r )r\ r� r� r] r] r^ � connect_unix� s � ��zTrioBackend.connect_unix�sockr� �abc.SocketListenerc C r� rY )r- �r\ r r] r] r^ �create_tcp_listener� ry z TrioBackend.create_tcp_listenerc C r� rY )r? r r] r] r^ �create_unix_listener� ry z TrioBackend.create_unix_listenerr� �socket.AddressFamily�remote_address�
267 reuse_port� UDPSocket | ConnectedUDPSocketc � sb �t jj|tjd�}|r|�tjtjd� |r |�|�I d H |r-|�|�I d H t|�S t |�S )N)r� �typer� )
268 ra r �
269 SOCK_DGRAMr8 r � SO_REUSEPORTr� r� rL r@ )r\ r� r� r r
270 r� r] r] r^ �create_udp_socket� s �z TrioBackend.create_udp_socketr. � remote_path�abc.UNIXDatagramSocketc � � �d S rY r] �r\ r. r r] r] r^ �create_unix_datagram_socket� � �z'TrioBackend.create_unix_datagram_socket� abc.ConnectedUNIXDatagramSocketc � r rY r] r r] r] r^ r � r �str | bytes | None�8abc.UNIXDatagramSocket | abc.ConnectedUNIXDatagramSocketc � s2 �t j�|�}|r|�|�I d H t|�S t|�S rY )ra r r2 r� rT rS )r\ r. r r� r] r] r^ r � s � r )r� r �protor$ �bytes | str | None�str | int | None�int | AddressFamilyr �int | SocketKindr r$ �uSequence[tuple[AddressFamily, SocketKind, int, str, tuple[str, int] | tuple[str, int, int, int] | tuple[int, bytes]]]c � s �t j�||||||�I d H S rY )ra r � getaddrinfo)r\ r� r� r� r r r$ r] r] r^ r � s � zTrioBackend.getaddrinfo�sockaddrr@ �tuple[str, str]c � s �t j�||�I d H S rY )ra r � getnameinfo)r\ r r$ r] r] r^ r" � s �zTrioBackend.getnameinfo�obj�HasFileno | intc
271 � �Z �z
272 t |�I d H W d S tjy } zt� �|j�d �d }~w tjy, td�d �w )Nr )r" ra r0 �with_traceback�
273 __traceback__r/ �r\ r# r� r] r] r^ r" � � ��
274 �zTrioBackend.wait_readablec
275 � r% )Nr )r# ra r0 r& r' r/ r( r] r] r^ r# � r) zTrioBackend.wait_writablec C s< zt �� W S ty ttj�� d�}t �|� | Y S w )Nrs )r� r� � LookupErrorr5 ra � to_thread� current_default_thread_limiterr\ )r\ r� r] r] r^ r, s
276 �
277 �z*TrioBackend.current_default_thread_limiterr� r �.AbstractContextManager[AsyncIterator[Signals]]c G r� rY )r� )r\ r� r] r] r^ r� s z TrioBackend.open_signal_receiverr* c C s t � }t|�S rY )r! rk )r\ r� r] r] r^ �get_current_task s z TrioBackend.get_current_task�Sequence[TaskInfo]c C sb t � }|sJ �t|�g}|j}|r/g }|D ]}|jD ]}|�t|�� |�|j� qq|}|s|S rY )r rk Zchild_nurseriesZ child_tasksr( �extend)r\ Z root_taskZ
278 task_infosZ nurseriesZ
279 new_nurseriesZnurseryr� r] r] r^ �get_running_tasks s
280
281 �� z TrioBackend.get_running_tasksc � s �ddl m} |� I d H d S )Nr )�wait_all_tasks_blocked)Z trio.testingr2 )r\ r2 r] r] r^ r2 + s � z"TrioBackend.wait_all_tasks_blockedr� c C s t di |��S r` )r� )r\ r� r] r] r^ �create_test_runner1 r� z TrioBackend.create_test_runner)
282 r� r� r� r� rU r� r� r� rW rO )rW rV r� )rW r� r� )r� ru rW r_ )rx ru r� rm rW r� )rW r� )rW r� rl )rp r� ro r� r^ rm rW r� )rt ru rW r5 )FN)
283 r� r� r� r� r� rm r� r� rW rO )r� r� r� r� r� rV rW rO )r� r� r� r� r� rV rW rO )rW r� ) r� r� r� r� r� r� r� r� rU r rW r� �r� r� rW r_ rY )r� r� r� r� r� r� rW r )r� r� rW r )r r� rW r )
284 r� r r� r� r r� r
285 rm rW r )r. r� r r_ rW r )r. r� r r� rW r )r. r� r r rW r )r� r r� r r� r r r r r� r$ r� rW r )r )r r@ r$ r� rW r! )r# r$ rW r_ )rW r5 )r� r rW r- )rW r* )rW r/ )r� r� rW r� )-r� r� r� � classmethodr� r� r� r� r rd 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. r1 r2 r3 r] r] r] r^ r� � s� � � � ( �
286 ��
287 r� r4 )��
288 __future__r r r� r� r �sys�typesr� �collections.abcr r r r r r r r
289 �concurrent.futuresr �
290 contextlibr � dataclassesr
291 � functoolsr �ior r r� r r r r �typingr r r r r r r r Ztrio.from_threadra Z
292 trio.lowlevelr� r r r r r! r" r# Z trio.socketr$ r� Ztrio.to_threadr% � r' r( r) r* r+ r, Z_core._eventloopr- Z_core._exceptionsr. r/ r0 r1 Z_core._socketsr2 Z_core._streamsr3 Z_core._synchronizationr5 ZBaseCapacityLimiterr7 Z BaseEventr9 ZBaseLockr: r; r= Z
293 BaseSemaphoreZ _core._tasksr? ZBaseCancelScoper@ rA rB Zabc._eventlooprC rD Zstreams.memoryrE Z _typeshedrF � version_inforI Ztyping_extensionsrK rL ZexceptiongrouprM rN rO r� rP rQ rR r� ZRunVarr� r� ZByteReceiveStreamr� ZByteSendStreamr� r� Z
294 Instrumentr� r� r� r� r� r r ZSocketListenerr- r? r@ rL rS rT r� r� r� rk r� Z
295 backend_classr] r] r] r^ �<module> s� (
296 ( 6; 8 �
297 !C
298
299 � 86L W Y