1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/imiptools/config.py Mon Jan 30 23:28:11 2017 +0100
1.3 @@ -0,0 +1,73 @@
1.4 +#!/usr/bin/env python
1.5 +
1.6 +"""
1.7 +Configuration management.
1.8 +
1.9 +Copyright (C) 2017 Paul Boddie <paul@boddie.org.uk>
1.10 +
1.11 +This program is free software; you can redistribute it and/or modify it under
1.12 +the terms of the GNU General Public License as published by the Free Software
1.13 +Foundation; either version 3 of the License, or (at your option) any later
1.14 +version.
1.15 +
1.16 +This program is distributed in the hope that it will be useful, but WITHOUT
1.17 +ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
1.18 +FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
1.19 +details.
1.20 +
1.21 +You should have received a copy of the GNU General Public License along with
1.22 +this program. If not, see <http://www.gnu.org/licenses/>.
1.23 +"""
1.24 +
1.25 +from compiler import parseFile
1.26 +from os.path import isfile, join, split
1.27 +from compiler.ast import Assign, AssName, Const, Discard, Module, Name, Stmt
1.28 +
1.29 +config_path = [".", "/etc/imip-agent", split(__file__)[0]]
1.30 +
1.31 +builtins = {
1.32 + "False" : False,
1.33 + "None" : None,
1.34 + "True" : True,
1.35 + }
1.36 +
1.37 +def get_config(filename="config.txt", path=config_path):
1.38 +
1.39 + "Obtain the configuration from 'filename', searching 'path' for the file."
1.40 +
1.41 + for dirname in path:
1.42 + pathname = join(dirname, filename)
1.43 + if isfile(pathname):
1.44 + module = parseFile(pathname)
1.45 + break
1.46 + else:
1.47 + return {}
1.48 +
1.49 + return get_config_data(module, {})
1.50 +
1.51 +def get_config_data(module, d):
1.52 +
1.53 + "Interpret the parsed 'module', storing mappings in 'd'."
1.54 +
1.55 + name = None
1.56 +
1.57 + for node in module.getChildNodes():
1.58 + if isinstance(node, (Module, Stmt)):
1.59 + get_config_data(node, d)
1.60 + elif isinstance(node, Discard):
1.61 + pass
1.62 + elif isinstance(node, Assign):
1.63 + if len(node.nodes) == 1 and isinstance(node.nodes[0], AssName):
1.64 + name = node.nodes[0].name
1.65 + if isinstance(node.expr, Const):
1.66 + d[name] = node.expr.value
1.67 + elif isinstance(node.expr, Name):
1.68 + d[name] = d.get(node.expr.name, builtins.get(node.expr.name))
1.69 +
1.70 + return d
1.71 +
1.72 +# Expose settings via a module-level name.
1.73 +
1.74 +settings = get_config()
1.75 +
1.76 +# vim: tabstop=4 expandtab shiftwidth=4