From 09827d8a0fcef2f43a74f0a1c4dddce7ca9f336a Mon Sep 17 00:00:00 2001 From: Nico Date: Wed, 4 Nov 2015 22:12:03 +0100 Subject: [PATCH] Add critical message if pub/sub operation is cancelled --- scripts/pub_script.py | 2 ++ scripts/sub_script.py | 2 ++ 2 files changed, 4 insertions(+) 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):