Improved typing

src/__init__.py

@@ -1,5 +1,7 @@
 import sys
 import struct
+import plugin
+import general
 
 class PulseWire:
     def __init__(self):
@@ -30,4 +32,29 @@ class PulseWire:
             if len(buffer) != length:
                 raise EOFError("Unexpected EOF while reading payload")
 
-            self.on_raw_engine_request(buffer)
+            self.on_raw_engine_request(buffer)
+
+class Strategy(PulseWire):
+    def on_request(self, req: plugin.StrategyEngineMessage):
+        pass
+
+    def send(self, req: plugin.StrategyMessage):
+        self.send_raw(req.to_com())
+
+    def log(self, log: general.EventLog):
+        msg = plugin.StrategyMessage.Log()
+        msg.log = log
+        self.send(msg)
+
+    def getWatchList(self):
+        self.send(plugin.StrategyMessage.GetWatchList())
+
+    def on_raw_engine_request(self, data: bytes):
+        com = plugin.StrategyEngineMessage.from_com(data)
+
+
+def test(req: plugin.StrategyMessage.type):
+    print(req, plugin.StrategyMessage.type)
+    pass
+
+test(plugin.StrategyMessage.GetWatchList())