use pathlib
This commit is contained in:
@@ -1,19 +1,19 @@
|
||||
import json
|
||||
import os
|
||||
from pathlib import Path
|
||||
|
||||
from untitled import model
|
||||
from untitled import PACKAGE_ROOT, model
|
||||
|
||||
SAVE_FOLDER = os.path.join("untitled", "saves")
|
||||
SAVE_FOLDER = PACKAGE_ROOT / "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)
|
||||
save_file = Path(folder) / file_name
|
||||
|
||||
data = save.to_dict()
|
||||
|
||||
os.makedirs(folder, exist_ok=True)
|
||||
Path.mkdir(folder, exist_ok=True)
|
||||
|
||||
with open(save_file, "w") as f:
|
||||
json.dump(data, f)
|
||||
@@ -22,7 +22,7 @@ def save(save: model.Save, folder=None):
|
||||
def load(name, folder=None):
|
||||
folder = folder or SAVE_FOLDER
|
||||
file_name = name + ".kitten"
|
||||
save_file = os.path.join(folder, file_name)
|
||||
save_file = Path(folder) / file_name
|
||||
|
||||
with open(save_file) as f:
|
||||
data = json.load(f)
|
||||
|
||||
Reference in New Issue
Block a user