Files
SupervisorAI/common/contants.py
2026-03-03 11:52:10 +08:00

31 lines
858 B
Python

import yaml
from utils.logger import get_logger
logger = get_logger(__name__)
ALERT_PUSH_URL = ""
HLS_ROOT_PATH = ""
HLS_SEGMENT_PATTERN = "segment_%09d.ts" # TS文件命名模式
def init_config(config_path: str = "config.yaml"):
"""
从配置文件初始化全局配置
Args:
config_path: 配置文件路径,默认为 config.yaml
"""
global ALERT_PUSH_URL, HLS_ROOT_PATH
try:
with open(config_path, "r", encoding="utf-8") as f:
cfg = yaml.safe_load(f)
ALERT_PUSH_URL = cfg.get("alert_push_url", "")
HLS_ROOT_PATH = cfg.get("hls_root_path", "")
logger.info(f"[INFO] Config initialized from {config_path}, alert_push_url={ALERT_PUSH_URL}")
except Exception as e:
logger.error(f"[ERROR] Failed to load config from {config_path}: {e}")