From 134fa96d045506638867a6971ef452316caadd58 Mon Sep 17 00:00:00 2001 From: Alfi Maulana Date: Wed, 18 Jun 2025 12:15:35 +0700 Subject: [PATCH] feat: utilize argparse for argument parsing Signed-off-by: Alfi Maulana --- src/my_fibonacci/__main__.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/src/my_fibonacci/__main__.py b/src/my_fibonacci/__main__.py index 72d096a..96ffd12 100644 --- a/src/my_fibonacci/__main__.py +++ b/src/my_fibonacci/__main__.py @@ -1,10 +1,18 @@ -import sys +import argparse from . import fibonacci_sequence def main() -> None: - sequence = fibonacci_sequence(int(sys.argv[1])) + parser = argparse.ArgumentParser( + prog="my_fibonacci", + description="Generate a Fibonacci sequence up to the given number of terms.", + ) + parser.add_argument("--version", action="version", version="0.0.0") + parser.add_argument("n", type=int, help="The number of terms") + args = parser.parse_args() + + sequence = fibonacci_sequence(args.n) print(" ".join(str(item) for item in sequence))