use yaml.CSafeDumper and yaml.CSafeLoader everywhere
[invirt/packages/invirt-base.git] / files / usr / sbin / invirt-getconf
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 authn.0.type
17 """
18
19 from invirt.config import load
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('-s', '--src',
30                 default = '/etc/invirt/master.yaml',
31                 help = 'the source YAML configuration file to read from')
32         parser.add_option('-c', '--cache',
33                 default = '/var/lib/invirt/invirt.json',
34                 help = 'path to the JSON cache')
35         parser.add_option('-r', '--refresh',
36                 action = 'store_true',
37                 help = 'force the cache to be regenerated')
38         parser.add_option('-l', '--ls',
39                 action = 'store_true',
40                 help = 'list node\'s children')
41         opts, args = parser.parse_args()
42
43         if len(args) > 1:
44             raise invirt_exception(__doc__.strip())
45         elif args and args[0]:
46             components = args[0].split('.')
47         else:
48             components = []
49
50         conf = load(opts.src, opts.cache, opts.refresh)
51         for i, component in enumerate(components):
52             progress = '.'.join(components[:i])
53             if type(conf) not in (dict, list):
54                 raise invirt_exception(
55                         '%s: node has no children (atomic datum)' % progress)
56             if type(conf) == list:
57                 try: component = int(component)
58                 except: raise invirt_exception(
59                         '%s: node a list; integer path component required, '
60                         'but got "%s"' % (progress, component))
61             try: conf = conf[component]
62             except KeyError: raise invirt_exception(
63                     '%s: key "%s" not found' % (progress, component))
64             except IndexError: raise invirt_exception(
65                     '%s: index %s out of range' % (progress, component))
66
67         if opts.ls:
68             if type(conf) not in (dict, list):
69                 raise invirt_exception(
70                         '%s: node has no children (atomic datum)' % progress)
71             if type(conf) == list:
72                 for i in xrange(len(conf)):
73                     print i
74             else:
75                 for k in conf.iterkeys():
76                     print k
77         else:
78             if type(conf) not in (dict, list):
79                 print conf
80             else:
81                 import yaml
82                 yaml.dump(conf, stdout,
83                           Dumper=yaml.CSafeDumper, default_flow_style=False)
84     except invirt_exception, ex:
85         print >> stderr, ex
86         return 1
87
88 if __name__ == '__main__':
89     exit(main(argv))
90
91 # vim:et:sw=4:ts=4