From 90911ae6112debd620904362b452dbdca04c0e44 Mon Sep 17 00:00:00 2001 From: Lars Wirzenius Date: Sat, 3 Jun 2017 16:49:25 +0300 Subject: Use libYAML's CSafeDumper and CSafeLoader, for speed --- obbenchlib/htmlgen.py | 2 +- obbenchlib/result.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) (limited to 'obbenchlib') diff --git a/obbenchlib/htmlgen.py b/obbenchlib/htmlgen.py index c0bc06f..778be9c 100644 --- a/obbenchlib/htmlgen.py +++ b/obbenchlib/htmlgen.py @@ -74,7 +74,7 @@ class HtmlGenerator(object): for i, filename in enumerate(filenames): print 'Loading', filename, i+1, 'of', len(filenames) with open(filename) as f: - yield yaml.safe_load(f) + yield yaml.load(f, Loader=yaml.CSafeLoader) def write_file(self, relative_path, text): filename = os.path.join(self.htmldir, relative_path) diff --git a/obbenchlib/result.py b/obbenchlib/result.py index 8ac24b8..d782680 100644 --- a/obbenchlib/result.py +++ b/obbenchlib/result.py @@ -54,4 +54,4 @@ class Result(object): } filename = os.path.join(dirname, '{}.yaml'.format(self._result_id)) with open(filename, 'w') as f: - yaml.safe_dump(o, stream=f) + yaml.dump(o, stream=f, Dumper=yaml.CSafeDumper) -- cgit v1.2.1