first commit

utils/__init__.py

@@ -0,0 +1,9 @@
+# -*- coding: utf-8 -*-
+"""
+工具模块 - 包含各种工具函数
+"""
+
+from .file_utils import load_alpha_list, save_results_to_file
+from .time_utils import format_time
+
+__all__ = ['load_alpha_list', 'save_results_to_file', 'format_time']

utils/file_utils.py

@@ -0,0 +1,59 @@
+# -*- coding: utf-8 -*-
+import os
+import json
+import time
+from typing import List, Any
+
+
+def load_alpha_list(file_path: str) -> List[str]:
+    """从文件加载Alpha因子列表"""
+    if not os.path.exists(file_path):
+        print(f"{file_path} 文件不存在")
+        with open(file_path, 'w', encoding='utf-8') as file:
+            file.write("")
+        print(f"已创建 {file_path} 文件, 请添加因子后重新运行, 一行一个因子")
+        return []
+
+    with open(file_path, 'r', encoding='utf-8') as file:
+        alpha_list = [line.strip() for line in file if line.strip()]
+
+    return alpha_list
+
+
+def save_results_to_file(results: List[Any], result_dir: str = 'result') -> str:
+    """保存结果到文件"""
+    # 转换为可序列化的格式
+    serializable_results = []
+    for result in results:
+        if hasattr(result, '__dict__'):
+            # 如果是dataclass对象
+            result_dict = result.__dict__.copy()
+        else:
+            # 如果是字典
+            result_dict = result.copy()
+
+        # 处理时间消耗
+        if 'time_consuming' in result_dict:
+            result_dict['time_consuming'] = round(result_dict['time_consuming'], 2)
+
+        # 处理metrics对象
+        for key in list(result_dict.keys()):
+            if hasattr(result_dict[key], '__dict__'):
+                result_dict[key] = result_dict[key].__dict__
+                # 处理浮点数精度
+                for metric_key, value in result_dict[key].items():
+                    if isinstance(value, float):
+                        result_dict[key][metric_key] = round(value, 6)
+
+        serializable_results.append(result_dict)
+
+    # 确保结果目录存在
+    if not os.path.exists(result_dir):
+        os.makedirs(result_dir)
+
+    result_name = f"{result_dir}/simulation_results-{str(int(time.time()))}.json"
+    with open(result_name, 'w', encoding='utf-8') as f:
+        json.dump(serializable_results, f, ensure_ascii=False, indent=2)
+
+    print(f"结果已保存到 {result_name}")
+    return result_name

utils/time_utils.py

@@ -0,0 +1,9 @@
+# -*- coding: utf-8 -*-
+def format_time(seconds: float) -> str:
+    """将秒数格式化为 xx分xx秒 格式"""
+    if seconds < 60:
+        return f"{seconds:.2f}秒"
+    else:
+        minutes = int(seconds // 60)
+        remaining_seconds = seconds % 60
+        return f"{minutes}分{remaining_seconds:.2f}秒"