fix: data_source_account reset error (cloudforet-io#199)

ImMin5 · May 13, 2024 · cb6c605 · cb6c605
1 parent bcc98a9
commit cb6c605
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/spaceone/cost_analysis/service/data_source_account_service.py b/src/spaceone/cost_analysis/service/data_source_account_service.py
@@ -111,9 +111,15 @@ def reset(self, params: DataSourceAccountResetRequest) -> None:
         domain_id = params.domain_id
 
         # Check if the data source exists
-        data_source_vo = self.data_source_mgr.get_data_source(
-            data_source_id, domain_id, workspace_id
-        )
+        role_type = self.transaction.get_meta("authorization.role_type")
+        if role_type != "DOMAIN_ADMIN":
+            data_source_vo = self.data_source_mgr.get_data_source(
+                data_source_id, domain_id, workspace_id
+            )
+        else:
+            data_source_vo = self.data_source_mgr.get_data_source(
+                data_source_id, domain_id
+            )
 
         conditions = {
             "data_source_id": data_source_id,