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

python: support SafetyChecks.PromptTemporarily

Browse Source
This commit is contained in:
Martin Milata 2020-09-03 16:53:29 +02:00 committed by Tomas Susanka
parent cff4955f93
commit 76a1760764
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
python/src/trezorlib/cli/settings.py
View File

@ -28,7 +28,7 @@ except ImportError:
ROTATION = {"north": 0, "east": 90, "south": 180, "west": 270} ROTATION = {"north": 0, "east": 90, "south": 180, "west": 270}
SAFETY_LEVELS = { SAFETY_LEVELS = {
"strict": messages.SafetyCheckLevel.Strict, "strict": messages.SafetyCheckLevel.Strict,
"prompt": messages.SafetyCheckLevel.Prompt, "prompt": messages.SafetyCheckLevel.PromptTemporarily,
} }
@ -190,18 +190,23 @@ def homescreen(client, filename):
@cli.command() @cli.command()
@click.option(
"--always", is_flag=True, help='Persist the "prompt" setting across Trezor reboots.'
)
@click.argument("level", type=ChoiceType(SAFETY_LEVELS)) @click.argument("level", type=ChoiceType(SAFETY_LEVELS))
@with_client @with_client
def safety_checks(client, level): def safety_checks(client, always, level):
"""Set safety check level. """Set safety check level.
Set to "strict" to get the full Trezor security. Set to "strict" to get the full Trezor security (default setting).
Set to "prompt" if you want to be able to allow potentially unsafe actions, such as Set to "prompt" if you want to be able to allow potentially unsafe actions, such as
mismatching coin keys or extreme fees. mismatching coin keys or extreme fees.
This is a power-user feature. Use with caution. This is a power-user feature. Use with caution.
""" """
if always and level == messages.SafetyCheckLevel.PromptTemporarily:
level = messages.SafetyCheckLevel.PromptAlways
return device.apply_settings(client, safety_checks=level) return device.apply_settings(client, safety_checks=level)