85 lines
2.8 KiB
Python
85 lines
2.8 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import logging.config
|
|
import os
|
|
import shutil
|
|
|
|
import yaml
|
|
|
|
|
|
class Config(object):
|
|
def __init__(self) -> None:
|
|
self.reload()
|
|
|
|
def _load_config(self) -> dict:
|
|
pwd = os.path.dirname(os.path.abspath(__file__))
|
|
try:
|
|
with open(f"{pwd}/config.yaml", "rb") as fp:
|
|
yconfig = yaml.safe_load(fp)
|
|
except FileNotFoundError:
|
|
shutil.copyfile(f"{pwd}/config.yaml.template", f"{pwd}/config.yaml")
|
|
with open(f"{pwd}/config.yaml", "rb") as fp:
|
|
yconfig = yaml.safe_load(fp)
|
|
|
|
return yconfig
|
|
|
|
def reload(self) -> None:
|
|
yconfig = self._load_config()
|
|
logging.config.dictConfig(yconfig["logging"])
|
|
self.GROUPS = yconfig["groups"]["enable"]
|
|
self.NEWS = yconfig["news"]["receivers"]
|
|
self.REPORT_REMINDERS = yconfig["report_reminder"]["receivers"]
|
|
|
|
self.CHATGPT = yconfig.get("chatgpt", {})
|
|
self.TIGERBOT = yconfig.get("tigerbot", {})
|
|
self.XINGHUO_WEB = yconfig.get("xinghuo_web", {})
|
|
self.CHATGLM = yconfig.get("chatglm", {})
|
|
self.BardAssistant = yconfig.get("bard", {})
|
|
self.ZhiPu = yconfig.get("zhipu", {})
|
|
self.CLAUDE = yconfig.get("claude", {})
|
|
self.DEEPSEEK = yconfig.get("deepseek", {})
|
|
self.DOUBAO = yconfig.get("doubao", {})
|
|
|
|
# DB config
|
|
self.mariadb = yconfig.get("db_config", {})
|
|
self.redis = yconfig.get("redis_config", {})
|
|
|
|
#gewechat config
|
|
gewechat_config = yconfig['gewechat']
|
|
self.BASE_URL = gewechat_config.get("base_url", "")
|
|
self.GEWECHAT_TOKEN = gewechat_config.get("gewechat_token", "")
|
|
self.APP_ID = gewechat_config.get("app_id", "")
|
|
self.CALLBACK_URL = gewechat_config.get("callback_url", "")
|
|
|
|
def update_config(self, section, key, value):
|
|
"""更新配置文件中指定部分的键值
|
|
|
|
Args:
|
|
section: 配置部分名称，如 'gewechat'
|
|
key: 键名，如 'app_id'
|
|
value: 要设置的值
|
|
"""
|
|
import yaml
|
|
import os
|
|
|
|
config_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'config.yaml')
|
|
|
|
# 读取当前配置
|
|
with open(config_path, 'r', encoding='utf-8') as f:
|
|
config_data = yaml.safe_load(f)
|
|
|
|
# 更新配置
|
|
if section in config_data:
|
|
config_data[section][key] = value
|
|
|
|
# 写回配置文件
|
|
with open(config_path, 'w', encoding='utf-8') as f:
|
|
yaml.dump(config_data, f, default_flow_style=False, allow_unicode=True)
|
|
|
|
# 更新当前实例的属性
|
|
if hasattr(self, key.upper()):
|
|
setattr(self, key.upper(), value)
|
|
|
|
return True
|
|
return False |