123 lines
3.5 KiB
Python
123 lines
3.5 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
计算策略文件中 propertyGroup 块的数量
|
|
|
|
用于解决 hyperopt 多阶段执行时，日志显示的总阶段数硬编码问题
|
|
现在可以根据实际的 propertyGroup 块数量动态确定总数
|
|
"""
|
|
|
|
import re
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
|
|
def count_property_groups(strategy_file_path: str) -> int:
|
|
"""
|
|
从策略文件中计算 propertyGroup 块的数量
|
|
|
|
Args:
|
|
strategy_file_path: 策略文件的路径
|
|
|
|
Returns:
|
|
propertyGroup 块的数量
|
|
"""
|
|
try:
|
|
with open(strategy_file_path, 'r', encoding='utf-8') as f:
|
|
content = f.read()
|
|
except FileNotFoundError:
|
|
print(f"❌ 文件不存在: {strategy_file_path}", file=sys.stderr)
|
|
return 0
|
|
except Exception as e:
|
|
print(f"❌ 读取文件失败: {e}", file=sys.stderr)
|
|
return 0
|
|
|
|
# 查找所有 [propertiesGrp step=... 的匹配
|
|
# 使用 (?<=...) 正向后查断言来匹配 step 属性
|
|
pattern = r'\[\s*propertiesGrp\s+step\s*=\s*["\']?\d+["\']?'
|
|
matches = re.findall(pattern, content)
|
|
|
|
count = len(matches)
|
|
|
|
if count == 0:
|
|
# 尝试另一种模式，没有 step 属性
|
|
pattern_alt = r'\[\s*propertiesGrp\b'
|
|
matches_alt = re.findall(pattern_alt, content)
|
|
count = len(matches_alt)
|
|
|
|
return count
|
|
|
|
|
|
def get_property_groups_details(strategy_file_path: str) -> list:
|
|
"""
|
|
获取所有 propertyGroup 块的详细信息
|
|
|
|
Args:
|
|
strategy_file_path: 策略文件的路径
|
|
|
|
Returns:
|
|
包含每个 propertyGroup 信息的列表
|
|
"""
|
|
try:
|
|
with open(strategy_file_path, 'r', encoding='utf-8') as f:
|
|
content = f.read()
|
|
except Exception as e:
|
|
print(f"❌ 读取文件失败: {e}", file=sys.stderr)
|
|
return []
|
|
|
|
# 匹配完整的 propertiesGrp 标签，包括所有属性
|
|
pattern = r'\[\s*propertiesGrp\s+([^\]]+)\s*\]'
|
|
matches = re.findall(pattern, content)
|
|
|
|
groups = []
|
|
for match in matches:
|
|
group_info = {
|
|
'raw': match,
|
|
'attributes': {}
|
|
}
|
|
|
|
# 解析各个属性
|
|
attr_pattern = r'(\w+)\s*=\s*["\']?([^"\']*)["\']?'
|
|
attrs = re.findall(attr_pattern, match)
|
|
for attr_name, attr_value in attrs:
|
|
group_info['attributes'][attr_name] = attr_value
|
|
|
|
groups.append(group_info)
|
|
|
|
return groups
|
|
|
|
|
|
def main():
|
|
"""主函数"""
|
|
if len(sys.argv) < 2:
|
|
print("用法: python count_property_groups.py <策略文件路径> [--details]")
|
|
print()
|
|
print("示例:")
|
|
print(" python count_property_groups.py /path/to/freqaiprimer.py")
|
|
print(" python count_property_groups.py /path/to/freqaiprimer.py --details")
|
|
sys.exit(1)
|
|
|
|
strategy_file = sys.argv[1]
|
|
show_details = '--details' in sys.argv
|
|
|
|
count = count_property_groups(strategy_file)
|
|
|
|
if show_details:
|
|
groups = get_property_groups_details(strategy_file)
|
|
print(f"📊 propertyGroup 总数: {count}")
|
|
print()
|
|
for idx, group in enumerate(groups, 1):
|
|
print(f"[{idx}] propertyGroup:")
|
|
attrs = group['attributes']
|
|
if attrs:
|
|
for key, value in attrs.items():
|
|
print(f" {key}: {value}")
|
|
else:
|
|
print(f" {group['raw']}")
|
|
print()
|
|
else:
|
|
print(f"{count}")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|