33 lines
661 B
Python
33 lines
661 B
Python
import json
|
|
import os
|
|
|
|
from untitled import model
|
|
|
|
SAVE_FOLDER = os.path.join("untitled", "saves")
|
|
|
|
|
|
def save(save: model.Save, folder=None):
|
|
folder = folder or SAVE_FOLDER
|
|
file_name = save.cat.name + ".kitten"
|
|
save_file = os.path.join(folder, file_name)
|
|
|
|
data = save.to_dict()
|
|
|
|
os.makedirs(folder, exist_ok=True)
|
|
|
|
with open(save_file, "w") as f:
|
|
json.dump(data, f)
|
|
|
|
|
|
def load(name, folder=None):
|
|
folder = folder or SAVE_FOLDER
|
|
file_name = name + ".kitten"
|
|
save_file = os.path.join(folder, file_name)
|
|
|
|
with open(save_file) as f:
|
|
data = json.load(f)
|
|
|
|
save = model.Save.from_dict(data)
|
|
|
|
return save
|