Make not passing args in actually default to word_count=7

micahflee · Oct 30, 2019 · 5074b9c · 5074b9c
1 parent 62805c1
commit 5074b9c
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/passphraseme/__init__.py b/passphraseme/__init__.py
@@ -23,7 +23,13 @@ def generate(filename, word_count=7):
 def main():
     parser = argparse.ArgumentParser()
     parser.add_argument(
-        "word_count", metavar="word_count", default=7, type=int, help="Word count"
+        "word_count",
+        metavar="word_count",
+        nargs="?",
+        const=1,
+        default=7,
+        type=int,
+        help="Word count",
     )
     group = parser.add_mutually_exclusive_group()
     group.add_argument(