| 1 | #!/usr/bin/env python |
|---|
| 2 | |
|---|
| 3 | """ |
|---|
| 4 | invirt-getconf loads an invirt configuration file (either the original YAML |
|---|
| 5 | source or the faster-to-load JSON cache) and prints the configuration option |
|---|
| 6 | with the given name (key). Keys are dot-separated paths into the YAML |
|---|
| 7 | configuration tree. List indexes (0-based) are also treated as path |
|---|
| 8 | components. |
|---|
| 9 | |
|---|
| 10 | (Due to this path language, certain restrictions are placed on the keys used in |
|---|
| 11 | the YAML configuration; e.g., they cannot contain dots.) |
|---|
| 12 | |
|---|
| 13 | Examples: |
|---|
| 14 | |
|---|
| 15 | invirt-getconf db.uri |
|---|
| 16 | invirt-getconf hosts.0.ip |
|---|
| 17 | """ |
|---|
| 18 | |
|---|
| 19 | from invirt import config |
|---|
| 20 | from sys import argv, exit, stderr, stdout |
|---|
| 21 | from optparse import OptionParser |
|---|
| 22 | |
|---|
| 23 | class invirt_exception(Exception): pass |
|---|
| 24 | |
|---|
| 25 | def main(argv): |
|---|
| 26 | try: |
|---|
| 27 | parser = OptionParser(usage = '%prog [options] key', |
|---|
| 28 | description = __doc__.strip().split('\n\n')[0]) |
|---|
| 29 | parser.add_option('-r', '--refresh', |
|---|
| 30 | action = 'store_true', |
|---|
| 31 | help = 'force the cache to be regenerated') |
|---|
| 32 | parser.add_option('-l', '--ls', |
|---|
| 33 | action = 'store_true', |
|---|
| 34 | help = 'list node\'s children') |
|---|
| 35 | opts, args = parser.parse_args() |
|---|
| 36 | |
|---|
| 37 | if len(args) > 1: |
|---|
| 38 | raise invirt_exception(__doc__.strip()) |
|---|
| 39 | elif args and args[0]: |
|---|
| 40 | components = args[0].split('.') |
|---|
| 41 | else: |
|---|
| 42 | components = [] |
|---|
| 43 | |
|---|
| 44 | conf = config.load(opts.refresh) |
|---|
| 45 | for i, component in enumerate(components): |
|---|
| 46 | progress = '.'.join(components[:i]) |
|---|
| 47 | if type(conf) not in (dict, list): |
|---|
| 48 | raise invirt_exception( |
|---|
| 49 | '%s: node has no children (atomic datum)' % progress) |
|---|
| 50 | if type(conf) == list: |
|---|
| 51 | try: component = int(component) |
|---|
| 52 | except: raise invirt_exception( |
|---|
| 53 | '%s: node a list; integer path component required, ' |
|---|
| 54 | 'but got "%s"' % (progress, component)) |
|---|
| 55 | try: conf = conf[component] |
|---|
| 56 | except KeyError: raise invirt_exception( |
|---|
| 57 | '%s: key "%s" not found' % (progress, component)) |
|---|
| 58 | except IndexError: raise invirt_exception( |
|---|
| 59 | '%s: index %s out of range' % (progress, component)) |
|---|
| 60 | |
|---|
| 61 | if opts.ls: |
|---|
| 62 | if type(conf) not in (dict, list): |
|---|
| 63 | raise invirt_exception( |
|---|
| 64 | '%s: node has no children (atomic datum)' |
|---|
| 65 | % '.'.join(components)) |
|---|
| 66 | if type(conf) == list: |
|---|
| 67 | for i in xrange(len(conf)): |
|---|
| 68 | print i |
|---|
| 69 | else: |
|---|
| 70 | for k in conf.iterkeys(): |
|---|
| 71 | print k |
|---|
| 72 | else: |
|---|
| 73 | if type(conf) not in (dict, list): |
|---|
| 74 | print conf |
|---|
| 75 | else: |
|---|
| 76 | import yaml |
|---|
| 77 | yaml.dump(conf, stdout, |
|---|
| 78 | Dumper=yaml.CSafeDumper, default_flow_style=False) |
|---|
| 79 | except invirt_exception, ex: |
|---|
| 80 | print >> stderr, ex |
|---|
| 81 | return 1 |
|---|
| 82 | |
|---|
| 83 | if __name__ == '__main__': |
|---|
| 84 | exit(main(argv)) |
|---|
| 85 | |
|---|
| 86 | # vim:et:sw=4:ts=4 |
|---|
