Problem 07 solution

Author: lukemcguire

07_gashlycrumb/gashlycrumb.py

@@ -0,0 +1,53 @@
+#!/usr/bin/env python3
+"""
+Author : Luke McGuire <[email protected]>
+Date   : 2024-06-21
+Purpose: Prints the line from a file starting with a given letter.
+"""
+
+import argparse
+
+
+# --------------------------------------------------
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description="Gashlycrumb",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+    )
+
+    parser.add_argument("letter", metavar="letter", nargs="+", help="Letter(s)")
+
+    parser.add_argument(
+        "-f",
+        "--file",
+        help="Input file",
+        metavar="FILE",
+        type=argparse.FileType("rt"),
+        default="gashlycrumb.txt",
+    )
+    return parser.parse_args()
+
+
+# --------------------------------------------------
+def main():
+    """Make a jazz noise here"""
+
+    args = get_args()
+
+    my_dict = {}
+    for i, line in enumerate(args.file):
+        my_dict[line[0].lower()] = line.strip()
+
+    for l in args.letter:
+        text = my_dict.get(l.lower())
+        if text:
+            print(my_dict[l.lower()])
+        else:
+            print(f'I do not know "{l}".')
+
+
+# --------------------------------------------------
+if __name__ == "__main__":
+    main()

07_gashlycrumb/test.py

@@ -7,14 +7,14 @@
 import string
 from subprocess import getstatusoutput
 
-prg = './gashlycrumb.py'
+prg = "./gashlycrumb.py"
 
 
 # --------------------------------------------------
 def file_flag():
     """Either -f or --file"""
 
-    return '-f' if random.randint(0, 1) else '--file'
+    return "-f" if random.randint(0, 1) else "--file"
 
 
 # --------------------------------------------------
@@ -28,8 +28,8 @@ 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"python3 {prg} {flag}")
         assert rv == 0
         assert re.match("usage", out, re.IGNORECASE)
 
@@ -40,7 +40,7 @@ def test_bad_file():
 
     bad = random_string()
     letter = random.choice(string.ascii_lowercase)
-    rv, out = getstatusoutput(f'{prg} {letter} -f {bad}')
+    rv, out = getstatusoutput(f"python3 {prg} {letter} -f {bad}")
     assert rv != 0
     expected = f"No such file or directory: '{bad}'"
     assert re.search(expected, out)
@@ -50,52 +50,53 @@ def test_bad_file():
 def test_a():
     """Test for 'a'"""
 
-    rv, out = getstatusoutput(f'{prg} a')
+    rv, out = getstatusoutput(f"python3 {prg} a")
     assert rv == 0
-    expected = 'A is for Amy who fell down the stairs.'
+    expected = "A is for Amy who fell down the stairs."
     assert out.strip() == expected
 
 
 # --------------------------------------------------
 def test_b_c():
     """Test for 'b c'"""
 
-    rv, out = getstatusoutput(f'{prg} b c')
+    rv, out = getstatusoutput(f"python3 {prg} b c")
     assert rv == 0
-    expected = ('B is for Basil assaulted by bears.\n'
-                'C is for Clara who wasted away.')
+    expected = "B is for Basil assaulted by bears.\n" "C is for Clara who wasted away."
     assert out.strip() == expected
 
 
 # --------------------------------------------------
 def test_y():
     """Test for 'y'"""
 
-    rv, out = getstatusoutput(f'{prg} Y')
+    rv, out = getstatusoutput(f"python3 {prg} Y")
     assert rv == 0
-    expected = 'Y is for Yorick whose head was bashed in.'
+    expected = "Y is for Yorick whose head was bashed in."
     assert out.strip() == expected
 
 
 # --------------------------------------------------
 def test_o_alternate():
-    """ Test for 'o' from 'alternate.txt' """
+    """Test for 'o' from 'alternate.txt'"""
 
-    rv, out = getstatusoutput(f'{prg} o P q -f alternate.txt')
+    rv, out = getstatusoutput(f"python3 {prg} o P q -f alternate.txt")
     assert rv == 0
-    expected = ('O is for Orville, who fell in a canyon.\n'
-                'P is for Paul, strangled by his banyan.\n'
-                'Q is for Quintanna, flayed in the night.')
+    expected = (
+        "O is for Orville, who fell in a canyon.\n"
+        "P is for Paul, strangled by his banyan.\n"
+        "Q is for Quintanna, flayed in the night."
+    )
     assert out.strip() == expected
 
 
 # --------------------------------------------------
 def test_bad_letter():
     """Test for bad input"""
 
-    rv, out = getstatusoutput(f'{prg} 5 CH')
+    rv, out = getstatusoutput(f"python3 {prg} 5 CH")
     assert rv == 0
-    expected = ('I do not know "5".\n' 'I do not know "CH".')
+    expected = 'I do not know "5".\n' 'I do not know "CH".'
     assert out.strip() == expected
 
 
@@ -104,4 +105,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))