From 7b9b4fc72b6d18cca36f9243172b55e288ba013b Mon Sep 17 00:00:00 2001 From: Lars Wirzenius Date: Tue, 24 Jul 2018 19:19:22 +0300 Subject: Change: use the libyaml bindings for yaml loading, dumping For speed. --- ick2/persistent.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'ick2') diff --git a/ick2/persistent.py b/ick2/persistent.py index 52c5aa0..c5e2840 100644 --- a/ick2/persistent.py +++ b/ick2/persistent.py @@ -97,7 +97,7 @@ class FilePersistentState(PersistentStateInterface): if not os.path.exists(filename): raise ick2.NotFound(kind=kind, rid=rid) with open(filename, 'r') as f: - as_dict = yaml.safe_load(f) + as_dict = yaml.load(f, Loader=yaml.CSafeLoader) return resource_from_dict(as_dict) def write_resource(self, kind, rid, resource): @@ -107,7 +107,8 @@ class FilePersistentState(PersistentStateInterface): filename = self._filename(kind, rid) with open(filename, 'w') as f: - yaml.safe_dump(resource.as_dict(), stream=f) + yaml.dump( + resource.as_dict(), stream=f, Dumper=yaml.CSafeDumper) def remove_resource(self, kind, rid): filename = self._filename(kind, rid) -- cgit v1.2.1