fixing bridge transport enumerate + init

This commit is contained in:
Karel Bilek 2015-12-21 18:18:06 +01:00
parent ca6f4c8ce6
commit f149e705f4
1 changed files with 22 additions and 11 deletions

View File

@ -15,32 +15,43 @@ def get_error(resp):
class BridgeTransport(Transport):
def __init__(self, device, *args, **kwargs):
self.configure()
self.path = device['path']
self.session = None
self.response = None
self.conn = requests.Session();
super(BridgeTransport, self).__init__(device, *args, **kwargs)
@staticmethod
def configure():
r = requests.get(CONFIG_URL)
if r.status_code != 200:
raise Exception('Could not fetch config from %s' % CONFIG_URL)
config = r.text
self.conn = requests.Session();
r = self.conn.post(TREZORD_HOST + '/configure', data=config)
r = requests.post(TREZORD_HOST + '/configure', data=config)
if r.status_code != 200:
raise Exception('trezord: Could not configure' + get_error(r))
r = self.conn.get(TREZORD_HOST + '/enumerate')
@classmethod
def enumerate(cls):
"""
Return a list of available TREZOR devices.
"""
devices = {}
cls.configure()
r = requests.get(TREZORD_HOST + '/enumerate')
if r.status_code != 200:
raise Exception('trezord: Could not enumerate devices' + get_error(r))
enum = r.json()
if len(enum) < 1:
raise Exception('trezord: No devices found')
return enum;
self.path = enum[0]['path']
self.session = None
self.response = None
super(BridgeTransport, self).__init__(device, *args, **kwargs)
def _open(self):
r = self.conn.post(TREZORD_HOST + '/acquire/%s' % self.path)