96 lines
3.6 KiB
Python
96 lines
3.6 KiB
Python
|
import os
|
|||
|
|
import json
|
|||
|
|
from typing import Dict, Any
|
|||
|
|
|
|||
|
|
class ConfigManager:
|
|||
|
|
"""配置管理器，支持配置的保存和验证"""
|
|||
|
|
|
|||
|
|
def __init__(self, config_file: str = None):
|
|||
|
|
if config_file is None:
|
|||
|
|
# 将配置文件存储在用户主目录下的.autoterminal目录中
|
|||
|
|
home_dir = os.path.expanduser("~")
|
|||
|
|
config_dir = os.path.join(home_dir, ".autoterminal")
|
|||
|
|
# 确保目录存在
|
|||
|
|
os.makedirs(config_dir, exist_ok=True)
|
|||
|
|
self.config_file = os.path.join(config_dir, "config.json")
|
|||
|
|
else:
|
|||
|
|
self.config_file = config_file
|
|||
|
|
|
|||
|
|
self.required_keys = ['api_key', 'base_url', 'model']
|
|||
|
|
self.default_config = {
|
|||
|
|
'base_url': 'https://api.openai.com/v1',
|
|||
|
|
'model': 'gpt-4o',
|
|||
|
|
'default_prompt': '你现在是一个终端助手,用户输入想要生成的命令,你来输出一个命令,不要任何多余的文本!'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
def save_config(self, config: Dict[str, Any]) -> bool:
|
|||
|
|
"""保存配置到文件"""
|
|||
|
|
try:
|
|||
|
|
# 确保目录存在
|
|||
|
|
os.makedirs(os.path.dirname(self.config_file) if os.path.dirname(self.config_file) else '.', exist_ok=True)
|
|||
|
|
|
|||
|
|
with open(self.config_file, 'w', encoding='utf-8') as f:
|
|||
|
|
json.dump(config, f, indent=2, ensure_ascii=False)
|
|||
|
|
return True
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"错误: 无法保存配置文件 {self.config_file}: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def validate_config(self, config: Dict[str, Any]) -> bool:
|
|||
|
|
"""验证配置是否完整"""
|
|||
|
|
for key in self.required_keys:
|
|||
|
|
if not config.get(key):
|
|||
|
|
return False
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
def initialize_config(self) -> Dict[str, Any]:
|
|||
|
|
"""初始化配置向导"""
|
|||
|
|
print("欢迎使用AutoTerminal配置向导！")
|
|||
|
|
print("请提供以下信息以完成配置：")
|
|||
|
|
|
|||
|
|
config = self.default_config.copy()
|
|||
|
|
|
|||
|
|
# 获取API密钥
|
|||
|
|
api_key = input("请输入您的API密钥: ").strip()
|
|||
|
|
if not api_key:
|
|||
|
|
print("错误: API密钥不能为空")
|
|||
|
|
return {}
|
|||
|
|
config['api_key'] = api_key
|
|||
|
|
|
|||
|
|
# 获取Base URL
|
|||
|
|
base_url = input(f"请输入Base URL (默认: {self.default_config['base_url']}): ").strip()
|
|||
|
|
if base_url:
|
|||
|
|
config['base_url'] = base_url
|
|||
|
|
|
|||
|
|
# 获取模型名称
|
|||
|
|
model = input(f"请输入模型名称 (默认: {self.default_config['model']}): ").strip()
|
|||
|
|
if model:
|
|||
|
|
config['model'] = model
|
|||
|
|
|
|||
|
|
# 保存配置
|
|||
|
|
if self.save_config(config):
|
|||
|
|
print(f"配置已保存到 {self.config_file}")
|
|||
|
|
return config
|
|||
|
|
else:
|
|||
|
|
print("配置保存失败")
|
|||
|
|
return {}
|
|||
|
|
|
|||
|
|
def get_or_create_config(self) -> Dict[str, Any]:
|
|||
|
|
"""获取现有配置或创建新配置"""
|
|||
|
|
# 尝试从文件加载配置
|
|||
|
|
if os.path.exists(self.config_file):
|
|||
|
|
try:
|
|||
|
|
with open(self.config_file, 'r', encoding='utf-8') as f:
|
|||
|
|
config = json.load(f)
|
|||
|
|
# 验证配置
|
|||
|
|
if self.validate_config(config):
|
|||
|
|
print("已加载现有配置")
|
|||
|
|
return config
|
|||
|
|
else:
|
|||
|
|
print("现有配置不完整")
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"警告: 无法读取配置文件 {self.config_file}: {e}")
|
|||
|
|
|
|||
|
|
# 如果配置不存在或不完整，启动初始化向导
|
|||
|
|
return self.initialize_config()
|