Fix detecting KeepKey USB device

The new KeepKey firmware uses WebUSB instead of HID
This commit is contained in:
Eric Siegel (Rocket Hazmat)
2020-03-13 13:05:08 -04:00
committed by GitHub
parent 15b0218bf2
commit 53b08f4968

View File

@@ -6,9 +6,16 @@ from keepkeylib.client import CallException, PinException
from keepkeylib.client import KeepKeyClient as Client
from keepkeylib.messages_pb2 import PassphraseAck, PinMatrixAck
from keepkeylib.transport_hid import HidTransport
from keepkeylib.transport_webusb import WebUsbTransport
from keepkeylib.types_pb2 import IdentityType
def find_device():
"""Returns first USB HID transport."""
return next(HidTransport(p) for p in HidTransport.enumerate())
"""Returns first WebUSB or HID transport."""
webusb = WebUsbTransport.enumerate()
hidusb = HidTransport.enumerate()
if len(webusb):
return next(WebUsbTransport(p) for p in webusb)
elif len(hidusb):
return next(HidTransport(p) for p in hidusb)