summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--ick2/persistent.py5
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)