From ee864e8c929f61506f827d8068ce1abee1e6b5fd Mon Sep 17 00:00:00 2001 From: Laurent Bachelier Date: Wed, 27 Mar 2019 16:02:02 +0100 Subject: [PATCH] config: Allow setting Loader and Dumper at class level --- weboob/tools/config/yamlconfig.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/weboob/tools/config/yamlconfig.py b/weboob/tools/config/yamlconfig.py index 38bf4bdfde..46c79c285d 100644 --- a/weboob/tools/config/yamlconfig.py +++ b/weboob/tools/config/yamlconfig.py @@ -51,6 +51,9 @@ class WeboobDumper(Dumper): class YamlConfig(IConfig): + DUMPER = WeboobDumper + LOADER = Loader + def __init__(self, path): self.path = path self.values = {} @@ -61,7 +64,7 @@ def load(self, default={}): logging.debug(u'Loading application configuration file: %s.' % self.path) try: with open(self.path, 'r') as f: - self.values = yaml.load(f, Loader=Loader) + self.values = yaml.load(f, Loader=self.LOADER) logging.debug(u'Application configuration file loaded: %s.' % self.path) except IOError: self.save() @@ -77,7 +80,7 @@ def save(self): else: f = tempfile.NamedTemporaryFile(mode='w', dir=os.path.dirname(self.path), delete=False, encoding='utf-8') with f: - yaml.dump(self.values, f, Dumper=WeboobDumper, default_flow_style=False) + yaml.dump(self.values, f, Dumper=self.DUMPER, default_flow_style=False) os.rename(f.name, self.path) def get(self, *args, **kwargs): -- GitLab