a
XC?hM; � @ s� d Z ddlZddlZddlmZ ddlmZ G dd� d�ZG dd � d e�ZG d
d � d �ZG d d
� d
e�Z G dd� de�Z
G dd� de
�Z dS ))�Lock�Event� Condition� Semaphore�BoundedSemaphore� N� )�events)�
exceptionsc @ s e Zd Zdd� Zdd� ZdS )�_ContextManagerMixinc � s | � � I d H d S �N)�acquire��self� r �2/opt/alt/python39/lib64/python3.9/asyncio/locks.py�
__aenter__
s z _ContextManagerMixin.__aenter__c � s | � � d S r )�release)r �exc_type�exc�tbr r r � __aexit__ s z _ContextManagerMixin.__aexit__N)�__name__�
__module__� __qualname__r r r r r r r
s r
c sJ e Zd Zdd�dd�Z� fdd�Zdd� Zd d
� Zd d � Zd
d� Z� Z S )r N��loopc C s: d | _ d| _|d u r t�� | _n|| _tjdtdd� d S �NF�[The loop argument is deprecated since Python 3.8, and scheduled for removal in Python 3.10.� ��
stacklevel)�_waiters�_lockedr �get_event_loop�_loop�warnings�warn�DeprecationWarning�r r r r r �__init__M s �z
Lock.__init__c sL t � �� }| jrdnd}| jr2|� dt| j�� �}d|dd� � d|� d�S � N�lockedZunlocked�
, waiters:�)�super�__repr__r" r! �len�r �resZextra�� __class__r r r2 X s
z
Lock.__repr__c C s | j S r )r" r
r r r r+ _ s z Lock.lockedc � s� | j s.| jd u s$tdd� | jD ��r.d| _ dS | jd u rBt�� | _| j�� }| j�|� z.z|I d H W | j�|� n| j�|� 0 W n$ t j
y� | j s�| � � � Y n0 d| _ dS )Nc s s | ]}|� � V qd S r )� cancelled)�.0�wr r r � j � z Lock.acquire..T) r" r! �all� collections�dequer$ �
create_future�append�remover �CancelledError�_wake_up_first�r �futr r r r c s& �
z Lock.acquirec C s" | j rd| _ | �� ntd��d S )NFzLock is not acquired.)r" rD � RuntimeErrorr
r r r r � s
z Lock.releasec C sH | j s
d S ztt| j ��}W n ty0 Y d S 0 |�� sD|�d� d S �NT)r! �next�iter�
StopIteration�done�
set_resultrE r r r rD � s zLock._wake_up_first)
r r r r) r2 r+ r r rD �
__classcell__r r r6 r r s 6 r c sJ e Zd Zdd�dd�Z� fdd�Zdd� Zd d
� Zd d � Zd
d� Z� Z S )r Nr c C s> t �� | _d| _|d u r$t�� | _n|| _tjdt dd� d S r )
r> r? r! �_valuer r# r$ r% r& r' r( r r r r) � s
�zEvent.__init__c sL t � �� }| jrdnd}| jr2|� dt| j�� �}d|dd� � d|� d�S ) N�setZunsetr, r- r r. r/ r0 )r1 r2 rO r! r3 r4 r6 r r r2 � s
zEvent.__repr__c C s | j S r �rO r
r r r �is_set� s z Event.is_setc C s. | j s*d| _ | jD ]}|�� s|�d� qd S rH )rO r! rL rM rE r r r rP � s
z Event.setc C s
d| _ d S )NFrQ r
r r r �clear� s z Event.clearc � sP | j r
dS | j�� }| j�|� z |I d H W | j�|� dS | j�|� 0 d S rH )rO r$ r@ r! rA rB rE r r r �wait� s
�z
Event.wait)
r r r r) r2 rR rP rS rT rN r r r6 r r � s r c sN e Zd Zddd�dd�Z� fdd�Zdd� Zd d
� Zdd d
�Zdd� Z� Z S )r Nr c C s~ |d u rt �� | _n|| _tjdtdd� |d u r>t|d�}n|j| jurRtd��|| _|j | _ |j
| _
|j | _ t �
� | _d S )Nr r r r z"loop argument must agree with lock)r r# r$ r% r& r' r �
ValueError�_lockr+ r r r> r? r! )r �lockr r r r r) � s � zCondition.__init__c sN t � �� }| �� rdnd}| jr4|� dt| j�� �}d|dd� � d|� d�S r* )r1 r2 r+ r! r3 r4 r6 r r r2 s
zCondition.__repr__c � s. | � � std��| �� z�| j�� }| j�|� z^|I d H W | j�|� W d}z| �� I d H W q�W qP t j
y~ d}Y qP0 qP|r�t j
�dS | j�|� 0 W d}z| �� I d H W q�W q� t j
y� d}Y q�0 q�|r�t j
�nHd}z| �� I d H W �q W q� t j
�y d}Y q�0 q�|�r(t j
�0 d S )Nz cannot wait on un-acquired lockFT) r+ rG r r$ r@ r! rA rB r r rC )r rF r8 r r r rT sJ
� � �
zCondition.waitc � s$ |� }|s | � � I d H |� }q|S r )rT )r Z predicate�resultr r r �wait_for4 s
zCondition.wait_forr c C sJ | � � std��d}| jD ]*}||kr* qF|�� s|d7 }|�d� qd S )Nz!cannot notify on un-acquired lockr r F)r+ rG r! rL rM )r �n�idxrF r r r �notifyA s
zCondition.notifyc C s | � t| j�� d S r )r\ r3 r! r
r r r �
notify_allY s zCondition.notify_all)N)r )
r r r r) r2 rT rY r\ r] rN r r r6 r r � s
%
r c sL e Zd Zddd�dd�Z� fdd�Zdd � Zd
d � Zd d
� Zdd� Z� Z S )r r Nr c C sT |dk rt d��|| _t�� | _|d u r4t�� | _n|| _tj dt
dd� d| _ d S )Nr z$Semaphore initial value must be >= 0r r r F) rU rO r> r? r! r r# r$ r% r& r' �_wakeup_scheduled�r �valuer r r r r) q s
�zSemaphore.__init__c sV t � �� }| �� rdn
d| j� �}| jr<|� dt| j�� �}d|dd� � d|� d�S ) Nr+ zunlocked, value:r, r- r r. r/ r0 )r1 r2 r+ rO r! r3 r4 r6 r r r2 s
zSemaphore.__repr__c C s2 | j r.| j �� }|�� s |�d � d| _d S q d S rH )r! �popleftrL rM r^ )r Zwaiterr r r �
_wake_up_next� s
zSemaphore._wake_up_nextc C s
| j dkS )Nr rQ r
r r r r+ � s zSemaphore.lockedc � sn | j s| jdkr\| j�� }| j�|� z|I d H d| _ W q tjyX | �� � Y q 0 q | jd8 _dS )Nr Fr T) r^ rO r$ r@ r! rA r rC rb rE r r r r � s
zSemaphore.acquirec C s | j d7 _ | �� d S )Nr )rO rb r
r r r r � s zSemaphore.release)r )
r r r r) r2 rb r+ r r rN r r r6 r r b s r c s0 e Zd Zddd�� fdd� Z� fdd�Z� ZS ) r r Nr c s. |rt jdtdd� || _t� j||d� d S )Nr r r r )r% r& r' � _bound_valuer1 r) r_ r6 r r r) � s �zBoundedSemaphore.__init__c s" | j | jkrtd��t� �� d S )Nz(BoundedSemaphore released too many times)rO rc rU r1 r r
r6 r r r � s zBoundedSemaphore.release)r )r r r r) r rN r r r6 r r � s r ) �__all__r> r% � r r r
r r r r r r r r r � s DzQ