how/script/utils.py

#!/usr/bin/env python3
"""
script 工具函数模块
提供配置访问等通用功能
"""
import sys
from pathlib import Path

# 添加项目根目录到路径
project_root = Path(__file__).parent.parent
sys.path.insert(0, str(project_root))

from lib.config import get_data_dir

__all__ = ['get_default_results_dir']


def get_default_results_dir(subdir: str) -> str:
    """
    获取默认的结果输出目录

    Args:
        subdir: 子目录名称，如 'search', 'detail', 'tools_list' 等

    Returns:
        完整的结果目录路径

    Examples:
        >>> get_default_results_dir('search')
        'data/search'

        >>> # 如果设置了 DATA_ROOT 环境变量
        >>> # export DATA_ROOT=/custom/data
        >>> get_default_results_dir('search')
        '/custom/data/search'
    """
    return get_data_dir(subdir)