from untitled import model
import os
import json

SAVE_FOLDER = os.path.join("untitled", "saves")


def save(save: model.Save):
    file_name = save.cat.name + ".kitten"
    save_file = os.path.join(SAVE_FOLDER, file_name)

    data = save.to_dict()

    os.makedirs(SAVE_FOLDER, exist_ok=True)

    with open(save_file, "w") as f:
        json.dump(data, f)


def load(name):
    file_name = name + ".kitten"
    save_file = os.path.join(SAVE_FOLDER, file_name)

    with open(save_file, "r") as f:
        data = json.load(f)

    save = model.Save.from_dict(data)

    return save