Utils

sereto.cli.utils

AliasedGroup

Bases: Group

A click Group subclass that allows for writing aliases and prefixes of any command.

Source code in sereto/cli/utils.py

class AliasedGroup(click.Group):
    """A click Group subclass that allows for writing aliases and prefixes of any command."""

    def get_command(self, ctx: click.core.Context, cmd_name: str) -> click.Command | None:
        """Retrieves the command with the given name.

        If the command is not found, it looks up an explicit command alias or a command prefix.

        Args:
            ctx: The click context object.
            cmd_name: The name of the command to retrieve.

        Returns:
            The command with the given name, or None if no command is found.
        """
        # built-in commands
        rv = click.Group.get_command(self, ctx, cmd_name)
        if rv is not None:
            return rv

        # look up an explicit command alias
        if cmd_name in cli_aliases:
            actual_cmd = cli_aliases[cmd_name]
            return click.Group.get_command(self, ctx, actual_cmd)

        # look up a command prefix
        matches = [x for x in self.list_commands(ctx) if x.lower().startswith(cmd_name.lower())]
        if not matches:
            return None
        elif len(matches) == 1:
            return click.Group.get_command(self, ctx, matches[0])
        ctx.fail(f"Too many matches: {', '.join(sorted(matches))}")

    def resolve_command(
        self, ctx: click.core.Context, args: list[str]
    ) -> tuple[str | None, click.Command | None, list[str]]:
        """Resolves the full command's name."""
        _, cmd, args = super().resolve_command(ctx, args)
        if cmd is None:
            ctx.fail("No such command")
        return cmd.name, cmd, args

get_command(ctx, cmd_name)

Retrieves the command with the given name.

If the command is not found, it looks up an explicit command alias or a command prefix.

Parameters:

Name	Type	Description	Default
ctx	Context	The click context object.	required
cmd_name	str	The name of the command to retrieve.	required

Returns:

Type	Description
Command | None	The command with the given name, or None if no command is found.

Source code in sereto/cli/utils.py

def get_command(self, ctx: click.core.Context, cmd_name: str) -> click.Command | None:
    """Retrieves the command with the given name.

    If the command is not found, it looks up an explicit command alias or a command prefix.

    Args:
        ctx: The click context object.
        cmd_name: The name of the command to retrieve.

    Returns:
        The command with the given name, or None if no command is found.
    """
    # built-in commands
    rv = click.Group.get_command(self, ctx, cmd_name)
    if rv is not None:
        return rv

    # look up an explicit command alias
    if cmd_name in cli_aliases:
        actual_cmd = cli_aliases[cmd_name]
        return click.Group.get_command(self, ctx, actual_cmd)

    # look up a command prefix
    matches = [x for x in self.list_commands(ctx) if x.lower().startswith(cmd_name.lower())]
    if not matches:
        return None
    elif len(matches) == 1:
        return click.Group.get_command(self, ctx, matches[0])
    ctx.fail(f"Too many matches: {', '.join(sorted(matches))}")

resolve_command(ctx, args)

Resolves the full command's name.

Source code in sereto/cli/utils.py

def resolve_command(
    self, ctx: click.core.Context, args: list[str]
) -> tuple[str | None, click.Command | None, list[str]]:
    """Resolves the full command's name."""
    _, cmd, args = super().resolve_command(ctx, args)
    if cmd is None:
        ctx.fail("No such command")
    return cmd.name, cmd, args

Console

Bases: Console

Singleton wrapper around Rich's Console.

Source code in sereto/cli/utils.py

class Console(RichConsole, metaclass=Singleton):
    """Singleton wrapper around Rich's Console."""

load_enum(enum, prompt)

Let user select a value from enum.

Source code in sereto/cli/utils.py

def load_enum(enum: type[EnumType], prompt: str) -> EnumType:
    """Let user select a value from enum."""
    choice = Prompt.ask(prompt=prompt, choices=[e.value for e in enum])
    return enum(choice)