Format: application/x-python-code
1 o
2
3 ;'h� � @ s� d dl mZ d dlmZmZmZ d dlmZ d dlm Z m
4 Z
5 m Z ddl m
6 Z
7 mZmZmZmZmZmZmZ e d�Ze d�Zed d
8 �G d d � d e��Zed d
9 �G d
10 d� de
11 e ee ��Zed d
12 �G dd� de
13 e ee ��ZdS )� )� annotations)�Callable�Mapping�Sequence)� dataclass)�Any�Generic�TypeVar� )�ByteReceiveStream�ByteSendStream�
14 ByteStream�Listener�ObjectReceiveStream�ObjectSendStream� ObjectStream� TaskGroup�T_Item�T_StreamF)�eqc @ sZ e Zd ZU dZded< ded< ddd d �Zddd�Zd dd�Zd dd�Ze d dd��Z
15 dS ) �StapledByteStreamaV
16 Combines two byte streams into a single, bidirectional byte stream.
17
18 Extra attributes will be provided from both streams, with the receive stream
19 providing the values in case of a conflict.
20
21 :param ByteSendStream send_stream: the sending byte stream
22 :param ByteReceiveStream receive_stream: the receiving byte stream
23 r � send_streamr �receive_stream� � max_bytes�int�return�bytesc � s �| j �|�I d H S �N�r �receive)�selfr � r" �KC:\Users\shubh\Videos\agent\nova\Lib\site-packages\anyio/streams/stapled.pyr % s �zStapledByteStream.receive�item�Nonec � � �| j �|�I d H d S r �r �send�r! r$ r" r" r# r( ( � �zStapledByteStream.sendc � � �| j �� I d H d S r �r �aclose�r! r" r" r# �send_eof+ � �zStapledByteStream.send_eofc � �& �| j �� I d H | j�� I d H d S r �r r- r r. r" r" r# r- . � �zStapledByteStream.aclose� Mapping[Any, Callable[[], Any]]c C � i | j j�| jj�S r �r �extra_attributesr r. r" r" r# r7 2 �
24 ��z"StapledByteStream.extra_attributesN)r )r r r r )r$ r r r% �r r% �r r4 � �__name__�
25 __module__� __qualname__�__doc__�__annotations__r r( r/ r- �propertyr7 r" r" r" r# r s
26
27
28
29
30 r c @ sX e Zd ZU dZded< ded< ddd �Zdd d
31 �Zddd�Zddd�Ze ddd��Z
32 dS )�StapledObjectStreamab
33 Combines two object streams into a single, bidirectional object stream.
34
35 Extra attributes will be provided from both streams, with the receive stream
36 providing the values in case of a conflict.
37
38 :param ObjectSendStream send_stream: the sending object stream
39 :param ObjectReceiveStream receive_stream: the receiving object stream
40 zObjectSendStream[T_Item]r zObjectReceiveStream[T_Item]r r r c � s �| j �� I d H S r r r. r" r" r# r I s �zStapledObjectStream.receiver$ r% c � r& r r' r) r" r" r# r( L r* zStapledObjectStream.sendc � r+ r r, r. r" r" r# r/ O r0 z StapledObjectStream.send_eofc � r1 r r2 r. r" r" r# r- R r3 zStapledObjectStream.acloser4 c C r5 r r6 r. r" r" r# r7 V r8 z$StapledObjectStream.extra_attributesN)r r )r$ r r r% r9 r: r; r" r" r" r# rB : s
41
42
43
44
45
46 rB c @ sJ e Zd ZU dZded< ddd�Z ddd
47 d�Zddd�Zeddd��Z dS )�
48 MultiListenera�
49 Combines multiple listeners into one, serving connections from all of them at once.
50
51 Any MultiListeners in the given collection of listeners will have their listeners
52 moved into this one.
53
54 Extra attributes are provided from each listener, with each successive listener
55 overriding any conflicting attributes from the previous one.
56
57 :param listeners: listeners to serve
58 :type listeners: Sequence[Listener[T_Stream]]
59 z Sequence[Listener[T_Stream]]� listenersr r% c C sH g }| j D ]}t|t�r|�|j � |j d d �= q|�|� q|| _ d S r )rD �
60 isinstancerC �extend�append)r! rD �listenerr" r" r# �
61 __post_init__o s
62
63
64 zMultiListener.__post_init__N�handler�Callable[[T_Stream], Any]�
65 task_group�TaskGroup | Nonec � sj �ddl m} |� 4 I d H �}| jD ]
66 }|�|j||� qW d �I d H d S 1 I d H s.w Y d S )Nr
67 )�create_task_group)� rN rD Z
68 start_soon�serve)r! rJ rL rN �tgrH r" r" r# rP z s �
69 �.�zMultiListener.servec � s �| j D ] }|�� I d H qd S r )rD r- )r! rH r" r" r# r- � s �
70 �zMultiListener.acloser4 c C s i }| j D ]}|�|j� q|S r )rD �updater7 )r! �
71 attributesrH r" r" r# r7 � s
72 z MultiListener.extra_attributesr9 r )rJ rK rL rM r r% r: )
73 r< r= r> r? r@ rI rP r- rA r7 r" r" r" r# rC ^ s
74
75 �
76 rC N)�
77 __future__r �collections.abcr r r � dataclassesr �typingr r r �abcr r r
78 r r r r r r r r rB rC r" r" r" r# �<module> s ( # #