From c5150be9f9f42f8928800ccf874fffa95649e7f4 Mon Sep 17 00:00:00 2001 From: Arjun Satarkar Date: Wed, 24 May 2023 22:17:46 +0530 Subject: Use nicer argument parsing --- main.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/main.py b/main.py index c9bfc7b..5bfc8b9 100755 --- a/main.py +++ b/main.py @@ -1,10 +1,11 @@ #!/usr/bin/env python3 +import argparse import sys -try: - word_num = int(sys.argv[1]) -except IndexError: - word_num = None +parser = argparse.ArgumentParser(prog="D&D&D&D", description="Dungeons & Dragons D...") +parser.add_argument("word_num", default=None, nargs="?", type=int) +args = parser.parse_args() +word_num = args.word_num WORDLIST_FILE_PATH = "corncob_lowercase.txt" -- cgit v1.2.3-57-g22cb