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
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
|
# mypy: allow-untyped-defs
import json
import os
import re
from collections import OrderedDict
from copy import deepcopy
import yaml
here = os.path.dirname(__file__)
def first(iterable):
# First item from a list or iterator
if not hasattr(iterable, "next"):
if hasattr(iterable, "__iter__"):
iterable = iter(iterable)
else:
raise ValueError("Object isn't iterable")
return next(iterable)
def load_task_file(path):
with open(path) as f:
return yaml.safe_load(f)
def update_recursive(data, update_data):
for key, value in update_data.items():
if key not in data:
data[key] = value
else:
initial_value = data[key]
if isinstance(value, dict):
if not isinstance(initial_value, dict):
raise ValueError("Variable %s has inconsistent types "
"(expected object)" % key)
update_recursive(initial_value, value)
elif isinstance(value, list):
if not isinstance(initial_value, list):
raise ValueError("Variable %s has inconsistent types "
"(expected list)" % key)
initial_value.extend(value)
else:
data[key] = value
def resolve_use(task_data, templates):
rv = {}
if "use" in task_data:
for template_name in task_data["use"]:
update_recursive(rv, deepcopy(templates[template_name]))
update_recursive(rv, task_data)
rv.pop("use", None)
return rv
def resolve_name(task_data, default_name):
if "name" not in task_data:
task_data["name"] = default_name
return task_data
def resolve_chunks(task_data):
if "chunks" not in task_data:
return [task_data]
rv = []
total_chunks = task_data["chunks"]
if "chunks-override" in task_data:
override = task_data["chunks-override"].get(task_data["vars"]["test-type"])
if override is not None:
total_chunks = override
for i in range(1, total_chunks + 1):
chunk_data = deepcopy(task_data)
chunk_data["chunks"] = {"id": i,
"total": total_chunks}
rv.append(chunk_data)
return rv
def replace_vars(input_string, variables):
# TODO: support replacing as a non-string type?
variable_re = re.compile(r"(?<!\\)\${([^}]+)}")
def replacer(m):
var = m.group(1).split(".")
repl = variables
for part in var:
try:
repl = repl[part]
except Exception:
# Don't substitute
return m.group(0)
return str(repl)
return variable_re.sub(replacer, input_string)
def sub_variables(data, variables):
if isinstance(data, str):
return replace_vars(data, variables)
if isinstance(data, list):
return [sub_variables(item, variables) for item in data]
if isinstance(data, dict):
return {key: sub_variables(value, variables)
for key, value in data.items()}
return data
def substitute_variables(task):
variables = {"vars": task.get("vars", {}),
"chunks": task.get("chunks", {})}
return sub_variables(task, variables)
def expand_maps(task):
name = first(task.keys())
if name != "$map":
return [task]
map_data = task["$map"]
if set(map_data.keys()) != {"for", "do"}:
raise ValueError("$map objects must have exactly two properties named 'for' "
"and 'do' (got %s)" % ("no properties" if not map_data.keys()
else ", ". join(map_data.keys())))
rv = []
for for_data in map_data["for"]:
do_items = map_data["do"]
if not isinstance(do_items, list):
do_items = expand_maps(do_items)
for do_data in do_items:
task_data = deepcopy(for_data)
if len(do_data.keys()) != 1:
raise ValueError("Each item in the 'do' list must be an object "
"with a single property")
name = first(do_data.keys())
update_recursive(task_data, deepcopy(do_data[name]))
rv.append({name: task_data})
return rv
def load_tasks(tasks_data):
map_resolved_tasks = OrderedDict()
tasks = []
for task in tasks_data["tasks"]:
if len(task.keys()) != 1:
raise ValueError("Each task must be an object with a single property")
for task in expand_maps(task):
if len(task.keys()) != 1:
raise ValueError("Each task must be an object with a single property")
name = first(task.keys())
data = task[name]
new_name = sub_variables(name, {"vars": data.get("vars", {})})
if new_name in map_resolved_tasks:
raise ValueError("Got duplicate task name %s" % new_name)
map_resolved_tasks[new_name] = substitute_variables(data)
for task_default_name, data in map_resolved_tasks.items():
task = resolve_use(data, tasks_data["components"])
task = resolve_name(task, task_default_name)
tasks.extend(resolve_chunks(task))
tasks = [substitute_variables(task_data) for task_data in tasks]
return OrderedDict([(t["name"], t) for t in tasks])
def load_tasks_from_path(path):
return load_tasks(load_task_file(path))
def run(venv, **kwargs):
print(json.dumps(load_tasks_from_path(os.path.join(here, "tasks", "test.yml")), indent=2))
|
