Fixed paths and imports

bitkeylib/client.py

@@ -1,6 +1,6 @@
 import os
 
-from bitkey_proto import bitkey_pb2 as proto
+import bitkey_pb2 as proto
 from transport_pipe import PipeTransport
 
 def show_message(message):

bitkeylib/transport.py

@@ -1,6 +1,6 @@
 import struct
-from bitkey_proto import bitkey_pb2 as proto
-from bitkey_proto import mapping
+import bitkey_pb2 as proto
+import mapping
 
 class Transport(object):
     def __init__(self, device, *args, **kwargs):
@@ -59,4 +59,4 @@ class Transport(object):
             except:
                 raise Exception("Cannot parse header length")
        
-        return (msg_type, datalen)
+        return (msg_type, datalen)

tests/test.py

@@ -1,14 +1,16 @@
 #!/usr/bin/python
+import sys
+sys.path = ['../',] + sys.path
 
 import time
 
-from transport_pipe import PipeTransport
-from transport_serial import SerialTransport
-from bitkey_proto import bitkey_pb2 as proto
+from bitkeylib.transport_pipe import PipeTransport
+from bitkeylib.transport_serial import SerialTransport
+import bitkeylib.bitkey_pb2 as proto
 
-from client import BitkeyClient
+from bitkeylib.client import BitkeyClient
 
-bitkey = BitkeyClient('../bitkey-python/device.socket', debug=True)
+bitkey = BitkeyClient('../../bitkey-python/device.socket', debug=True)
 bitkey.open()
 bitkey.call(proto.Ping(message='ahoj!'))
 bitkey.call(proto.SetMaxFeeKb(maxfee_kb=200000))