76 lines
2.4 KiB
Python
76 lines
2.4 KiB
Python
import sys
|
||
from pathlib import Path
|
||
|
||
|
||
def get_project_root():
|
||
"""
|
||
获取项目根路径（strategy_manager 的父目录）
|
||
项目结构:
|
||
project/
|
||
├── futures_trading_strategies/
|
||
└── strategy_manager/
|
||
"""
|
||
# strategy_manager/core/path_utils.py -> strategy_manager -> project
|
||
return Path(__file__).parent.parent.parent
|
||
|
||
|
||
def get_strategy_root(config_path: str = "config/main.json"):
|
||
"""
|
||
动态获取策略代码根路径
|
||
优先级：
|
||
1. config/main.json 中配置的绝对路径
|
||
2. config/main.json 中配置的相对路径（相对于项目根）
|
||
3. AUTO_DETECT: 自动探测项目根目录下的 futures_trading_strategies
|
||
"""
|
||
config_file = Path(config_path)
|
||
strategy_root = None
|
||
|
||
# 读取配置
|
||
if config_file.exists():
|
||
import json
|
||
try:
|
||
with open(config_file, 'r') as f:
|
||
config = json.load(f)
|
||
strategy_root = config.get("strategy_root", "AUTO_DETECT")
|
||
except:
|
||
strategy_root = "AUTO_DETECT"
|
||
else:
|
||
strategy_root = "AUTO_DETECT"
|
||
|
||
# 如果是绝对路径，直接返回
|
||
if strategy_root and Path(strategy_root).is_absolute():
|
||
return Path(strategy_root)
|
||
|
||
# 如果是相对路径，相对于项目根
|
||
project_root = get_project_root()
|
||
if strategy_root and strategy_root != "AUTO_DETECT":
|
||
return project_root / strategy_root
|
||
|
||
# AUTO_DETECT模式: 探测项目根目录下的 futures_trading_strategies
|
||
auto_detect = project_root / "futures_trading_strategies"
|
||
if auto_detect.exists():
|
||
return auto_detect
|
||
|
||
# 如果失败，抛出错误
|
||
raise RuntimeError(
|
||
"无法自动探测策略代码路径，请在 config/main.json 中配置:\n"
|
||
'"strategy_root": "/path/to/your/futures_trading_strategies"'
|
||
)
|
||
|
||
|
||
def add_project_root_to_path():
|
||
"""将项目根路径添加到 sys.path"""
|
||
project_root = get_project_root()
|
||
if str(project_root) not in sys.path:
|
||
sys.path.insert(0, str(project_root))
|
||
print(f"[INFO] 已添加项目根路径到sys.path: {project_root}")
|
||
return project_root
|
||
|
||
|
||
def add_strategy_root_to_path():
|
||
"""将策略根路径添加到 sys.path"""
|
||
strategy_root = get_strategy_root()
|
||
if str(strategy_root) not in sys.path:
|
||
sys.path.insert(0, str(strategy_root))
|
||
print(f"[INFO] 已添加策略根路径到sys.path: {strategy_root}")
|
||
return strategy_root |