robot-systest/resources/basic_01-crud_operations_on_vim_targets_data.py

   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
   4 #
   5 #       http://www.apache.org/licenses/LICENSE-2.0
   6 #
   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.
  12
  13 import os
  14 from pathlib import Path
  15 import yaml
  16 from common_helpers import get_prometheus_info
  17
  18
  19 # Prometheus host and port
  20 (
  21     prometheus_host,
  22     prometheus_port,
  23     prometheus_user,
  24     prometheus_password,
  25 ) = get_prometheus_info()
  26
  27 # VIM configuration
  28 cloud_type = os.environ.get("CLOUD_TYPE", "openstack")
  29 if cloud_type == "openstack":
  30     # Openstack VIM
  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")
  35     clouds_file_paths = [
  36         "./clouds.yaml",
  37         str(Path.home()) + "/.config/openstack/clouds.yaml",
  38         "/etc/openstack/clouds.yaml",
  39     ]
  40     for path in clouds_file_paths:
  41         clouds_file_path = Path(path)
  42         if clouds_file_path.exists():
  43             break
  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 os_cloud not in clouds["clouds"]:
  49             raise Exception("Openstack cloud '" + os_cloud + "' not found")
  50         cloud = clouds["clouds"][os_cloud]
  51         if "username" not in cloud["auth"]:
  52             raise Exception("Username not found in Openstack cloud '" + os_cloud + "'")
  53         vim_user = cloud["auth"]["username"]
  54         if "password" not in cloud["auth"]:
  55             raise Exception("Password not found in Openstack cloud '" + os_cloud + "'")
  56         vim_password = cloud["auth"]["password"]
  57         if "auth_url" not 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 "project_name" not in cloud["auth"]:
  61             raise Exception(
  62                 "Project name not found in Openstack cloud '" + os_cloud + "'"
  63             )
  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"]
  68             else None
  69         )
  70         vim_project_domain_name = (
  71             cloud["auth"]["project_domain_name"]
  72             if "project_domain_name" in cloud["auth"]
  73             else None
  74         )
  75         vim_insecure = True if "verify" in cloud and not cloud["verify"] else None
  76
  77     # Extra config
  78     vim_config_dict = {}
  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
  84     if vim_insecure:
  85         vim_config_dict["insecure"] = True
  86     vim_config = "'{}'".format(
  87         yaml.safe_dump(vim_config_dict, default_flow_style=True, width=10000).rstrip(
  88             "\r\n"
  89         )
  90     )
  91
  92 elif cloud_type == "azure":
  93     # Azure VIM
  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")
 100
 101     # Extra config
 102     vim_config_dict = {}
 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(
 115             "\r\n"
 116         )
 117     )
 118
 119 else:
 120     raise Exception("VIM type not supported: '" + cloud_type + "'")