mirror of
https://github.com/Genshin-bots/gsuid_core.git
synced 2025-05-12 06:55:49 +08:00
93 lines
2.3 KiB
Python
93 lines
2.3 KiB
Python
import json
|
|
from pathlib import Path
|
|
from typing import Dict, List, Union, Literal, overload
|
|
|
|
CONFIG_PATH = Path(__file__).parent / 'config.json'
|
|
|
|
CONFIG_DEFAULT = {
|
|
'HOST': 'localhost',
|
|
'PORT': '8765',
|
|
'masters': [],
|
|
'superusers': [],
|
|
'log': {
|
|
'level': 'INFO',
|
|
# ...
|
|
},
|
|
'command_start': [],
|
|
'sv': {},
|
|
}
|
|
STR_CONFIG = Literal['HOST', 'PORT']
|
|
LIST_CONFIG = Literal['superusers', 'masters', 'command_start']
|
|
DICT_CONFIG = Literal['sv', 'log']
|
|
|
|
|
|
class CoreConfig:
|
|
def __init__(self) -> None:
|
|
if not CONFIG_PATH.exists():
|
|
with open(CONFIG_PATH, 'w', encoding='UTF-8') as file:
|
|
json.dump(CONFIG_DEFAULT, file, indent=4, ensure_ascii=False)
|
|
|
|
self.update_config()
|
|
|
|
def write_config(self):
|
|
with open(CONFIG_PATH, 'w', encoding='UTF-8') as file:
|
|
json.dump(self.config, file, indent=4, ensure_ascii=False)
|
|
|
|
def update_config(self):
|
|
# 打开config.json
|
|
with open(CONFIG_PATH, 'r', encoding='UTF-8') as f:
|
|
self.config = json.load(f)
|
|
# 对没有的值，添加默认值
|
|
for key in CONFIG_DEFAULT:
|
|
if key not in self.config:
|
|
self.config[key] = CONFIG_DEFAULT[key]
|
|
|
|
# 重新写回
|
|
self.write_config()
|
|
|
|
@overload
|
|
def get_config(self, key: STR_CONFIG) -> str:
|
|
...
|
|
|
|
@overload
|
|
def get_config(self, key: DICT_CONFIG) -> Dict:
|
|
...
|
|
|
|
@overload
|
|
def get_config(self, key: LIST_CONFIG) -> List:
|
|
...
|
|
|
|
def get_config(self, key: str) -> Union[str, Dict, List]:
|
|
if key in self.config:
|
|
return self.config[key]
|
|
elif key in CONFIG_DEFAULT:
|
|
self.update_config()
|
|
return self.config[key]
|
|
else:
|
|
return {}
|
|
|
|
@overload
|
|
def set_config(self, key: STR_CONFIG, value: str) -> bool:
|
|
...
|
|
|
|
@overload
|
|
def set_config(self, key: LIST_CONFIG, value: List) -> bool:
|
|
...
|
|
|
|
@overload
|
|
def set_config(self, key: DICT_CONFIG, value: Dict) -> bool:
|
|
...
|
|
|
|
def set_config(self, key: str, value: Union[str, List, Dict]) -> bool:
|
|
if key in CONFIG_DEFAULT:
|
|
# 设置值
|
|
self.config[key] = value
|
|
# 重新写回
|
|
self.write_config()
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
|
|
core_config = CoreConfig()
|