feat: modify get_tasks for getting benefit cost data

Signed-off-by: ImMin5 <[email protected]>
cloudforet-io · Jul 31, 2024 · 009c91b · 009c91b
1 parent 8b58562
commit 009c91b
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/cloudforet/cost_analysis/manager/job_manager.py b/src/cloudforet/cost_analysis/manager/job_manager.py
@@ -96,6 +96,19 @@ def get_tasks(
                         synced_accounts = self._extend_synced_accounts(
                             synced_accounts, first_sync_tenants
                         )
+                # Benefit Job Task
+                if options.get("cost_metric") == "AmortizedCost":
+                    tasks.append(
+                        {
+                            "task_options": {
+                                "start": start_month,
+                                "account_agreement_type": billing_account_agreement_type,
+                                "collect_scope": "billing_account_id",
+                                "is_benefit_job": True,
+                            }
+                        }
+                    )
+
             else:
                 tasks = [
                     {