try: from ujson import JSONDecodeError as JSONDecodeError from ujson import loads, dumps except ModuleNotFoundError: from json import JSONDecodeError as JSONDecodeError from json import loads, dumps from sys import exit from os import sep, kill from os import name as osname from traceback import print_exc from typing import Any 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)) 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") 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")): """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()