46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
from typing import Any, Union
|
|
|
|
from discord import Member, User
|
|
from ujson import dumps, loads
|
|
|
|
|
|
def json_read_sync(path: str) -> Any:
|
|
with open(path, mode="r", encoding="utf-8") as f:
|
|
data = f.read()
|
|
return loads(data)
|
|
|
|
|
|
def json_write_sync(data: Any, path: str) -> None:
|
|
with open(path, mode="w", encoding="utf-8") as f:
|
|
f.write(dumps(data, ensure_ascii=False, escape_forward_slashes=False, indent=4))
|
|
|
|
|
|
def config_get_sync(key: str, *path: str) -> Any:
|
|
this_key = json_read_sync("config.json")
|
|
for dict_key in path:
|
|
this_key = this_key[dict_key]
|
|
return this_key[key]
|
|
|
|
|
|
def config_set_sync(key: str, value: Any, *path: str) -> None:
|
|
this_dict = json_read_sync("config.json")
|
|
string = "this_dict"
|
|
for arg in path:
|
|
string += f'["{arg}"]'
|
|
if type(value) in [str]:
|
|
string += f'["{key}"] = "{value}"'
|
|
else:
|
|
string += f'["{key}"] = {value}'
|
|
exec(string)
|
|
json_write_sync(this_dict, "config.json")
|
|
return
|
|
|
|
|
|
def guild_name(member: Union[Member, User]) -> str:
|
|
if isinstance(member, User):
|
|
return member.name
|
|
if member.nick == None:
|
|
return member.name
|
|
else:
|
|
return member.nick
|