Code cleaned up

Author: Rashaad-Ebrahim

implement-shell-tools/wc/wc.py

@@ -15,7 +15,8 @@
 file_paths = args.path
 lines_flag, words_flag, bytes_flag = args.lines, args.words, args.bytes
 
-def format_output(details_list, totals=None):
+def format_output(details_list, totals=None, flags=None):
+    lines_flag, words_flag, bytes_flag = flags
     show_all = not (lines_flag or words_flag or bytes_flag)
     output_lines = []
 
@@ -43,7 +44,7 @@ def format_output(details_list, totals=None):
             total_line += f"{totals['line_count']:>3} "
 
         if show_all or words_flag:
-            total_line += f"{totals['word_count']:>3} "    
+            total_line += f"{totals['word_count']:>3} "
 
         if show_all or bytes_flag:
             total_line += f"{totals['file_size']:>3} "
@@ -53,12 +54,13 @@ def format_output(details_list, totals=None):
 
     return "\n".join(output_lines)
 
+
+# Collect file details
 file_details_list = []
 line_count_total = 0
 word_count_total = 0
 file_size_total = 0
 
-# Get file details
 for file_path in file_paths:
     with open(file_path, "r", encoding="utf-8") as f:
         content = f.read()
@@ -69,11 +71,10 @@ def format_output(details_list, totals=None):
         "file_size": os.path.getsize(file_path),
         "file_path": file_path,
     }
-    # Update totals
+
     line_count_total += details["line_count"]
     word_count_total += details["word_count"]
     file_size_total += details["file_size"]
-    total_path = "Total"
 
     file_details_list.append(details)
 
@@ -83,11 +84,6 @@ def format_output(details_list, totals=None):
     "file_size": file_size_total,
 }
 
-formatted_output = format_output(file_details_list, totals_details)
-print(formatted_output)
-
-
-
-# print(args)
-
-#   1   4  20 sample-files/1.txt
+# Final output
+flags = (lines_flag, words_flag, bytes_flag)
+print(format_output(file_details_list, totals_details, flags))