logic completed

Author: RaihanSharif

implement-shell-tools/wc/wc.py

@@ -6,10 +6,69 @@
     prog="a simple version of wc. Takes in one or more files.",
     description="ls command line tool which can accept -l -w -c cflags")
 
-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")
+parser.add_argument("-l", action="store_true", help="show line count", default="l")
+parser.add_argument("-w", action="store_true", help="show word count", default='w')
+parser.add_argument("-c", action="store_true", help="show byte count", default="c")
 
 parser.add_argument("paths", nargs="*", help="file(s) for which to show data")
 
-args = parser.parse_args()
+args = parser.parse_args()
+
+totals = {"l": 0, "w": 0, "c": 0}
+
+file_count = 0
+
+for path in args.paths:
+    try:
+        if (os.path.isdir(path)):
+            print(f"wc: {path}: read: Is a directory")
+    except:
+        print(f"wc: {path} open: No such file or directory", file=sys.stderr)
+        file_count += 1
+        continue
+
+    if (os.path.isfile(path)):
+        file_count += 1
+        output_str = ""
+
+        with open(path, "r", encoding="utf-8") as file:
+            lines = file.readlines()
+
+            if (args.l):
+                if (len(lines) > 0 and lines[-1] == ""):
+                    lines.pop()
+
+                line_count = len(lines)
+                totals["l"] += line_count
+                output_str += f"\t{line_count}"
+
+            if (args.w):
+                word_count = 0
+                for line in lines:
+                    # python string.split splits on any white space
+                    word_count += len(line.split())
+                totals["w"] += word_count
+                output_str += f"\t{word_count}"
+
+            if (args.c):
+                bytes = os.path.getsize(path)
+                totals["c"] += bytes
+                output_str += f"\t{bytes}"
+
+        output_str += f" {path}"
+        print(output_str)
+
+if (file_count > 1):
+    res = {key : val for key, val in totals.items() 
+           if val != 0}
+    total_str = ""
+    for v in res.values():
+        total_str += f"\t{v}"
+
+    total_str += " total"
+    print(total_str)
+            
+
+                
+
+