mirror of
https://github.com/woodchen-ink/dnspod-yxip.git
synced 2025-07-18 05:42:08 +08:00
47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
import os
|
|
from typing import Dict, List
|
|
import yaml
|
|
from loguru import logger
|
|
|
|
# 加载YAML配置
|
|
def load_config() -> Dict:
|
|
"""从YAML文件加载配置"""
|
|
yaml_files = ["config.yaml", "config.example.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", [])
|