mirror of
https://github.com/JDM170/SaveWizard
synced 2025-04-20 22:30:42 +07:00
6094f785a5
* Update README.md * UI improvements: ** Removed unnessecary checkboxes on main window ** Removed "Analyze" button for cities, dealers and agencies; Now the are analyzing on second window startup ** And other small fixes * Code improvements: ** Typo fixes ** Added "dataIO.py" for interaction with configs (taken from "cog-creators/Red-DiscordBot" repo) ** Added dealers and agencies configs for both games (ATS, ETS2) ** Added "textEdited" signal for all edits on main window Signed-off-by: JDM170 <30170278+JDM170@users.noreply.github.com>
78 lines
2.2 KiB
Python
78 lines
2.2 KiB
Python
from json import decoder, load, dump
|
|
from os import replace
|
|
from os.path import splitext
|
|
from random import randint
|
|
|
|
|
|
class InvalidFileIO(Exception):
|
|
pass
|
|
|
|
|
|
class DataIO:
|
|
@staticmethod
|
|
def _read_json(filename):
|
|
with open(filename, encoding="utf-8", mode="r") as f:
|
|
data = load(f)
|
|
return data
|
|
|
|
@staticmethod
|
|
def _save_json(filename, data):
|
|
with open(filename, encoding="utf-8", mode="w") as f:
|
|
dump(data, f, indent=4, sort_keys=True, separators=(",", " : "))
|
|
return data
|
|
|
|
def is_valid_json(self, filename):
|
|
"""Verifies if json file exists / is readable"""
|
|
try:
|
|
self._read_json(filename)
|
|
return True
|
|
except FileNotFoundError:
|
|
return False
|
|
except decoder.JSONDecodeError:
|
|
return False
|
|
|
|
def load_json(self, filename):
|
|
"""Loads json file"""
|
|
return self._read_json(filename)
|
|
|
|
def save_json(self, filename, data):
|
|
"""Atomically saves json file"""
|
|
rnd = randint(1000, 9999)
|
|
path, ext = splitext(filename)
|
|
tmp_file = "{}-{}.tmp".format(path, rnd)
|
|
self._save_json(tmp_file, data)
|
|
try:
|
|
self._read_json(tmp_file)
|
|
except decoder.JSONDecodeError:
|
|
return False
|
|
replace(tmp_file, filename)
|
|
return True
|
|
|
|
def _legacy_fileio(self, filename, IO, data=None):
|
|
"""Old fileIO provided for backwards compatibility"""
|
|
if (IO == "save") and data is not None:
|
|
return self.save_json(filename, data)
|
|
elif (IO == "load") and data is None:
|
|
return self.load_json(filename)
|
|
elif (IO == "check") and data is None:
|
|
return self.is_valid_json(filename)
|
|
else:
|
|
raise InvalidFileIO("FileIO was called with invalid parameters")
|
|
|
|
|
|
def get_value(filename, key):
|
|
with open(filename, encoding="utf-8", mode="r") as f:
|
|
data = load(f)
|
|
return data[key]
|
|
|
|
|
|
def set_value(filename, key, value):
|
|
data = fileIO(filename, "load")
|
|
data[key] = value
|
|
fileIO(filename, "save", data)
|
|
return True
|
|
|
|
|
|
dataIO = DataIO()
|
|
fileIO = dataIO._legacy_fileio # backwards compatibility
|