finished wc.py

Author: M-Abdoon

implement-shell-tools/wc/wc.py

@@ -0,0 +1,64 @@
+import argparse
+
+parser = argparse.ArgumentParser()
+parser.add_argument("files", nargs="+")
+parser.add_argument("-l", action="store_true", help="show line count")
+parser.add_argument("-w", action="store_true", help="show word count")
+parser.add_argument("-c", action="store_true", help="show byte count")
+
+args = parser.parse_args()
+
+show_lines = args.l
+show_words = args.w
+show_bytes = args.c
+
+if not show_lines and not show_words and not show_bytes:
+    show_lines = True
+    show_words = True
+    show_bytes = True
+
+total_lines = 0
+total_words = 0
+total_bytes = 0
+
+for file in args.files:
+    try:
+        f = open(file, "rb")
+        content = f.read()
+        f.close()
+
+        line_count = content.count(b"\n")
+        word_count = len(content.split())
+        byte_count = len(content)
+
+        total_lines += line_count
+        total_words += word_count
+        total_bytes += byte_count
+
+        output = []
+
+        if show_lines:
+            output.append(str(line_count))
+        if show_words:
+            output.append(str(word_count))
+        if show_bytes:
+            output.append(str(byte_count))
+
+        output.append(file)
+        print(" ".join(output))
+
+    except OSError:
+        print("wc:", file, "not found")
+
+if len(args.files) > 1:
+    output = []
+
+    if show_lines:
+        output.append(str(total_lines))
+    if show_words:
+        output.append(str(total_words))
+    if show_bytes:
+        output.append(str(total_bytes))
+
+    output.append("total")
+    print(" ".join(output))