feat(qmt): 新增 Pydantic 配置模型并重构引擎架构
- 新增 config_models.py: 使用 Pydantic 提供强类型配置校验 - QMTConfig, QMTTerminalConfig, StrategyConfig 等数据模型 - 支持 slots/percentage 两种下单模式 - 兼容旧版配置格式迁移 - 新增 validate_config.py: 配置检测 CLI 工具 - 重构 TradingUnit 和 MultiEngineManager 使用新配置模型 - 新增百分比模式买卖逻辑 (_execute_percentage_buy/sell) - 完善日志记录和错误处理 - 删除 TODO_FIX.md: 清理已完成的缺陷修复任务清单
This commit is contained in:
95
qmt/main.py
95
qmt/main.py
@@ -6,6 +6,101 @@ import logging
|
||||
import datetime
|
||||
import uvicorn
|
||||
|
||||
from .qmt_engine import QMTEngine, ConfigError
|
||||
from .api_server import create_api_server
|
||||
|
||||
|
||||
def setup_logger():
|
||||
"""配置日志系统"""
|
||||
log_dir = "logs"
|
||||
if not os.path.exists(log_dir):
|
||||
os.makedirs(log_dir)
|
||||
|
||||
log_file = os.path.join(log_dir, f"{datetime.date.today().strftime('%Y-%m-%d')}.log")
|
||||
|
||||
logger = logging.getLogger("QMT_Main")
|
||||
logger.setLevel(logging.INFO)
|
||||
|
||||
formatter = logging.Formatter(
|
||||
'[%(asctime)s] [%(levelname)s] [%(filename)s:%(lineno)d] %(message)s',
|
||||
datefmt='%Y-%m-%d %H:%M:%S'
|
||||
)
|
||||
|
||||
file_handler = logging.FileHandler(log_file, mode='a', encoding='utf-8')
|
||||
file_handler.setFormatter(formatter)
|
||||
|
||||
stream_handler = logging.StreamHandler(sys.stdout)
|
||||
stream_handler.setFormatter(formatter)
|
||||
|
||||
logger.addHandler(file_handler)
|
||||
logger.addHandler(stream_handler)
|
||||
|
||||
return logger
|
||||
|
||||
|
||||
def main():
|
||||
"""主函数 - 启动QMT交易引擎和API服务器"""
|
||||
logger = setup_logger()
|
||||
logger.info("="*50)
|
||||
logger.info(">>> QMT交易系统启动中...")
|
||||
logger.info("="*50)
|
||||
|
||||
# 创建QMT引擎实例
|
||||
engine = QMTEngine()
|
||||
logger.info("QMT引擎实例创建成功")
|
||||
|
||||
try:
|
||||
# 初始化引擎
|
||||
engine.initialize('config.json')
|
||||
logger.info("✅ QMT引擎初始化成功")
|
||||
except ConfigError as e:
|
||||
logger.error(f"❌ 配置校验失败: {str(e)}")
|
||||
logger.error("请检查 config.json 配置文件")
|
||||
sys.exit(1)
|
||||
except Exception as e:
|
||||
logger.error(f"❌ QMT引擎初始化失败: {str(e)}", exc_info=True)
|
||||
sys.exit(1)
|
||||
|
||||
# 启动交易线程
|
||||
trading_thread = threading.Thread(target=engine.run_trading_loop, daemon=True)
|
||||
trading_thread.start()
|
||||
logger.info("✅ 交易线程启动成功")
|
||||
|
||||
# 创建API服务器
|
||||
app = create_api_server(engine)
|
||||
logger.info("✅ API服务器创建成功")
|
||||
|
||||
# 启动Web服务
|
||||
logger.info(">>> Web服务启动: http://localhost:8001")
|
||||
try:
|
||||
uvicorn.run(
|
||||
app,
|
||||
host="0.0.0.0",
|
||||
port=8001,
|
||||
log_level="warning",
|
||||
access_log=False
|
||||
)
|
||||
except KeyboardInterrupt:
|
||||
logger.info(">>> 正在关闭系统...")
|
||||
engine.stop()
|
||||
logger.info(">>> 系统已关闭")
|
||||
except Exception as e:
|
||||
logger.error(f">>> 系统异常: {str(e)}", exc_info=True)
|
||||
engine.stop()
|
||||
logger.info(">>> 系统已关闭")
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
# 使用 -u 参数运行是最佳实践: python -u main.py
|
||||
# 但这里也在代码里强制 flush 了
|
||||
main()
|
||||
import threading
|
||||
import sys
|
||||
import os
|
||||
import logging
|
||||
import datetime
|
||||
import uvicorn
|
||||
|
||||
from .qmt_engine import QMTEngine
|
||||
from .api_server import create_api_server
|
||||
|
||||
|
||||
Reference in New Issue
Block a user