blob: 353e87df7ea44bc34a050f2469160a10cce015f2 (
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
|
#!/usr/bin/env python
"""Utility to return YAML value from key in YAML file."""
from __future__ import print_function
import optparse
import yaml
def get_yaml_value(yaml_file, yaml_key):
"""Return string value for 'yaml_key' from 'yaml_file.'"""
with open(yaml_file, "r") as ystream:
yaml_dict = yaml.safe_load(ystream)
return str(yaml_dict.get(yaml_key, ""))
def main():
"""Execute Main program."""
parser = optparse.OptionParser(description=__doc__)
parser.add_option("--yamlFile", dest="yaml_file", default=None, help="YAML file to read")
parser.add_option(
"--yamlKey", dest="yaml_key", default=None, help="Top level YAML key to provide the value")
(options, _) = parser.parse_args()
if not options.yaml_file:
parser.error("Must specifiy '--yamlFile'")
if not options.yaml_key:
parser.error("Must specifiy '--yamlKey'")
print(get_yaml_value(options.yaml_file, options.yaml_key))
if __name__ == "__main__":
main()
|