Polished binding to protobuf, msg.

emu.sh

@@ -1,4 +1,5 @@
 #!/bin/bash
 cd `dirname $0`/src
 
-../vendor/micropython/unix/micropython -i -O0 -X heapsize=100000 main.py
+rm -f ../pipe.*
+../vendor/micropython/unix/micropython -O0 -X heapsize=100000 main.py

src/playground/__init__.py

@@ -13,8 +13,8 @@ import gc
 from uasyncio import core
 # import transport_pipe as pipe
 
-from trezor import ui, io
-from trezor import msg2 as msg
+from trezor import ui
+from trezor import msg
 
 logging.basicConfig(level=logging.INFO)
 loop = core.get_event_loop()

src/trezor/__init__.py

@@ -1,5 +0,0 @@
-from TrezorMsg import Msg
-from TrezorProtobuf import Protobuf
-
-msg = Msg()
-protobuf = Protobuf()

src/trezor/msg.py

@@ -3,7 +3,7 @@ import sys
 if sys.platform == 'linux':
     import transport_pipe as pipe
 
-    def write(msg):
+    def send(msg):
         return pipe.write(msg)
         
     def read():
@@ -15,4 +15,14 @@ if sys.platform == 'linux':
     pipe.init('../pipe')
 
 else:
-    NotImplemented("HID transport")
+    from TrezorMsg import Msg
+
+    def send(msg):
+        return Msg.send(msg)
+
+    def read():
+        raise NotImplemented
+        return Msg.receive()
+
+    def set_notify(_on_read):
+        raise NotImplemented

src/trezor/protobuf.py

@@ -0,0 +1,9 @@
+from TrezorProtobuf import Protobuf
+
+_protobuf = Protobuf()
+
+def encode(data):
+    return _protobuf.encode(data)
+
+def decode(data):
+    return _protobuf.decode(data)