U
e5d|C @ s d Z ddlZddlZddlZddlmZ ddlmZ ddlmZ ddlmZ G dd d Z G d
d d Z
G d d
d
e
Z G dd dZ G dd de
Z
G dd de
ZG dd deZdS ))LockEvent Condition SemaphoreBoundedSemaphore N )events)futures)
exceptions)
coroutinesc @ s$ e Zd Zdd Zdd Zdd ZdS )_ContextManagerc C s
|| _ d S N)_lock)selflock r %/usr/lib64/python3.8/asyncio/locks.py__init__" s z_ContextManager.__init__c C s d S r
r r r r r __enter__% s z_ContextManager.__enter__c G s z| j W 5 d | _ X d S r
)r releaser argsr r r __exit__* s z_ContextManager.__exit__N)__name__
__module__ __qualname__r r r r r r r r s r c @ sR e Zd Zdd Zdd Zejdd Zej e_ dd Z
d d
Z d d Z d
d Z
dS )_ContextManagerMixinc C s t dd S )Nz9"yield from" should be used as context manager expression) RuntimeErrorr r r r r 2 s z _ContextManagerMixin.__enter__c G s d S r
r r r r r r 6 s z _ContextManagerMixin.__exit__c c s& t jdtdd | E d H t| S )NzD'with (yield from lock)' is deprecated use 'async with lock' instead
stacklevel)warningswarnDeprecationWarningacquirer r r r r __iter__; s z _ContextManagerMixin.__iter__c s | I d H t| S r
)r% r r r r r Z
__acquire_ctxU s z"_ContextManagerMixin.__acquire_ctxc C s t jdtdd | S )Nz='with await lock' is deprecated use 'async with lock' insteadr r )r" r# r$ !_ContextManagerMixin__acquire_ctx __await__r r r r r( Y s
z _ContextManagerMixin.__await__c s | I d H d S r
)r% r r r r
__aenter__` s z _ContextManagerMixin.__aenter__c s | d S r
)r )r exc_typeexctbr r r __aexit__f s z _ContextManagerMixin.__aexit__N)r r r r r types coroutiner& r Z
_is_coroutiner' r( r) r- r r r r r 1 s
r c sJ e Zd ZddddZ fddZdd Zd d
Zd d Zd
d Z Z S )r Nloopc C s: d | _ d| _|d kr t | _n|| _tjdtdd d S NF[The loop argument is deprecated since Python 3.8, and scheduled for removal in Python 3.10.r r )_waiters_lockedr get_event_loop_loopr" r# r$ r r1 r r r r s z
Lock.__init__c sL t }| jrdnd}| jr2| dt| j }d|dd d| dS NlockedZunlocked
, waiters:)super__repr__r5 r4 lenr resZextra __class__r r rA s
z
Lock.__repr__c C s | j S r
)r5 r r r r r: s z Lock.lockedc s | j s.| jd ks$tdd | jD r.d| _ dS | jd krBt | _| j }| j| z"z|I d H W 5 | j| X W n& t j
k
r | j s| Y nX d| _ dS )Nc s s | ]}| V qd S r
) cancelled).0wr r r s z Lock.acquire..T) r5 r4 all collectionsdequer7
create_futureappendremover
CancelledError_wake_up_firstr futr r r r% s&
z Lock.acquirec C s" | j rd| _ | ntdd S )NFzLock is not acquired.)r5 rR r r r r r r s
z Lock.releasec C sJ | j s
d S ztt| j }W n tk
r2 Y d S X | sF|d d S NT)r4 nextiter
StopIterationdone
set_resultrS r r r rR s zLock._wake_up_first)
r r r r rA r: r% r rR
__classcell__r r rE r r j s 6 r c sJ e Zd ZddddZ fddZdd Zd d
Zd d Zd
d Z Z S )r Nr0 c C s> t | _d| _|d kr$t | _n|| _tjdt dd d S r2 )
rL rM r4 _valuer r6 r7 r" r# r$ r8 r r r r s
zEvent.__init__c sL t }| jrdnd}| jr2| dt| j }d|dd d| dS ) NsetZunsetr; r< r r= r> r? )r@ rA r\ r4 rB rC rE r r rA s
zEvent.__repr__c C s | j S r
r\ r r r r is_set s z Event.is_setc C s. | j s*d| _ | jD ]}| s|d qd S rU )r\ r4 rY rZ rS r r r r] s
z Event.setc C s
d| _ d S )NFr^ r r r r clear" s z Event.clearc sF | j r
dS | j }| j| z|I d H W dS | j| X d S rU )r\ r7 rN r4 rO rP rS r r r wait( s
z
Event.wait)
r r r r rA r_ r] r` ra r[ r r rE r r s r c sN e Zd ZdddddZ fddZdd Zd d
Zdd d
Zdd Z Z S )r Nr0 c C s~ |d krt | _n|| _tjdtdd |d kr>t|d}n|j| jk rRtd|| _|j | _ |j
| _
|j | _ t
| _d S )Nr3 r r r0 z"loop argument must agree with lock)r r6 r7 r" r# r$ r
ValueErrorr r: r% r rL rM r4 )r r r1 r r r r E s zCondition.__init__c sN t }| rdnd}| jr4| dt| j }d|dd d| dS r9 )r@ rA r: r4 rB rC rE r r rA [ s
zCondition.__repr__c s | std| z@| j }| j | z|I d H W W dS | j
| X W 5 d}z| I d H W qW q^ tjk
r d}Y q^X q^|rtjX d S )Nz cannot wait on un-acquired lockFT) r: r r r% r
rQ r7 rN r4 rO rP )r rG rT r r r ra b s$
zCondition.waitc s$ | }|s | I d H | }q|S r
)ra )r Z predicateresultr r r wait_for s
zCondition.wait_forr c C sJ | stdd}| jD ]*}||kr* qF| s|d7 }|d qd S )Nz!cannot notify on un-acquired lockr r F)r: r r4 rY rZ )r nidxrT r r r notify s
zCondition.notifyc C s | t| j d S r
)rg rB r4 r r r r
notify_all s zCondition.notify_all)N)r )
r r r r rA ra rd rg rh r[ r r rE r r ; s
%
r c sL e Zd ZdddddZ fddZdd Zd
d Zd d
Zdd Z Z S )r r Nr0 c C sN |dk rt d|| _t | _|d kr4t | _n|| _tj dt
dd d S )Nr z$Semaphore initial value must be >= 0r3 r r ) rb r\ rL rM r4 r r6 r7 r" r# r$ r valuer1 r r r r s
zSemaphore.__init__c sV t }| rdn
d| j }| jr<| dt| j }d|dd d| dS ) Nr: zunlocked, value:r; r< r r= r> r? )r@ rA r: r\ r4 rB rC rE r r rA s
zSemaphore.__repr__c C s, | j r(| j }| s |d d S q d S r
)r4 popleftrY rZ )r Zwaiterr r r
_wake_up_next s
zSemaphore._wake_up_nextc C s
| j dkS )Nr r^ r r r r r: s zSemaphore.lockedc st | j dkrb| j }| j| z|I d H W q | | j dkrX| sX| Y q X q | j d8 _ dS )Nr r T)r\ r7 rN r4 rO ZcancelrG rl rS r r r r% s
zSemaphore.acquirec C s | j d7 _ | d S )Nr )r\ rl r r r r r s zSemaphore.release)r )
r r r r rA rl r: r% r r[ r r rE r r s
r c s0 e Zd Zddd fdd Z fddZ ZS ) r r Nr0 c s. |rt jdtdd || _t j||d d S )Nr3 r r r0 )r" r# r$ _bound_valuer@ r ri rE r r r
s zBoundedSemaphore.__init__c s" | j | jkrtdt d S )Nz(BoundedSemaphore released too many times)r\ rm rb r@ r r rE r r r s zBoundedSemaphore.release)r )r r r r r r[ r r rE r r s r )__all__rL r. r" r r r
r r r r r r r r r r r r s "9 DzN