!A?hFJ ` d Z ddlZddlZddlmZ ddlmZ ddlmZ G d d Z G d d
eej Z G d d ej Z
G d
deej Z G d deej Z G d de Z
G d dej Z G d dej ZdS ))LockEvent Condition SemaphoreBoundedSemaphoreBarrier N )
exceptions)mixins)tasksc e Zd Zd Zd ZdS )_ContextManagerMixinc > K | d {V d S N)acquireselfs :/opt/alt/python-internal/lib64/python3.11/asyncio/locks.py
__aenter__z _ContextManagerMixin.__aenter__ s- llnn t c 2 K | d S r )release)r exc_typeexctbs r __aexit__z _ContextManagerMixin.__aexit__ s r N)__name__
__module__ __qualname__r r r r r r
s2 r r c > e Zd Z d Z fdZd Zd Zd Zd Z xZ S )r c " d | _ d| _ d S NF)_waiters_lockedr s r __init__z
Lock.__init__N s
r c t }| j rdnd}| j r| dt | j }d|dd d| dS Nlockedunlocked
, waiters:<r []>)super__repr__r% r$ lenr resextra __class__s r r1 z
Lock.__repr__R sp gg L8j = = <<DM(:(:<<E)3qt9))))))r c | j S r )r% r s r r) z Lock.lockedY s
.|r c * K | j s.| j t d | j D r d| _ dS | j t j | _ | }| j | | d {V | j | n # | j | w xY wn/# t j
$ r | j s| w xY wd| _ dS )Nc 3 > K | ]}| V d S r cancelled.0ws r <genexpr>z Lock.acquire.<locals>.<genexpr>d s* 99aAKKMM999999r T) r% r$ all collectionsdeque _get_loop
create_futureappendremover
CancelledError_wake_up_firstr futs r r z Lock.acquire] s2
$-"7994=99999 #8 DL4 = ' - / /DMnn ,,..
S!!!
*
$$S))))
$$S)))))( <
&##%%%
ts B= "C = CC ,D c j | j r d| _ | d S t d )NFzLock is not acquired.)r% rH RuntimeErrorr s r r z Lock.release} sA < 8 DL ! ! ! ! ! 677 7r c | j sd S t t | j }n# t $ r Y d S w xY w| s| d d S d S NT)r$ nextiter
StopIterationdone
set_resultrI s r rH zLock._wake_up_first s 8} F tDM**++CC FF xxzz ! NN4 ! !s !.
<<)
r r r r& r1 r) r r rH
__classcell__r6 s @r r r s 3j * * * * * @8 8 8"
!
!
!
!
!
!
!r r c > e Zd Z d Z fdZd Zd Zd Zd Z xZ S )r c D t j | _ d| _ d S r# )rA rB r$ _valuer s r r& zEvent.__init__ s #)++
r c t }| j rdnd}| j r| dt | j }d|dd d| dS ) Nsetunsetr+ r, r r- r. r/ )r0 r1 rX r$ r2 r3 s r r1 zEvent.__repr__ sp gg 1' = = <<DM(:(:<<E)3qt9))))))r c | j S r rX r s r is_setz Event.is_set s
C{r c | j s:d| _ | j D ]-}| s| d ,d S d S rN )rX r$ rR rS rI s r rZ z Event.set sb { ) DK}
)
)xxzz )NN4((( ) )
)
)r c d| _ d S r# r] r s r clearz Event.clear s r c $ K | j rdS | }| j | | d {V | j | dS # | j | w xY wrN )rX rC rD r$ rE rF rI s r waitz
Event.wait s ; 4nn ,,..
S!!! &IIIIIII M % % % % %DM % % % %s A3 3 B)
r r r r& r1 r^ rZ ra rc rT rU s @r r r s * * * * *
)
)
) & & & & & & &r r c B e Zd Z d dZ fdZd Zd Zd
dZd Z xZ S ) r Nc |t }|| _ |j | _ |j | _ |j | _ t j | _ d S r )r _lockr) r r rA rB r$ )r locks r r& zCondition.__init__ sF <66D
k | | #)++
r c t }| rdnd}| j r| dt | j }d|dd d| dS r( )r0 r1 r) r$ r2 r3 s r r1 zCondition.__repr__ sv gg KKMM9z = = <<DM(:(:<<E)3qt9))))))r c K | st d | | }| j | | d {V | j | d} | d {V n# t j
$ r d}Y nw xY w4|r t j
dS # | j | w xY w# d} | d {V n# t j
$ r d}Y nw xY w4|r t j
w xY w)Nz cannot wait on un-acquired lockFT) r) rL r rC rD r$ rE rF r r
rG )r rJ r; s r rc zCondition.wait s {{}} B @AA A 0..""0022C M % % %
*
$$S))) I
%% ,,..(((((((!0 % % % $III%
%
0 //
0
0
$$S)))) I
%% ,,..(((((((!0 % % % $III%
%
0 //////sZ A D =C' D $B? ?CC' DD E
D&%E
&D:7E
9D::E
c p K | }|s&| d {V | }|&|S r rc )r predicateresults r wait_forzCondition.wait_for s\ !))++ Y[[F !
r r c | st d d}| j D ]9}||k r d S | s|dz
}| d :d S )Nz!cannot notify on un-acquired lockr r F)r) rL r$ rR rS )r nidxrJ s r notifyzCondition.notify* s
{{}} D BCC C= & &Caxx88::
&qu%%%
& &r c V | t | j d S r )rr r2 r$ r s r
notify_allzCondition.notify_allB s+
C
&&'''''r r r )
r r r r& r1 rc rn rr rt rT rU s @r r r s
,
,
,
,* * * * *#0 #0 #0J & & & &0( ( ( ( ( ( (r r c @ e Zd Z ddZ fdZd Zd Zd Zd Z xZ S ) r r c L |dk rt d d | _ || _ d S )Nr z$Semaphore initial value must be >= 0)
ValueErrorr$ rX )r values r r& zSemaphore.__init__Z s, 199 CDD D
r c t }| rdn d| j }| j r| dt | j }d|dd d| dS ) Nr) zunlocked, value:r+ r, r r- r. r/ )r0 r1 r) rX r$ r2 r3 s r r1 zSemaphore.__repr__` s gg KKMMO/O$+/O/O = = <<DM(:(:<<E)3qt9))))))r c Z | j dk p t d | j pdD S )Nr c 3 @ K | ]}| V d S r r: r< s r r? z#Semaphore.locked.<locals>.<genexpr>j s- AAaAKKMM!AAAAAAr r )rX anyr$ r s r r) zSemaphore.lockedg s< G{a C AADM,?RAAA A A Cr c v K | s| xj dz c_ dS | j t j | _ | }| j | | d {V | j | n # | j | w xY wnL# t j
$ r: | s$| xj dz
c_ | w xY w| j dk r| dS )Nr Tr )
r) rX r$ rA rB rC rD rE rF r
rG r;
_wake_up_nextrI s r r zSemaphore.acquirel sE {{}} KK1 KK4 = ' - / /DMnn ,,..
S!!!
*
$$S))))
$$S)))))( ==??
% q ""$$$ ;?? ts B. C . C
C A Dc P | xj dz
c_ | d S Nr )rX r r s r r zSemaphore.release s1
q r c | j sd S | j D ]>}| s(| xj dz c_ | d d S ?d S )Nr T)r$ rR rX rS rI s r r zSemaphore._wake_up_next sm 7} F= C88::
q t$$$
r ru )
r r r r& r1 r) r r r rT rU s @r r r K s * * * * *C C C
" " "H r r c , e Zd Z d fd Z fdZ xZS )r r c X || _ t | d S r ) _bound_valuer0 r& )r ry r6 s r r& zBoundedSemaphore.__init__ s) !
r c | j | j k rt d t d S )Nz(BoundedSemaphore released too many times)rX r rx r0 r )r r6 s r r zBoundedSemaphore.release s< ;$+ + + GHH H
r ru )r r r r& r rT rU s @r r r s[ r r c e Zd ZdZdZdZdZdS )
_BarrierStatefillingdraining resettingbrokenN)r r r FILLINGDRAINING RESETTINGBROKENr r r r r s" GHI
FFFr r c e Zd Z d Z fdZd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
ed Zed
Zed Z xZS )r c |dk rt d t | _ || _ t j | _ d| _ d S )Nr zparties must be > 0r )rx r _cond_partiesr r _state_count)r partiess r r& zBarrier.__init__ sD ? Q;; 233 3 [[
#+ r c t }| j j }| j s|d| j d| j z
}d|dd d| dS )Nr+ /r, r r- r. r/ )r0 r1 r ry r n_waitingr r3 s r r1 zBarrier.__repr__ sp gg ;$&{ B A$.AA4<AA AE)3qt9))))))r c : K | d {V S r rk r s r r zBarrier.__aenter__ s( YY[[ r c
K d S r r )r argss r r zBarrier.__aexit__ s r c K | j 4 d {V | d {V | j }| xj dz
c_ |dz | j k r| d {V n| d {V || xj dz c_ | cd d d d {V S # | xj dz c_ | w xY w# 1 d {V swxY w Y d S r )r _blockr r _release_wait_exit)r indexs r rc z Barrier.wait s : ++--
q 19
-- --//)))))))) **,,&&&&&&& q
q
s) C(AB?$C(?&C%%C((
C25C2c K j fd d {V j t j u rt j d d S )Nc B j t j t j fvS r )r r r r r s r <lambda>z Barrier._block.<locals>.<lambda> s DK &
(?( r zBarrier aborted)r rn r r r r
BrokenBarrierErrorr s `r r zBarrier._block s j!!
;-. . . /0ABB B / .r c ^ K t j | _ | j d S r )r r r r rt r s r r zBarrier._release s,
$,
r c K j fd d {V j t j t j fv rt
j d d S )Nc * j t j uS r )r r r r s r r z Barrier._wait.<locals>.<lambda> s $+]=R*R r zAbort or reset of barrier)r rn r r r r r
r r s `r r z
Barrier._wait sn j!!"R"R"R"RSSSSSSSSS ;=/ 1HI I I /0KLL L J Ir c | j dk rK| j t j t j fv rt j | _ | j d S d S Nr )r r r r r r r rt r s r r z
Barrier._exit sU ;! {} 6
8N OOO + 3 J ! ! # # # # # r c $ K | j 4 d {V | j dk r%| j t j urt j | _ nt j | _ | j d d d d {V d S # 1 d {V swxY w Y d S r )r r r r r r rt r s r resetz
Barrier.reset# s
: $ $ $ $ $ $ $ ${Q;m&==="/"9DK + 3 J ! ! # # # $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $s AA??
B B c K | j 4 d {V t j | _ | j d d d d {V d S # 1 d {V swxY w Y d S r )r r r r rt r s r abortz
Barrier.abort2 s
: $ $ $ $ $ $ $ $'.DK J ! ! # # # $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $s +A
A Ac | j S r )r r s r r zBarrier.parties< s
F} r c < | j t j u r| j S dS r )r r r r r s r r zBarrier.n_waitingA s" J ;-/ / /; qr c * | j t j u S r )r r r r s r r zBarrier.brokenH s >{m 222r )r r r r& r1 r r rc r r r r r r propertyr r r rT rU s @r r r s* * * * * *! ! !
.C C C M M M$ $ $
$
$
$ $ $ $ X X 3 3 X3 3 3 3 3r r )__all__rA enum r
r r r _LoopBoundMixinr r r r r Enumr r r r r <module>r s ! * C! C! C! C! C! !7 C! C! C!L:& :& :& :& :&F " :& :& :&zm( m( m( m( m($f&< m( m( m(`W W W W W$f&< W W Wt y $ DI M3 M3 M3 M3 M3f$ M3 M3 M3 M3 M3r