o "h@sddlmZddlZddlmZmZmZddlmZm Z ddl m Z m Z ddl mZmZmZmZddlmZmZmZmZmZmZejd krLdd l mZndd lmZed d Gd ddeZGdddeeZGdddeZdS)) annotationsN)CallableIterableMapping) dataclassfield)Any SupportsIndex)ClosedResourceErrorDelimiterNotFound EndOfStreamIncompleteRead)AnyByteReceiveStream AnyByteStreamAnyByteStreamConnectableByteReceiveStream ByteStreamByteStreamConnectable) )overrideF)eqc@seZdZUdZded<ededZded<edddZd ed <d%d dZ e d&ddZ e d'ddZ d(ddZ d)d*ddZd+dd Zd,d"d#Zd$S)-BufferedByteReceiveStreamz Wraps any bytes-based receive stream and uses a buffer to provide sophisticated receiving capabilities in the form of a byte stream. rreceive_streamF)initdefault_factory bytearray_buffer)rdefaultbool_closedreturnNonecs|jIdHd|_dS)NT)racloser!selfr'I/var/www/html/venv/lib/python3.10/site-packages/anyio/streams/buffered.pyr$#s z BufferedByteReceiveStream.aclosebytescCs t|jS)z"The bytes currently in the buffer.)r)rr%r'r'r(buffer's z BufferedByteReceiveStream.bufferMapping[Any, Callable[[], Any]]cCs|jjSN)rextra_attributesr%r'r'r(r-,sz*BufferedByteReceiveStream.extra_attributesdataIterable[SupportsIndex]cCs|j|dS)a; Append data directly into the buffer. Any data in the buffer will be consumed by receive operations before receiving anything from the wrapped stream. :param data: the data to append to the buffer (can be bytes or anything else that supports ``__index__()``) N)rextend)r&r.r'r'r( feed_data0s z#BufferedByteReceiveStream.feed_data max_bytesintcs|jrt|jrt|jd|}|jd|=|St|jtr)|j|IdHS|jIdH}t||krG|j ||d|d|S|Sr,) r!r rr) isinstancerrreceivelenr0)r&r3chunkr'r'r(r6=s    z!BufferedByteReceiveStream.receivenbytesc s |t|j}|dkr|jd|}|jd|=t|Szt|jtr/|j|IdH}n|jIdH}WntyG}zt|d}~ww|j |q)a- Read exactly the given amount of bytes from the stream. :param nbytes: the number of bytes to read :return: the bytes read :raises ~anyio.IncompleteRead: if the stream was closed before the requested amount of bytes could be read from the stream TrN) r7rr)r5rrr6r rr0)r&r9 remainingretvalr8excr'r'r(receive_exactlyRs$    z)BufferedByteReceiveStream.receive_exactly delimiterc st|}d} |j||}|dkr(|jd|}|jd|t|=t|St|j|kr3t|z |jIdH}WntyL}zt|d}~wwt t|j|dd}|j |q)aM Read from the stream until the delimiter is found or max_bytes have been read. :param delimiter: the marker to look for in the stream :param max_bytes: maximum number of bytes that will be read before raising :exc:`~anyio.DelimiterNotFound` :return: the bytes read (not including the delimiter) :raises ~anyio.IncompleteRead: if the stream was closed before the delimiter was found :raises ~anyio.DelimiterNotFound: if the delimiter is not found within the bytes read up to the maximum allowed rTN) r7rfindr)r rr6r rmaxr0) r&r>r3delimiter_sizeoffsetindexfoundr.r<r'r'r( receive_untilms( z'BufferedByteReceiveStream.receive_untilNr"r#)r"r))r"r+)r.r/r"r#)r2)r3r4r"r))r9r4r"r))r>r)r3r4r"r))__name__ __module__ __qualname____doc____annotations__rrrr!r$propertyr*r-r1r6r=rFr'r'r'r(rs      rcs>eZdZdZdfdd Zeddd Zedd d ZZS)BufferedByteStreamz A full-duplex variant of :class:`BufferedByteReceiveStream`. All writes are passed through to the wrapped stream as-is. streamrcst|||_dS)z: :param stream: the stream to be wrapped N)super__init___streamr&rO __class__r'r(rQs  zBufferedByteStream.__init__r"r#cs|jIdHdSr,)rRsend_eofr%r'r'r(rVszBufferedByteStream.send_eofitemr)cs|j|IdHdSr,)rRsend)r&rWr'r'r(rXszBufferedByteStream.send)rOrrG)rWr)r"r#) rHrIrJrKrQrrVrX __classcell__r'r'rTr(rNs rNc@s$eZdZd ddZed ddZd S) BufferedConnectable connectablercCs ||_dS)z> :param connectable: the connectable to wrap N)r[)r&r[r'r'r(rQs zBufferedConnectable.__init__r"rNcs|jIdH}t|Sr,)r[connectrNrSr'r'r(r\szBufferedConnectable.connectN)r[r)r"rN)rHrIrJrQrr\r'r'r'r(rZs rZ) __future__rsyscollections.abcrrr dataclassesrrtypingrr r r r rabcrrrrrr version_infortyping_extensionsrrNrZr'r'r'r(s    {