ruạṛ
a �u:j�) � @ sX d dl mZ dZd dlZd dlmZmZ d dlmZm Z d dl m Z d dl m Z mZmZ dd lmZmZmZmZ dd lmZmZ dd lmZmZmZ dd l m Z ed �Z eddd�Z eddd�Z G dd� de�Z!edd�G dd� de e ��Z"edd�G dd� de e ��Z#edd�G dd � d e e ee ��Z$edd�G d d � d e e ee ��Z%dS ) � )� annotations)�MemoryObjectReceiveStream�MemoryObjectSendStream� MemoryObjectStreamStatisticsN)� OrderedDict�deque)� dataclass�field)� TracebackType)�Generic� NamedTuple�TypeVar� )�BrokenResourceError�ClosedResourceError� EndOfStream� WouldBlock)�TaskInfo�get_current_task)�Event�ObjectReceiveStream�ObjectSendStream)� checkpoint�T_Item�T_coT)� covariant�T_contra)� contravariantc @ s> e Zd ZU ded< ded< ded< ded< ded< ded< d S ) r �intZcurrent_buffer_used�float�max_buffer_sizeZopen_send_streamsZopen_receive_streamsZtasks_waiting_sendZtasks_waiting_receiveN)�__name__� __module__� __qualname__�__annotations__� r% r% �G/opt/bart/bart_venv/lib/python3.9/site-packages/anyio/streams/memory.pyr s r F)�eqc @ sB e Zd ZU eded�Zded< edd�Zded< dd �d d �Zd S ) �_MemoryObjectItemReceiverF��init�default_factoryr � task_info)r* r �item�str��returnc C s( t | dd �}| jj� d| j� d|�d�S )Nr- z (task_info=z, item=�))�getattr� __class__r! r, )�selfr- r% r% r& �__repr__/ s z"_MemoryObjectItemReceiver.__repr__N) r! r" r# r r r, r$ r- r5 r% r% r% r& r( * s r( c @ s� e Zd ZU e� Zded< eded�Zded< eddd�Zd ed < eddd�Z d ed < ede d�Z d ed < ede d�Z ded< dd�dd�Z dS )�_MemoryObjectStreamStater r Fr) z deque[T_Item]�bufferr �r* �defaultr �open_send_channels�open_receive_channelsz5OrderedDict[Event, _MemoryObjectItemReceiver[T_Item]]�waiting_receiverszOrderedDict[Event, T_Item]�waiting_sendersr r/ c C s* t t| j�| j| j| jt| j�t| j��S �N)r �lenr7 r r: r; r= r< �r4 r% r% r& � statisticsC s �z#_MemoryObjectStreamState.statisticsN)r! r" r# r r r$ r r7 r: r; r r< r= rA r% r% r% r& r6 6 s ��r6 c @ s� e Zd ZU ded< eddd�Zded< dd�d d �Zd d�d d �Zd d�dd�Zdd�dd�Z dd�dd�Z dd�dd�Z dd�dd�Z dd�dd�Z d d d dd �d d!�Zdd�d"d#�Zd$S )%r z _MemoryObjectStreamState[T_co]�_stateFr8 �bool�_closed�Noner/ c C s | j jd7 _d S �N� )rB r; r@ r% r% r& � __post_init__S s z'MemoryObjectReceiveStream.__post_init__r c C sd | j r t�| jjr<| jjjdd�\}}| jj�|� |�� | jjrP| jj�� S | jj s\t �t �dS )a� Receive the next item if it can be done without waiting. :return: the received item :raises ~anyio.ClosedResourceError: if this send stream has been closed :raises ~anyio.EndOfStream: if the buffer is empty and this stream has been closed from the sending end :raises ~anyio.WouldBlock: if there are no items in the buffer and no tasks waiting to send F��lastN) rD r rB r= �popitemr7 �append�set�popleftr: r r )r4 � send_eventr- r% r% r& �receive_nowaitV s z(MemoryObjectReceiveStream.receive_nowaitc � s� t � I d H z | �� W S ty� t� }tt � }|| jj|< z"|�� I d H W | jj� |d � n| jj� |d � 0 z |j W Y S t y� t d �Y n0 Y n0 d S r> ) r rP r r r( r rB r< �wait�popr- �AttributeErrorr )r4 � receive_event�receiverr% r% r& �receiver s $ z!MemoryObjectReceiveStream.receivez MemoryObjectReceiveStream[T_co]c C s | j r t�t| jd�S )a Create a clone of this receive stream. Each clone can be closed separately. Only when all clones have been closed will the receiving end of the memory stream be considered closed by the sending ends. :return: the cloned stream �rB )rD r r rB r@ r% r% r& �clone� s z MemoryObjectReceiveStream.clonec C sN | j sJd| _ | j jd8 _| jjdkrJt| jj�� �}|D ] }|�� q<dS �z� Close the stream. This works the exact same way as :meth:`aclose`, but is provided as a special case for the benefit of synchronous callbacks. TrG r N)rD rB r; �listr= �keysrM )r4 Z send_events�eventr% r% r& �close� s z MemoryObjectReceiveStream.closec � s | � � d S r> �r] r@ r% r% r& �aclose� s z MemoryObjectReceiveStream.acloser c C s | j �� S �zj Return statistics about the current state of this stream. .. versionadded:: 3.0 �rB rA r@ r% r% r& rA � s z$MemoryObjectReceiveStream.statisticsc C s | S r> r% r@ r% r% r& � __enter__� s z#MemoryObjectReceiveStream.__enter__�type[BaseException] | None�BaseException | None�TracebackType | None��exc_type�exc_val�exc_tbr0 c C s | � � d S r> r^ �r4 rg rh ri r% r% r& �__exit__� s z"MemoryObjectReceiveStream.__exit__c C s4 | j s0tjd| jj� dt| �d�d�td| d� d S �Nz Unclosed <z at �x�>rG )� stacklevel�source�rD �warnings�warnr3 r! �id�ResourceWarningr@ r% r% r& �__del__� s �z!MemoryObjectReceiveStream.__del__N)r! r" r# r$ r rD rH rP rV rX r] r_ rA rb rk rv r% r% r% r& r N s r c @ s� e Zd ZU ded< eddd�Zded< dd�d d �Zd dd �d d�Zd dd �dd�Zdd�dd�Z dd�dd�Z dd�dd�Z dd�dd�Z dd�dd �Z d d d dd �d!d"�Zdd�d#d$�Zd%S )&r z"_MemoryObjectStreamState[T_contra]rB Fr8 rC rD rE r/ c C s | j jd7 _d S rF )rB r: r@ r% r% r& rH � s z$MemoryObjectSendStream.__post_init__r )r- r0 c C s| | j r t�| jjst�| jjrP| jjjdd�\}}|j�� s||_ |� � dS qt | jj �| jj k rt| jj �|� nt�dS )a� Send an item immediately if it can be done without waiting. :param item: the item to send :raises ~anyio.ClosedResourceError: if this send stream has been closed :raises ~anyio.BrokenResourceError: if the stream has been closed from the receiving end :raises ~anyio.WouldBlock: if the buffer is full and there are no tasks waiting to receive FrI N)rD r rB r; r r< rK r, Zhas_pending_cancellationr- rM r? r7 r rL r )r4 r- rT rU r% r% r& � send_nowait� s z"MemoryObjectSendStream.send_nowaitc � s� t � I dH z| �|� W nx ty� t� }|| jj|< z|�� I dH W n$ typ | jj�|d� � Y n0 || jjv r�| jj|= t d�Y n0 dS )a� Send an item to the stream. If the buffer is full, this method blocks until there is again room in the buffer or the item can be sent directly to a receiver. :param item: the item to send :raises ~anyio.ClosedResourceError: if this send stream has been closed :raises ~anyio.BrokenResourceError: if the stream has been closed from the receiving end N) r rw r r rB r= rQ � BaseExceptionrR r )r4 r- rO r% r% r& �send� s zMemoryObjectSendStream.sendz MemoryObjectSendStream[T_contra]c C s | j r t�t| jd�S )a Create a clone of this send stream. Each clone can be closed separately. Only when all clones have been closed will the sending end of the memory stream be considered closed by the receiving ends. :return: the cloned stream rW )rD r r rB r@ r% r% r& rX s z MemoryObjectSendStream.clonec C sZ | j sVd| _ | j jd8 _| jjdkrVt| jj�� �}| jj�� |D ] }|�� qHdS rY )rD rB r: rZ r< r[ �clearrM )r4 Zreceive_eventsr\ r% r% r& r] s z MemoryObjectSendStream.closec � s | � � d S r> r^ r@ r% r% r& r_ ( s z MemoryObjectSendStream.acloser c C s | j �� S r` ra r@ r% r% r& rA + s z!MemoryObjectSendStream.statisticsc C s | S r> r% r@ r% r% r& rb 3 s z MemoryObjectSendStream.__enter__rc rd re rf c C s | � � d S r> r^ rj r% r% r& rk 6 s z MemoryObjectSendStream.__exit__c C s4 | j s0tjd| jj� dt| �d�d�td| d� d S rl rq r@ r% r% r& rv > s �z MemoryObjectSendStream.__del__N)r! r" r# r$ r rD rH rw ry rX r] r_ rA rb rk rv r% r% r% r& r � s r )&� __future__r �__all__rr � collectionsr r � dataclassesr r �typesr �typingr r r � r r r r Z_core._testingr r �abcr r r Zlowlevelr r r r r r( r6 r r r% r% r% r& �<module> s, v
cải xoăn