fixing bridge transport enumerate + init
This commit is contained in:
parent
ca6f4c8ce6
commit
f149e705f4
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user