Format: application/x-python-code
1 o
2
3 ;'h@) � @ sT d dl mZ d dlZd dlmZmZ d dlmZmZ d dl m
4 Z
5 d dl m Z m
6 Z
7 mZ ddlmZmZmZmZ dd lmZmZ dd
8 lmZmZmZ dd lm Z ed �Z ed
9 dd�Z eddd�Z G dd� de
10 �Z edd�G dd� de e ��Z!edd�G dd� de e ��Z"edd�G dd� de e ee ��Z#edd�G d d � d e e ee ��Z$dS ) � )� annotationsN)� OrderedDict�deque)� dataclass�field)�
11 TracebackType)�Generic�
12 NamedTuple�TypeVar� )�BrokenResourceError�ClosedResourceError� EndOfStream�
13 WouldBlock)�TaskInfo�get_current_task)�Event�ObjectReceiveStream�ObjectSendStream)�
14 checkpoint�T_Item�T_coT)� covariant�T_contra)�
15 contravariantc @ s> e Zd ZU ded< ded< ded< ded< ded< ded< d S )
16 � MemoryObjectStreamStatistics�intZcurrent_buffer_used�float�max_buffer_sizeZopen_send_streamsZopen_receive_streamsZtasks_waiting_sendZtasks_waiting_receiveN)�__name__�
17 __module__� __qualname__�__annotations__� r# r# �JC:\Users\shubh\Videos\agent\nova\Lib\site-packages\anyio/streams/memory.pyr s
18 r F)�eqc @ s> e Zd ZU eded�Zded< edd�Zded< d
19 d
20 d �Zd S )�MemoryObjectItemReceiverF��init�default_factoryr � task_info)r( r �item�return�strc 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* r" r+ r2 r# r# r# r$ r& $ s
21 r& c @ s� e Zd ZU e� Zded< eded�Zded< eddd�Zd ed
22 < eddd�Z d ed < ede
23 d�Z d ed
24 < ede
25 d�Z ded< ddd�Z
26 dS )�MemoryObjectStreamStater r Fr' z
27 deque[T_Item]�bufferr �r( �defaultr �open_send_channels�open_receive_channelsz4OrderedDict[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 �lenr4 r r7 r8 r: r9 �r1 r# r# r$ �
28 statistics= s �z"MemoryObjectStreamState.statisticsN�r, r )r r r! r r r" r r4 r7 r8 r r9 r: r> r# r# r# r$ r3 0 s
29 ��r3 c @ s� e Zd ZU ded< eddd�Zded< d'd d
30 �Zd(d d
31 �Zd(dd�Zd)dd�Z d'dd�Z
32 d'dd�Z d*dd�Z d)dd�Z
33 d+d"d#�Zd'd$d%�Zd&S ),�MemoryObjectReceiveStreamz MemoryObjectStreamState[T_co]�_stateFr5 �bool�_closedr, �Nonec C � | j jd7 _d S �N� )rA r8 r= r# r# r$ �
34 __post_init__M � z'MemoryObjectReceiveStream.__post_init__r c C s` | j rt�| jjr | jjjdd�\}}| jj�|� |�� | jjr(| jj�� S | jj s.t
35 �t �)a�
36 Receive the next item if it can be done without waiting.
37
38 :return: the received item
39 :raises ~anyio.ClosedResourceError: if this send stream has been closed
40 :raises ~anyio.EndOfStream: if the buffer is empty and this stream has been
41 closed from the sending end
42 :raises ~anyio.WouldBlock: if there are no items in the buffer and no tasks
43 waiting to send
44
45 F��last) rC r
46 rA r: �popitemr4 �append�set�popleftr7 r r )r1 �
47 send_eventr+ r# r# r$ �receive_nowaitP s z(MemoryObjectReceiveStream.receive_nowaitc
48 � s� �t � I d H z| �� W S tyM t� }tt � }|| jj|< z|�� I d H W | jj� |d � n | jj� |d � w z|j
49 W Y S t yL t d �w w r; )
50 r rQ r r r& r rA r9 �wait�popr+ �AttributeErrorr )r1 �
51 receive_event�receiverr# r# r$ �receivel s" �
52
53 $ ��z!MemoryObjectReceiveStream.receive� MemoryObjectReceiveStream[T_co]c C � | j rt�t| jd�S )a
54 Create a clone of this receive stream.
55
56 Each clone can be closed separately. Only when all clones have been closed will
57 the receiving end of the memory stream be considered closed by the sending ends.
58
59 :return: the cloned stream
60
61 �rA )rC r
62 r@ rA r= r# r# r$ �clone� �
63 z MemoryObjectReceiveStream.clonec C sV | j s%d| _ | j jd8 _| jjdkr't| jj�� �}|D ]
64 }|�� q dS dS dS �z�
65 Close the stream.
66
67 This works the exact same way as :meth:`aclose`, but is provided as a special
68 case for the benefit of synchronous callbacks.
69
70 TrG r N)rC rA r8 �listr: �keysrN )r1 Z send_events�eventr# r# r$ �close� s
71 �z MemoryObjectReceiveStream.closec � � �| � � d S r; �ra r= r# r# r$ �aclose� � � z MemoryObjectReceiveStream.acloser c C �
72 | j �� S �zj
73 Return statistics about the current state of this stream.
74
75 .. versionadded:: 3.0
76 �rA r> r= r# r# r$ r> � �
77 z$MemoryObjectReceiveStream.statisticsc C � | S r; r# r= r# r# r$ � __enter__� � z#MemoryObjectReceiveStream.__enter__�exc_type�type[BaseException] | None�exc_val�BaseException | None�exc_tb�TracebackType | Nonec C � | � � d S r; rc �r1 rm ro rq r# r# r$ �__exit__� � z"MemoryObjectReceiveStream.__exit__c C �6 | j stjd| jj� dt| �d�d�t| d� d S d S �Nz
78 Unclosed <z at �x�>)�source�rC �warnings�warnr0 r �id�ResourceWarningr= r# r# r$ �__del__� �
79 ��z!MemoryObjectReceiveStream.__del__N�r, rD )r, r )r, rX r? �rm rn ro rp rq rr r, rD )r r r! r" r rC rH rQ rW r[ ra rd r> rk ru r� r# r# r# r$ r@ H s
80
81
82
83
84
85
86
87
88
89 r@ c @ s� e Zd ZU ded< eddd�Zded< d(d d
90 �Zd)d
91 d�Zd)dd�Zd*dd�Z d(dd�Z
92 d(dd�Z d+dd�Z d*dd �Z
93 d,d#d$�Zd(d%d&�Zd'S )-�MemoryObjectSendStreamz!MemoryObjectStreamState[T_contra]rA Fr5 rB rC r, rD c C rE rF )rA r7 r= r# r# r$ rH � rI z$MemoryObjectSendStream.__post_init__r+ r c C s� | j rt�| jjs t�| jjr+| jjjdd�\}}|j�� s'||_ |�
94 � dS | jjst | jj �| jj
95 k r>| jj �|� dS t�)a�
96 Send an item immediately if it can be done without waiting.
97
98 :param item: the item to send
99 :raises ~anyio.ClosedResourceError: if this send stream has been closed
100 :raises ~anyio.BrokenResourceError: if the stream has been closed from the
101 receiving end
102 :raises ~anyio.WouldBlock: if the buffer is full and there are no tasks waiting
103 to receive
104
105 FrJ N)rC r
106 rA r8 r r9 rL r* Zhas_pending_cancellationr+ rN r< r4 r rM r )r1 r+ rU rV r# r# r$ � send_nowait� s
107 �z"MemoryObjectSendStream.send_nowaitc � s� �t � I dH z| �|� W dS tyJ t� }|| jj|< z |�� I dH W n ty8 | jj�|d� � w || jjv rG| jj|= t d�Y dS w )a�
108 Send an item to the stream.
109
110 If the buffer is full, this method blocks until there is again room in the
111 buffer or the item can be sent directly to a receiver.
112
113 :param item: the item to send
114 :raises ~anyio.ClosedResourceError: if this send stream has been closed
115 :raises ~anyio.BrokenResourceError: if the stream has been closed from the
116 receiving end
117
118 N)
119 r r� r r rA r: rR �
120 BaseExceptionrS r )r1 r+ rP r# r# r$ �send� s$ �
121 �
122 ��zMemoryObjectSendStream.send� MemoryObjectSendStream[T_contra]c C rY )a
123 Create a clone of this send stream.
124
125 Each clone can be closed separately. Only when all clones have been closed will
126 the sending end of the memory stream be considered closed by the receiving ends.
127
128 :return: the cloned stream
129
130 rZ )rC r
131 r� rA r= r# r# r$ r[ r\ z MemoryObjectSendStream.clonec C sb | j s+d| _ | j jd8 _| jjdkr-t| jj�� �}| jj�� |D ]
132 }|�� q$dS dS dS r] )rC rA r7 r^ r9 r_ �clearrN )r1 Zreceive_eventsr` r# r# r$ ra s
133 �z MemoryObjectSendStream.closec � rb r; rc r= r# r# r$ rd ! re z MemoryObjectSendStream.acloser c C rf rg rh r= r# r# r$ r> $ ri z!MemoryObjectSendStream.statisticsc C rj r; r# r= r# r# r$ rk , rl z MemoryObjectSendStream.__enter__rm rn ro rp rq rr c C rs r; rc rt r# r# r$ ru / rv z MemoryObjectSendStream.__exit__c C rw rx r| r= r# r# r$ r� 7 r� z MemoryObjectSendStream.__del__Nr� )r+ r r, rD )r, r� r? r� )r r r! r" r rC rH r� r� r[ ra rd r> rk ru r� r# r# r# r$ r� � s
134
135
136
137
138
139
140
141
142
143 r� )%�
144 __future__r r} � collectionsr r � dataclassesr r �typesr �typingr r r
145 � r r
146 r r Z_core._testingr r �abcr r r Zlowlevelr r r r r r& r3 r@ r� r# r# r# r$ �<module> s, u