Add total line for multiple files in wc

Author: zohrehKazemianpour

implement-shell-tools/wc/wc.py

@@ -12,6 +12,11 @@
 
 args = parser.parse_args()
 
+# Track totals
+total_lines = 0
+total_words = 0
+total_chars = 0
+
 for path in args.paths:
     # Read the file
     with open(path, "r") as f:
@@ -22,11 +27,29 @@
     words = len(content.split())
     chars = len(content)
 
+    # Add to totals
+    total_lines += lines
+    total_words += words
+    total_chars += chars
+
     if args.lines:
         print(f"{lines:8} {path}")
     elif args.words:
         print(f"{words:8} {path}")
     elif args.chars:
         print(f"{chars:8} {path}")
     else:
-        print(f"{lines:8}{words:8}{chars:8} {path}")  
+        print(f"{lines:8}{words:8}{chars:8} {path}")  
+
+        
+# Print totals if multiple files
+if len(args.paths) > 1:
+    if args.lines:
+        print(f"{total_lines:8} total")
+    elif args.words:
+        print(f"{total_words:8} total")
+    elif args.chars:
+        print(f"{total_chars:8} total")
+    else:
+        print(f"{total_lines:8}{total_words:8}{total_chars:8} total")        
+