B
XZ @ s d Z ddlZddlZddlmZ ddlmZ ddlmZ G dd dZG d d
d
Z G d d d e Z
G d
d dZ G dd de Z G dd de Z
G dd de
ZdS ))LockEvent Condition SemaphoreBoundedSemaphore N )events)futures) coroutinec @ s$ e Zd Zdd Zdd Zdd ZdS )_ContextManagerc C s
|| _ d S )N)_lock)selflock r 2/opt/alt/python37/lib64/python3.7/asyncio/locks.py__init__ s z_ContextManager.__init__c C s d S )Nr )r
r r r __enter__# s z_ContextManager.__enter__c G s z| j W d d | _ X d S )N)r release)r
argsr r r __exit__( s z_ContextManager.__exit__N)__name__
__module__ __qualname__r r r r r r r r
s r c @ sH e Zd Zdd Zdd Zedd Zdd Zd d
Zd d Z d
d Z
dS )_ContextManagerMixinc C s t dd S )Nz9"yield from" should be used as context manager expression) RuntimeError)r
r r r r 0 s z _ContextManagerMixin.__enter__c G s d S )Nr )r
r r r r r 4 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__9 s
z _ContextManagerMixin.__iter__c s | I d H t| S )N)r r )r
r r r Z
__acquire_ctxP 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# T s
z _ContextManagerMixin.__await__c s | I d H d S )N)r )r
r r r
__aenter__[ s z _ContextManagerMixin.__aenter__c s | d S )N)r )r
exc_typeexctbr r r __aexit__a s z _ContextManagerMixin.__aexit__N) r r r r r r
r! r" r# r$ r( r r r r r / s r c sJ e Zd ZddddZ fddZdd Zd d
Zd d Zd
d Z Z S )r N)loopc C s. t | _d| _|d k r || _n
t | _d S )NF) collectionsdeque_waiters_locked_loopr get_event_loop)r
r) r r r r s
z
Lock.__init__c sL t }| jrdnd}| jr2| dt| j }d|dd d| dS ) Nlockedunlockedz
, waiters:)super__repr__r- r, len)r
resextra) __class__r r r5 s
z
Lock.__repr__c C s | j S )N)r- )r
r r r r0 s z Lock.lockedc s | j s$tdd | jD r$d| _ dS | j }| j| y"z|I d H W d | j| X W n& tjk
r | j s|| Y nX d| _ dS )Nc s s | ]}| V qd S )N) cancelled).0wr r r s z Lock.acquire..T)
r- allr, r.
create_futureappendremover CancelledError_wake_up_first)r
futr r r r s
z Lock.acquirec C s" | j rd| _ | ntdd S )NFzLock is not acquired.)r- rC r )r
r r r r s
z Lock.releasec C s> yt t| j}W n tk
r& d S X | s:|d d S )NT)nextiterr,
StopIterationdone
set_result)r
rD r r r rC s zLock._wake_up_first)
r r r r r5 r0 r r rC
__classcell__r r )r9 r r e s 6 r c sJ e Zd ZddddZ fddZdd Zd d
Zd d Zd
d Z Z S )r N)r) c C s. t | _d| _|d k r || _n
t | _d S )NF)r* r+ r, _valuer. r r/ )r
r) r r r r s
zEvent.__init__c sL t }| jrdnd}| jr2| dt| j }d|dd d| dS ) NsetZunsetz
, waiters:r2 r r3 z [z]>)r4 r5 rK r, r6 )r
r7 r8 )r9 r r r5 s
zEvent.__repr__c C s | j S )N)rK )r
r r r is_set s z Event.is_setc C s2 | j s.d| _ x | jD ]}| s|d qW d S )NT)rK r, rH rI )r
rD r r r rL s
z Event.setc C s
d| _ d S )NF)rK )r
r r r clear s z Event.clearc sB | j r
dS | j }| j| z|I d H dS | j| X d S )NT)rK r. r? r, r@ rA )r
rD r r r wait s
z
Event.wait)
r r r r r5 rM rL rN rO rJ r r )r9 r r s r c sN e Zd ZdddddZ fddZdd Zd d
Zdd d
Zdd Z Z S )r N)r) c C sp |d k r|| _ n
t | _ |d kr0t| j d}n|j | j k rDtd|| _|j| _|j| _|j| _t
| _ d S )N)r) z"loop argument must agree with lock) r. r r/ r
ValueErrorr r0 r r r* r+ r, )r
r r) r r r r 5 s
zCondition.__init__c sN t }| rdnd}| jr4| dt| j }d|dd d| dS ) Nr0 r1 z
, waiters:r2 r r3 z [z]>)r4 r5 r0 r, r6 )r
r7 r8 )r9 r r r5 H s
zCondition.__repr__c s | std| z8| j }| j| z|I d H dS | j| X W d d}x4y| I d H P W qX t j
k
r d}Y qXX qXW |rt j
X d S )Nz cannot wait on un-acquired lockTF) r0 r r r. r? r, r@ rA r r rB )r
rD r: r r r rO O s&
zCondition.waitc s( | }x |s"| I d H | }qW |S )N)rO )r
Z predicateresultr r r wait_fort s
zCondition.wait_forr c C sL | stdd}x2| jD ](}||kr*P | s |d7 }|d q W d S )Nz!cannot notify on un-acquired lockr r F)r0 r r, rH rI )r
nidxrD r r r notify s zCondition.notifyc C s | t| j d S )N)rU r6 r, )r
r r r
notify_all s zCondition.notify_all)N)r )
r r r r r5 rO rR rU rV rJ r r )r9 r r + s
%
r c sL e Zd ZdddddZ fddZdd Zd
d Zd d
Zdd Z Z S )r r N)r) c C s> |dk rt d|| _t | _|d k r0|| _n
t | _d S )Nr z$Semaphore initial value must be >= 0)rP rK r* r+ r, r. r r/ )r
valuer) r r r r s
zSemaphore.__init__c sV t }| rdn
d| j }| jr<| dt| j }d|dd d| dS ) Nr0 zunlocked, value:z
, waiters:r2 r r3 z [z]>)r4 r5 r0 rK r, r6 )r
r7 r8 )r9 r r r5 s
zSemaphore.__repr__c C s0 x*| j r*| j }| s|d d S qW d S )N)r, popleftrH rI )r
Zwaiterr r r
_wake_up_next s
zSemaphore._wake_up_nextc C s
| j dkS )Nr )rK )r
r r r r0 s zSemaphore.lockedc sx xd| j dkrd| j }| j| y|I d H W q | | j dkrZ| sZ| Y qX qW | j d8 _ dS )Nr r T)rK r. r? r, r@ Zcancelr: rY )r
rD r r r r s
zSemaphore.acquirec C s | j d7 _ | d S )Nr )rK rY )r
r r r r s zSemaphore.release)r )
r r r r r5 rY r0 r r rJ r r )r9 r r s
r c s0 e Zd Zddd fdd Z fddZ ZS ) r r N)r) c s || _ t j||d d S )N)r) ) _bound_valuer4 r )r
rW r) )r9 r r r s zBoundedSemaphore.__init__c s" | j | jkrtdt d S )Nz(BoundedSemaphore released too many times)rK rZ rP r4 r )r
)r9 r r r s zBoundedSemaphore.release)r )r r r r r rJ r r )r9 r r s r )__all__r* r r r Z
coroutinesr
r r r r r r r r r r r s "6 AwK