Refactored choosing of devices

pull/1/head
slush0 10 years ago
parent 461380e9d2
commit 430362bd31

@ -27,7 +27,7 @@ def wait_for_devices():
return devices return devices
def list_devices(devices): def choose_device(devices):
i = 0 i = 0
sys.stderr.write("----------------------------\n") sys.stderr.write("----------------------------\n")
sys.stderr.write("Available devices:\n") sys.stderr.write("Available devices:\n")
@ -52,23 +52,21 @@ def list_devices(devices):
try: try:
device_id = int(raw_input()) device_id = int(raw_input())
t = HidTransport(devices[device_id]) transport = HidTransport(devices[device_id])
t.close()
except: except:
raise Exception("Invalid choice, exiting...") raise Exception("Invalid choice, exiting...")
return device_id return transport
def main(): def main():
devices = wait_for_devices() devices = wait_for_devices()
if len(devices) > 1: if len(devices) > 1:
device_id = list_devices(devices) transport = choose_device(devices)
else: else:
device_id = 0 transport = HidTransport(devices[0])
transport = HidTransport(devices[device_id])
client = TrezorClient(transport) client = TrezorClient(transport)
rootdir = os.environ['encfs_root'] # Read "man encfs" for more rootdir = os.environ['encfs_root'] # Read "man encfs" for more

Loading…
Cancel
Save