About to break everything with a merge

[ipdf/code.git] / tools / saveload.py

diff --git a/tools/saveload.py b/tools/saveload.py
new file mode 100644 (file)
index 0000000..7f43a43
--- /dev/null
+++ b/tools/saveload.py
@@ -0,0 +1,16 @@
+import pickle
+
+def save_obj(obj, name ):
+    with open('obj/'+ name + '.pkl', 'wb') as f:
+        pickle.dump(obj, f, pickle.HIGHEST_PROTOCOL)
+
+def load_obj(name ):
+    with open('obj/' + name + '.pkl', 'r') as f:
+        return pickle.load(f)
+
+
+def load_dict(name):
+       try:
+               return load_obj(name)
+       except:
+               return {}