o dU@sjddlmZmZddlmZmZmZddlmZm Z m Z m Z ddl m Z mZeddGdd d eZd S) ) dataclassfield)AnyCallableMapping)ClosedResourceErrorDelimiterNotFound EndOfStreamIncompleteRead)AnyByteReceiveStreamByteReceiveStreamF)eqc@seZdZUdZeed<ededZeed<edddZ e ed<dd d Z e de fd d Ze deeegefffddZddede fddZdede fddZde dede fddZd S)BufferedByteReceiveStreamz Wraps any bytes-based receive stream and uses a buffer to provide sophisticated receiving capabilities in the form of a byte stream. receive_streamF)initdefault_factory_buffer)rdefault_closedreturnNcs|jIdHd|_dS)NT)racloserselfrA/usr/local/lib/python3.10/dist-packages/anyio/streams/buffered.pyrs z BufferedByteReceiveStream.aclosecCs t|jS)z"The bytes currently in the buffer.)bytesrrrrrbuffers z BufferedByteReceiveStream.buffercCs|jjSN)rextra_attributesrrrrrsz*BufferedByteReceiveStream.extra_attributes max_bytescs|jrt|jrt|jd|}|jd|=|St|jtr)|j|IdHS|jIdH}t||krG|j ||d|d|S|Sr) rrrr isinstancerr receivelenextend)rr!chunkrrrr# 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) r$rrr"rr r#r r r%)rr' remainingretvalr&excrrrreceive_exactly5s$    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) r$rfindrr rr#r r maxr%) rr,r!delimiter_sizeoffsetindexfounddatar*rrr receive_untilPs( z'BufferedByteReceiveStream.receive_until)rN)r )__name__ __module__ __qualname____doc__r __annotations__r bytearrayrrboolrpropertyrrrrrrintr#r+r5rrrrrs   rN) dataclassesrrtypingrrrrr r r abcr r rrrrrs