feat: add error context when download cost data failed

Signed-off-by: ImMin5 <[email protected]>
cloudforet-io · Oct 19, 2023 · 2685273 · 2685273
1 parent 43e72af
commit 2685273
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/cloudforet/cost_analysis/connector/azure_cost_mgmt_connector.py b/src/cloudforet/cost_analysis/connector/azure_cost_mgmt_connector.py
@@ -187,6 +187,8 @@ def convert_nested_dictionary(self, cloud_svc_object):
     def _download_cost_data(blob: dict) -> str:
         try:
             response = requests.get(blob.get('blob_link'))
+            if response.status_code != 200:
+                raise ERROR_CONNECTOR_CALL_API(reason=f'{response.reason}')
             return response.text
         except Exception as e:
             _LOGGER.error(f'[_download_cost_data] download error: {e}', exc_info=True)

diff --git a/src/cloudforet/cost_analysis/error/cost.py b/src/cloudforet/cost_analysis/error/cost.py
@@ -19,3 +19,6 @@ class ERROR_EMPTY_CUSTOMER_TENANTS(ERROR_INVALID_ARGUMENT):
 
 class ERROR_INVALID_TOKEN(ERROR_INVALID_ARGUMENT):
     _message = 'Invalid token: {token}'
+
+class ERROR_CONNECTOR_CALL_API(ERROR_UNKNOWN):
+    _message = 'API Call Error: {reason}'