diff options
| -rwxr-xr-x | bitbotd | 1 | ||||
| -rw-r--r-- | src/Config.py | 21 |
2 files changed, 17 insertions, 5 deletions
@@ -117,6 +117,7 @@ if args.add_server: cache = Cache.Cache() config = Config.Config(args.config) +config.load() events = EventManager.EventRoot(log).wrap() exports = Exports.Exports() timers = Timers.Timers(database, events, log) diff --git a/src/Config.py b/src/Config.py index e9209d2c..6d80b1ca 100644 --- a/src/Config.py +++ b/src/Config.py @@ -1,17 +1,28 @@ -import configparser, os, typing +import collections, configparser, os, typing class Config(object): def __init__(self, location: str): self.location = location - self._config = {} # type: typing.Dict[str, str] - self.load() + self._config = collections.OrderedDict() + + def _parser(self) -> configparser.ConfigParser: + return configparser.ConfigParser(dict_type=collections.OrderedDict) def load(self): if os.path.isfile(self.location): with open(self.location) as config_file: - parser = configparser.ConfigParser() + parser = self._parser() parser.read_string(config_file.read()) - self._config = {k: v for k, v in parser["bot"].items() if v} + self._config.clear() + for k, v in parser["bot"].items(): + if v: + self._config[k] = v + + def save(self): + with open(self.location, "w") as config_file: + parser = self._parser() + parser["bot"] = self._config.copy() + parser.write(config_file) def __getitem__(self, key: str) -> typing.Any: return self._config[key] |