Source code for super_pocket.settings

import functools

import rich_click as click
from rich.panel import Panel
from rich.text import Text
from rich.spinner import Spinner
from rich.align import Align

from super_pocket.utils import console

click.rich_click.STYLE_OPTION = "bold blue"
click.rich_click.STYLE_ARGUMENT = "bold orange_red1"
click.rich_click.STYLE_COMMAND = "bold orange_red1"
click.rich_click.STYLE_SWITCH = "blue"
click.rich_click.STYLE_METAVAR = "dim orange_red1"
click.rich_click.STYLE_METAVAR_SEPARATOR = ""
click.rich_click.STYLE_USAGE = "bold grey100"
click.rich_click.STYLE_USAGE_COMMAND = "bold bright_blue"
click.rich_click.STYLE_HELPTEXT_FIRST_LINE = "bold"
click.rich_click.STYLE_HELPTEXT = "dim italic"
click.rich_click.STYLE_OPTION_DEFAULT = "blue"
click.rich_click.STYLE_REQUIRED_SHORT = "italic orange_red1"
click.rich_click.STYLE_REQUIRED_LONG = "orange_red1"
click.rich_click.STYLE_OPTIONS_PANEL_BORDER = "orange_red1"
click.rich_click.STYLE_COMMANDS_PANEL_BORDER = "blue"
click.rich_click.STYLE_OPTIONS_TABLE_BOX = "SIMPLE"


click.rich_click.STYLE_OPTIONS_PANEL_BOX = "SIMPLE_HEAD"

click.rich_click.STYLE_OPTIONS_PANEL_TITLE_STYLE = "bold grey100"

click.rich_click.STYLE_OPTIONS_TABLE_LEADING: int = 1
click.rich_click.STYLE_COMMANDS_PANEL_BOX = "SIMPLE_HEAD"
click.rich_click.STYLE_COMMANDS_PANEL_TITLE_STYLE = "bold grey100"
click.rich_click.STYLE_COMMANDS_TABLE_BOX = "SIMPLE"


# Context settings to enable -h in addition to --help
CONTEXT_SETTINGS = {"help_option_names": ["-h", "--help"]}




[docs]
def add_help_command(group):
    """
    Add a 'help' subcommand to a Click group.

    This allows users to get help via 'pocket <group> help' in addition
    to 'pocket <group> --help' and 'pocket <group> -h'.

    Args:
        group: A Click group to add the help command to.

    Returns:
        The group with the help command added.
    """
    @group.command(name="help", context_settings=CONTEXT_SETTINGS)
    @click.pass_context
    def help_cmd(ctx):
        """Show help for this command group."""
        click.echo(ctx.parent.get_help())
    return group





[docs]
def add_help_argument(command):
    """
    Make a command support 'help' as a positional argument.

    For commands, this allows users to type 'pocket <cmd> help' to get help,
    in addition to '-h' and '--help'.

    Args:
        command: A Click command to modify.

    Returns:
        The modified command.
    """
    original_make_context = command.make_context

    @functools.wraps(original_make_context)
    def new_make_context(info_name, args, parent=None, **extra):
        # Check if 'help' is in arguments (first position or alone)
        if args and 'help' in args:
            # Replace 'help' with '--help' so Click handles it properly
            args = ['--help'] + [a for a in args if a != 'help']
        return original_make_context(info_name, args, parent=parent, **extra)

    command.make_context = new_make_context
    return command





[docs]
def centered_spinner(message: str = "Loading...", style: str = "bold blue"):
    text = Text(message, style=style)
    return Align.center(Spinner("dots12", text=text, style=style))