arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-22 07:28:10 +00:00
Code Issues Releases Wiki Activity

python: add backup_type to reset device

Browse Source
This commit is contained in:
Tomas Susanka 2019-07-22 11:37:44 +02:00
parent 539379bd55
commit e663b7cb54
2 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
python/trezorctl
View File

@ -113,6 +113,14 @@ CHOICE_OUTPUT_SCRIPT_TYPE = ChoiceType(
}
)
CHOICE_RESET_DEVICE_TYPE = ChoiceType(
{
"single": proto.ResetDeviceBackupType.Bip39,
"shamir": proto.ResetDeviceBackupType.Slip39_Single_Group,
"advanced": proto.ResetDeviceBackupType.Slip39_Multiple_Groups,
}
)
class UnderscoreAgnosticGroup(click.Group):
"""Command group that normalizes dashes and underscores.
@ -505,7 +513,7 @@ def recovery_device(
@click.option("-u", "--u2f-counter", default=0)
@click.option("-s", "--skip-backup", is_flag=True)
@click.option("-n", "--no-backup", is_flag=True)
@click.option("-x", "--slip39", is_flag=True)
@click.option("-b", "--backup-type", type=CHOICE_RESET_DEVICE_TYPE, default="single")
@click.pass_obj
def reset_device(
connect,
@ -517,7 +525,7 @@ def reset_device(
u2f_counter,
skip_backup,
no_backup,
slip39,
backup_type,
):
if strength:
strength = int(strength)
@ -532,7 +540,7 @@ def reset_device(
u2f_counter=u2f_counter,
skip_backup=skip_backup,
no_backup=no_backup,
slip39=slip39,
backup_type=backup_type,
)

4
python/trezorlib/device.py
View File

@ -167,7 +167,7 @@ def reset(
u2f_counter=0,
skip_backup=False,
no_backup=False,
slip39=False,
backup_type=proto.ResetDeviceBackupType.Bip39,
):
if client.features.initialized:
raise RuntimeError(
@ -191,7 +191,7 @@ def reset(
u2f_counter=u2f_counter,
skip_backup=bool(skip_backup),
no_backup=bool(no_backup),
slip39=bool(slip39),
backup_type=backup_type,
)
resp = client.call(msg)