Improved pwp

src/__init__.py

@@ -49,5 +49,48 @@ class Strategy(PulseWire):
     def getWatchList(self):
         self.send(plugin.StrategyMessage.GetWatchList())
 
+    def getCandleStick(self, symbol: str, interval: general.CandleInterval, count: general.u32):
+        msg = plugin.StrategyMessage.RequestCandlestick()
+        msg.symbol = symbol
+        msg.interval = interval
+        msg.count = count
+        self.send(msg)
+
+    def unsubscribeAll(self):
+        self.send(plugin.StrategyMessage.UnsubscribeAll())
+
+    def signal(self, signal: plugin.StrategySignal):
+        msg = plugin.StrategyMessage.Signal()
+        msg.signal = signal
+        self.send(msg)
+
     def on_raw_engine_request(self, data: bytes):
-        com = plugin.StrategyEngineMessage.from_com(data=data)
+        self.on_request(plugin.StrategyEngineMessage.from_com(data=data)) #type: ignore
+
+class Risk(PulseWire):
+    def on_request(self, req: plugin.RiskEngineMessage):
+        pass
+
+    def send(self, req: plugin.RiskMessageType):
+        self.send_raw(req.to_com())
+
+    def log(self, log: general.EventLog):
+        msg = plugin.RiskMessage.Log()
+        msg.log = log
+        self.send(msg)
+
+    def getWatchList(self):
+        self.send(plugin.RiskMessage.GetWatchList())
+
+    def approveSignal(self, signal: general.Signal):
+        msg = plugin.RiskMessage.Approve()
+        msg.signal = signal
+        self.send(msg)
+
+    def rejectSignal(self, reason: str):
+        msg = plugin.RiskMessage.Reject()
+        msg.reason = reason
+        self.send(msg)
+
+    def on_raw_engine_request(self, data: bytes):
+        com = plugin.RiskEngineMessage.from_com(data=data)