市场数据跟踪

core/sfgrid_strategy.py

@@ -2,7 +2,7 @@ from peewee import IntegerField
 
 
 from core import strategy_db
-from core.eventbus import MarketDataUpdate, event_bus
+from core.eventbus import EventTradeTargetUpdate, event_bus
 from core.strategy_db import OrderTypeBuy, OrderTypeInit, OrderTypeSell, TradeTarget
 from core.util import queryPendingOrder, is_trading_time
 
@@ -20,9 +20,14 @@ class SFGridStrategy:
         self.account:StockAccount = account
         self.enabledTrading(bool(tradeTarget.enabled))  # 修复类型兼容性问题
         
-        event_bus.publish(MarketDataUpdate, self.tradeTarget)
+        event_bus.publish(EventTradeTargetUpdate, self.tradeTarget)
         self.dataUpdateLock = threading.Lock()
 
+    def updateGridIndex(self, grid_index: int):
+        """更新网格索引"""
+        self.tradeTarget.grid_index = grid_index  # type: ignore
+        self.refreshPlanPrice()
+        self.saveProxy()
 
     def enabledTrading(self, enabled: bool) -> TradeTarget:
         self.tradeTarget.enabled = enabled  # type: ignore
@@ -216,5 +221,5 @@ class SFGridStrategy:
     
     def saveProxy(self):
         rc = self.tradeTarget.save()
-        event_bus.publish(MarketDataUpdate, self.tradeTarget)
+        event_bus.publish(EventTradeTargetUpdate, self.tradeTarget)
         return rc