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()
|