diff --git a/megacom/__init__.py b/megacom/__init__.py index 8663f0a..ca39941 100644 --- a/megacom/__init__.py +++ b/megacom/__init__.py @@ -203,8 +203,11 @@ async def megacom_main(stdin: asyncio.StreamReader, stdout: asyncio.StreamWriter elif handle_done(serial_to_stdout): pass + time_to_exit.cancel() stdin_to_serial.cancel() serial_to_stdout.cancel() + with contextlib.suppress(asyncio.CancelledError): + await time_to_exit with contextlib.suppress(asyncio.CancelledError): with contextlib.suppress(serial.SerialException): await stdin_to_serial