diff --git a/scripts/pub_script.py b/scripts/pub_script.py index 9f03b8b..fb6c3e7 100644 --- a/scripts/pub_script.py +++ b/scripts/pub_script.py @@ -111,6 +111,8 @@ def do_pub(client, arguments): logger.info("%s Disconnected from broker" % client.client_id) except ConnectException as ce: logger.fatal("connection to '%s' failed: %r" % (arguments['--url'], ce)) + except asyncio.CancelledError as cae: + logger.fatal("Publish canceled due to prvious error") def main(*args, **kwargs): diff --git a/scripts/sub_script.py b/scripts/sub_script.py index 6d6db44..1d47df1 100644 --- a/scripts/sub_script.py +++ b/scripts/sub_script.py @@ -85,6 +85,8 @@ def do_sub(client, arguments): yield from client.disconnect() except ConnectException as ce: logger.fatal("connection to '%s' failed: %r" % (arguments['--url'], ce)) + except asyncio.CancelledError as cae: + logger.fatal("Publish canceled due to prvious error") def main(*args, **kwargs):