lili/amodem
1
0
Fork 0
mirror of https://github.com/romanz/amodem.git synced 2026-04-18 20:25:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

trezor: better logging and exception handling

Browse Source
This commit is contained in:
Roman Zeyde
2015-06-15 09:50:51 +03:00
parent 7617b08b88
commit 3e1af2e2b2
1 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
trezor.py
View File

@@ -1,6 +1,7 @@
import io import io
import base64 import base64
import logging import logging
import binascii
from trezorlib.client import TrezorClient from trezorlib.client import TrezorClient
from trezorlib.transport_hid import HidTransport from trezorlib.transport_hid import HidTransport
@@ -59,9 +60,18 @@ class Client(object):
proto = 'ssh' proto = 'ssh'
def __init__(self): def __init__(self):
device, = HidTransport.enumerate() devices = HidTransport.enumerate()
client = TrezorClient(HidTransport(device)) if len(devices) != 1:
log.debug('connected to Trezor #%s', client.get_device_id()) raise ValueError('{:d} Trezor devices found'.format(len(devices)))
client = TrezorClient(HidTransport(devices[0]))
f = client.features
log.info('connected to Trezor')
log.debug('ID : {}'.format(f.device_id))
log.debug('label : {}'.format(f.label))
log.debug('vendor : {}'.format(f.vendor))
version = [f.major_version, f.minor_version, f.patch_version]
log.debug('version : {}'.format('.'.join(map(str, version))))
log.debug('revision : {}'.format(binascii.hexlify(f.revision)))
self.client = client self.client = client
def close(self): def close(self):