blob: 01f5e355233b0e4f36232127dfa30fc0fb118153 (
plain)
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
|
#!/usr/bin/env python
"""Loads a "target" YAML file and overwrites its values with values from
"override" YAML files.
Override files are processed in the order given.
Usage:
merge_settings.py <target yaml> [override yamls]
"""
import sys
import yaml
import copy
def dict_merge(target, source):
"""Deep merge for dicts.
Works like dict.update() that recursively updates any dict values present in
both parameters.
Args:
target (dict): Values to be overwritten by corresponding values from
`source`.
source (dict): Overriding values. Not changed by call.
Returns:
`target` with values overwritten from those in `source` at any and all
levels of nested dicts.
"""
if not isinstance(source, dict):
return source
for k, v in source.iteritems():
if k in target and isinstance(target[k], dict):
dict_merge(target[k], v)
else:
target[k] = copy.deepcopy(v)
return target
if len(sys.argv) < 2:
sys.exit('Argument required: target yaml')
if len(sys.argv) == 2:
# No overrides to handle
sys.exit()
target_filename = sys.argv[1]
with open(target_filename) as target_file:
data = yaml.safe_load(target_file)
print('Loaded target YAML file ' + target_filename)
for override_filename in sys.argv[2:]:
with open(override_filename) as override_file:
override = yaml.safe_load(override_file)
dict_merge(data, override)
print('Merged override YAML file ' + override_filename)
with open(target_filename, 'w') as target_file:
yaml.dump(data, target_file)
print('Wrote merged target YAML file ' + target_filename)
|
