o _c@s*ddlmZddlmZGdddZdS)) annotations) Generatorc@s`eZdZdZdddZdd d Zdd d ZdddZdddZd ddZ dddZ dddZ dS)! StreamReaderz Generator-based stream reader. This class doesn't support concurrent calls to :meth:`read_line`, :meth:`read_exact`, or :meth:`read_to_eof`. Make sure calls are serialized. returnNonecCst|_d|_dS)NF) bytearraybuffereofselfr =/usr/local/lib/python3.10/dist-packages/websockets/streams.py__init__s zStreamReader.__init__mintGenerator[None, None, bytes]ccsd}d} |jd|d}|dkrn#t|j}||kr(td|d|d|jr3td|d d Vq||krFtd|d|d|jd |}|jd |=|S) a Read a LF-terminated line from the stream. This is a generator-based coroutine. The return value includes the LF character. Args: m: maximum number bytes to read; this is a security limit. Raises: EOFError: if the stream ends without a LF. RuntimeError: if the stream ends in more than ``m`` bytes. rT read  bytes, expected no more than  bytesstream ends after z bytes, before end of lineN)rfindlen RuntimeErrorr EOFError)r rnprr r r read_lines&   zStreamReader.read_linerccst|dksJt|j|kr+|jr!t|j}td|d|ddVt|j|ks|jd|}|jd|=|S)z Read a given number of bytes from the stream. This is a generator-based coroutine. Args: n: how many bytes to read. Raises: EOFError: if the stream ends in less than ``n`` bytes. rrz bytes, expected rN)rrr r)r rrrr r r read_exact6s   zStreamReader.read_exactccsZ|jst|j}||krtd|d|ddV|jr|jdd}|jdd=|S)a Read all bytes from the stream. This is a generator-based coroutine. Args: m: maximum number bytes to read; this is a security limit. Raises: RuntimeError: if the stream ends in more than ``m`` bytes. rrrN)r rrr)r rrrr r r read_to_eofMs  zStreamReader.read_to_eofGenerator[None, None, bool]ccs |jrdS|jr dSdVq)zy Tell whether the stream has ended and all data was read. This is a generator-based coroutine. TFN)rr r r r r at_eofcszStreamReader.at_eofdatabytescCs |jrtd|j|7_dS)z Write data to the stream. :meth:`feed_data` cannot be called after :meth:`feed_eof`. Args: data: data to write. Raises: EOFError: if the stream has ended. stream endedN)r rr)r r$r r r feed_datass zStreamReader.feed_datacCs|jrtdd|_dS)z End the stream. :meth:`feed_eof` cannot be called more than once. Raises: EOFError: if the stream has ended. r&TN)r rr r r r feed_eofs  zStreamReader.feed_eofcCs|jdd=dS)zG Discard all buffered data, but don't end the stream. N)rr r r r discardszStreamReader.discardN)rr)rrrr)rrrr)rr")r$r%rr) __name__ __module__ __qualname____doc__rrr r!r#r'r(r)r r r r rs  "    rN) __future__rtypingrrr r r r s