Remove duplicate wc print code for readability

Author: sheida-shab

implement-shell-tools/wc/wc.py

@@ -1,6 +1,8 @@
 import argparse
 import os
 
+
+# Set up argument parser for wc command
 parser=argparse.ArgumentParser(prog="wc",usage="implement a simple wc in python")
 parser.add_argument("-l",action="store_true",help="count of lines")
 parser.add_argument("-w",action="store_true",help="count of words")
@@ -15,28 +17,42 @@
 total_words=0
 total_bytes=0
 
+# Function to print counts for a file or total
+def print_wc(lines, words, bytes_, name):
+    # Print selected counts in formatted columns
+    print(
+        (f"{lines:<5}" if args.l else "") +
+        (f"{words:<5}" if args.w else "") +
+        (f"{bytes_:<5}" if args.c else "") +
+        f"{name:<20}"
+    )
+
+# Loop through all files
 for file in paths :
     with open(file,"r") as f :
+        # Get file size in bytes
         bytes_count=os.path.getsize(file)
+         # Read all lines
         lines=f.readlines()
         lines_count=len(lines)
         for line in lines :
+            # Count words in file
             words_count+=len(line.split())
+        
+        # Update totals
         total_lines +=lines_count
         total_bytes += bytes_count
         total_words +=words_count   
+
+        # If no flags provided, default to showing all counts
         if not (args.l or args.w or args.c):
             args.l = True
             args.w = True
             args.c = True 
-        print((f"{lines_count:<5}" if args.l else "")+
-              (f"{words_count:<5}" if args.w else "")+
-              (f"{bytes_count:<5}" if args.c else "")+
-              f"{file:<20}")
-        words_count=0
+        # Print counts for this file    
+        print_wc(lines_count, words_count, bytes_count, file)
 
+# If multiple files, print totals at the end
 if len(paths)>1 :        
-    print((f"{total_lines:<5}" if args.l else "")+
-        (f"{total_words:<5}" if args.w else "")+
-        (f"{total_bytes:<5}" if args.c else "")+
-        "total")        
+   print_wc(total_lines, total_words, total_bytes, "total")      
+