add query_order to cta_engine & tv_engine

dyx52 · Jul 12, 2022 · cf7c771 · cf7c771
1 parent 90dfd8b
commit cf7c771
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 0 deletions.
diff --git a/howtrader/app/cta_strategy/engine.py b/howtrader/app/cta_strategy/engine.py
@@ -17,6 +17,7 @@
 from howtrader.trader.engine import BaseEngine, MainEngine
 from howtrader.trader.object import (
     OrderRequest,
+    OrderQueryRequest,
     SubscribeRequest,
     HistoryRequest,
     CancelRequest,
@@ -526,6 +527,12 @@ def get_pricetick(self, strategy: CtaTemplate) -> Optional[Decimal]:
         else:
             return None
 
+    def query_order(self, vt_orderid: str) -> None:
+        order: Optional[OrderData] = self.main_engine.get_order(vt_orderid)
+        if order:
+            req: OrderQueryRequest = order.create_query_request()
+            self.main_engine.query_order(req, order.gateway_name)
+
     def load_bar(
         self,
         vt_symbol: str,

diff --git a/howtrader/app/cta_strategy/template.py b/howtrader/app/cta_strategy/template.py
@@ -259,6 +259,10 @@ def cancel_order(self, vt_orderid: str) -> None:
         if self.trading:
             self.cta_engine.cancel_order(self, vt_orderid)
 
+    def query_order(self, vt_orderid: str) -> None:
+        """query order"""
+        self.cta_engine.query_order(vt_orderid)
+
     def cancel_all(self) -> None:
         """
         Cancel all orders sent by strategy.

diff --git a/howtrader/app/tradingview/engine.py b/howtrader/app/tradingview/engine.py
@@ -15,6 +15,7 @@
 from howtrader.trader.engine import BaseEngine, MainEngine
 from howtrader.trader.object import (
     OrderRequest,
+    OrderQueryRequest,
     SubscribeRequest,
     CancelRequest,
     LogData,
@@ -247,6 +248,12 @@ def cancel_order(self, strategy: TVTemplate, vt_orderid: str) -> None:
         req: CancelRequest = order.create_cancel_request()
         self.main_engine.cancel_order(req, order.gateway_name)
 
+    def query_order(self, vt_orderid: str) -> None:
+        order: Optional[OrderData] = self.main_engine.get_order(vt_orderid)
+        if order:
+            req: OrderQueryRequest = order.create_query_request()
+            self.main_engine.query_order(req, order.gateway_name)
+
     def cancel_all(self, strategy: TVTemplate) -> None:
         """
         Cancel all active orders of a strategy.

diff --git a/howtrader/app/tradingview/template.py b/howtrader/app/tradingview/template.py
@@ -224,6 +224,9 @@ def cancel_order(self, vt_orderid: str) -> None:
         if self.trading:
             self.tv_engine.cancel_order(self, vt_orderid)
 
+    def query_order(self, vt_orderid: str) -> None:
+        self.tv_engine.query_order(vt_orderid)
+
     def cancel_all(self) -> None:
         """
         Cancel all orders sent by strategy.