3
2a< @ s d ddddgZ ddlZddlmZ dd lmZ dd
lmZ dd lmZ G d d
d
ZG dd dZ G dd d e Z
G dd dZ G dd de Z G dd de Z
G dd de
ZdS )LockEvent Condition SemaphoreBoundedSemaphore N )compat)events)futures) coroutinec @ s$ e Zd Zdd Zdd Zdd ZdS )_ContextManagerc C s
|| _ d S )N)_lock)selflock r */opt/alt/python36/lib64/python3.6/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 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 @ sN e Zd Zdd Zdd Zedd ZejrJdd Z ed d
Z
ed d Z d
S )_ContextManagerMixinc C s t dd S )Nz9"yield from" should be used as context manager expression) RuntimeError)r r r r r , s z _ContextManagerMixin.__enter__c G s d S )Nr )r r r r r r 0 s z _ContextManagerMixin.__exit__c c s | j E d H t| S )N)acquirer )r r r r __iter__5 s z _ContextManagerMixin.__iter__c c s | j E d H t| S )N)r r )r r r r __await__H s z _ContextManagerMixin.__await__c c s | j E d H d S )N)r )r r r r
__aenter__M s z _ContextManagerMixin.__aenter__c C s | j d S )N)r )r exc_typeexctbr r r __aexit__T s z _ContextManagerMixin.__aexit__N) r r r r r r r r ZPY35r r r# r r r r r + s r c sN e Zd ZddddZ fddZdd Zed d
Zd d Zd
d Z Z
S )r N)loopc C s. t j | _d| _|d k r || _n
tj | _d S )NF) collectionsdeque_waiters_locked_loopr get_event_loop)r r$ r r r r s
z
Lock.__init__c sD t j }| jrdnd}| jr0dj|t| j}dj|dd |S )Nlockedunlockedz
{},waiters:{}z <{} [{}]>r )super__repr__r( r' formatlen)r resextra) __class__r r r/ s
z
Lock.__repr__c C s | j S )N)r( )r r r r r+ s z Lock.lockedc c s | j r&tdd | jD r&d| _ dS | jj }| jj| y"z|E d H W d | jj| X W n& tjk
r | j s~| j Y nX d| _ dS )Nc s s | ]}|j 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| _ | j ntdd S )NFzLock is not acquired.)r( r> r )r r r r r s
z Lock.releasec C s> yt t| j}W n tk
r& d S X |j s:|jd d S )NT)nextiterr'
StopIterationdone
set_result)r r? r r r r> s zLock._wake_up_first) r r r r r/ r+ r r r r>
__classcell__r r )r4 r r Y s 6 c sN e Zd ZddddZ fddZdd Zd d
Zd d Zed
d Z Z
S )r N)r$ c C s. t j | _d| _|d k r || _n
tj | _d S )NF)r% r& r' _valuer) r r* )r r$ r r r r s
zEvent.__init__c sD t j }| jrdnd}| jr0dj|t| j}dj|dd |S )NsetZunsetz
{},waiters:{}z <{} [{}]>r r- )r. r/ rF r' r0 r1 )r r2 r3 )r4 r r r/ s
zEvent.__repr__c C s | j S )N)rF )r r r r is_set s z Event.is_setc C s2 | j s.d| _ x | jD ]}|j s|jd qW d S )NT)rF r' rC rD )r r? r r r rG s
z Event.setc C s
d| _ d S )NF)rF )r r r r clear s z Event.clearc c sB | j r
dS | jj }| jj| z|E d H dS | jj| X d S )NT)rF r) r: r' r; r< )r r? r r r wait
s
z
Event.wait) r r r r r/ rH rG rI r rJ rE r r )r4 r r s c sV e Zd ZdddddZ fddZedd Zed d
Zdd d
Zdd Z Z
S )r N)r$ c C sp |d k r|| _ n
tj | _ |d kr0t| j d}n|j | j k rDtd|| _|j| _|j| _|j| _t j
| _ d S )N)r$ z"loop argument must agree with lock) r) r r* r
ValueErrorr
r+ r r r% r& r' )r r r$ r r r r + s
zCondition.__init__c sF t j }| j rdnd}| jr2dj|t| j}dj|dd |S )Nr+ r, z
{},waiters:{}z <{} [{}]>r r- )r. r/ r+ r' r0 r1 )r r2 r3 )r4 r r r/ > s
zCondition.__repr__c c s | j std| j z8| jj }| jj| z|E d H dS | jj| X W d d}x4y| j E 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) r+ r r r) r: r' r; r< r r
r= )r r? r5 r r r rJ E s&
zCondition.waitc c s( | }x |s"| j E d H | }qW |S )N)rJ )r predicateresultr r r wait_fork s
zCondition.wait_forr c C sL | j stdd}x2| jD ](}||kr*P |j s |d7 }|jd q W d S )Nz!cannot notify on un-acquired lockr r F)r+ r r' rC rD )r nidxr? r r r notifyy s zCondition.notifyc C s | j t| j d S )N)rQ r1 r' )r r r r
notify_all s zCondition.notify_all)N)r ) r r r r r/ r rJ rN rQ rR rE r r )r4 r r ! s
&
c sP e Zd ZdddddZ fddZdd Zd
d Zed d
Zdd Z Z
S )r r N)r$ c C s> |dk rt d|| _tj | _|d k r0|| _n
tj | _d S )Nr z$Semaphore initial value must be >= 0)rK rF r% r& r' r) r r* )r valuer$ r r r r s
zSemaphore.__init__c sN t j }| j rdn
dj| j}| jr:dj|t| j}dj|dd |S )Nr+ zunlocked,value:{}z
{},waiters:{}z <{} [{}]>r r- )r. r/ r+ r0 rF r' r1 )r r2 r3 )r4 r r r/ s
zSemaphore.__repr__c C s0 x*| j r*| j j }|j s|jd d S qW d S )N)r' popleftrC rD )r Zwaiterr r r
_wake_up_next s
zSemaphore._wake_up_nextc C s
| j dkS )Nr )rF )r r r r r+ s zSemaphore.lockedc c sz xf| j dkrf| jj }| jj| y|E d H W q |j | j dkr\|j r\| j Y qX qW | j d8 _ dS )Nr r T)rF r) r: r' r; Zcancelr5 rU )r r? r r r r s
zSemaphore.acquirec C s | j d7 _ | j d S )Nr )rF rU )r r r r r s zSemaphore.release)r ) r r r r r/ rU r+ r r r rE r r )r4 r r s
c s0 e Zd Zddd fdd Z fddZ ZS ) r r N)r$ c s || _ t j||d d S )N)r$ ) _bound_valuer. r )r rS r$ )r4 r r r s zBoundedSemaphore.__init__c s" | j | jkrtdt j d S )Nz(BoundedSemaphore released too many times)rF rV rK r. r )r )r4 r r r s zBoundedSemaphore.release)r )r r r r r rE r r )r4 r r s )__all__r% r r r
Z
coroutinesr r r r r r r r r r r r s . ByM