raise error when EOF is hit

This commit is contained in:
Anthony Sottile 2020-05-04 12:28:30 -07:00
parent bed864a459
commit 038373f96e
1 changed files with 2 additions and 0 deletions

2
bot.py
View File

@ -66,6 +66,8 @@ async def recv(
quiet: bool = False,
) -> bytes:
data = await reader.readline()
if not data:
raise SystemExit('unexpected EOF')
if not quiet:
sys.stderr.buffer.write(b'> ')
sys.stderr.buffer.write(data)