File: //home/arjun/.local/lib/python3.10/site-packages/anyio/streams/__pycache__/text.cpython-310.pyc
o
���g� � @ s� d dl mZ d dlZd dlmZmZ d dlmZmZm Z d dl
m Z ddl m
Z
mZmZmZmZmZ edd �G d
d � d ee ��Zedd �G d d
� d
ee ��Zedd �G dd� dee ��ZdS )� )� annotationsN)�Callable�Mapping)�InitVar� dataclass�field)�Any� )�AnyByteReceiveStream�AnyByteSendStream�
AnyByteStream�ObjectReceiveStream�ObjectSendStream� ObjectStreamF)�eqc @ sp e Zd ZU dZded< dZded< dZded< ed d
�Zd ed < ddd�Z ddd�Z
d dd�Z e d dd��Z
dS ) �TextReceiveStreama�
Stream wrapper that decodes bytes to strings using the given encoding.
Decoding is done using :class:`~codecs.IncrementalDecoder` which returns any
completely received unicode characters as soon as they come in.
:param transport_stream: any bytes-based receive stream
:param encoding: character encoding to use for decoding bytes to strings (defaults
to ``utf-8``)
:param errors: handling scheme for decoding errors (defaults to ``strict``; see the
`codecs module documentation`_ for a comprehensive list of options)
.. _codecs module documentation:
https://docs.python.org/3/library/codecs.html#codec-objects
r
�transport_stream�utf-8� InitVar[str]�encoding�strict�errorsF��initzcodecs.IncrementalDecoder�_decoder�str�return�Nonec C s t �|�}||d�| _d S )N)r )�codecs�getincrementaldecoderr )�selfr r �
decoder_class� r" �E/home/arjun/.local/lib/python3.10/site-packages/anyio/streams/text.py�
__post_init__) s
z TextReceiveStream.__post_init__c � s* � | j �� I d H }| j�|�}|r|S q�N)r �receiver �decode)r �chunk�decodedr" r" r# r& - s � �zTextReceiveStream.receivec � s �| j �� I d H | j�� d S r% )r �acloser �reset�r r" r" r# r* 4 s �zTextReceiveStream.aclose� Mapping[Any, Callable[[], Any]]c C � | j jS r% �r �extra_attributesr, r" r" r# r0 8 � z"TextReceiveStream.extra_attributesN�r r r r r r �r r �r r �r r- )�__name__�
__module__� __qualname__�__doc__�__annotations__r r r r r$ r& r* �propertyr0 r" r" r" r# r s
r c @ sp e Zd ZU dZded< dZded< dZded < ed
d �Zd ed
< ddd�Z d dd�Z
d dd�Z e d dd��Z
dS ) �TextSendStreama
Sends strings to the wrapped stream as bytes using the given encoding.
:param AnyByteSendStream transport_stream: any bytes-based send stream
:param str encoding: character encoding to use for encoding strings to bytes
(defaults to ``utf-8``)
:param str errors: handling scheme for encoding errors (defaults to ``strict``; see
the `codecs module documentation`_ for a comprehensive list of options)
.. _codecs module documentation:
https://docs.python.org/3/library/codecs.html#codec-objects
r r r r r r r r Fr z Callable[..., tuple[bytes, int]]�_encoderr r c C s t �|�| _d S r% )r �
getencoderr= )r r r" r" r# r$ Q s z TextSendStream.__post_init__�itemc � s* �| � || j�d }| j�|�I d H d S )Nr )r= r r �send)r r? �encodedr" r" r# r@ T s �zTextSendStream.sendc � � �| j �� I d H d S r% )r r* r, r" r" r# r* X � �zTextSendStream.acloser- c C r. r% r/ r, r" r" r# r0 [ r1 z TextSendStream.extra_attributesN)r r r r �r? r r r r4 r5 )r6 r7 r8 r9 r: r r r r= r$ r@ r* r; r0 r" r" r" r# r< = s
r< c @ s� e Zd ZU dZded< dZded< dZded< ed d
�Zd ed < ed d
�Z d
ed< d!dd�Z
d"dd�Z d#dd�Z d$dd�Z
d$dd �Zed%d d ��Zd S )&�
TextStreama�
A bidirectional stream that decodes bytes to strings on receive and encodes strings
to bytes on send.
Extra attributes will be provided from both streams, with the receive stream
providing the values in case of a conflict.
:param AnyByteStream transport_stream: any bytes-based stream
:param str encoding: character encoding to use for encoding/decoding strings to/from
bytes (defaults to ``utf-8``)
:param str errors: handling scheme for encoding errors (defaults to ``strict``; see
the `codecs module documentation`_ for a comprehensive list of options)
.. _codecs module documentation:
https://docs.python.org/3/library/codecs.html#codec-objects
r r r r r r r Fr r �_receive_streamr< � _send_streamr r r c C s( t | j||d�| _t| j||d�| _d S )N)r r )r r rF r< rG )r r r r" r" r# r$ y s � �zTextStream.__post_init__c � s �| j �� I d H S r% )rF r& r, r" r" r# r& � s �zTextStream.receiver? c � s �| j �|�I d H d S r% )rG r@ )r r? r" r" r# r@ � s �zTextStream.sendc � rB r% )r �send_eofr, r" r" r# rH � rC zTextStream.send_eofc � s& �| j �� I d H | j�� I d H d S r% )rG r* rF r, r" r" r# r* � s �zTextStream.acloser- c C s i | j j�| jj�S r% )rG r0 rF r, r" r" r# r0 � s
��zTextStream.extra_attributesNr2 r3 rD r4 r5 )r6 r7 r8 r9 r: r r r rF rG r$ r& r@ rH r* r; r0 r" r" r" r# rE ` s
rE )�
__future__r r �collections.abcr r � dataclassesr r r �typingr �abcr
r r r
r r r r r< rE r" r" r" r# �<module> s
*"