Files
amodem/libagent/device/keepkey_defs.py
Eric Siegel (Rocket Hazmat) 7bdfa7609d Upgrade KeepKey for new libagent code
Add get_public_node for KeepKey
2020-03-13 13:50:09 -04:00

25 lines
809 B
Python

"""KeepKey-related definitions."""
# pylint: disable=unused-import,import-error
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
get_public_node = Client.get_public_node
sign_identity = Client.sign_identity
Client.state = None
def find_device():
"""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)