Problem 17 solution

Author: lukemcguire

17_mad_libs/mad.py

@@ -0,0 +1,63 @@
+#!/usr/bin/env python3
+"""
+Author : Luke McGuire <[email protected]>
+Date   : 2024-07-01
+Purpose: Mad Libs
+"""
+
+import argparse
+import re
+import sys
+
+
+# --------------------------------------------------
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description="Mad Libs", formatter_class=argparse.ArgumentDefaultsHelpFormatter
+    )
+
+    parser.add_argument(
+        "-i",
+        "--inputs",
+        help="A named string argument",
+        metavar="str",
+        type=str,
+        nargs="*",
+    )
+
+    parser.add_argument(
+        "file", metavar="FILE", help="Input file", type=argparse.FileType("rt")
+    )
+
+    return parser.parse_args()
+
+
+# --------------------------------------------------
+def main():
+    """Make a jazz noise here"""
+
+    args = get_args()
+
+    text = args.file.read().rstrip()
+
+    placeholders = re.findall("(<([^<>]+?)>)", text)
+    if not placeholders:
+        print(f'"{args.file.name}" has no placeholders.', file=sys.stderr)
+        sys.exit(1)
+
+    inputs = iter(args.inputs)
+
+    template = "Give me {} {}: "
+    for placeholder, word in placeholders:
+        article: str = "an" if word[0].lower() in "aeiou" else "a"
+        answer = next(inputs) if inputs else input(template.format(article, word))
+        text = re.sub(placeholder, answer, text, count=1)
+
+    print(text)
+
+
+# --------------------------------------------------
+if __name__ == "__main__":
+    main()

17_mad_libs/test.py

@@ -7,11 +7,11 @@
 import string
 from subprocess import getstatusoutput
 
-prg = './mad.py'
-no_blanks = 'inputs/no_blanks.txt'
-fox = 'inputs/fox.txt'
-hlp = 'inputs/help.txt'
-verona = 'inputs/romeo_juliet.txt'
+prg = "./mad.py"
+no_blanks = "inputs/no_blanks.txt"
+fox = "inputs/fox.txt"
+hlp = "inputs/help.txt"
+verona = "inputs/romeo_juliet.txt"
 
 
 # --------------------------------------------------
@@ -25,18 +25,18 @@ def test_exists():
 def test_usage():
     """usage"""
 
-    for flag in ['-h', '--help']:
-        rv, out = getstatusoutput(f'{prg} {flag}')
+    for flag in ["-h", "--help"]:
+        rv, out = getstatusoutput(f"python {prg} {flag}")
         assert rv == 0
-        assert out.lower().startswith('usage')
+        assert out.lower().startswith("usage")
 
 
 # --------------------------------------------------
 def test_bad_file():
     """Test bad input file"""
 
     bad = random_string()
-    rv, out = getstatusoutput(f'{prg} {bad}')
+    rv, out = getstatusoutput(f"python {prg} {bad}")
     assert rv != 0
     assert re.search(f"No such file or directory: '{bad}'", out)
 
@@ -45,7 +45,7 @@ def test_bad_file():
 def test_no_blanks():
     """Test no blanks"""
 
-    rv, out = getstatusoutput(f'{prg} {no_blanks}')
+    rv, out = getstatusoutput(f"python {prg} {no_blanks}")
     assert rv != 0
     assert out == f'"{no_blanks}" has no placeholders.'
 
@@ -54,10 +54,10 @@ def test_no_blanks():
 def test_fox():
     """test fox"""
 
-    args = f'{fox} -i surly car under bicycle'
-    rv, out = getstatusoutput(f'{prg} {args}')
+    args = f"{fox} -i surly car under bicycle"
+    rv, out = getstatusoutput(f"python {prg} {args}")
     assert rv == 0
-    assert out.strip() == 'The quick surly car jumps under the lazy bicycle.'
+    assert out.strip() == "The quick surly car jumps under the lazy bicycle."
 
 
 # --------------------------------------------------
@@ -71,8 +71,8 @@ def test_help():
 Arriba!
     """.strip()
 
-    args = f'{hlp} -i Hey tacos Oi salsa Hola queso Arriba'
-    rv, out = getstatusoutput(f'{prg} {args}')
+    args = f"{hlp} -i Hey tacos Oi salsa Hola queso Arriba"
+    rv, out = getstatusoutput(f"python {prg} {args}")
     assert rv == 0
     assert out.strip() == expected.strip()
 
@@ -98,9 +98,11 @@ def test_verona():
 What here shall hammer, our toil shall strive to mend.
     """.strip()
 
-    args = (f'{verona} --inputs cars Detroit oil pistons '
-            '"stick shift" furious accelerate 42 foot hammer')
-    rv, out = getstatusoutput(f'{prg} {args}')
+    args = (
+        f"{verona} --inputs cars Detroit oil pistons "
+        '"stick shift" furious accelerate 42 foot hammer'
+    )
+    rv, out = getstatusoutput(f"python {prg} {args}")
     assert rv == 0
     assert out.strip() == expected.strip()
 
@@ -110,4 +112,4 @@ def random_string():
     """generate a random string"""
 
     k = random.randint(5, 10)
-    return ''.join(random.choices(string.ascii_letters + string.digits, k=k))
+    return "".join(random.choices(string.ascii_letters + string.digits, k=k))