307 lines
14 KiB
Python
307 lines
14 KiB
Python
# coding:utf-8
|
|
from core.sfgrid.model import TradeTarget
|
|
from core.eventbus import ActionDisableMarketData, ActionEnableMarketData, ActionEventAddTradeTarget, ActionEventDeleteTradeTarget, ActionEventDisableTrade, ActionEventEnableTrade, EventTradeTargetUpdate, MarketDataUpdate, MarketDataEnabled, MarketDataDisabled, ResultEventTradeDisabled, ResultEventTradeEnabled, ResultEventTradeTargetDeleted, ActionEventGridFix, event_bus
|
|
from xtquant.xttrader import XtQuantTrader
|
|
import time
|
|
from peewee import ModelSelect
|
|
|
|
import core.sfgrid.model as model
|
|
import sfgrid_config
|
|
from core.sfgrid.sfgrid_strategy import SFGridStrategy
|
|
from core.util import getInstrumentName, getStockPosition, queryPendingOrder
|
|
from xtquant.xttrader import XtQuantTrader
|
|
from xtquant.xttype import StockAccount, XtAsset, XtOrder, XtPosition, XtTrade
|
|
from xtquant import xtdata
|
|
from xtquant.xttrader import XtQuantTraderCallback
|
|
import datetime
|
|
import core.sfgrid.ui as ui
|
|
from core.logger import PrintLog, LogLevel
|
|
from core.sfgrid.objects import GridFixData
|
|
|
|
# 量化核心控制对象
|
|
class MainController(XtQuantTraderCallback):
|
|
def __init__(self, account_no: str, miniQmtPath: str):
|
|
super().__init__()
|
|
|
|
self.registerEventHandler()
|
|
# self.appUi = ui.TradeTargetUI()
|
|
|
|
xtdata.enable_hello = False
|
|
|
|
session_id = int(time.time())
|
|
|
|
self.xt_trader: XtQuantTrader = XtQuantTrader(miniQmtPath, session_id)
|
|
self.xt_trader.register_callback(self)
|
|
self.xt_trader.start()
|
|
self.xt_trader.connect()
|
|
PrintLog(LogLevel.INFO, f'- [{'成功' if self.xt_trader.connected else '失败'}]市场交易连接: {miniQmtPath}')
|
|
if self.xt_trader.connected == False:
|
|
self.inited: bool = False
|
|
return
|
|
else:
|
|
self.inited = True
|
|
|
|
self.account= StockAccount(account_no, 'STOCK')
|
|
PrintLog(LogLevel.INFO, f'- [成功]交易账号对象初始化完成, 账号: {self.account.account_id}') # pyright: ignore[reportAttributeAccessIssue]
|
|
subscribe_result = self.xt_trader.subscribe(self.account)
|
|
PrintLog(LogLevel.INFO, f'- [{'成功' if subscribe_result == 0 else '失败'}:{subscribe_result}]交易状态订阅')
|
|
if subscribe_result == 0:
|
|
self.inited = True
|
|
else:
|
|
self.inited = False
|
|
return
|
|
self.listening_stock = []
|
|
self.stock_trade_ctrl = {}
|
|
self.init_instrument_pool(self.xt_trader, self.account) # type: ignore
|
|
|
|
self.seq = None
|
|
PrintLog(LogLevel.INFO, '- [成功]三疯交易系统初始化完成')
|
|
self.startMarketData()
|
|
|
|
def registerEventHandler(self):
|
|
event_bus.subscribe(ActionEventEnableTrade, self.onEnableTrade)
|
|
event_bus.subscribe(ActionEventDisableTrade, self.onDisableTrade)
|
|
event_bus.subscribe(ActionEnableMarketData, self.onMarketDataEnabled)
|
|
event_bus.subscribe(ActionDisableMarketData, self.onMarketDataDisabled)
|
|
event_bus.subscribe(ActionEventAddTradeTarget, self.onAddTradeTarget)
|
|
event_bus.subscribe(ActionEventDeleteTradeTarget, self.onDeleteTradeTarget)
|
|
event_bus.subscribe(ActionEventGridFix, self.onGridFix)
|
|
|
|
def onDeleteTradeTarget(self, id: int):
|
|
"""处理删除交易标的事件"""
|
|
self.del_trade_target(id)
|
|
# 发布删除完成事件
|
|
event_bus.publish(ResultEventTradeTargetDeleted, id)
|
|
|
|
def onAddTradeTarget(self, stock_code: str):
|
|
"""处理添加交易标的事件"""
|
|
self.add_trade_target(stock_code)
|
|
|
|
def onMarketDataEnabled(self, data):
|
|
"""处理市场数据监听启用事件"""
|
|
self.startMarketData()
|
|
|
|
def onMarketDataDisabled(self, data):
|
|
"""处理市场数据监听禁用事件"""
|
|
self.stopMarketData()
|
|
|
|
def onEnableTrade(self, id: int):
|
|
self.start_stock_trade(id)
|
|
|
|
def onDisableTrade(self, id: int):
|
|
self.pause_stock_trade(id)
|
|
|
|
def onGridFix(self, data: GridFixData):
|
|
"""处理网格修正事件"""
|
|
self.update_trade_target_grid(data)
|
|
|
|
def update_trade_target_grid(self, data: GridFixData):
|
|
"""更新交易标的网格信息"""
|
|
try:
|
|
target = data.tradeTarget
|
|
grid_index = data.grid_index
|
|
|
|
# 更新数据库中的网格索引
|
|
target.grid_index = grid_index
|
|
target.save()
|
|
|
|
# 更新内存中的交易标的
|
|
if target.get_id() in self.instrument_pool:
|
|
self.instrument_pool[target.get_id()] = target
|
|
|
|
# 更新交易控制器中的网格信息
|
|
if target.stock_code in self.stock_trade_ctrl:
|
|
trade_controller: SFGridStrategy = self.stock_trade_ctrl[target.stock_code]
|
|
trade_controller.updateGridIndex(grid_index) # type: ignore
|
|
|
|
PrintLog(LogLevel.INFO, f"网格修正已应用: {target.stock_code} - {target.stock_name}, 网格索引: {grid_index}")
|
|
except Exception as e:
|
|
PrintLog(LogLevel.ERROR, f"网格修正更新失败: {str(e)}")
|
|
|
|
def startMarketData(self):
|
|
PrintLog(LogLevel.INFO, '- 启动市场数据订阅')
|
|
|
|
# self.seq = xtdata.subscribe_whole_quote(['SH', 'SZ'], callback=self.onDataUpdate)
|
|
if self.seq == -1:
|
|
PrintLog(LogLevel.ERROR, '- 市场数据订阅失败')
|
|
else:
|
|
event_bus.publish(MarketDataEnabled, True)
|
|
PrintLog(LogLevel.INFO, f'- 市场数据订阅成功, 订阅号={self.seq}')
|
|
|
|
|
|
|
|
def stopMarketData(self):
|
|
PrintLog(LogLevel.INFO, '- 停止市场数据订阅')
|
|
|
|
if self.seq is not None and self.seq > 0:
|
|
xtdata.unsubscribe_quote(self.seq)
|
|
event_bus.publish(MarketDataDisabled, False)
|
|
|
|
|
|
def add_trade_target(self, stock_code: str):
|
|
try:
|
|
stock_name = getInstrumentName(stock_code)
|
|
if not stock_name:
|
|
PrintLog(LogLevel.ERROR, f'无法获取股票代码 {stock_code} 的名称，请检查代码是否正确')
|
|
return
|
|
|
|
# 检查是否已存在该标的
|
|
existing_target = model.TradeTarget.get_or_none(model.TradeTarget.stock_code == stock_code)
|
|
if existing_target:
|
|
PrintLog(LogLevel.INFO, f'交易标的 {stock_code} {stock_name} 已存在')
|
|
return
|
|
|
|
new_target = model.TradeTarget.create(
|
|
stock_name=stock_name,
|
|
stock_code=stock_code,
|
|
market_price=0.0,
|
|
current_position=0,
|
|
grid_index=0,
|
|
last_trade_price=0.0,
|
|
plan_buy_price=0.0,
|
|
plan_sell_price=0.0,
|
|
current_order_price=0.0,
|
|
current_order_no='',
|
|
current_order_type=''
|
|
)
|
|
new_target.save()
|
|
PrintLog(LogLevel.INFO, f'新增交易标的 {stock_code} {stock_name}, {new_target.id}')
|
|
# 刷新标的持仓
|
|
pos = getStockPosition(stock_code, self.xt_trader, self.account) # type: ignore
|
|
model.TradeTarget.update(current_position=pos).where(model.TradeTarget.stock_code == stock_code).execute()
|
|
# 更新标的池
|
|
self.refresh_targets()
|
|
# 添加交易控制器
|
|
stockTradeController = SFGridStrategy(new_target, self.xt_trader, self.account) # type: ignore
|
|
self.stock_trade_ctrl[stock_code] = stockTradeController
|
|
|
|
except Exception as e:
|
|
PrintLog(LogLevel.ERROR, f'新增交易标的失败 {stock_code} {e}')
|
|
|
|
|
|
def del_trade_target(self, id:int):
|
|
try:
|
|
# 检查标的是否存在
|
|
if id not in self.instrument_pool:
|
|
PrintLog(LogLevel.ERROR, f"交易标的 ID {id} 不存在")
|
|
return
|
|
|
|
target: model.TradeTarget = self.instrument_pool[id]
|
|
|
|
# 如果存在交易控制器，先停止交易
|
|
if target.stock_code in self.stock_trade_ctrl:
|
|
# 停止交易控制器
|
|
del self.stock_trade_ctrl[target.stock_code]
|
|
|
|
# 从数据库中删除
|
|
target.delete_instance()
|
|
|
|
# 从内存中删除
|
|
del self.instrument_pool[id]
|
|
|
|
# 刷新标的池
|
|
self.refresh_targets()
|
|
|
|
PrintLog(LogLevel.INFO, f"已删除交易标的: {target.stock_code} - {target.stock_name}")
|
|
except Exception as e:
|
|
PrintLog(LogLevel.ERROR, f"删除交易标的失败 ID {id}: {str(e)}")
|
|
|
|
def init_instrument_pool(self, xtTrader:XtQuantTrader, account:StockAccount):
|
|
self.refresh_targets()
|
|
|
|
for id in self.instrument_pool:
|
|
target:TradeTarget = self.instrument_pool[id]
|
|
status = "新建" if target.status == 0 else "已建初始仓"
|
|
PrintLog(LogLevel.INFO, f' [序号-{id}] 股票代码: {target.stock_code}-{target.stock_name} 当前持仓: {getStockPosition(target.stock_code, self.xt_trader, self.account)} 网格索引: {target.grid_index} 基准价格 {sfgrid_config.grid_price[target.grid_index]} 状态: {status} 启用交易线程: {'自动交易中' if target.enabled else '交易已停止'}') # type: ignore
|
|
|
|
tradeTarget:model.TradeTarget = self.instrument_pool[id]
|
|
tradeTarget.current_position = getStockPosition(tradeTarget.stock_code, xtTrader, account) # type: ignore
|
|
result = tradeTarget.save()
|
|
PrintLog(LogLevel.INFO, f' |- 同步[{target.stock_code}-{target.stock_name}]持仓信息[{'成功' if result == 1 else '失败'}]')
|
|
stockTradeController = SFGridStrategy(tradeTarget, self.xt_trader, self.account) # type: ignore
|
|
self.stock_trade_ctrl[tradeTarget.stock_code] = stockTradeController
|
|
event_bus.publish(EventTradeTargetUpdate, tradeTarget)
|
|
|
|
PrintLog(LogLevel.INFO, f'- [成功]交易标的信息初始化, 共 {len(self.instrument_pool)} 个标的')
|
|
|
|
|
|
def refresh_targets(self):
|
|
# 更新标的池
|
|
results:ModelSelect = model.TradeTarget.select()
|
|
self.instrument_pool: dict[int, model.TradeTarget] = {}
|
|
for temp in results:
|
|
result :model.TradeTarget = temp
|
|
self.instrument_pool[result.get_id()] = result
|
|
|
|
def print_position_info(self):
|
|
positions:list[XtPosition] = self.xt_trader.query_stock_positions(self.account)
|
|
if positions:
|
|
PrintLog(LogLevel.INFO, "\n- 持仓信息")
|
|
for temp in positions:
|
|
pos : XtPosition = temp
|
|
if pos.volume <=0:
|
|
continue
|
|
PrintLog(LogLevel.INFO, f"股票代码: {pos.stock_code}-{getInstrumentName(pos.stock_code)}")
|
|
PrintLog(LogLevel.INFO, f"总持仓: {pos.volume}")
|
|
PrintLog(LogLevel.INFO, f"可用持仓: {pos.can_use_volume}")
|
|
PrintLog(LogLevel.INFO, f"持仓成本: {pos.avg_price}")
|
|
PrintLog(LogLevel.INFO, "---")
|
|
else:
|
|
PrintLog(LogLevel.INFO, "\n当前无持仓")
|
|
|
|
def print_account_info(self):
|
|
temp = self.xt_trader.query_stock_asset(self.account)
|
|
asset: XtAsset = temp # type: ignore
|
|
|
|
PrintLog(LogLevel.INFO, f"=== 账户信息 {self.account.account_id} ===") # type: ignore
|
|
PrintLog(LogLevel.INFO, f"可用资金: {asset.cash}")
|
|
PrintLog(LogLevel.INFO, f"总资产: {asset.total_asset}")
|
|
PrintLog(LogLevel.INFO, f"证券市值: {asset.market_value}")
|
|
|
|
def print_stock_orders(self):
|
|
orders = self.xt_trader.query_stock_orders(self.account, cancelable_only=True)
|
|
if orders:
|
|
PrintLog(LogLevel.INFO, "\n=== 委托信息 ===")
|
|
for order in orders:
|
|
PrintLog(LogLevel.INFO, f"委托编号: {order.order_id}")
|
|
PrintLog(LogLevel.INFO, f"股票代码: {order.stock_code} {getInstrumentName(order.stock_code)}")
|
|
PrintLog(LogLevel.INFO, f"委托方向: {order.offset_flag} ")
|
|
PrintLog(LogLevel.INFO, f"委托价格: {order.price}")
|
|
PrintLog(LogLevel.INFO, f"委托数量: {order.order_volume}")
|
|
PrintLog(LogLevel.INFO, f"已成交数量: {order.traded_volume}")
|
|
PrintLog(LogLevel.INFO, f"委托状态: {order.order_status} ")
|
|
PrintLog(LogLevel.INFO, "---")
|
|
else:
|
|
PrintLog(LogLevel.INFO, "\n当前无委托记录")
|
|
|
|
|
|
# 初始化指定标的交易控制器
|
|
def start_stock_trade(self, id: int):
|
|
tradeTarget: TradeTarget = self.instrument_pool[id]
|
|
# check existing thread
|
|
if tradeTarget.stock_code in self.stock_trade_ctrl:
|
|
tradeController: SFGridStrategy = self.stock_trade_ctrl[tradeTarget.stock_code]
|
|
|
|
tradeTarget = tradeController.enabledTrading(True)
|
|
self.instrument_pool[id] = tradeTarget
|
|
event_bus.publish(ResultEventTradeEnabled, tradeTarget)
|
|
else:
|
|
PrintLog(LogLevel.INFO, f"\t创建标的交易控制器 {tradeTarget.stock_code} {getInstrumentName(tradeTarget.stock_code)}")
|
|
|
|
|
|
def pause_stock_trade(self, id: int):
|
|
localTarget: model.TradeTarget = self.instrument_pool[id]
|
|
print(f'暂停标的交易 {localTarget.stock_code} - enabled {localTarget.enabled}')
|
|
if localTarget.stock_code in self.stock_trade_ctrl:
|
|
tradeController: SFGridStrategy = self.stock_trade_ctrl[localTarget.stock_code]
|
|
tradeTarget = tradeController.enabledTrading(False)
|
|
orders = queryPendingOrder(localTarget.stock_code, tradeController.getName(), self.xt_trader, self.account) # type: ignore
|
|
for order in orders:
|
|
self.xt_trader.cancel_order_stock_async(self.account, order.order_id)
|
|
print(f'取消未成交订单 {len(orders)}')
|
|
self.instrument_pool[id] = tradeTarget
|
|
event_bus.publish(ResultEventTradeDisabled, tradeTarget)
|
|
else:
|
|
print(f"标的交易控制器不存在 {localTarget.stock_code} {localTarget.stock_name}\n")
|
|
|