subiquity/scripts/check-yaml-fields.py

28 lines
516 B
Python

#!/usr/bin/python3
import sys
import yaml
config = yaml.safe_load(open(sys.argv[1]))
def main():
for arg in sys.argv[2:]:
if '=' in arg:
k, expected = arg.split('=', 1)
expected = yaml.safe_load(expected)
else:
k, expected = arg, None
v = config
for part in k.split('.'):
v = v[part]
if expected is None:
print(v)
else:
assert v == expected, "{!r} != {!r}".format(v, expected)
main()