B ]b@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) bytearraybuffereof)selfr 1lib/python3.7/site-packages/websockets/streams.py__init__szStreamReader.__init__intzGenerator[None, None, bytes])mrccsd}d}xd|jd|d}|dkr&Pt|j}||krNtd|d|d|jrdtd|dd Vq W||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 zread z bytes, expected no more than z byteszstream ends after z bytes, before end of lineN)rfindlen RuntimeErrorr EOFError)r rnprr r r read_lines"   zStreamReader.read_line)rrccsj|dks tx>t|j|krJ|jrBt|j}td|d|ddVqW|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. rzstream ends after z bytes, expected z bytesN)AssertionErrorrrr r)r rrrr r r read_exact6s    zStreamReader.read_exactccsXx8|js8t|j}||kr0td|d|ddVqW|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. zread z bytes, expected no more than z bytesN)r rrr)r rrrr r r read_to_eofMs    zStreamReader.read_to_eofzGenerator[None, None, bool]ccs$x|jr dS|jrdSdVqWdS)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. z stream endedN)r rr)r rr 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. z stream endedTN)r r)r r r r feed_eofs zStreamReader.feed_eofcCs|jdd=dS)zG Discard all buffered data, but don't end the stream. N)r)r r r r discardszStreamReader.discardN) __name__ __module__ __qualname____doc__r rrrrr r!r"r r r r rs"rN)Z __future__rtypingrrr r r r s