staking-deposit-cli/staking_deposit/deposit.py
2023-02-10 23:28:25 +08:00

64 lines
1.9 KiB
Python

import click
import sys
from staking_deposit.cli.existing_mnemonic import existing_mnemonic
from staking_deposit.cli.generate_bls_to_execution_change import generate_bls_to_execution_change
from staking_deposit.cli.new_mnemonic import new_mnemonic
from staking_deposit.utils.click import (
captive_prompt_callback,
choice_prompt_func,
jit_option,
)
from staking_deposit.utils import config
from staking_deposit.utils.constants import INTL_LANG_OPTIONS
from staking_deposit.utils.intl import (
get_first_options,
fuzzy_reverse_dict_lookup,
load_text,
)
def check_python_version() -> None:
'''
Checks that the python version running is sufficient and exits if not.
'''
if sys.version_info < (3, 7):
click.pause(load_text(['err_python_version']))
sys.exit()
@click.group()
@click.pass_context
@jit_option(
'--language',
callback=captive_prompt_callback(
lambda language: fuzzy_reverse_dict_lookup(language, INTL_LANG_OPTIONS),
choice_prompt_func(lambda: 'Please choose your language', get_first_options(INTL_LANG_OPTIONS)),
),
default='English',
help='The language you wish to use the CLI in.',
prompt=choice_prompt_func(lambda: 'Please choose your language', get_first_options(INTL_LANG_OPTIONS))(),
type=str,
)
@click.option(
'--non_interactive',
default=False,
is_flag=True,
help='Disables interactive prompts.',
hidden=True,
)
def cli(ctx: click.Context, language: str, non_interactive: bool) -> None:
config.language = language
config.non_interactive = non_interactive # Remove interactive commands
cli.add_command(existing_mnemonic)
cli.add_command(new_mnemonic)
cli.add_command(generate_bls_to_execution_change)
if __name__ == '__main__':
check_python_version()
print('\n***Using the tool on an offline and secure device is highly recommended to keep your mnemonic safe.***\n')
cli()