[azure][fix] Finish in case of exception (#2035)

someengineering · Apr 23, 2024 · 43d0a4c · 43d0a4c
1 parent 0eae95c
commit 43d0a4c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/plugins/azure/fix_plugin_azure/__init__.py b/plugins/azure/fix_plugin_azure/__init__.py
@@ -91,8 +91,12 @@ def collect_account_proxy(subscription_collector_arg: AzureSubscriptionArg, queu
     subscription_collector = AzureSubscriptionCollector(
         config, cloud, subscription, account_config.credentials(), core_feedback, task_data
     )
-    subscription_collector.collect()
-    queue.put((subscription_collector_arg.subscription, subscription_collector.graph))
+    try:
+        subscription_collector.collect()
+        queue.put((subscription_collector_arg.subscription, subscription_collector.graph))
+    except Exception as e:
+        log.exception(f"Error collecting subscription {subscription.subscription_id}: {e}. Give up.")
+        queue.put((subscription_collector_arg.subscription, None))  # signal done
 
 
 def collect_in_process(