summaryrefslogtreecommitdiff
path: root/src/manage_saves.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/manage_saves.py')
-rw-r--r--src/manage_saves.py21
1 files changed, 21 insertions, 0 deletions
diff --git a/src/manage_saves.py b/src/manage_saves.py
new file mode 100644
index 0000000..988b9a2
--- /dev/null
+++ b/src/manage_saves.py
@@ -0,0 +1,21 @@
+import pickle
+from rich import print
+
+def save_color_palette(hex_colors, hex_compliments, save_directory):
+
+ colors = [hex_colors, hex_compliments]
+
+ with open(save_directory, 'wb') as file:
+ pickle.dump(colors, file, pickle.HIGHEST_PROTOCOL)
+
+ print('[bold green]color palette saved successfully')
+
+def load_color_palette(save_location):
+ with open(save_location, 'rb') as file:
+ try:
+ data = pickle.load(file)
+ except:
+ print('invalid arguments. -p should be followed by a color palette save')
+ exit(2)
+
+ return data