arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-18 04:18:10 +00:00
Code Issues Releases Wiki Activity

fix(python/trezorctl): improve docs and behavior around unexpected messages

Browse Source 
closes #2254
This commit is contained in:
matejcik 2022-06-21 15:40:31 +02:00
parent 8c3d3c6548
commit c12e4edd70
1 changed files with 23 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
python/src/trezorlib/cli/device.py
View File

@ -53,7 +53,10 @@ def cli() -> None:
@cli.command() @cli.command()
@with_client @with_client
def self_test(client: "TrezorClient") -> str: def self_test(client: "TrezorClient") -> str:
"""Perform a self-test.""" """Perform a factory self-test.
Only available on PRODTEST firmware.
"""
return debuglink.self_test(client) return debuglink.self_test(client)
@ -126,6 +129,7 @@ def load(
if not label: if not label:
label = "SLIP-0014" label = "SLIP-0014"
try:
return debuglink.load_device( return debuglink.load_device(
client, client,
mnemonic=list(mnemonic), mnemonic=list(mnemonic),
@ -137,6 +141,13 @@ def load(
needs_backup=needs_backup, needs_backup=needs_backup,
no_backup=no_backup, no_backup=no_backup,
) )
except exceptions.TrezorFailure as e:
if e.code == messages.FailureType.UnexpectedMessage:
raise click.ClickException(
"Unrecognized message. Make sure your Trezor is using debug firmware."
)
else:
raise
@cli.command() @cli.command()