1 # Licensed under the Apache License, Version 2.0 (the "License");
2 # you may not use this file except in compliance with the License.
3 # You may obtain a copy of the License at
5 # http://www.apache.org/licenses/LICENSE-2.0
7 # Unless required by applicable law or agreed to in writing, software
8 # distributed under the License is distributed on an "AS IS" BASIS,
9 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
10 # See the License for the specific language governing permissions and
11 # limitations under the License.
14 from pathlib
import Path
16 from common_helpers
import get_prometheus_info
19 # Prometheus host and port
25 ) = get_prometheus_info()
28 cloud_type
= os
.environ
.get("CLOUD_TYPE", "openstack")
29 if cloud_type
== "openstack":
31 vim_account_type
= "openstack"
32 vim_name_prefix
= "basic01"
33 # Get credentias from Openstack Clouds file
34 os_cloud
= os
.environ
.get("OS_CLOUD")
37 str(Path
.home()) + "/.config/openstack/clouds.yaml",
38 "/etc/openstack/clouds.yaml",
40 for path
in clouds_file_paths
:
41 clouds_file_path
= Path(path
)
42 if clouds_file_path
.exists():
44 if not clouds_file_path
.exists():
45 raise Exception("Openstack clouds file not found")
46 with clouds_file_path
.open() as clouds_file
:
47 clouds
= yaml
.safe_load(clouds_file
)
48 if not os_cloud
in clouds
["clouds"]:
49 raise Exception("Openstack cloud '" + os_cloud
+ "' not found")
50 cloud
= clouds
["clouds"][os_cloud
]
51 if not "username" in cloud
["auth"]:
52 raise Exception("Username not found in Openstack cloud '" + os_cloud
+ "'")
53 vim_user
= cloud
["auth"]["username"]
54 if not "password" in cloud
["auth"]:
55 raise Exception("Password not found in Openstack cloud '" + os_cloud
+ "'")
56 vim_password
= cloud
["auth"]["password"]
57 if not "auth_url" in cloud
["auth"]:
58 raise Exception("Auth url not found in Openstack cloud '" + os_cloud
+ "'")
59 vim_auth_url
= cloud
["auth"]["auth_url"]
60 if not "project_name" in cloud
["auth"]:
62 "Project name not found in Openstack cloud '" + os_cloud
+ "'"
64 vim_tenant
= cloud
["auth"]["project_name"]
65 vim_user_domain_name
= (
66 cloud
["auth"]["user_domain_name"]
67 if "user_domain_name" in cloud
["auth"]
70 vim_project_domain_name
= (
71 cloud
["auth"]["project_domain_name"]
72 if "project_domain_name" in cloud
["auth"]
75 vim_insecure
= True if "verify" in cloud
and not cloud
["verify"] else None
79 vim_config_dict
["vim_network_name"] = os
.environ
.get("VIM_MGMT_NET")
80 if vim_project_domain_name
:
81 vim_config_dict
["project_domain_name"] = vim_project_domain_name
82 if vim_user_domain_name
:
83 vim_config_dict
["user_domain_name"] = vim_user_domain_name
85 vim_config_dict
["insecure"] = True
86 vim_config
= "'{}'".format(
87 yaml
.safe_dump(vim_config_dict
, default_flow_style
=True, width
=10000).rstrip(
92 elif cloud_type
== "azure":
94 vim_account_type
= "azure"
95 vim_name_prefix
= "basic01"
96 vim_auth_url
= "http://www.azure.com"
97 vim_user
= os
.environ
.get("AZURE_CLIENT_ID")
98 vim_password
= os
.environ
.get("AZURE_SECRET")
99 vim_tenant
= os
.environ
.get("AZURE_TENANT")
103 if "RESOURCE_GROUP" in os
.environ
:
104 vim_config_dict
["resource_group"] = os
.environ
.get("RESOURCE_GROUP")
105 if "AZURE_REGION" in os
.environ
:
106 vim_config_dict
["region_name"] = os
.environ
.get("AZURE_REGION")
107 if "AZURE_SUBSCRIPTION_ID" in os
.environ
:
108 vim_config_dict
["subscription_id"] = os
.environ
.get("AZURE_SUBSCRIPTION_ID")
109 if "VNET_NAME" in os
.environ
:
110 vim_config_dict
["vnet_name"] = os
.environ
.get("VNET_NAME")
111 if "AZURE_FLAVORS_PATTERN" in os
.environ
:
112 vim_config_dict
["flavors_pattern"] = os
.environ
.get("AZURE_FLAVORS_PATTERN")
113 vim_config
= "'{}'".format(
114 yaml
.safe_dump(vim_config_dict
, default_flow_style
=True, width
=10000).rstrip(
120 raise Exception("VIM type not supported: '" + cloud_type
+ "'")