From 0f6fd394f39406b76e8f18fe4c2ddb791194f978 Mon Sep 17 00:00:00 2001 From: Chandler J Date: Mon, 18 Mar 2024 13:14:40 -0600 Subject: began implementating save states for color palettes and MOAR POLISH --- src/manage_saves.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 src/manage_saves.py (limited to 'src/manage_saves.py') 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 -- cgit v1.2.3