Implement wc command-line tool in python

Author: ike-agu

implement-shell-tools/wc/README.md

@@ -1,4 +1,4 @@
-# Implement `wc`
+# Implement `wc` in python
 
 You should already be familiar with the `wc` command line tool.
 
@@ -15,3 +15,20 @@ It must act the same as `wc` would, if run from the directory containing this RE
 Matching any additional behaviours or flags are optional stretch goals.
 
 We recommend you start off supporting no flags for one file, then add support for multiple files, then add support for the flags.
+
+*======Command for testing the script========
+
+* All sample files
+python wc.py sample-files/*
+
+* Just lines
+python wc.py -l sample-files/3.txt
+
+* Just words
+python wc.py -w sample-files/3.txt
+
+* Just characters
+python wc.py -c sample-files/3.txt
+
+* Lines with multiple files (to see totals)
+python wc.py -l sample-files/*

implement-shell-tools/wc/wc.py

@@ -0,0 +1,93 @@
+import argparse
+import sys
+import re
+
+
+def main():
+    # --------------------------------------------------------
+    # 1. Set up argparse
+    # --------------------------------------------------------
+    parser= argparse.ArgumentParser(
+        prog="wc",
+        description ="wc command implementation in python"
+    )
+
+    parser.add_argument("-l", action="store_true", help ="show number of lines only")
+    parser.add_argument("-w", action="store_true", help="show number of words only")
+    parser.add_argument("-c", action="store_true", help="show number of characters only")
+
+    parser.add_argument("paths", nargs="*", help="file paths to process")
+
+    args = parser.parse_args()
+
+    # --------------------------------------------------------
+    # 2. Ensures at least one path exists
+    # --------------------------------------------------------
+    if len(args.paths) == 0:
+        print("wc: no file specified", file=sys.stderr)
+        sys.exit(1)
+
+    totals= {"lines": 0, "words": 0, "chars": 0}
+
+    # --------------------------------------------------------
+    # 3. Loop over each file path and process it
+    # --------------------------------------------------------
+    for file_path in args.paths:
+        try:
+            with open(file_path, "r", encoding="utf-8") as f:
+                content = f.read()
+        except OSError as err:
+            print(f"wc: cannot read file'{file_path}': {err}", file=sys.stderr)
+            continue
+
+        # --------------------------------------------------------
+        # 4. Count values
+        # --------------------------------------------------------
+        line_count = len(content.split("\n"))
+
+        words = [w for w in re.split(r"\s+", content) if w]
+        word_count = len(words)
+
+        char_count = len(content)
+
+        totals["lines"] += line_count
+        totals["words"] += word_count
+        totals["chars"] +=char_count
+
+        # --------------------------------------------------------
+        # 5. Decide what to print based on flags
+        # --------------------------------------------------------
+        no_flags = not args.l and not args.w and not args.c
+
+        if no_flags:
+            print(f"{line_count} {word_count} {char_count} {file_path}")
+            continue
+
+        if args.l:
+            print(f"{line_count} {file_path}" )
+
+        if args.w:
+            print(f"{word_count} {file_path}")
+
+        if args.c:
+            print(f"{char_count} {file_path}")
+
+
+    # --------------------------------------------------------
+    # 6. Print totals if there are multiple files
+    # --------------------------------------------------------
+    if len(args.paths) > 1:
+        no_flags = not args.l and not args.w and not args.c
+
+        if no_flags:
+            print(f"{totals['lines']} {totals['words']} {totals['chars']} total")
+
+        if args.l:
+            print(f"{totals['lines']} total")
+        if args.w:
+            print(f"{totals['words']} total")
+        if args.c:
+            print(f"{totals['chars']} total")
+
+if __name__ == "__main__":
+    main()