143 lines
5.1 KiB
Python
143 lines
5.1 KiB
Python
from typing import Any, Union
|
|
from ujson import JSONDecodeError as JSONDecodeError
|
|
from ujson import loads, dumps
|
|
|
|
from sys import exit
|
|
from os import kill, sep
|
|
from os import name as osname
|
|
from traceback import print_exc
|
|
|
|
from modules.logging import logWrite
|
|
|
|
def jsonLoad(filename):
|
|
"""Loads arg1 as json and returns its contents"""
|
|
with open(filename, "r", encoding='utf8') as file:
|
|
try:
|
|
output = loads(file.read())
|
|
except JSONDecodeError:
|
|
logWrite(f"Could not load json file {filename}: file seems to be incorrect!\n{print_exc()}")
|
|
raise
|
|
except FileNotFoundError:
|
|
logWrite(f"Could not load json file {filename}: file does not seem to exist!\n{print_exc()}")
|
|
raise
|
|
return output
|
|
|
|
def jsonSave(contents, filename):
|
|
"""Dumps dict/list arg1 to file arg2"""
|
|
try:
|
|
with open(filename, "w", encoding='utf8') as file:
|
|
file.write(dumps(contents, ensure_ascii=False, indent=4))
|
|
except Exception as exp:
|
|
logWrite(f"Could not save json file {filename}: {exp}\n{print_exc()}")
|
|
return
|
|
|
|
|
|
def nested_set(dic, keys, value, create_missing=True):
|
|
d = dic
|
|
for key in keys[:-1]:
|
|
if key in d:
|
|
d = d[key]
|
|
elif create_missing:
|
|
d = d.setdefault(key, {})
|
|
else:
|
|
return dic
|
|
if keys[-1] in d or create_missing:
|
|
d[keys[-1]] = value
|
|
return dic
|
|
|
|
def configSet(keys: list, value: Any, file: str = "config", create_missing=True):
|
|
"""Set config's value to provided one
|
|
|
|
Args:
|
|
* keys (list): List of keys from the highest one to target
|
|
* value (Any): Needed value
|
|
* file (str, optional): File (if not config). Defaults to "config".
|
|
* create_missing (bool, optional): Create missing items on the way. Defaults to True.
|
|
"""
|
|
if file == "config":
|
|
filepath = ""
|
|
this_dict = jsonLoad(f"{filepath}{file}.json")
|
|
if this_dict["debug"] is True:
|
|
try:
|
|
this_dict = jsonLoad("config_debug.json")
|
|
file = "config_debug"
|
|
except FileNotFoundError:
|
|
print("Debug mode is set but config_debug.json is not there! Falling back to config.json", flush=True)
|
|
else:
|
|
filepath = f"data{sep}users{sep}"
|
|
this_dict = jsonLoad(f"{filepath}{file}.json")
|
|
|
|
this_dict = nested_set(this_dict, keys, value, create_missing=create_missing)
|
|
jsonSave(this_dict, f"{filepath}{file}.json")
|
|
return
|
|
|
|
def configGet(key: str, *args: str, file: str = "config"):
|
|
"""Get value of the config key
|
|
Args:
|
|
* key (str): The last key of the keys path.
|
|
* *args (str): Path to key like: dict[args][key].
|
|
* file (str): User ID to load. Loads config if not provided. Defaults to "config".
|
|
Returns:
|
|
* any: Value of provided key
|
|
"""
|
|
if file == "config":
|
|
try:
|
|
this_dict = jsonLoad("config.json")
|
|
except FileNotFoundError:
|
|
print("Config file not found! Copy config_example.json to config.json, configure it and rerun the bot!", flush=True)
|
|
exit()
|
|
if this_dict["debug"] is True:
|
|
try:
|
|
this_dict = jsonLoad("config_debug.json")
|
|
except FileNotFoundError:
|
|
print("Debug mode is set but config_debug.json is not there! Falling back to config.json", flush=True)
|
|
else:
|
|
this_dict = jsonLoad(f"data{sep}users{sep}{file}.json")
|
|
this_key = this_dict
|
|
for dict_key in args:
|
|
this_key = this_key[dict_key]
|
|
return this_key[key]
|
|
|
|
def locale(key: str, *args: str, locale=configGet("locale")) -> Union[str, list, dict]:
|
|
"""Get value of locale string
|
|
Args:
|
|
* key (str): The last key of the locale's keys path.
|
|
* *args (list): Path to key like: dict[args][key].
|
|
* locale (str): Locale to looked up in. Defaults to config's locale value.
|
|
Returns:
|
|
* any: Value of provided locale key
|
|
"""
|
|
if (locale == None):
|
|
locale = configGet("locale")
|
|
|
|
try:
|
|
this_dict = jsonLoad(f'{configGet("locale", "locations")}{sep}{locale}.json')
|
|
except FileNotFoundError:
|
|
try:
|
|
this_dict = jsonLoad(f'{configGet("locale", "locations")}{sep}{configGet("locale")}.json')
|
|
except FileNotFoundError:
|
|
return f'⚠️ Locale in config is invalid: could not get "{key}" in {str(args)} from locale "{locale}"'
|
|
|
|
this_key = this_dict
|
|
for dict_key in args:
|
|
this_key = this_key[dict_key]
|
|
|
|
try:
|
|
return this_key[key]
|
|
except KeyError:
|
|
return f'⚠️ Locale in config is invalid: could not get "{key}" in {str(args)} from locale "{locale}"'
|
|
|
|
try:
|
|
from psutil import Process
|
|
except ModuleNotFoundError:
|
|
# print(locale("deps_missing", "console", locale=configGet("locale")), flush=True)
|
|
print("Missing dependencies! Please install all needed dependencies and run the bot again!")
|
|
exit()
|
|
|
|
def killProc(pid):
|
|
if osname == "posix":
|
|
from signal import SIGKILL # type: ignore
|
|
kill(pid, SIGKILL)
|
|
else:
|
|
p = Process(pid)
|
|
p.kill() |