a h@sddlmZddlZddlmZmZmZddlmZm Z ddl m Z m Z ddl mZmZmZmZddlmZmZmZmZmZmZejd krdd l mZn dd lmZed d Gd ddeZGdddeeZGdddeZdS)) annotationsN)CallableIterableMapping) dataclassfield)Any SupportsIndex)ClosedResourceErrorDelimiterNotFound EndOfStreamIncompleteRead)AnyByteReceiveStream AnyByteStreamAnyByteStreamConnectableByteReceiveStream ByteStreamByteStreamConnectable) )overrideF)eqc@seZdZUdZded<ededZded<edddZd ed <d d d dZ e dd ddZ e dd ddZ dd dddZ d%dddddZddddd Zdddd!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_closedNonereturncs|jIdHd|_dS)NT)racloser!selfr(R/var/www/html/assistant/venv/lib/python3.9/site-packages/anyio/streams/buffered.pyr%#sz BufferedByteReceiveStream.aclosebytescCs t|jS)z"The bytes currently in the buffer.)r*rr&r(r(r)buffer'sz BufferedByteReceiveStream.bufferzMapping[Any, Callable[[], Any]]cCs|jjSN)rextra_attributesr&r(r(r)r-,sz*BufferedByteReceiveStream.extra_attributeszIterable[SupportsIndex])datar$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_dataint) max_bytesr$cs|jr t|jr2t|jd|}|jd|=|St|jtrP|j|IdHS|jIdH}t||kr|j ||d|d|S|SdSr,) r!r rr* isinstancerrreceivelenr/)r'r3chunkr(r(r)r5=s    z!BufferedByteReceiveStream.receive)nbytesr$c s|t|j}|dkr8|jd|}|jd|=t|Sz4t|jtrZ|j|IdH}n|jIdH}Wn*ty}zt|WYd}~n d}~00|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) r6rr*r4rrr5r rr/)r'r8 remainingretvalr7excr(r(r)receive_exactlyRs   z)BufferedByteReceiveStream.receive_exactly) delimiterr3r$c st|}d}|j||}|dkrL|jd|}|jd|t|=t|St|j|krbt|z|jIdH}Wn*ty}zt|WYd}~n d}~00t 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) r6rfindr*r rr5r rmaxr/) r'r=r3Zdelimiter_sizeoffsetindexfoundr.r;r(r(r) receive_untilmsz'BufferedByteReceiveStream.receive_untilN)r1)__name__ __module__ __qualname____doc____annotations__rrrr!r%propertyr+r-r0r5r<rDr(r(r(r)rs  rcsLeZdZdZddfdd Zedddd Zed dd d d ZZS)BufferedByteStreamz A full-duplex variant of :class:`BufferedByteReceiveStream`. All writes are passed through to the wrapped stream as-is. r)streamcst|||_dS)z: :param stream: the stream to be wrapped N)super__init___streamr'rL __class__r(r)rNs zBufferedByteStream.__init__r"r#cs|jIdHdSr,)rOsend_eofr&r(r(r)rSszBufferedByteStream.send_eofr*)itemr$cs|j|IdHdSr,)rOsend)r'rTr(r(r)rUszBufferedByteStream.send) rErFrGrHrNrrSrU __classcell__r(r(rQr)rKs rKc@s,eZdZddddZeddddZd S) BufferedConnectabler connectablecCs ||_dS)z> :param connectable: the connectable to wrap NrX)r'rYr(r(r)rNszBufferedConnectable.__init__rKr#cs|jIdH}t|Sr,)rYconnectrKrPr(r(r)rZszBufferedConnectable.connectN)rErFrGrNrrZr(r(r(r)rWsrW) __future__rsyscollections.abcrrrZ dataclassesrrtypingrr r r r rabcrrrrrr version_inforZtyping_extensionsrrKrWr(r(r(r)s   {