RGBCube/JsonWrapper
1
Fork 0
mirror of https://github.com/RGBCube/JsonWrapper synced 2025-07-27 19:57:44 +00:00
Code Issues Activity Actions

pathmagic for get

Browse source
This commit is contained in:
RGBCube 2022-01-16 10:32:57 +03:00 • committed by GitHub
parent 298b0d95a2
commit 2894fc59e9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

19
db.py
View file

@ -13,7 +13,7 @@ class Utils:
with open(path_to_json, "w") as json_file: with open(path_to_json, "w") as json_file:
json.dump({}, json_file) json.dump({}, json_file)
def path_magic(self, main_dict: dict, path: str, *, key: str, value): def path_magic_set(self, main_dict: dict, path: str, *, key: str, value):
def magic(alt_dict: dict, key: str): def magic(alt_dict: dict, key: str):
if key in alt_dict.keys(): if key in alt_dict.keys():
return alt_dict return alt_dict
@ -28,6 +28,14 @@ class Utils:
main_dict[key] = value main_dict[key] = value
return main_dict_ref return main_dict_ref
def path_magic_get(self, main_dict: dict, path: str, *, key, default=None):
for dict_name in path.split("+"):
try:
main_dict = main_dict[dict_name]
except (KeyError, AttributeError):
return default
return main_dict.get(key, default)
class ClutterDB: class ClutterDB:
def __init__(self, path_to_json: str): def __init__(self, path_to_json: str):
@ -44,13 +52,16 @@ class ClutterDB:
json_data[key] = value json_data[key] = value
json.dump(json_data, json_file, indent=4) json.dump(json_data, json_file, indent=4)
else: else:
json.dump(self.utils.path_magic(json_data, pathmagic, key=key, value=value), json_file, indent=4) json.dump(self.utils.path_magic_set(json_data, pathmagic, key=key, value=value), json_file, indent=4)
def get(self, key: str, *, default=None): def get(self, key: str, *, default=None, pathmagic=""):
self.utils.validate_json(self.path_to_json) self.utils.validate_json(self.path_to_json)
with open(self.path_to_json, mode="r") as json_file: with open(self.path_to_json, mode="r") as json_file:
json_data = json.load(json_file) json_data = json.load(json_file)
return json_data.get(key, default) if pathmagic == "":
return json_data.get(key, default)
else:
return self.utils.path_magic_get(json_data, pathmagic, key=key, default=default)
def all(self): def all(self):
self.utils.validate_json(self.path_to_json) self.utils.validate_json(self.path_to_json)