Removed duplicates in the file

Author: Priscilla-EM

implement-shell-tools/wc/mywc.py

@@ -19,6 +19,27 @@ def count_file(file_path, count_lines=False, count_words=False, count_bytes=Fals
         bytes_count if count_bytes else None
     )
 
+
+def calculate_output(counts, flags, label):
+    """Helper to build consistent formatted output lines."""
+    l_flag, w_flag, c_flag = flags
+    count_strings = []
+
+    # If no flags are set, show all counts
+    if not (l_flag or w_flag or c_flag):
+        lines, words, bytes_count = counts
+        count_strings = [f"{lines:>7}", f"{words:>7}", f"{bytes_count:>7}"]
+    else:
+        if counts[0] is not None:
+            count_strings.append(f"{counts[0]:>7}")
+        if counts[1] is not None:
+            count_strings.append(f"{counts[1]:>7}")
+        if counts[2] is not None:
+            count_strings.append(f"{counts[2]:>7}")
+
+    return f"{' '.join(count_strings)} {label}"
+
+
 def main():
     parser = argparse.ArgumentParser(description="count lines, words, and bytes like the wc command")
     parser.add_argument("-l", action="store_true", help="Count lines")
@@ -35,38 +56,20 @@ def main():
 
     for file_path in file_list:
         counts = count_file(file_path, args.l, args.w, args.c)
+        total_lines += counts[0] or 0
+        total_words += counts[1] or 0
+        total_bytes += counts[2] or 0
 
-        if not (args.l or args.w or args.c):
-            lines, words, bytes_count = counts
-            print(f"{lines:>7} {words:>7} {bytes_count:>7} {file_path}")
-            total_lines += lines
-            total_words += words
-            total_bytes += bytes_count
-        else:
-            count_strings = []
-            if counts[0] is not None:
-                count_strings.append(f"{counts[0]:>7}")
-                total_lines += counts[0]
-            if counts[1] is not None:
-                count_strings.append(f"{counts[1]:>7}")
-                total_words += counts[1]
-            if counts[2] is not None:
-                count_strings.append(f"{counts[2]:>7}")
-                total_bytes += counts[2]
-            print(" ".join(count_strings), file_path)
+        print(calculate_output(counts, (args.l, args.w, args.c), file_path))
 
     if len(file_list) > 1:
-        if not (args.l or args.w or args.c):
-            print(f"{total_lines:>7} {total_words:>7} {total_bytes:>7} total")
-        else:
-            total_strings = []
-            if args.l:
-                total_strings.append(f"{total_lines:>7}")
-            if args.w:
-                total_strings.append(f"{total_words:>7}")
-            if args.c:
-                total_strings.append(f"{total_bytes:>7}")
-            print(" ".join(total_strings), "total")
+        total_counts = (
+            total_lines if args.l or not (args.l or args.w or args.c) else None,
+            total_words if args.w or not (args.l or args.w or args.c) else None,
+            total_bytes if args.c or not (args.l or args.w or args.c) else None,
+        )
+        print(calculate_output(total_counts, (args.l, args.w, args.c), "total"))
+
 
 if __name__ == "__main__":
     main()