Problem 05 solution

Author: lukemcguire

05_howler/howler.py

@@ -0,0 +1,48 @@
+#!/usr/bin/env python3
+"""
+Author : Luke McGuire <[email protected]>
+Date   : 2024-06-18
+Purpose: Howler (upper-cases input)
+"""
+
+import argparse
+import sys
+import os
+
+
+# --------------------------------------------------
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description="Howler (upper-cases input)",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+    )
+
+    parser.add_argument("text", metavar="text", help="Input string or file")
+    parser.add_argument(
+        "-o", "--outfile", help="Output filename", metavar="str", type=str, default=""
+    )
+
+    args = parser.parse_args()
+    if os.path.isfile(args.text):
+        with open(args.text, "rt", encoding="utf-8") as fh:
+            args.text = fh.read().rstrip()
+
+    return args
+
+
+# --------------------------------------------------
+def main():
+    """Make a jazz noise here"""
+
+    args = get_args()
+    text = args.text
+    outfile = args.outfile
+    with open(outfile, "wt", encoding="utf-8") if outfile else sys.stdout as fh:
+        fh.write(text.upper() + "\n")
+
+
+# --------------------------------------------------
+if __name__ == "__main__":
+    main()

05_howler/test.py

@@ -7,37 +7,37 @@
 import string
 from subprocess import getstatusoutput, getoutput
 
-prg = './howler.py'
+PRG = "./howler.py"
 
 
 # --------------------------------------------------
 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))
 
 
 # --------------------------------------------------
 def out_flag():
     """Either -o or --outfile"""
 
-    return '-o' if random.randint(0, 1) else '--outfile'
+    return "-o" if random.randint(0, 1) else "--outfile"
 
 
 # --------------------------------------------------
 def test_exists():
     """exists"""
 
-    assert os.path.isfile(prg)
+    assert os.path.isfile(PRG)
 
 
 # --------------------------------------------------
 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)
 
@@ -46,8 +46,8 @@ def test_usage():
 def test_text_stdout():
     """Test STDIN/STDOUT"""
 
-    out = getoutput(f'{prg} "foo bar baz"')
-    assert out.strip() == 'FOO BAR BAZ'
+    out = getoutput(f'python3 {PRG} "foo bar baz"')
+    assert out.strip() == "FOO BAR BAZ"
 
 
 # --------------------------------------------------
@@ -59,11 +59,11 @@ def test_text_outfile():
         os.remove(out_file)
 
     try:
-        out = getoutput(f'{prg} {out_flag()} {out_file} "foo bar baz"')
-        assert out.strip() == ''
+        out = getoutput(f'python3 {PRG} {out_flag()} {out_file} "foo bar baz"')
+        assert out.strip() == ""
         assert os.path.isfile(out_file)
         text = open(out_file).read().rstrip()
-        assert text == 'FOO BAR BAZ'
+        assert text == "FOO BAR BAZ"
     finally:
         if os.path.isfile(out_file):
             os.remove(out_file)
@@ -73,19 +73,18 @@ def test_text_outfile():
 def test_file():
     """Test file in/out"""
 
-    for expected_file in os.listdir('test-outs'):
+    for expected_file in os.listdir("test-outs"):
         try:
             out_file = random_string()
             if os.path.isfile(out_file):
                 os.remove(out_file)
 
             basename = os.path.basename(expected_file)
-            in_file = os.path.join('../inputs', basename)
-            out = getoutput(f'{prg} {out_flag()} {out_file} {in_file}')
-            assert out.strip() == ''
+            in_file = os.path.join("../inputs", basename)
+            out = getoutput(f"python3 {PRG} {out_flag()} {out_file} {in_file}")
+            assert out.strip() == ""
             produced = open(out_file).read().rstrip()
-            expected = open(os.path.join('test-outs',
-                                         expected_file)).read().strip()
+            expected = open(os.path.join("test-outs", expected_file)).read().strip()
             assert expected == produced
         finally:
             if os.path.isfile(out_file):