Skip to content

Commit

Permalink
Merge pull request #78 from remyroy/commands-sorted
Browse files Browse the repository at this point in the history
Order click commands
  • Loading branch information
valefar-on-discord authored Jul 20, 2024
2 parents 49a7ac1 + be24468 commit c30adab
Showing 1 changed file with 20 additions and 6 deletions.
26 changes: 20 additions & 6 deletions ethstaker_deposit/deposit.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
import socket
import sys
from multiprocessing import freeze_support
from typing import List

from ethstaker_deposit.cli.existing_mnemonic import existing_mnemonic
from ethstaker_deposit.cli.exit_transaction_keystore import exit_transaction_keystore
Expand Down Expand Up @@ -44,7 +45,23 @@ def check_connectivity() -> None:
return None


@click.group()
# Define commands available to the user and their order
commands = [
new_mnemonic,
existing_mnemonic,
generate_bls_to_execution_change,
exit_transaction_keystore,
exit_transaction_mnemonic,
]


class SortedGroup(click.Group):

def list_commands(self, ctx: click.Context) -> List[str]:
return [x.name for x in commands]


@click.group(cls=SortedGroup)
@click.pass_context
@jit_option(
'--language',
Expand Down Expand Up @@ -85,11 +102,8 @@ def cli(ctx: click.Context, language: str, non_interactive: bool, ignore_connect
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)
cli.add_command(exit_transaction_keystore)
cli.add_command(exit_transaction_mnemonic)
for command in commands:
cli.add_command(command)


def run() -> None:
Expand Down

0 comments on commit c30adab

Please sign in to comment.