Problem 06 solution

Author: lukemcguire

06_wc/test.py

@@ -7,12 +7,12 @@
 import string
 from subprocess import getstatusoutput
 
-prg = './wc.py'
-empty = './inputs/empty.txt'
-one_line = './inputs/one.txt'
-two_lines = './inputs/two.txt'
-fox = '../inputs/fox.txt'
-sonnet = '../inputs/sonnet-29.txt'
+prg = "./wc.py"
+empty = "./inputs/empty.txt"
+one_line = "./inputs/one.txt"
+two_lines = "./inputs/two.txt"
+fox = "../inputs/fox.txt"
+sonnet = "../inputs/sonnet-29.txt"
 
 
 # --------------------------------------------------
@@ -26,8 +26,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)
 
@@ -37,15 +37,15 @@ 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 test_bad_file():
     """bad_file"""
 
     bad = random_string()
-    rv, out = getstatusoutput(f'{prg} {bad}')
+    rv, out = getstatusoutput(f"python3 {prg} {bad}")
     assert rv != 0
     assert re.search(f"No such file or directory: '{bad}'", out)
 
@@ -54,46 +54,48 @@ def test_bad_file():
 def test_empty():
     """Test on empty"""
 
-    rv, out = getstatusoutput(f'{prg} {empty}')
+    rv, out = getstatusoutput(f"python3 {prg} {empty}")
     assert rv == 0
-    assert out.rstrip() == '       0       0       0 ./inputs/empty.txt'
+    assert out.rstrip() == "       0       0       0 ./inputs/empty.txt"
 
 
 # --------------------------------------------------
 def test_one():
     """Test on one"""
 
-    rv, out = getstatusoutput(f'{prg} {one_line}')
+    rv, out = getstatusoutput(f"python3 {prg} {one_line}")
     assert rv == 0
-    assert out.rstrip() == '       1       1       2 ./inputs/one.txt'
+    assert out.rstrip() == "       1       1       2 ./inputs/one.txt"
 
 
 # --------------------------------------------------
 def test_two():
     """Test on two"""
 
-    rv, out = getstatusoutput(f'{prg} {two_lines}')
+    rv, out = getstatusoutput(f"python3 {prg} {two_lines}")
     assert rv == 0
-    assert out.rstrip() == '       2       2       4 ./inputs/two.txt'
+    assert out.rstrip() == "       2       2       4 ./inputs/two.txt"
 
 
 # --------------------------------------------------
 def test_fox():
     """Test on fox"""
 
-    rv, out = getstatusoutput(f'{prg} {fox}')
+    rv, out = getstatusoutput(f"python3 {prg} {fox}")
     assert rv == 0
-    assert out.rstrip() == '       1       9      45 ../inputs/fox.txt'
+    assert out.rstrip() == "       1       9      45 ../inputs/fox.txt"
 
 
 # --------------------------------------------------
 def test_more():
     """Test on more than one file"""
 
-    rv, out = getstatusoutput(f'{prg} {fox} {sonnet}')
-    expected = ('       1       9      45 ../inputs/fox.txt\n'
-                '      17     118     661 ../inputs/sonnet-29.txt\n'
-                '      18     127     706 total')
+    rv, out = getstatusoutput(f"python3 {prg} {fox} {sonnet}")
+    expected = (
+        "       1       9      45 ../inputs/fox.txt\n"
+        "      17     118     661 ../inputs/sonnet-29.txt\n"
+        "      18     127     706 total"
+    )
     assert rv == 0
     assert out.rstrip() == expected
 
@@ -102,6 +104,6 @@ def test_more():
 def test_stdin():
     """Test on stdin"""
 
-    rv, out = getstatusoutput(f'{prg} < {fox}')
+    rv, out = getstatusoutput(f"python3 {prg} < {fox}")
     assert rv == 0
-    assert out.rstrip() == '       1       9      45 <stdin>'
+    assert out.rstrip() == "       1       9      45 <stdin>"

06_wc/wc.py

@@ -0,0 +1,71 @@
+#!/usr/bin/env python3
+"""
+Author : Luke McGuire <[email protected]>
+Date   : 2024-06-21
+Purpose: Emulate wc (word count)
+"""
+
+import argparse
+import sys
+import os
+
+
+# --------------------------------------------------
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description="Emulate wc (word count)",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+    )
+
+    parser.add_argument(
+        "file",
+        help="Input file(s)",
+        nargs="*",
+        metavar="FILE",
+        type=argparse.FileType("rt"),
+        default=[sys.stdin],
+    )
+
+    args = parser.parse_args()
+
+    # if args.file != [sys.stdin]:
+    #     for path in args.file:
+    #         if not os.path.isfile(path.name):
+    #             print(f"No such file or directory: '{path.name}'")
+    #             sys.exit()
+
+    return args
+
+
+# --------------------------------------------------
+def main():
+    """Make a jazz noise here"""
+
+    args = get_args()
+    total_lines = 0
+    total_words = 0
+    total_bytes = 0
+
+    for fh in args.file:
+        num_lines = 0
+        num_words = 0
+        num_bytes = 0
+        for line in fh:
+            num_lines += 1
+            num_words += len(line.split())
+            num_bytes += len(line)
+
+        print(f"{num_lines:>8}{num_words:>8}{num_bytes:>8} {fh.name}")
+        total_lines += num_lines
+        total_words += num_words
+        total_bytes += num_bytes
+
+    if len(args.file) > 1:
+        print(f"{total_lines:>8}{total_words:>8}{total_bytes:>8} total")
+
+
+# --------------------------------------------------
+if __name__ == "__main__":
+    main()