Fix exception handling

debian/changelog

@@ -1,3 +1,9 @@
+wb-cloud-agent (1.5.6) stable; urgency=medium
+
+  * Fix exception handling
+
+ -- Nikolay Korotkiy <[email protected]>  Mon, 22 Jul 2024 15:45:00 +0400
+
 wb-cloud-agent (1.5.5) stable; urgency=medium
 
   * Fix preinst hook

wb-cloud-agent

@@ -6,4 +6,7 @@ import sys
 from wb.cloud_agent import main
 
 if __name__ == "__main__":
-    sys.exit(main.main())
+    try:
+        sys.exit(main.main())
+    except KeyboardInterrupt:
+        sys.exit(0)

wb/cloud_agent/main.py

@@ -5,6 +5,7 @@
 import logging
 import os
 import subprocess
+import sys
 import threading
 import time
 from contextlib import ExitStack
@@ -153,7 +154,12 @@ def do_curl(settings: AppSettings, method="get", endpoint="", params=None):
         url,
     ]
 
-    result = subprocess.run(command, timeout=360, check=True, capture_output=True)
+    try:
+        result = subprocess.run(command, timeout=360, check=True, capture_output=True)
+    except subprocess.CalledProcessError as e:
+        raise ValueError("Error during request: " + str(e))
+    except subprocess.TimeoutExpired as e:
+        raise ValueError("Timeout expired: " + str(e))
 
     decoded_result = result.stdout.decode("utf-8")
     split_result = decoded_result.split(data_delimiter)
@@ -473,20 +479,28 @@ def main():
     mqtt.on_connect = on_connect
     mqtt.on_message = on_message
 
-    if hasattr(options, "func"):
-        mqtt.start()
-        return options.func(options, settings, mqtt)
+    try:
+        if hasattr(options, "func"):
+            mqtt.start()
+            return options.func(options, settings, mqtt)
 
-    if not options.daemon:
+        if not options.daemon:
+            mqtt.start()
+            make_start_up_request(settings, mqtt)
+            return show_activation_link(settings)
+
+        mqtt.will_set(settings.MQTT_PREFIX + "/controls/status", "stopped", retain=True, qos=2)
         mqtt.start()
+        publish_ctrl(settings, mqtt, "status", "starting")
         make_start_up_request(settings, mqtt)
-        return show_activation_link(settings)
+        send_agent_version(settings)
+        update_providers_list(settings, mqtt)
 
-    mqtt.will_set(settings.MQTT_PREFIX + "/controls/status", "stopped", retain=True, qos=2)
-    mqtt.start()
-    publish_ctrl(settings, mqtt, "status", "starting")
-    make_start_up_request(settings, mqtt)
-    send_agent_version(settings)
-    update_providers_list(settings, mqtt)
+        run_daemon(mqtt, settings)
 
-    run_daemon(mqtt, settings)
+    except (ConnectionError, ConnectionRefusedError):
+        logging.error(f"Cannot connect to broker {settings.BROKER_URL}")
+        sys.exit(1)
+    except Exception as e:
+        logging.error(e)
+        sys.exit(1)