dnspod-yxip/config.py

import os
from typing import Dict, List
import yaml
from loguru import logger

# 加载YAML配置
def load_config() -> Dict:
    """从YAML文件加载配置"""
    yaml_files = ["config.yaml"]
    for yaml_file in yaml_files:
        if os.path.exists(yaml_file):
            try:
                with open(yaml_file, "r", encoding="utf-8") as f:
                    config = yaml.safe_load(f)
                    logger.info(f"成功加载配置文件: {yaml_file}")
                    # 验证必要的配置项
                    if not config.get("tencent", {}).get("secret_id"):
                        logger.error(f"配置文件 {yaml_file} 中缺少必要的配置项: tencent.secret_id")
                        return {}
                    if not config.get("tencent", {}).get("secret_key"):
                        logger.error(f"配置文件 {yaml_file} 中缺少必要的配置项: tencent.secret_key")
                        return {}
                    return config
            except Exception as e:
                logger.error(f"加载配置文件 {yaml_file} 失败: {str(e)}")
    logger.error("未找到有效的配置文件")
    return {}

# 加载配置
config_data = load_config()

# 腾讯云API配置
SECRET_ID = config_data.get("tencent", {}).get("secret_id")
SECRET_KEY = config_data.get("tencent", {}).get("secret_key")

# API接口配置
API_URL = "https://api.vvhan.com/tool/cf_ip"

# 日志级别
LOG_LEVEL = config_data.get("log_level", "INFO")

# 更新检查间隔（分钟）
check_interval = config_data.get("check_interval", 15)

# 获取所有域名配置
DOMAINS = config_data.get("domains", [])