2 # ============LICENSE_START=======================================================
3 # org.onap.vvp/validation-scripts
4 # ===================================================================
5 # Copyright © 2019 AT&T Intellectual Property. All rights reserved.
6 # ===================================================================
8 # Unless otherwise specified, all software contained herein is licensed
9 # under the Apache License, Version 2.0 (the "License");
10 # you may not use this software except in compliance with the License.
11 # You may obtain a copy of the License at
13 # http://www.apache.org/licenses/LICENSE-2.0
15 # Unless required by applicable law or agreed to in writing, software
16 # distributed under the License is distributed on an "AS IS" BASIS,
17 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
18 # See the License for the specific language governing permissions and
19 # limitations under the License.
23 # Unless otherwise specified, all documentation contained herein is licensed
24 # under the Creative Commons License, Attribution 4.0 Intl. (the "License");
25 # you may not use this documentation except in compliance with the License.
26 # You may obtain a copy of the License at
28 # https://creativecommons.org/licenses/by/4.0/
30 # Unless required by applicable law or agreed to in writing, documentation
31 # distributed under the License is distributed on an "AS IS" BASIS,
32 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
33 # See the License for the specific language governing permissions and
34 # limitations under the License.
36 # ============LICENSE_END============================================
41 from tests import cached_yaml as yaml
44 def get_vm_types_for_resource(resource):
46 Get all unique vm_types for a resource
48 - Returns set([]) if the resource is not formatted
49 properly, the passed resource is not a nova server
50 - If more than one vm_type is detected all vm_types will
53 if not isinstance(resource, dict):
55 if "type" not in resource:
57 if resource["type"] != "OS::Nova::Server":
59 if "properties" not in resource:
62 key_values = ["name", "flavor", "image"]
64 ["name", "string", re.compile(r"(.+?)_name_\d+")],
65 ["name", "comma_delimited_list", re.compile(r"(.+?)_names")],
66 ["flavor", "string", re.compile(r"(.+?)_flavor_name")],
67 ["image", "string", re.compile(r"(.+?)_image_name")],
71 for k2, v2 in resource["properties"].items():
72 if k2 not in key_values:
74 if "get_param" not in v2:
76 formats = [v for v in key_value_formats if v[0] == k2]
78 param = v2["get_param"]
79 if isinstance(param, list):
81 m = v3[2].match(param)
83 vm_types.append(m.group(1))
88 def get_vm_type_for_nova_server(resource):
90 Get the vm_type for a resource
91 Note: Returns None if not exactly one vm_type
92 is detected, if the resource is not formatted properly, or
93 the passed resource is not a nova server
95 vm_types = get_vm_types_for_resource(resource)
97 # if more than one vm_type was identified, return None
101 return vm_types.pop()
104 def get_vm_types(resources):
106 Get all vm_types for a list of heat resources, do note that
107 some of the values retrieved may be invalid
110 for v in resources.values():
111 vm_types.extend(list(get_vm_types_for_resource(v)))
116 def get_all_vm_types(yaml_files):
118 Get all vm_types for a list of yaml files
121 for yaml_file in yaml_files:
122 with open(yaml_file, "r") as f:
125 if "resources" not in yml:
128 vm_types.extend(get_vm_types(yml["resources"]))