wc

Author: aiahibrahim

implement-shell-tools/wc/wc.py

@@ -1,25 +1,78 @@
 import argparse
+import glob
 
 def count_file(filename):
+    line_count = 0
+    word_count = 0
+    byte_count = 0
+
     try:
-        with open(filename, "r", encoding="utf-8") as f:
+        with open(filename, "rb") as f:
             content = f.read()
-            lines = content.count("\n")
-            words = len(content.split())
-            bytes_ = len(content.encode("utf-8"))
-            return lines, words, bytes_
+            byte_count = len(content)
+            text = content.decode("utf-8", errors="replace")
+            lines = text.splitlines()
+            line_count = len(lines)
+            word_count = sum(len(line.split()) for line in lines)
+
+        return (line_count, word_count, byte_count)
     except FileNotFoundError:
         print(f"wc: {filename}: No such file or directory")
-        return 0, 0, 0
+        return None
 
 def main():
     parser = argparse.ArgumentParser(description="Python version of wc")
-    parser.add_argument("files", nargs="+", help="Files to count")
+    parser.add_argument("-l", action="store_true", help="Count lines")
+    parser.add_argument("-w", action="store_true", help="Count words")
+    parser.add_argument("-c", action="store_true", help="Count bytes")
+    parser.add_argument("files", nargs="+", help="Files to read (supports wildcards)")
     args = parser.parse_args()
 
-    for filename in args.files:
-        lines, words, bytes_ = count_file(filename)
-        print(f"{lines:7} {words:7} {bytes_:7} {filename}")
+    # If no flag is given, show all
+    show_lines = args.l
+    show_words = args.w
+    show_bytes = args.c
+    if not (show_lines or show_words or show_bytes):
+        show_lines = show_words = show_bytes = True
+
+    file_list = []
+    for pattern in args.files:
+        matched = glob.glob(pattern)
+        if matched:
+            file_list.extend(matched)
+        else:
+            file_list.append(pattern)
+
+    total_lines = total_words = total_bytes = 0
+    for filename in file_list:
+        result = count_file(filename)
+        if result is None:
+            continue
+        lines, words, bytes_ = result
+        total_lines += lines
+        total_words += words
+        total_bytes += bytes_
+
+        output = []
+        if show_lines:
+            output.append(f"{lines:7}")
+        if show_words:
+            output.append(f"{words:7}")
+        if show_bytes:
+            output.append(f"{bytes_:7}")
+        output.append(filename)
+        print(" ".join(output))
+
+    if len(file_list) > 1:
+        output = []
+        if show_lines:
+            output.append(f"{total_lines:7}")
+        if show_words:
+            output.append(f"{total_words:7}")
+        if show_bytes:
+            output.append(f"{total_bytes:7}")
+        output.append("total")
+        print(" ".join(output))
 
 if __name__ == "__main__":
     main()