from enum import Enum

from arger import Arger
from tests.utils import _reprint


class Choice(Enum):
    one = "1. One"
    two = "2. Two"

    def __str__(self):
        """By implementing this function, one can control the enum display in the help string."""
        return self.name


arger = Arger(prog="pytest", description="App Description goes here")

container = []


@arger.add_cmd
def cmd1(
    an_int: int,
    an_str: str,
    a_tuple: tuple[str, str, str],  # nargs: 3 -> consumes 3
    a_var_tuple: tuple[str, ...],  # nargs: + -> consumes one or more
    an_enum=Choice.one,
    optional_str="",
    optional_int=0,
    optional_tpl=(),
):
    """Example function with types documented in the docstring."""
    _reprint(**locals())


@arger.add_cmd
def cmd2(
    a_list: list[int],  # nargs='*': -> captures many args
    m_opt=False,
    y_opt=False,
    my=False,
):
    """A script with three optional values.

    :param a_list: Catch all positional arguments
    :param m_opt: The m_opt helptext
    :param y_opt: The y_opt helptext
    :param my: The my helptext
    """
    _reprint(**locals())


if __name__ == "__main__":
    arger.run()