149 lines
4.4 KiB
Python
149 lines
4.4 KiB
Python
from os import kill, path
|
|
from os import name as osname
|
|
from os import sep
|
|
from sys import exit
|
|
from traceback import print_exc
|
|
from typing import Any
|
|
|
|
from ujson import JSONDecodeError, dumps, loads
|
|
|
|
from modules.logger import logWrite
|
|
|
|
|
|
def jsonLoad(filename: str) -> Any:
|
|
"""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
|
|
file.close()
|
|
return output
|
|
|
|
|
|
def jsonSave(contents: Any, filename: str) -> None:
|
|
"""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, escape_forward_slashes=False
|
|
)
|
|
)
|
|
file.close()
|
|
except Exception as exp:
|
|
logWrite(f"Could not save json file {filename}: {exp}\n{print_exc()}")
|
|
return
|
|
|
|
|
|
def configSet(key: str, value, *args: str):
|
|
"""Set key to a value
|
|
Args:
|
|
* key (str): The last key of the keys path.
|
|
* value (str/int/float/list/dict/None): Some needed value.
|
|
* *args (str): Path to key like: dict[args][key].
|
|
"""
|
|
this_dict = jsonLoad("config.json")
|
|
string = "this_dict"
|
|
for arg in args:
|
|
string += f'["{arg}"]'
|
|
if type(value) in [str]:
|
|
string += f'["{key}"] = "{value}"'
|
|
else:
|
|
string += f'["{key}"] = {value}'
|
|
exec(string)
|
|
jsonSave(this_dict, "config.json")
|
|
return
|
|
|
|
|
|
def configGet(key: str, *args: str):
|
|
"""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].
|
|
Returns:
|
|
* any: Value of provided key
|
|
"""
|
|
this_dict = jsonLoad("config.json")
|
|
try:
|
|
this_key = this_dict
|
|
for dict_key in args:
|
|
this_key = this_key[dict_key]
|
|
except KeyError:
|
|
print(
|
|
f"Could not find config key '{key}' under path {args}: falling back to default config",
|
|
flush=True,
|
|
)
|
|
fallback_dict = jsonLoad(path.join("modules", "default_config.json"))
|
|
this_key = fallback_dict
|
|
for dict_key in args:
|
|
this_key = this_key[dict_key]
|
|
configSet(key, this_key[key], args)
|
|
return this_key[key]
|
|
|
|
|
|
def locale(key: str, *args: str, locale=configGet("locale")):
|
|
"""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:
|
|
try:
|
|
this_dict = jsonLoad(
|
|
f'{configGet("locale_fallback", "locations")}{sep}{configGet("locale")}.json'
|
|
)
|
|
except:
|
|
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)
|
|
exit()
|
|
|
|
|
|
def killProc(pid: int) -> None:
|
|
"""Kill process by its PID. Meant to be used to kill the main process of bot itself.
|
|
|
|
### Args:
|
|
* pid (`int`): PID of the target
|
|
"""
|
|
if osname == "posix":
|
|
from signal import SIGKILL
|
|
|
|
kill(pid, SIGKILL)
|
|
else:
|
|
Process(pid).kill()
|