Merge pull request #107 from ImMin5/master

Fix update cost manager to use generator for data retrieval

Author: ImMin5

src/cloudforet/cost_analysis/main.py

@@ -201,14 +201,13 @@ def __get_secret_data(secret_data: dict, task_options: dict) -> dict:
     if len(secrets) == 1:
         return secrets[0]
 
-    tenant_id = task_options.get(
-        "billing_tenant_id",
-    )
+    tenant_id = task_options["billing_tenant_id"]
 
     for _secret_data in secrets:
         if _secret_data["tenant_id"] == tenant_id:
-            secret_data = _secret_data
+            return _secret_data
+
         elif _secret_data.get("subscription_id") == task_options.get("subscription_id"):
-            secret_data = _secret_data
+            return _secret_data
 
     return secret_data

src/cloudforet/cost_analysis/manager/cost_manager.py

@@ -70,7 +70,7 @@ def get_data(
         schema: str,
         task_options: dict,
         domain_id: str,
-    ) -> list:
+    ) -> Generator[list, Any, None]:
         self.azure_cm_connector.create_session(options, secret_data, schema)
         self._check_task_options(task_options)
 
@@ -132,7 +132,6 @@ def get_data(
             _LOGGER.info(
                 f"[get_data] all collect is done in {int(end_time - start_time)} seconds"
             )
-        yield []
 
     def _make_cost_data(
         self,