arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-02-17 01:52:02 +00:00
Code Issues Releases Wiki Activity

Introduce pin_input message to trigger PinMatrix widget

Browse Source
This commit is contained in:
slush0 2013-10-07 23:27:56 +02:00
parent 12955a0c0c
commit f7f04b0482
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
trezorlib/client.py
View File

@ -11,6 +11,9 @@ def show_input(input_text, message=None):
print "QUESTION FROM DEVICE:", message print "QUESTION FROM DEVICE:", message
return raw_input(input_text) return raw_input(input_text)
def pin_func(input_text, message=None):
return show_input(input_text, message)
class CallException(Exception): class CallException(Exception):
pass pass
@ -20,12 +23,13 @@ class PinException(CallException):
class TrezorClient(object): class TrezorClient(object):
def __init__(self, transport, debuglink=None, def __init__(self, transport, debuglink=None,
message_func=show_message, input_func=show_input, debug=False): message_func=show_message, input_func=show_input, pin_func=pin_func, debug=False):
self.transport = transport self.transport = transport
self.debuglink = debuglink self.debuglink = debuglink
self.message_func = message_func self.message_func = message_func
self.input_func = input_func self.input_func = input_func
self.pin_func = pin_func
self.debug = debug self.debug = debug
self.setup_debuglink() self.setup_debuglink()
@ -104,7 +108,7 @@ class TrezorClient(object):
else: else:
msg2 = proto.PinMatrixAck(pin='444444222222') msg2 = proto.PinMatrixAck(pin='444444222222')
else: else:
pin = self.input_func("PIN required: ", resp.message) pin = self.pin_func("PIN required: ", resp.message)
msg2 = proto.PinMatrixAck(pin=pin) msg2 = proto.PinMatrixAck(pin=pin)
return self.call(msg2) return self.call(msg2)