From 64bf35a7e963ed82547a19f8cf6158c75425e912 Mon Sep 17 00:00:00 2001 From: Laurent Bachelier Date: Tue, 6 Mar 2012 19:08:50 +0100 Subject: [PATCH] Use libyaml for faster loading and saving if installed --- weboob/tools/config/yamlconfig.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/weboob/tools/config/yamlconfig.py b/weboob/tools/config/yamlconfig.py index f859e922..899176e8 100644 --- a/weboob/tools/config/yamlconfig.py +++ b/weboob/tools/config/yamlconfig.py @@ -25,6 +25,13 @@ import tempfile import logging import yaml +try: + from yaml import CLoader as Loader + from yaml import CDumper as Dumper +except ImportError: + from yaml import Loader + from yaml import Dumper + from .iconfig import IConfig, ConfigError @@ -42,7 +49,7 @@ class YamlConfig(IConfig): logging.debug(u'Loading application configuration file: %s.' % self.path) try: with open(self.path, 'r') as f: - self.values = yaml.load(f) + self.values = yaml.load(f, Loader=Loader) logging.debug(u'Application configuration file loaded: %s.' % self.path) except IOError: self.save() @@ -55,7 +62,7 @@ class YamlConfig(IConfig): # write in a temporary file to avoid corruption problems fd, path = tempfile.mkstemp(dir=os.path.dirname(self.path)) with os.fdopen(fd, 'w') as f: - yaml.dump(self.values, f) + yaml.dump(self.values, f, Dumper=Dumper) os.rename(path, self.path) def get(self, *args, **kwargs):