diff --git a/trading_system/main.py b/trading_system/main.py index 77ebae8..1e0b1bd 100644 --- a/trading_system/main.py +++ b/trading_system/main.py @@ -6,9 +6,10 @@ import logging import sys from pathlib import Path -# 支持直接运行和作为模块导入 -if __name__ == '__main__': - # 直接运行时,使用相对导入 +# 启动方式兼容: +# - python trading_system/main.py(__package__ 为空,需从同目录导入) +# - python -m trading_system.main(__package__='trading_system',必须用相对导入) +if __package__ in (None, ""): from binance_client import BinanceClient from market_scanner import MarketScanner from risk_manager import RiskManager @@ -16,7 +17,6 @@ if __name__ == '__main__': from strategy import TradingStrategy import config else: - # 作为模块导入时,使用绝对导入 from .binance_client import BinanceClient from .market_scanner import MarketScanner from .risk_manager import RiskManager diff --git a/trading_system/recommendations_main.py b/trading_system/recommendations_main.py index e9c3936..49323ac 100644 --- a/trading_system/recommendations_main.py +++ b/trading_system/recommendations_main.py @@ -14,8 +14,10 @@ from pathlib import Path from datetime import datetime, timezone, timedelta -# 支持直接运行和作为模块导入 -if __name__ == "__main__": +# 启动方式兼容: +# - python trading_system/recommendations_main.py(__package__ 为空,需从同目录导入) +# - python -m trading_system.recommendations_main(__package__='trading_system',必须用相对导入) +if __package__ in (None, ""): from binance_client import BinanceClient from market_scanner import MarketScanner from risk_manager import RiskManager