yangxiaohui 1 week ago
parent
commit
ae2898ba50
1 changed files with 37 additions and 0 deletions
  1. 37 0
      script/utils.py

+ 37 - 0
script/utils.py

@@ -0,0 +1,37 @@
+#!/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)