File: //usr/local/lib/python3.10/dist-packages/anyio/streams/__pycache__/stapled.cpython-310.pyc
o
���g� � @ s� d dl mZ d dlmZmZmZ d dlmZ d dlm Z m
Z
m Z ddl m
Z
mZmZmZmZmZmZmZ e d�Ze d�Zed d
�G d d � d e��Zed d
�G d
d� de
e ee ��Zed d
�G dd� de
e ee ��ZdS )� )� annotations)�Callable�Mapping�Sequence)� dataclass)�Any�Generic�TypeVar� )�ByteReceiveStream�ByteSendStream�
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
dS ) �StapledByteStreamaV
Combines two byte streams into a single, bidirectional byte stream.
Extra attributes will be provided from both streams, with the receive stream
providing the values in case of a conflict.
:param ByteSendStream send_stream: the sending byte stream
:param ByteReceiveStream receive_stream: the receiving byte stream
r � send_streamr �receive_stream� � max_bytes�int�return�bytesc � s �| j �|�I d H S �N�r �receive)�selfr � r" �@/usr/local/lib/python3.10/dist-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 �
��z"StapledByteStream.extra_attributesN)r )r r r r )r$ r r r% �r r% �r r4 � �__name__�
__module__� __qualname__�__doc__�__annotations__r r( r/ r- �propertyr7 r" r" r" r# r s
r c @ sX e Zd ZU dZded< ded< ddd �Zdd d
�Zddd�Zddd�Ze ddd��Z
dS )�StapledObjectStreamab
Combines two object streams into a single, bidirectional object stream.
Extra attributes will be provided from both streams, with the receive stream
providing the values in case of a conflict.
:param ObjectSendStream send_stream: the sending object stream
:param ObjectReceiveStream receive_stream: the receiving object stream
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
rB c @ sJ e Zd ZU dZded< ddd�Z ddd
d�Zddd�Zeddd��Z dS )�
MultiListenera�
Combines multiple listeners into one, serving connections from all of them at once.
Any MultiListeners in the given collection of listeners will have their listeners
moved into this one.
Extra attributes are provided from each listener, with each successive listener
overriding any conflicting attributes from the previous one.
:param listeners: listeners to serve
:type listeners: Sequence[Listener[T_Stream]]
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 �
isinstancerC �extend�append)r! rD �listenerr" r" r# �
__post_init__o s
zMultiListener.__post_init__N�handler�Callable[[T_Stream], Any]�
task_group�TaskGroup | Nonec � sj �ddl m} |� 4 I d H �}| jD ]
}|�|j||� qW d �I d H d S 1 I d H s.w Y d S )Nr
)�create_task_group)� rN rD �
start_soon�serve)r! rJ rL rN �tgrH r" r" r# rQ z s �
�.�zMultiListener.servec � s �| j D ] }|�� I d H qd S r )rD r- )r! rH r" r" r# r- � s �
�zMultiListener.acloser4 c C s i }| j D ]}|�|j� q|S r )rD �updater7 )r! �
attributesrH r" r" r# r7 � s
z MultiListener.extra_attributesr9 r )rJ rK rL rM r r% r: )
r< r= r> r? r@ rI rQ r- rA r7 r" r" r" r# rC ^ s
�
rC N)�
__future__r �collections.abcr r r � dataclassesr �typingr r r �abcr r r
r r r r r r r r rB rC r" r" r" r# �<module> s ( # #