Update atst to atat

atat/domain/reports.py

@@ -0,0 +1,33 @@
+from flask import current_app
+from atat.domain.csp.cloud.models import (
+    ReportingCSPPayload,
+    CostManagementQueryCSPResult,
+)
+from atat.domain.csp.reports import prepare_azure_reporting_data
+import pendulum
+
+
+class Reports:
+    @classmethod
+    def expired_task_orders(cls, portfolio):
+        return [
+            task_order for task_order in portfolio.task_orders if task_order.is_expired
+        ]
+
+    @classmethod
+    def get_portfolio_spending(cls, portfolio):
+        # TODO: Extend this function to make from_date and to_date configurable
+        from_date = pendulum.now().subtract(years=1).add(days=1).format("YYYY-MM-DD")
+        to_date = pendulum.now().format("YYYY-MM-DD")
+        rows = []
+
+        if portfolio.csp_data:
+            payload = ReportingCSPPayload(
+                from_date=from_date, to_date=to_date, **portfolio.csp_data
+            )
+            response: CostManagementQueryCSPResult = current_app.csp.cloud.get_reporting_data(
+                payload
+            )
+            rows = response.properties.rows
+
+        return prepare_azure_reporting_data(rows)