38 lines
1021 B
Python
38 lines
1021 B
Python
import tomllib
|
|
from pathlib import Path
|
|
from typing import Any, Dict
|
|
|
|
class Config:
|
|
def __init__(self, file_path: str = "config.toml"):
|
|
self.path = Path(file_path)
|
|
self._data: Dict[str, Any] = {}
|
|
self.load()
|
|
def load(self):
|
|
if not self.path.exists():
|
|
raise FileNotFoundError(f"配置文件 {self.path} 未找到！")
|
|
|
|
with open(self.path, "rb") as f:
|
|
self._data = tomllib.load(f)
|
|
|
|
# 通过属性访问配置
|
|
@property
|
|
def napcat_ws(self) -> dict:
|
|
return self._data.get("napcat_ws", {})
|
|
|
|
@property
|
|
def bot(self) -> dict:
|
|
return self._data.get("bot", {})
|
|
|
|
@property
|
|
def features(self) -> dict:
|
|
return self._data.get("features", {})
|
|
|
|
# 实例化全局配置对象
|
|
global_config = Config()
|
|
|
|
if __name__ == "__main__":
|
|
print(global_config.napcat_ws)
|
|
print(global_config.bot.get("command"))
|
|
print(type(global_config.bot.get("command")) is list)
|
|
print(global_config.features)
|
|
|