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
20 (prometheus_host
, prometheus_port
) = get_prometheus_info()
23 cloud_type
= os
.environ
.get("CLOUD_TYPE", "openstack")
24 if cloud_type
== "openstack":
26 vim_account_type
= "openstack"
27 vim_name_prefix
= "basic01"
28 # Get credentias from Openstack Clouds file
29 os_cloud
= os
.environ
.get("OS_CLOUD")
30 clouds_file_paths
= ["./clouds.yaml", str(Path
.home()) + "/.config/openstack/clouds.yaml", "/etc/openstack/clouds.yaml"]
31 for path
in clouds_file_paths
:
32 clouds_file_path
= Path(path
)
33 if clouds_file_path
.exists():
35 if not clouds_file_path
.exists():
36 raise Exception("Openstack clouds file not found")
37 with clouds_file_path
.open() as clouds_file
:
38 clouds
= yaml
.safe_load(clouds_file
)
39 if not os_cloud
in clouds
["clouds"]:
40 raise Exception("Openstack cloud '" + os_cloud
+ "' not found")
41 cloud
= clouds
["clouds"][os_cloud
]
42 if not "username" in cloud
["auth"]:
43 raise Exception("Username not found in Openstack cloud '" + os_cloud
+ "'")
44 vim_user
= cloud
["auth"]["username"]
45 if not "password" in cloud
["auth"]:
46 raise Exception("Password not found in Openstack cloud '" + os_cloud
+ "'")
47 vim_password
= cloud
["auth"]["password"]
48 if not "auth_url" in cloud
["auth"]:
49 raise Exception("Auth url not found in Openstack cloud '" + os_cloud
+ "'")
50 vim_auth_url
= cloud
["auth"]["auth_url"]
51 if not "project_name" in cloud
["auth"]:
52 raise Exception("Project name not found in Openstack cloud '" + os_cloud
+ "'")
53 vim_tenant
= cloud
["auth"]["project_name"]
54 vim_user_domain_name
= cloud
["auth"]["user_domain_name"] if "user_domain_name" in cloud
["auth"] else None
55 vim_project_domain_name
= cloud
["auth"]["project_domain_name"] if "project_domain_name" in cloud
["auth"] else None
56 vim_insecure
= True if "verify" in cloud
and not cloud
["verify"] else None
60 vim_config_dict
["vim_network_name"] = os
.environ
.get("VIM_MGMT_NET")
61 if vim_project_domain_name
:
62 vim_config_dict
["project_domain_name"] = vim_project_domain_name
63 if vim_user_domain_name
:
64 vim_config_dict
["user_domain_name"] = vim_user_domain_name
66 vim_config_dict
["insecure"] = True
67 vim_config
= "'{}'".format(yaml
.safe_dump(vim_config_dict
, default_flow_style
=True, width
=10000).rstrip('\r\n'))
69 elif cloud_type
== "azure":
71 vim_account_type
= "azure"
72 vim_name_prefix
= "basic01"
73 vim_auth_url
= "http://www.azure.com"
74 vim_user
= os
.environ
.get("AZURE_CLIENT_ID")
75 vim_password
= os
.environ
.get("AZURE_SECRET")
76 vim_tenant
= os
.environ
.get("AZURE_TENANT")
80 if "RESOURCE_GROUP" in os
.environ
:
81 vim_config_dict
["resource_group"] = os
.environ
.get("RESOURCE_GROUP")
82 if "AZURE_REGION" in os
.environ
:
83 vim_config_dict
["region_name"] = os
.environ
.get("AZURE_REGION")
84 if "AZURE_SUBSCRIPTION_ID" in os
.environ
:
85 vim_config_dict
["subscription_id"] = os
.environ
.get("AZURE_SUBSCRIPTION_ID")
86 if "VNET_NAME" in os
.environ
:
87 vim_config_dict
["vnet_name"] = os
.environ
.get("VNET_NAME")
88 if "AZURE_FLAVORS_PATTERN" in os
.environ
:
89 vim_config_dict
["flavors_pattern"] = os
.environ
.get("AZURE_FLAVORS_PATTERN")
90 vim_config
= "'{}'".format(yaml
.safe_dump(vim_config_dict
, default_flow_style
=True, width
=10000).rstrip('\r\n'))
93 raise Exception("VIM type not supported: '" + cloud_type
+ "'")