U g@s*ddlmZddlmZGdddZdS)) annotations) Generatorc@seZdZdZddddZdddd d Zddd d d ZdddddZddddZdddddZ ddddZ ddddZ 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. None)returncCst|_d|_dS)NF) bytearraybuffereofselfr 6/tmp/pip-unpacked-wheel-dx_q7dq3/websockets/streams.py__init__szStreamReader.__init__intzGenerator[None, None, bytes])mrccsd}d}|jd|d}|dkr$qjt|j}||krLtd|d|d|jrbtd|dd Vq||krtd|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. r 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_line)rrccsf|dks tt|j|krH|jr@t|j}td|d|ddVq |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)AssertionErrorrrr r)r rrrr r r read_exact6s   zStreamReader.read_exactccsT|js6t|j}||kr.td|d|ddVq|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_eofzGenerator[None, None, bool]ccs |jr dS|jrdSdVqdS)zy Tell whether the stream has ended and all data was read. This is a generator-based coroutine. FTN)rr r r r r at_eofcs zStreamReader.at_eofbytes)datarcCs |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) __name__ __module__ __qualname____doc__rrr r!r"r&r'r(r r r r rs "rN) __future__rtypingrrr r r r s