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
  17
  18 # VIM configuration
  19 cloud_type = os.environ.get("CLOUD_TYPE", "openstack")
  20 if cloud_type == "openstack":
  21     # Openstack VIM
  22     vim_account_type = "openstack"
  23     vim_name_prefix = "basic01"
  24     # Get credentias from Openstack Clouds file
  25     os_cloud = os.environ.get("OS_CLOUD")
  26     clouds_file_paths = [
  27         "./clouds.yaml",
  28         str(Path.home()) + "/.config/openstack/clouds.yaml",
  29         "/etc/openstack/clouds.yaml",
  30     ]
  31     for path in clouds_file_paths:
  32         clouds_file_path = Path(path)
  33         if clouds_file_path.exists():
  34             break
  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 os_cloud not in clouds["clouds"]:
  40             raise Exception("Openstack cloud '" + os_cloud + "' not found")
  41         cloud = clouds["clouds"][os_cloud]
  42         if "username" not in cloud["auth"]:
  43             raise Exception("Username not found in Openstack cloud '" + os_cloud + "'")
  44         vim_user = cloud["auth"]["username"]
  45         if "password" not in cloud["auth"]:
  46             raise Exception("Password not found in Openstack cloud '" + os_cloud + "'")
  47         vim_password = cloud["auth"]["password"]
  48         if "auth_url" not 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 "project_name" not in cloud["auth"]:
  52             raise Exception(
  53                 "Project name not found in Openstack cloud '" + os_cloud + "'"
  54             )
  55         vim_tenant = cloud["auth"]["project_name"]
  56         vim_user_domain_name = (
  57             cloud["auth"]["user_domain_name"]
  58             if "user_domain_name" in cloud["auth"]
  59             else None
  60         )
  61         vim_project_domain_name = (
  62             cloud["auth"]["project_domain_name"]
  63             if "project_domain_name" in cloud["auth"]
  64             else None
  65         )
  66         vim_insecure = True if "verify" in cloud and not cloud["verify"] else None
  67
  68     # Extra config
  69     vim_config_dict = {}
  70     vim_config_dict["vim_network_name"] = os.environ.get("VIM_MGMT_NET")
  71     if vim_project_domain_name:
  72         vim_config_dict["project_domain_name"] = vim_project_domain_name
  73     if vim_user_domain_name:
  74         vim_config_dict["user_domain_name"] = vim_user_domain_name
  75     if vim_insecure:
  76         vim_config_dict["insecure"] = True
  77     vim_config = "'{}'".format(
  78         yaml.safe_dump(vim_config_dict, default_flow_style=True, width=10000).rstrip(
  79             "\r\n"
  80         )
  81     )
  82
  83 elif cloud_type == "azure":
  84     # Azure VIM
  85     vim_account_type = "azure"
  86     vim_name_prefix = "basic01"
  87     vim_auth_url = "http://www.azure.com"
  88     vim_user = os.environ.get("AZURE_CLIENT_ID")
  89     vim_password = os.environ.get("AZURE_SECRET")
  90     vim_tenant = os.environ.get("AZURE_TENANT")
  91
  92     # Extra config
  93     vim_config_dict = {}
  94     if "RESOURCE_GROUP" in os.environ:
  95         vim_config_dict["resource_group"] = os.environ.get("RESOURCE_GROUP")
  96     if "AZURE_REGION" in os.environ:
  97         vim_config_dict["region_name"] = os.environ.get("AZURE_REGION")
  98     if "AZURE_SUBSCRIPTION_ID" in os.environ:
  99         vim_config_dict["subscription_id"] = os.environ.get("AZURE_SUBSCRIPTION_ID")
 100     if "VNET_NAME" in os.environ:
 101         vim_config_dict["vnet_name"] = os.environ.get("VNET_NAME")
 102     if "AZURE_FLAVORS_PATTERN" in os.environ:
 103         vim_config_dict["flavors_pattern"] = os.environ.get("AZURE_FLAVORS_PATTERN")
 104     vim_config = "'{}'".format(
 105         yaml.safe_dump(vim_config_dict, default_flow_style=True, width=10000).rstrip(
 106             "\r\n"
 107         )
 108     )
 109
 110 else:
 111     raise Exception("VIM type not supported: '" + cloud_type + "'")