mirror of https://github.com/mongodb/mongo
36 lines
993 B
Python
Executable File
36 lines
993 B
Python
Executable File
#!/usr/bin/env python3
|
|
"""Utility to return YAML value from key in YAML file."""
|
|
|
|
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", encoding="utf8") 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()
|