"""A small command-line greeting utility.""" from __future__ import annotations import argparse from collections.abc import Sequence def positive_int(value: str) -> int: """Convert a string to a positive integer.""" number = int(value) if number < 1: raise argparse.ArgumentTypeError("must be a positive integer") return number def build_parser() -> argparse.ArgumentParser: parser = argparse.ArgumentParser(description="Print a personalized greeting.") parser.add_argument("name", help="name of the person to greet") parser.add_argument( "-n", "--count", type=positive_int, default=1, help="number of greetings to print (default: 1)", ) parser.add_argument( "--uppercase", action="store_true", help="print the greeting in uppercase", ) return parser def format_greeting(name: str, uppercase: bool = False) -> str: greeting = f"Hello, {name}!" return greeting.upper() if uppercase else greeting def main(argv: Sequence[str] | None = None) -> int: args = build_parser().parse_args(argv) greeting = format_greeting(args.name, args.uppercase) for _ in range(args.count): print(greeting) return 0 # Lightweight tests: run with `python -m unittest this_script.py`. def test_format_greeting() -> None: assert format_greeting("Ada") == "Hello, Ada!" assert format_greeting("Ada", uppercase=True) == "HELLO, ADA!" def test_parser() -> None: args = build_parser().parse_args(["Grace", "--count", "2", "--uppercase"]) assert (args.name, args.count, args.uppercase) == ("Grace", 2, True) if __name__ == "__main__": raise SystemExit(main())