From 6591424162b76dd75f5b5a5f596450e8ea9eac43 Mon Sep 17 00:00:00 2001
From: JakHolzer <53743814+JakHolzer@users.noreply.github.com>
Date: Thu, 10 Sep 2020 15:26:12 +0200
Subject: [PATCH] simple pickle function to store the dict

---
 pyzebra/app/save_load_dict.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)
 create mode 100644 pyzebra/app/save_load_dict.py

diff --git a/pyzebra/app/save_load_dict.py b/pyzebra/app/save_load_dict.py
new file mode 100644
index 0000000..5e31442
--- /dev/null
+++ b/pyzebra/app/save_load_dict.py
@@ -0,0 +1,19 @@
+import pickle
+
+
+def save_dict(obj, name):
+    """ saves dictionary as pickle file in binary format
+    :arg obj - object to save
+    :arg name - name of the file
+    NOTE: path should be added later"""
+    with open(name + '.pkl', 'wb') as f:
+        pickle.dump(obj, f, pickle.HIGHEST_PROTOCOL)
+
+
+def load_dict(name):
+    """load dictionary from picle file
+    :arg name - name of the file to load
+    NOTE: expect the file in the same folder, path should be added later
+    :return dictionary"""
+    with open(name + '.pkl', 'rb') as f:
+        return pickle.load(f)