fix: fix budget alarm

cloudforet-io · Jan 22, 2024 · 61e322b · 61e322b
1 parent 3ea2417
commit 61e322b
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/spaceone/cost_analysis/manager/budget_usage_manager.py b/src/spaceone/cost_analysis/manager/budget_usage_manager.py
@@ -117,7 +117,7 @@ def notify_budget_usage(self, budget_vo: Budget):
                 notification_type = notification.notification_type
                 is_notify = False
 
-                if budget_vo.time_unit == "MONTHLY":
+                if budget_vo.time_unit == "TOTAL":
                     budget_usage_vos = self.filter_budget_usages(
                         budget_id=budget_id,
                         workspace_id=workspace_id,
@@ -134,6 +134,13 @@ def notify_budget_usage(self, budget_vo: Budget):
                         domain_id=domain_id,
                         date=current_month,
                     )
+
+                    if budget_usage_vos.count() == 0:
+                        _LOGGER.debug(
+                            f"[notify_budget_usage] budget_usage_vos is empty: {budget_id}"
+                        )
+                        continue
+
                     total_budget_usage = budget_usage_vos[0].cost
                     budget_limit = budget_usage_vos[0].limit