Skip to content
Snippets Groups Projects
Commit 1c06c220 authored by Olena Holub (EPAM)'s avatar Olena Holub (EPAM)
Browse files

Added whitelisted cloud provider names

parent de6fdee3
No related branches found
No related tags found
1 merge request!572Added whitelisted cloud provider names
Pipeline #312594 failed
......@@ -55,12 +55,19 @@ def get_storage() -> BundleStorageClient:
logger.critical("Error: No CLOUD_PROVIDER ENV VAR SET")
sys.exit(1)
allowed_values = ["aws", "gc", "az", "local", "ibm", "baremetal"]
cloud_env = os.environ.get("CLOUD_PROVIDER")
try:
provider_module = _import_provider_specific_module(cloud_env)
except ModuleNotFoundError as exc:
logger.critical(f"Error occurred while importing module for {cloud_env}")
logger.critical(f"Exception: {exc}")
if cloud_env in allowed_values:
try:
provider_module = _import_provider_specific_module(cloud_env)
except ModuleNotFoundError as exc:
logger.critical(f"Error occurred while importing module for {cloud_env}")
logger.critical(f"Exception: {exc}")
else:
logger.critical("Error: specified CLOUD_PROVIDER wasn't in allowed list")
sys.exit(1)
logger.info(f"get_storage provider for cloud_env {cloud_env}")
return provider_module.storage_client()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment