| """ |
| 通用工具函数 |
| 提供项目中常用的基础功能 |
| """ |
|
|
| 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) |
