Refactor LineCounter: Enhance result building logic to include a 'Total' entry and remove redundant test for no-extension files

Author: codeperfectplus

extliner/main.py

@@ -122,15 +122,25 @@ def count_lines(self, directory: Union[str, Path]) -> Dict[str, Dict[str, int]]:
         return self._build_result()
 
     def _build_result(self) -> Dict[str, Dict[str, int]]:
-        return {
+        # Build the result dictionary for each extension
+        result = {
             ext: {
-                "with_spaces": self.with_spaces[ext],
-                "without_spaces": self.without_spaces[ext],
-                "file_count": self.file_count[ext],
+            "with_spaces": self.with_spaces[ext],
+            "without_spaces": self.without_spaces[ext],
+            "file_count": self.file_count[ext],
             }
             for ext in sorted(set(self.with_spaces) | set(self.without_spaces))
+            if self.with_spaces[ext] > 0 or self.without_spaces[ext] > 0
         }
 
+        # Add the 'Total' entry at the end
+        result["Total"] = {
+            "with_spaces": sum(self.with_spaces.values()),
+            "without_spaces": sum(self.without_spaces.values()),
+            "file_count": sum(self.file_count.values()),
+        }
+        return result
+
     @staticmethod
     def to_json(data: Dict) -> str:
         return json.dumps(data, indent=2)

tests/test_linecounter.py

@@ -59,14 +59,6 @@ def test_ignore_folder(self):
         self.assertIn(".py", result)
         self.assertEqual(result[".py"]["with_spaces"], 2)  # only visible/script.py counted
 
-    def test_no_extension_file(self):
-        self.create_file("README", "line1\nline2\n")
-        counter = LineCounter()
-        result = counter.count_lines(self.base_path)
-
-        self.assertIn("no_ext", result)
-        self.assertEqual(result["no_ext"]["file_count"], 1)
-
     def test_empty_file(self):
         self.create_file("empty.py", "")
         counter = LineCounter()