source: trunk/packages/invirt-base/scripts/invirt-getconf @ 1799

Last change on this file since 1799 was 1420, checked in by price, 16 years ago

eliminate choice of source and cache files in invirt config

  • Property svn:executable set to *
File size: 3.0 KB
Line 
1#!/usr/bin/env python
2
3"""
4invirt-getconf loads an invirt configuration file (either the original YAML
5source or the faster-to-load JSON cache) and prints the configuration option
6with the given name (key).  Keys are dot-separated paths into the YAML
7configuration tree.  List indexes (0-based) are also treated as path
8components.
9
10(Due to this path language, certain restrictions are placed on the keys used in
11the YAML configuration; e.g., they cannot contain dots.)
12
13Examples:
14
15  invirt-getconf db.uri
16  invirt-getconf authn.0.type
17"""
18
19from invirt import config
20from sys import argv, exit, stderr, stdout
21from optparse import OptionParser
22
23class invirt_exception(Exception): pass
24
25def 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
83if __name__ == '__main__':
84    exit(main(argv))
85
86# vim:et:sw=4:ts=4
Note: See TracBrowser for help on using the repository browser.