zhiyinya/AIStudioBuildWS

1"""2通用工具函数3提供项目中常用的基础功能4"""5 6import os7from pathlib import Path8 9def clean_env_value(value):10    """11    清理环境变量值，去除首尾空白字符12 13    Args:14        value: 环境变量的原始值15 16    Returns:17        str or None: 清理后的值，如果为空或None则返回None18    """19    if value is None:20        return None21    stripped = value.strip()22    return stripped or None23 24 25def parse_headless_mode(headless_setting):26    """27    解析headless模式配置28 29    Args:30        headless_setting: headless配置值31 32    Returns:33        bool or str: True表示headless，False表示有界面，'virtual'表示虚拟模式34    """35    if str(headless_setting).lower() == 'true':36        return True37    elif str(headless_setting).lower() == 'false':38        return False39    else:40        return 'virtual'41 42 43def ensure_dir(path):44    """45    确保目录存在，如果不存在则创建46 47    Args:48        path: 目录路径（可以是字符串或Path对象）49    """50    if isinstance(path, str):51        path = Path(path)52    os.makedirs(path, exist_ok=True)