1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
# Copyright (c) 2016 Weitian LI <liweitianux@live.com>
# MIT license
#
# References:
# [1] https://configobj.readthedocs.io/en/latest/configobj.html
# [2] https://github.com/pazz/alot/blob/master/alot/settings/manager.py
"""
Configuration manager.
"""
import os
from glob import glob
from errors import ConfigError
from configobj import ConfigObj, ConfigObjError, flatten_errors
from validate import Validator
CONFIGS_PATH = os.path.dirname(__file__)
class ConfigManager:
"""Manager the configurations"""
def __init__(self, configs=None):
"""
:param configs: (optional) list of user configs to load
"""
configs_spec = sorted(glob(os.path.join(CONFIGS_PATH, "*.conf.spec")))
spec = "\n".join([open(f).read() for f in configs_spec]).split("\n")
self._configspec = ConfigObj(spec, interpolation=False,
list_values=False, _inspec=True)
self._validator = Validator()
configs_default = ConfigObj(configspec=self._configspec)
self._config = self.validate(configs_default)
if configs:
for config in configs:
self.read_config(config)
def read_config(self, config):
newconfig = ConfigObj(config, configspec=self._configspec)
newconfig = self.validate(newconfig)
self._config.merge(newconfig)
def validate(self, config):
try:
results = config.validate(self._validator, preserve_errors=True)
except ConfigObjError as e:
raise ConfigError(e.message)
if not results:
error_msg = ''
for (section_list, key, res) in flatten_errors(config, results):
if key is not None:
if res is False:
msg = 'key "%s" in section "%s" is missing.'
msg = msg % (key, ', '.join(section_list))
else:
msg = 'key "%s" in section "%s" failed validation: %s'
msg = msg % (key, ', '.join(section_list), res)
else:
msg = 'section "%s" is missing' % '.'.join(section_list)
error_msg += msg + '\n'
raise ConfigError(error_msg)
return config
def get(self, key, fallback=None):
if key in self._config:
value = self._config[key]
else:
value = fallback
return value
def set(self, key, value):
self._config[key] = value
|
