1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
# -*- coding: utf8 -*-
# ============LICENSE_START=======================================================
# org.onap.vvp/validation-scripts
# ===================================================================
# Copyright © 2019 AT&T Intellectual Property. All rights reserved.
# ===================================================================
#
# Unless otherwise specified, all software contained herein is licensed
# under the Apache License, Version 2.0 (the "License");
# you may not use this software except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
#
#
# Unless otherwise specified, all documentation contained herein is licensed
# under the Creative Commons License, Attribution 4.0 Intl. (the "License");
# you may not use this documentation except in compliance with the License.
# You may obtain a copy of the License at
#
# https://creativecommons.org/licenses/by/4.0/
#
# Unless required by applicable law or agreed to in writing, documentation
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
# ============LICENSE_END============================================
#
#
import re
from tests import cached_yaml as yaml
def get_vm_types_for_resource(resource):
"""
Get all unique vm_types for a resource
Notes:
- Returns set([]) if the resource is not formatted
properly, the passed resource is not a nova server
- If more than one vm_type is detected all vm_types will
be returned
"""
if not is_nova_server(resource):
return set()
key_values = ["name", "flavor", "image"]
key_value_formats = [
["name", "string", re.compile(r"(.+?)_name_\d+")],
["name", "comma_delimited_list", re.compile(r"(.+?)_names")],
["flavor", "string", re.compile(r"(.+?)_flavor_name")],
["image", "string", re.compile(r"(.+?)_image_name")],
]
vm_types = []
for k2, v2 in resource["properties"].items():
if not isinstance(v2, dict) or any(
[k2 not in key_values, "get_param" not in v2]
):
continue
formats = [v for v in key_value_formats if v[0] == k2]
for v3 in formats:
param = v2["get_param"]
if isinstance(param, list):
param = param[0]
m = v3[2].match(param)
if m and m.group(1):
vm_types.append(m.group(1))
return set(vm_types)
def is_nova_server(resource):
return (
isinstance(resource, dict)
and "type" in resource
and "properties" in resource
and resource.get("type") == "OS::Nova::Server"
)
def get_vm_type_for_nova_server(resource):
"""
Get the vm_type for a resource
Note: Returns None if not exactly one vm_type
is detected, if the resource is not formatted properly, or
the passed resource is not a nova server
"""
vm_types = get_vm_types_for_resource(resource)
# if more than one vm_type was identified, return None
if len(vm_types) > 1:
return None
return vm_types.pop()
def get_vm_types(resources):
"""
Get all vm_types for a list of heat resources, do note that
some of the values retrieved may be invalid
"""
vm_types = []
for v in resources.values():
vm_types.extend(list(get_vm_types_for_resource(v)))
return set(vm_types)
def get_all_vm_types(yaml_files):
"""
Get all vm_types for a list of yaml files
"""
vm_types = []
for yaml_file in yaml_files:
with open(yaml_file, "r") as f:
yml = yaml.load(f)
if "resources" not in yml:
continue
vm_types.extend(get_vm_types(yml["resources"]))
return set(vm_types)
|
