o
0i*7 @ s d Z ddlZddlmZ ddlmZ ddlmZ G dd dZG d d
d
eejZG d d d ejZ G d
d deejZ
G dd deejZ G dd de Z dS ))LockEvent Condition SemaphoreBoundedSemaphore N )
exceptions)mixins)tasksc @ s e Zd Zdd Zdd ZdS )_ContextManagerMixinc s | I d H d S N)acquireself r 4/opt/alt/python310/lib64/python3.10/asyncio/locks.py
__aenter__
s z _ContextManagerMixin.__aenter__c s | d S r )release)r exc_typeexctbr r r __aexit__ s z _ContextManagerMixin.__aexit__N)__name__
__module__ __qualname__r r r r r r r s r c R e Zd Z ejd fdd
Z fddZdd Zdd Zd
d Z d d
Z
Z S )r loopc s t j|d d | _d| _d S Nr F)super__init___waiters_lockedr r __class__r r r M s
z
Lock.__init__c L t }| jr
dnd}| jr| dt| j }d|dd d| dS NlockedZunlocked
, waiters:)r __repr__r" r! lenr resZextrar$ r r r. R
z
Lock.__repr__c C | j S r )r" r r r r r( Y z Lock.lockedc s | j s| jd u stdd | jD rd| _ dS | jd u r#t | _| }| j| zz
|I d H W | j| n| j| w W n t j
yV | j sU| w d| _ dS )Nc s s | ]}| V qd S r cancelled.0wr r r d s z Lock.acquire..T) r" r! all collectionsdeque _get_loop
create_futureappendremover CancelledError_wake_up_firstr futr r r r
] s,
z Lock.acquirec C s" | j r
d| _ | d S td)NFzLock is not acquired.)r" rC RuntimeErrorr r r r r } s
z Lock.releasec C sN | j sd S z tt| j }W n
ty Y d S w | s%|d d S d S NT)r! nextiter
StopIterationdone
set_resultrD r r r rC s zLock._wake_up_first) r r r r _markerr r. r( r
r rC
__classcell__r r r$ r r s 5 r c r )r r c s" t j|d t | _d| _d S r )r r r< r= r! _valuer# r$ r r r s
zEvent.__init__c r& ) NsetZunsetr) r* r r+ r, r- )r r. rO r! r/ r0 r$ r r r. r2 zEvent.__repr__c C r3 r rO r r r r is_set r4 z Event.is_setc C s4 | j sd| _ | jD ]
}| s|d q
d S d S rG )rO r! rK rL rD r r r rP s
z Event.setc C s d| _ d S )NFrQ r r r r clear s
z Event.clearc sR | j rdS | }| j| z|I d H W | j| dS | j| w rG )rO r> r? r! r@ rA rD r r r wait s
z
Event.wait) r r r r rM r r. rR rP rS rT rN r r r$ r r s r c sV e Zd Z dejd fdd Z fddZdd Zd d
Zdd d
Z dd Z
Z S )r Nr c sH t j|d |d u rt }|| _|j| _|j| _|j| _t | _ d S Nr )
r r r _lockr( r
r r< r= r! )r lockr r$ r r r s zCondition.__init__c sN t }| r dnd}| jr| dt| j }d|dd d| dS r' )r r. r( r! r/ r0 r$ r r r. s
zCondition.__repr__c s | s
td| zB| }| j| z.|I d H W | j| W d} z | I d H W n
t j
yA d}Y nw q,|rHt j
dS | j| w d} z | I d H W n
t j
yi d}Y nw qT|rpt j
w )Nz cannot wait on un-acquired lockFT) r( rF r r> r? r! r@ rA r
r rB )r rE r6 r r r rT sH
zCondition.waitc s* | }|s| I d H | }|r|S r )rT )r Z predicateresultr r r wait_for s zCondition.wait_forr c C sN | s tdd}| jD ]}||kr d S | s$|d7 }|d qd S )Nz!cannot notify on un-acquired lockr r F)r( rF r! rK rL )r nidxrE r r r notify, s
zCondition.notifyc C s | t| j d S r )r\ r/ r! r r r r
notify_allD s zCondition.notify_allr r ) r r r r rM r r. rT rY r\ r] rN r r r$ r r s
%
r c sT e Zd Z dejd fdd Z fddZdd Zd d
Zd d Z d
d Z
Z S )r r r c s. t j|d |dk rtdd | _|| _d S )Nr r z$Semaphore initial value must be >= 0)r r
ValueErrorr! rO r valuer r$ r r r \ s
zSemaphore.__init__c sV t }| r dnd| j }| jr | dt| j }d|dd d| dS ) Nr( zunlocked, value:r) r* r r+ r, r- )r r. r( rO r! r/ r0 r$ r r r. c s
zSemaphore.__repr__c C s$ | j dkptdd | jpdD S )Nr c s s | ]}| V qd S r r5 r7 r r r r: m s z#Semaphore.locked..r )rO anyr! r r r r r( j s
zSemaphore.lockedc s | s| jd8 _dS | jd u rt | _| }| j| zz
|I d H W | j| n| j| w W n t j
yT | sS| jd7 _| w | jdkr^| dS )Nr Tr )
r( rO r! r< r= r> r? r@ rA r rB r6
_wake_up_nextrD r r r r
o s,
zSemaphore.acquirec C s | j d7 _ | d S )Nr )rO rc r r r r r s zSemaphore.releasec C sB | j sd S | j D ]}| s | jd8 _|d d S q d S )Nr T)r! rK rO rL rD r r r rc s
zSemaphore._wake_up_nextr^ ) r r r r rM r r. r( r
r rc rN r r r$ r r M s $ r c s4 e Zd Z dejd fdd Z fddZ ZS )r r r c s || _ t j||d d S rU ) _bound_valuer r r` r$ r r r s zBoundedSemaphore.__init__c s" | j | jkr
tdt d S )Nz(BoundedSemaphore released too many times)rO rd r_ r r r r$ r r r s zBoundedSemaphore.releaser^ )r r r r rM r r rN r r r$ r r s r )
__all__r< r r r
r Z_LoopBoundMixinr r r r r r r r r s >q[