Change: use the libyaml bindings for yaml loading, dumping

For speed.

1 files changed, 3 insertions, 2 deletions

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)