Format: application/x-python-code
1 o
2
3 ;'h`O � @ s� d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dl m
4 Z
5 ddl m Z d d
6 l
7 mZ d d lmZ d d lmZ d d
8 lmZmZ edd�G dd� d��Zedd�G dd� d��Zedd�G dd� d��Zedd�G dd� d��Zedd�G dd� d��ZG dd� d�ZG d d � d e�Z G d d � d �Z G d d!� d!e �Z G d"d#� d#�Z G d$d%� d%�Z G d&d'� d'e �Z!G d(d)� d)�Z"G d*d+� d+e"�Z#G d,d-� d-�Z$dS ).� )� annotationsN)�deque)� dataclass)�
9 TracebackType)�AsyncLibraryNotFoundError� )�
10 checkpoint� )�get_async_backend)�BusyResourceError)� CancelScope)�TaskInfo�get_current_taskT)�frozenc @ � e Zd ZU dZded< dS )�EventStatisticszR
11 :ivar int tasks_waiting: number of tasks waiting on :meth:`~.Event.wait`
12 �int�
13 tasks_waitingN��__name__�
14 __module__� __qualname__�__doc__�__annotations__� r r �RC:\Users\shubh\Videos\agent\nova\Lib\site-packages\anyio/_core/_synchronization.pyr s
15 r c @ s2 e Zd ZU dZded< ded< ded< ded< d S )
16 �CapacityLimiterStatisticsa�
17 :ivar int borrowed_tokens: number of tokens currently borrowed by tasks
18 :ivar float total_tokens: total number of available tokens
19 :ivar tuple borrowers: tasks or other objects currently holding tokens borrowed from
20 this limiter
21 :ivar int tasks_waiting: number of tasks waiting on
22 :meth:`~.CapacityLimiter.acquire` or
23 :meth:`~.CapacityLimiter.acquire_on_behalf_of`
24 r �borrowed_tokens�float� total_tokensztuple[object, ...]� borrowersr Nr r r r r r s
25
26 r c @ s* e Zd ZU dZded< ded< ded< dS ) �LockStatisticsa
27 :ivar bool locked: flag indicating if this lock is locked or not
28 :ivar ~anyio.TaskInfo owner: task currently holding the lock (or ``None`` if the
29 lock is not held by any task)
30 :ivar int tasks_waiting: number of tasks waiting on :meth:`~.Lock.acquire`
31 �bool�locked�TaskInfo | None�ownerr r Nr r r r r r! , s
32
33 r! c @ s" e Zd ZU dZded< ded< dS )�ConditionStatisticsz�
34 :ivar int tasks_waiting: number of tasks blocked on :meth:`~.Condition.wait`
35 :ivar ~anyio.LockStatistics lock_statistics: statistics of the underlying
36 :class:`~.Lock`
37 r r r! Zlock_statisticsNr r r r r r& : s
38 r& c @ r )�SemaphoreStatisticszZ
39 :ivar int tasks_waiting: number of tasks waiting on :meth:`~.Semaphore.acquire`
40
41 r r Nr r r r r r' F s
42 r' c @ s> e Zd Zddd�Zddd�Zddd �Zdd
43 d �Zdd
44 d�ZdS )�Event�returnc C s& zt � �� W S ty t� Y S w �N)r
45 � create_eventr � EventAdapter��clsr r r �__new__Q s
46
47 �z
48 Event.__new__�Nonec C � t �)z&Set the flag, notifying all listeners.��NotImplementedError��selfr r r �setW � z Event.setr" c C r1 )z5Return ``True`` if the flag is set, ``False`` if not.r2 r4 r r r �is_set[ r7 z Event.is_setc � � �t �)z�
49 Wait until the flag has been set.
50
51 If the flag has already been set when this method is called, it returns
52 immediately.
53
54 r2 r4 r r r �wait_ s �z
55 Event.waitr c C r1 )z8Return statistics about the current state of this event.r2 r4 r r r �
56 statisticsi r7 zEvent.statisticsN�r) r( �r) r0 �r) r" �r) r )r r r r/ r6 r8 r: r; r r r r r( P s
57
58
59
60
61 r( c @ sf e Zd ZU dZded< dZded< ddd �Zedd d ��Zddd�Z ddd�Z
62 ddd�Z ddd�Z dS ) r, Nz Event | None�_internal_eventFr" �_is_setr) c C �
63 t �| �S r* ��objectr/ r- r r r r/ r �
64 zEventAdapter.__new__r( c C s, | j d u rt� �� | _ | jr| j �� | j S r* )r@ r
65 r+ rA r6 r4 r r r �_eventu s
66
67
68 zEventAdapter._eventr0 c C s" | j d u r
69 d| _d S | j�� d S �NT)r@ rA rF r6 r4 r r r r6 ~ s
70
71 zEventAdapter.setc C s | j d u r| jS | j �� S r* )r@ rA r8 r4 r r r r8 � s
72
73 zEventAdapter.is_setc � � �| j �� I d H d S r* )rF r: r4 r r r r: � � �zEventAdapter.waitr c C s | j d u r
74 tdd�S | j �� S �Nr )r )r@ r r; r4 r r r r; � �
75
76
77 zEventAdapter.statistics)r) r, r< r= r> r? )
78 r r r r@ r rA r/ �propertyrF r6 r8 r: r; r r r r r, n s
79
80
81
82
83 r, c @ sb e Zd Zdd�d dd�Zd d d
84 �Zd!dd�Zd dd�Zd dd�Zd dd�Zd"dd�Z d#d d �Z
85 d S )$�LockF�� fast_acquirerO r" r) c C s. zt � j|d�W S ty t|d� Y S w �NrN )r
86 � create_lockr � LockAdapter�r. rO r r r r/ � s
87 �z Lock.__new__r0 c � � �| � � I d H d S r* ��acquirer4 r r r �
88 __aenter__� � �zLock.__aenter__�exc_type�type[BaseException] | None�exc_val�BaseException | None�exc_tb�TracebackType | Nonec � � �| � � d S r* ��release�r5 rY r[ r] r r r � __aexit__� � � zLock.__aexit__c � r9 )�Acquire the lock.r2 r4 r r r rV � � �z Lock.acquirec C r1 )�w
89 Acquire the lock, without blocking.
90
91 :raises ~anyio.WouldBlock: if the operation would block
92
93 r2 r4 r r r �acquire_nowait� � zLock.acquire_nowaitc C r1 )�Release the lock.r2 r4 r r r ra � r7 z Lock.releasec C r1 �z*Return True if the lock is currently held.r2 r4 r r r r# � r7 z Lock.lockedr! c C r1 )zh
94 Return statistics about the current state of this lock.
95
96 .. versionadded:: 3.0
97 r2 r4 r r r r; � � zLock.statisticsN)rO r" r) rM r= �rY rZ r[ r\ r] r^ r) r0 r> �r) r! ) r r r r/ rW rc rV rh ra r# r; r r r r rM � s
98
99
100
101
102
103 rM c @ s� e Zd ZU dZded< dd�d&d d
104 �Zdd�d'd d �Zed(dd��Zd)dd�Z d*dd�Z
105 d)dd �Z d)d d �Z d)d d �Z
106 d+d!d"�Zd,d$d%�ZdS )-rR N� Lock | None�_internal_lockFrN rO r" r) c C rB r* rC rS r r r r/ � rE zLockAdapter.__new__c C �
107 || _ d S r* )�
108 _fast_acquire)r5 rO r r r �__init__� rE zLockAdapter.__init__rM c C s" | j d u rt� j| jd�| _ | j S rP )rp r
109 rQ rr r4 r r r �_lock� s
110
111 �zLockAdapter._lockr0 c � rH r* �rt rV r4 r r r rW � rI zLockAdapter.__aenter__rY rZ r[ r\ r] r^ c � s �| j d ur
112 | j �� d S d S r* )rp ra rb r r r rc � s �
113 �zLockAdapter.__aexit__c � s �| j �� I dH dS )re Nru r4 r r r rV � s �zLockAdapter.acquirec C � | j �� dS )rg N)rt rh r4 r r r rh � s zLockAdapter.acquire_nowaitc C rv )rj N�rt ra r4 r r r ra � � zLockAdapter.releasec C �
114 | j �� S rk �rt r# r4 r r r r# � �
115 zLockAdapter.lockedr! c C s | j du r tddd�S | j �� S )zi
116 Return statistics about the current state of this lock.
117
118 .. versionadded:: 3.0
119
120 NFr )rp r! r; r4 r r r r; � s
121
122 zLockAdapter.statistics)rO r" r) rR )rO r" )r) rM r= rm r> rn )r r r rp r r/ rs rL rt rW rc rV rh ra r# r; r r r r rR � s
123
124
125
126
127
128
129 rR c @ s� e Zd ZU dZded< d+d,dd�Zd-d
130 d �Zd.dd�Zd-dd�Zd-dd�Z d-dd�Z
131 d-dd�Z d/d d �Z d0d1d"d#�Z
132 d-d$d%�Zd-d&d'�Zd2d)d*�ZdS )3� ConditionNr$ � _owner_task�lockro c C s |pt � | _t� | _d S r* )rM rt r �_waiters)r5 r~ r r r rs s zCondition.__init__r) r0 c � rT r* rU r4 r r r rW rX zCondition.__aenter__rY rZ r[ r\ r] r^ c � r_ r* r` rb r r r rc rd zCondition.__aexit__c C s | j t� kr
133 td��d S )Nz3The current task is not holding the underlying lock)r} r � RuntimeErrorr4 r r r �_check_acquired s �zCondition._check_acquiredc � s �| j �� I dH t� | _dS )z Acquire the underlying lock.N)rt rV r r} r4 r r r rV s � zCondition.acquirec C s | j �� t� | _dS )��
134 Acquire the underlying lock, without blocking.
135
136 :raises ~anyio.WouldBlock: if the operation would block
137
138 N)rt rh r r} r4 r r r rh s
139 zCondition.acquire_nowaitc C rv )z Release the underlying lock.Nrw r4 r r r ra ) rx zCondition.releaser" c C ry )z Return True if the lock is set.rz r4 r r r r# - r{ zCondition.lockedr �nr c C sH | � � t|�D ]}z| j�� }W n ty Y dS w |�� qdS )zNotify exactly n listeners.N)r� �ranger �popleft�
140 IndexErrorr6 )r5 r� �_�eventr r r �notify1 s �
141 �zCondition.notifyc C s* | � � | jD ]}|�� q| j�� dS )zNotify all the listeners.N)r� r r6 �clear�r5 r� r r r �
142 notify_all< s
143
144 zCondition.notify_allc � s� �t � I dH t� }| j�|� | �� z<z |�� I dH W n ty0 |�� s/| j�|� � w W t dd�� | �
145 � I dH W d � dS 1 sJw Y dS t dd�� | �
146 � I dH W d � w 1 shw Y w )zWait for a notification.NT)�shield) r r( r �appendra r: �
147 BaseExceptionr8 �remover rV r� r r r r: D s$ � �� .� �zCondition.waitr& c C s t t| j�| j�� �S )zm
148 Return statistics about the current state of this condition.
149
150 .. versionadded:: 3.0
151 )r& �lenr rt r; r4 r r r r; U s zCondition.statisticsr* )r~ ro r= rm r> )r )r� r r) r0 )r) r& )r r r r} r rs rW rc r� rV rh ra r# r� r� r: r; r r r r r| s
152
153
154
155
156
157
158
159
160
161
162 r| c @ s� e Zd Zddd�d'd d �Zddd�d(d
163 d�Zd)dd�Zd*dd�Zd+dd�Zd+d d �Zd+d d �Z e
164 d,d d!��Z e
165 d-d"d#��Z d.d%d&�Z
166 dS )/� SemaphoreNF�� max_valuerO �
167 initial_valuer r� �
168 int | NonerO r" r) c C s4 z
169 t � j|||d�W S ty t||d� Y S w )Nr� �r� )r
170 �create_semaphorer �SemaphoreAdapter�r. r� r� rO r r r r/ _ s � �zSemaphore.__new__c C sV t |t�s td��|dk rtd��|d ur&t |t�s td��||k r&td��|| _d S )Nz initial_value must be an integerr zinitial_value must be >= 0z$max_value must be an integer or Nonez7max_value must be equal to or higher than initial_value)�
171 isinstancer � TypeError�
172 ValueErrorrr �r5 r� r� rO r r r rs m s
173
174 �
175 zSemaphore.__init__c � s �| � � I d H | S r* rU r4 r r r rW � s �zSemaphore.__aenter__rY rZ r[ r\ r] r^ r0 c � r_ r* r` rb r r r rc � rd zSemaphore.__aexit__c � r9 )z5Decrement the semaphore value, blocking if necessary.r2 r4 r r r rV � rf zSemaphore.acquirec C r1 )r� r2 r4 r r r rh � ri zSemaphore.acquire_nowaitc C r1 )z Increment the semaphore value.r2 r4 r r r ra � r7 zSemaphore.releasec C r1 )z#The current value of the semaphore.r2 r4 r r r �value� � zSemaphore.valuec C r1 )z#The maximum value of the semaphore.r2 r4 r r r r� � r� zSemaphore.max_valuer' c C r1 )zm
176 Return statistics about the current state of this semaphore.
177
178 .. versionadded:: 3.0
179 r2 r4 r r r r; � rl zSemaphore.statistics)r� r r� r� rO r" r) r� )r� r r� r� rO r" �r) r� rm r= �r) r �r) r� �r) r' )r r r r/ rs rW rc rV rh ra rL r� r� r; r r r r r� ^ s" ��
180
181
182
183
184 r� c s� e Zd ZU dZded< ddd�d"d
185 d�Zddd�d#� fdd�Zed$dd��Zd%dd�Z d%dd�Z
186 d%dd�Z ed&dd ��Z ed'd d ��Z
187 d(d d!�Z� ZS ))r� NzSemaphore | None�_internal_semaphoreFr� r� r r� r� rO r" r) c C rB r* rC r� r r r r/ � s
188 zSemaphoreAdapter.__new__r0 c s" t � j|||d� || _|| _d S )Nr� )�superrs �_initial_value�
189 _max_valuer� �� __class__r r rs � s
190 zSemaphoreAdapter.__init__r� c C s& | j d u rt� j| j| jd�| _ | j S )Nr� )r� r
191 r� r� r� r4 r r r �
192 _semaphore� s
193
194 �zSemaphoreAdapter._semaphorec � rH r* )r� rV r4 r r r rV � rI zSemaphoreAdapter.acquirec C � | j �� d S r* )r� rh r4 r r r rh � � z SemaphoreAdapter.acquire_nowaitc C r� r* )r� ra r4 r r r ra � r� zSemaphoreAdapter.releasec C s | j d u r| jS | jjS r* )r� r� r� r� r4 r r r r� � �
195 zSemaphoreAdapter.valuec C s | j S r* )r� r4 r r r r� � s zSemaphoreAdapter.max_valuer' c C s | j d u r
196 tdd�S | j�� S rJ )r� r' r� r; r4 r r r r; � rK zSemaphoreAdapter.statistics)r� r r� r� rO r" r) r� )r� r r� r� rO r" r) r0 r� r= r� r� r� )r r r r� r r/ rs rL r� rV rh ra r� r� r; �
197 __classcell__r r r� r r� � s$
198 �
199 �
200
201
202 r� c @ s� e Zd Zd-dd�Zd.dd�Zd/dd�Zed0dd��Zejd1dd��Zed2dd��Z ed0dd��Z
203 d.dd �Z d3d d �Z d.d!d"�Z
204 d3d#d$�Zd.d%d&�Zd3d'd(�Zd4d*d+�Zd,S )5�CapacityLimiterr r r) c C s* zt � �|�W S ty t|� Y S w r* )r
205 �create_capacity_limiterr �CapacityLimiterAdapter�r. r r r r r/ � s
206 �zCapacityLimiter.__new__r0 c � r9 r* r2 r4 r r r rW � s �zCapacityLimiter.__aenter__rY rZ r[ r\ r] r^ � bool | Nonec � r9 r* r2 rb r r r rc � � �zCapacityLimiter.__aexit__c C r1 )aM
207 The total number of tokens available for borrowing.
208
209 This is a read-write property. If the total number of tokens is increased, the
210 proportionate number of tasks waiting on this limiter will be granted their
211 tokens.
212
213 .. versionchanged:: 3.0
214 The property is now writable.
215
216 r2 r4 r r r r � s
217 z CapacityLimiter.total_tokensr� c C r1 r* r2 �r5 r� r r r r r7 r c C r1 )z7The number of tokens that have currently been borrowed.r2 r4 r r r r r� z CapacityLimiter.borrowed_tokensc C r1 )z7The number of tokens currently available to be borrowedr2 r4 r r r �available_tokens r� z CapacityLimiter.available_tokensc C r1 )z�
218 Acquire a token for the current task without waiting for one to become
219 available.
220
221 :raises ~anyio.WouldBlock: if there are no tokens available for borrowing
222
223 r2 r4 r r r rh � z CapacityLimiter.acquire_nowait�borrowerrD c C r1 )z�
224 Acquire a token without waiting for one to become available.
225
226 :param borrower: the entity borrowing a token
227 :raises ~anyio.WouldBlock: if there are no tokens available for borrowing
228
229 r2 �r5 r� r r r �acquire_on_behalf_of_nowait& r� z+CapacityLimiter.acquire_on_behalf_of_nowaitc � r9 )zr
230 Acquire a token for the current task, waiting if necessary for one to become
231 available.
232
233 r2 r4 r r r rV 0 r� zCapacityLimiter.acquirec � r9 )z�
234 Acquire a token, waiting if necessary for one to become available.
235
236 :param borrower: the entity borrowing a token
237
238 r2 r� r r r �acquire_on_behalf_of8 s �z$CapacityLimiter.acquire_on_behalf_ofc C r1 )z�
239 Release the token held by the current task.
240
241 :raises RuntimeError: if the current task has not borrowed a token from this
242 limiter.
243
244 r2 r4 r r r ra A r� zCapacityLimiter.releasec C r1 )z�
245 Release the token held by the given borrower.
246
247 :raises RuntimeError: if the borrower has not borrowed a token from this
248 limiter.
249
250 r2 r� r r r �release_on_behalf_ofK r� z$CapacityLimiter.release_on_behalf_ofr c C r1 )zl
251 Return statistics about the current state of this limiter.
252
253 .. versionadded:: 3.0
254
255 r2 r4 r r r r; U ri zCapacityLimiter.statisticsN)r r r) r� r= �rY rZ r[ r\ r] r^ r) r� �r) r �r� r r) r0 r� �r� rD r) r0 �r) r )r r r r/ rW rc rL r �setterr r� rh r� rV r� ra r� r; r r r r r� � s&
256
257
258
259
260
261
262
263
264
265
266
267
268 r� c @ s� e Zd ZU dZded< d4dd�Zd5d
269 d �Zed6d
270 d��Zd7dd�Z d8dd�Z
271 ed9dd��Z e j d:d d��Z ed;d d ��Z
272 ed9d!d"��Zd7d#d$�Zd<d'd(�Zd7d)d*�Zd<d+d,�Zd7d-d.�Zd<d/d0�Zd=d2d3�ZdS )>r� NzCapacityLimiter | None�_internal_limiterr r r) c C rB r* rC r� r r r r/ b rE z CapacityLimiterAdapter.__new__r0 c C rq r* )r )r5 r r r r rs e rE z CapacityLimiterAdapter.__init__r� c C s | j d u r
273 t� �| j�| _ | j S r* )r� r
274 r� �
275 _total_tokensr4 r r r �_limiterh s
276
277 �z CapacityLimiterAdapter._limiterc � rH r* )r� rW r4 r r r rW q rI z!CapacityLimiterAdapter.__aenter__rY rZ r[ r\ r] r^ r� c � s �| j �|||�I d H S r* )r� rc rb r r r rc t s �z CapacityLimiterAdapter.__aexit__c C � | j d u r| jS | j jS r* )r� r� r r4 r r r r | r� z#CapacityLimiterAdapter.total_tokensr� c C sL t |t�s|tjurtd��|dk rtd��| jd u r || _d S || j_ d S )Nz'total_tokens must be an int or math.infr ztotal_tokens must be >= 1)
278 r� r �math�infr� r� r� r� r� r r� r r r r � s
279 r c C s | j d u rdS | j jS )Nr )r� r r4 r r r r � s
280 z&CapacityLimiterAdapter.borrowed_tokensc C r� r* )r� r� r� r4 r r r r� � r� z'CapacityLimiterAdapter.available_tokensc C r� r* )r� rh r4 r r r rh � r� z%CapacityLimiterAdapter.acquire_nowaitr� rD c C � | j �|� d S r* )r� r� r� r r r r� � � z2CapacityLimiterAdapter.acquire_on_behalf_of_nowaitc � rH r* )r� rV r4 r r r rV � rI z CapacityLimiterAdapter.acquirec � s �| j �|�I d H d S r* )r� r� r� r r r r� � s �z+CapacityLimiterAdapter.acquire_on_behalf_ofc C r� r* )r� ra r4 r r r ra � r� z CapacityLimiterAdapter.releasec C r� r* )r� r� r� r r r r� � r� z+CapacityLimiterAdapter.release_on_behalf_ofr c C s& | j d u rtd| jddd�S | j �� S )Nr r )r r r r )r� r r r; r4 r r r r; � s
281 �
282 z!CapacityLimiterAdapter.statistics)r r r) r� )r r r) r0 )r) r� r= r� r� r� r� r� r� )r r r r� r r/ rs rL r� rW rc r r� r r� rh r� rV r� ra r� r; r r r r r� _ s.
283
284
285
286
287
288
289
290
291
292
293 r� c @ s4 e Zd ZdZdZdddd�Zdd
294 d �Zddd�ZdS )�
295 ResourceGuarda�
296 A context manager for ensuring that a resource is only used by a single task at a
297 time.
298
299 Entering this context manager while the previous has not exited it yet will trigger
300 :exc:`BusyResourceError`.
301
302 :param action: the action to guard against (visible in the :exc:`BusyResourceError`
303 when triggered, e.g. "Another task is already {action} this resource")
304
305 .. versionadded:: 4.1
306 ��action�_guarded�usingr� �strc C s || _ d| _d S �NFr� )r5 r� r r r rs � s
307 zResourceGuard.__init__r) r0 c C s | j rt| j��d| _ d S rG )r� r r� r4 r r r � __enter__� s
308
309 zResourceGuard.__enter__rY rZ r[ r\ r] r^ c C s
310 d| _ d S r� )r� rb r r r �__exit__� s
311 zResourceGuard.__exit__N)r� )r� r� r= rm )r r r r � __slots__rs r� r� r r r r r� � s
312
313 r� )%�
314 __future__r r� � collectionsr � dataclassesr �typesr Zsniffior Zlowlevelr Z
315 _eventloopr
316 � _exceptionsr �_tasksr Z_testingr
317 r r r r! r& r' r( r, rM rR r| r� r� r� r� r� r r r r �<module> s>
318 &0@ZT;r]