saving

untitled/persistence.py

@@ -0,0 +1,29 @@
+from untitled import model
+import os
+import json
+
+SAVE_FOLDER = os.path.join("untitled", "saves")
+
+
+def save(save: model.Save):
+    file_name = save.cat.name + ".kitten"
+    save_file = os.path.join(SAVE_FOLDER, file_name)
+
+    data = save.to_dict()
+
+    os.makedirs(SAVE_FOLDER, exist_ok=True)
+
+    with open(save_file, "w") as f:
+        json.dump(data, f)
+
+
+def load(name):
+    file_name = name + ".kitten"
+    save_file = os.path.join(SAVE_FOLDER, file_name)
+
+    with open(save_file, "r") as f:
+        data = json.load(f)
+
+    save = model.Save.from_dict(data)
+
+    return save