utils/common.py

"""
通用工具函数
提供项目中常用的基础功能
"""

import os
from pathlib import Path

def clean_env_value(value):
    """
    清理环境变量值，去除首尾空白字符

    Args:
        value: 环境变量的原始值

    Returns:
        str or None: 清理后的值，如果为空或None则返回None
    """
    if value is None:
        return None
    stripped = value.strip()
    return stripped or None


def parse_headless_mode(headless_setting):
    """
    解析headless模式配置

    Args:
        headless_setting: headless配置值

    Returns:
        bool or str: True表示headless，False表示有界面，'virtual'表示虚拟模式
    """
    if str(headless_setting).lower() == 'true':
        return True
    elif str(headless_setting).lower() == 'false':
        return False
    else:
        return 'virtual'


def ensure_dir(path):
    """
    确保目录存在，如果不存在则创建

    Args:
        path: 目录路径（可以是字符串或Path对象）
    """
    if isinstance(path, str):
        path = Path(path)
    os.makedirs(path, exist_ok=True)