src/manage_saves.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
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