U g@s~ddlmZddlmZmZddlmZmZddlm Z ddl m Z m Z m Z mZddlmZmZedd Gd d d eZd S) ) annotations)CallableMapping) dataclassfield)Any)ClosedResourceErrorDelimiterNotFound EndOfStreamIncompleteRead)AnyByteReceiveStreamByteReceiveStreamF)eqc@seZdZUdZded<ededZded<edddZd ed <d d d dZ e dd ddZ e dd ddZ d!dddddZ dddddZddddddZd S)"BufferedByteReceiveStreamz Wraps any bytes-based receive stream and uses a buffer to provide sophisticated receiving capabilities in the form of a byte stream. r receive_streamF)initdefault_factory bytearray_buffer)rdefaultbool_closedNone)returncs|jIdHd|_dS)NT)racloserselfr:/tmp/pip-unpacked-wheel-3u0cc9gf/anyio/streams/buffered.pyrsz BufferedByteReceiveStream.aclosebytescCs t|jS)z"The bytes currently in the buffer.)r rrrrrbuffersz BufferedByteReceiveStream.bufferzMapping[Any, Callable[[], Any]]cCs|jjSN)rextra_attributesrrrrr#sz*BufferedByteReceiveStream.extra_attributesint) max_bytesrcs|jr t|jr2t|jd|}|jd|=|St|jtrP|j|IdHS|jIdH}t||kr|j ||d|d|S|SdSr") rr rr isinstancerrreceivelenextend)rr&chunkrrrr(#s    z!BufferedByteReceiveStream.receive)nbytesrc s|t|j}|dkr8|jd|}|jd|=t|Sz4t|jtrZ|j|IdH}n|jIdH}Wn(tk r}z t|W5d}~XYnX|j |qdS)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 rN) r)rr r'rrr(r r r*)rr, remainingretvalr+excrrrreceive_exactly8s   z)BufferedByteReceiveStream.receive_exactly) delimiterr&rc st|}d}|j||}|dkrL|jd|}|jd|t|=t|St|j|krbt|z|jIdH}Wn(tk r}z t|W5d}~XYnXt t|j|dd}|j |q dS)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 rN) r)rfindr r rr(r r maxr*) rr1r&Zdelimiter_sizeoffsetindexfounddatar/rrr receive_untilSsz'BufferedByteReceiveStream.receive_untilN)r$)__name__ __module__ __qualname____doc____annotations__rrrrrpropertyr!r#r(r0r9rrrrr s rN) __future__rcollections.abcrrZ dataclassesrrtypingrr r r r abcr rrrrrrs