diff options
Diffstat (limited to 'ick2/persistent.py')
-rw-r--r-- | ick2/persistent.py | 5 |
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) |