Skip to content

Commit 3555353

Browse files
authored
Merge pull request #203 from ImMin5/feature-cost-router
Fix plugin data access error
2 parents bdd3101 + f0f708e commit 3555353

File tree

1 file changed

+42
-42
lines changed

1 file changed

+42
-42
lines changed

src/spaceone/cost_analysis/service/job_service.py

+42-42
Original file line numberDiff line numberDiff line change
@@ -260,7 +260,7 @@ def get_cost_data(self, params):
260260
is_canceled = False
261261

262262
for costs_data in self.ds_plugin_mgr.get_cost_data(
263-
options, secret_data, schema, task_options, domain_id
263+
options, secret_data, schema, task_options, domain_id
264264
):
265265
results = costs_data.get("results", [])
266266
for cost_data in results:
@@ -434,11 +434,11 @@ def create_cost_job(self, data_source_vo: DataSource, job_options):
434434
return job_vo
435435

436436
def _list_secret_ids_from_secret_type(
437-
self,
438-
data_source_vo: DataSource,
439-
secret_type: str,
440-
workspace_id: str,
441-
domain_id: str,
437+
self,
438+
data_source_vo: DataSource,
439+
secret_type: str,
440+
workspace_id: str,
441+
domain_id: str,
442442
):
443443
secret_ids = []
444444

@@ -459,7 +459,7 @@ def _list_secret_ids_from_secret_type(
459459
return secret_ids
460460

461461
def _list_secret_ids_from_secret_filter(
462-
self, secret_filter, provider: str, workspace_id: str, domain_id: str
462+
self, secret_filter, provider: str, workspace_id: str, domain_id: str
463463
):
464464
secret_manager: SecretManager = self.locator.get_manager(SecretManager)
465465

@@ -474,7 +474,7 @@ def _list_secret_ids_from_secret_filter(
474474

475475
@staticmethod
476476
def _set_secret_filter(
477-
secret_filter, provider: str, workspace_id: str, domain_id: str
477+
secret_filter, provider: str, workspace_id: str, domain_id: str
478478
):
479479
_filter = [{"k": "domain_id", "v": domain_id, "o": "eq"}]
480480

@@ -489,8 +489,8 @@ def _set_secret_filter(
489489
{"k": "secret_id", "v": secret_filter["secrets"], "o": "in"}
490490
)
491491
if (
492-
"service_accounts" in secret_filter
493-
and secret_filter["service_accounts"]
492+
"service_accounts" in secret_filter
493+
and secret_filter["service_accounts"]
494494
):
495495
_filter.append(
496496
{
@@ -586,10 +586,10 @@ def _create_cost_data(self, cost_data, job_task_vo, cost_options):
586586
self.cost_mgr.create_cost(cost_data, execute_rollback=False)
587587

588588
def _is_job_failed(
589-
self,
590-
job_id: str,
591-
domain_id: str,
592-
workspace_id: str,
589+
self,
590+
job_id: str,
591+
domain_id: str,
592+
workspace_id: str,
593593
):
594594
job_vo: Job = self.job_mgr.get_job(job_id, domain_id, workspace_id)
595595

@@ -599,12 +599,12 @@ def _is_job_failed(
599599
return False
600600

601601
def _close_job(
602-
self,
603-
job_id: str,
604-
data_source_id: str,
605-
domain_id: str,
606-
data_keys: list,
607-
workspace_id: str = None,
602+
self,
603+
job_id: str,
604+
data_source_id: str,
605+
domain_id: str,
606+
data_keys: list,
607+
workspace_id: str = None,
608608
) -> None:
609609
job_vo: Job = self.job_mgr.get_job(job_id, domain_id, workspace_id)
610610
no_preload_cache = job_vo.options.get("no_preload_cache", False)
@@ -754,7 +754,7 @@ def _delete_old_cost_data(self, data_source_id, domain_id):
754754
monthly_cost_vos.delete()
755755

756756
def _delete_changed_cost_data(
757-
self, job_vo: Job, start, end, change_filter, domain_id
757+
self, job_vo: Job, start, end, change_filter, domain_id
758758
):
759759
query = {
760760
"filter": [
@@ -800,7 +800,7 @@ def _aggregate_cost_data(self, job_vo: Job, data_keys: list):
800800

801801
for job_task_id in job_task_ids:
802802
for billed_month in self._distinct_billed_month(
803-
data_source_id, domain_id, job_id, job_task_id
803+
data_source_id, domain_id, job_id, job_task_id
804804
):
805805
self._aggregate_monthly_cost_data(
806806
data_source_id,
@@ -832,14 +832,14 @@ def _distinct_billed_month(self, data_source_id, domain_id, job_id, job_task_id)
832832
return values
833833

834834
def _aggregate_monthly_cost_data(
835-
self,
836-
data_source_id: str,
837-
domain_id: str,
838-
job_id: str,
839-
job_task_id: str,
840-
billed_month: str,
841-
data_keys: list,
842-
workspace_id: str = None,
835+
self,
836+
data_source_id: str,
837+
domain_id: str,
838+
job_id: str,
839+
job_task_id: str,
840+
billed_month: str,
841+
data_keys: list,
842+
workspace_id: str = None,
843843
):
844844
query = {
845845
"group_by": [
@@ -906,7 +906,7 @@ def _get_all_data_sources(self):
906906
)
907907

908908
def _check_duplicate_job(
909-
self, data_source_id: str, domain_id: str, this_job_vo: Job
909+
self, data_source_id: str, domain_id: str, this_job_vo: Job
910910
):
911911
query = {
912912
"filter": [
@@ -952,11 +952,11 @@ def _get_job_task_ids(self, job_id, domain_id):
952952
return job_task_ids
953953

954954
def _get_data_source_account_map(
955-
self,
956-
data_source_id: str,
957-
domain_id: str,
958-
workspace_id: str,
959-
resource_group: str,
955+
self,
956+
data_source_id: str,
957+
domain_id: str,
958+
workspace_id: str,
959+
resource_group: str,
960960
) -> Dict[str, DataSourceAccount]:
961961
data_source_account_map = {}
962962
conditions = {
@@ -978,11 +978,11 @@ def _get_data_source_account_map(
978978
return data_source_account_map
979979

980980
def _get_linked_accounts_from_data_source_vo(
981-
self,
982-
data_source_vo: DataSource,
983-
options: dict,
984-
secret_data: dict,
985-
schema: dict = None,
981+
self,
982+
data_source_vo: DataSource,
983+
options: dict,
984+
secret_data: dict,
985+
schema: dict = None,
986986
) -> list:
987987
linked_accounts = []
988988

@@ -1046,7 +1046,7 @@ def _update_data_source_is_synced(self, job_vo: Job) -> None:
10461046

10471047
@staticmethod
10481048
def _check_use_account_routing(data_source_vo: DataSource) -> bool:
1049-
plugin_info = data_source_vo.plugin_info or {}
1049+
plugin_info = data_source_vo.plugin_info.to_dict() or {}
10501050
metadata = plugin_info.get("metadata", {})
10511051

10521052
if metadata.get("use_account_routing", False):

0 commit comments

Comments
 (0)