78 lines
2.6 KiB
Python
78 lines
2.6 KiB
Python
import os
|
|
import json
|
|
|
|
class ConfigManager:
|
|
def __init__(self):
|
|
# 插件目录和配置文件路径
|
|
self.dir_path = os.path.dirname(os.path.abspath(__file__))
|
|
self.config_path = os.path.join(self.dir_path, "config.json")
|
|
|
|
# 默认配置
|
|
self.default_config = {
|
|
"__comment": "请在下方配置百度翻译和智谱AI的API密钥",
|
|
"baidu_translate": {
|
|
"app_id": "",
|
|
"secret_key": ""
|
|
},
|
|
"llm": {
|
|
"api_key": ""
|
|
}
|
|
}
|
|
|
|
# 确保配置文件存在
|
|
self.ensure_config_exists()
|
|
|
|
def ensure_config_exists(self):
|
|
"""确保配置文件存在，不存在则创建默认配置"""
|
|
if not os.path.exists(self.config_path):
|
|
print("[PromptAssistant] 配置文件不存在，创建默认配置文件...")
|
|
self.save_config(self.default_config)
|
|
|
|
def load_config(self):
|
|
"""加载配置文件"""
|
|
try:
|
|
with open(self.config_path, "r", encoding="utf-8") as f:
|
|
return json.load(f)
|
|
except Exception as e:
|
|
print(f"[PromptAssistant] 加载配置文件失败: {str(e)}")
|
|
return self.default_config
|
|
|
|
def save_config(self, config):
|
|
"""保存配置文件"""
|
|
try:
|
|
with open(self.config_path, "w", encoding="utf-8") as f:
|
|
json.dump(config, f, ensure_ascii=False, indent=2)
|
|
return True
|
|
except Exception as e:
|
|
print(f"[PromptAssistant] 保存配置文件失败: {str(e)}")
|
|
return False
|
|
|
|
def get_baidu_translate_config(self):
|
|
"""获取百度翻译配置"""
|
|
config = self.load_config()
|
|
return config.get("baidu_translate", self.default_config["baidu_translate"])
|
|
|
|
def get_llm_config(self):
|
|
"""获取LLM配置"""
|
|
config = self.load_config()
|
|
return config.get("llm", self.default_config["llm"])
|
|
|
|
def update_baidu_translate_config(self, app_id, secret_key):
|
|
"""更新百度翻译配置"""
|
|
config = self.load_config()
|
|
config["baidu_translate"] = {
|
|
"app_id": app_id,
|
|
"secret_key": secret_key
|
|
}
|
|
return self.save_config(config)
|
|
|
|
def update_llm_config(self, api_key):
|
|
"""更新LLM配置"""
|
|
config = self.load_config()
|
|
config["llm"] = {
|
|
"api_key": api_key
|
|
}
|
|
return self.save_config(config)
|
|
|
|
# 创建全局配置管理器实例
|
|
config_manager = ConfigManager() |