FEAT: Add deletion statistics tracking to main.py

Author: BrenoFariasdaSilva

Folder Size Tagger/main.py

@@ -89,6 +89,9 @@ class BackgroundColors:  # Colors for the terminal
     "Play Sound": True,  # Set to True to play a sound when the program finishes
 }
 
+# Deletion statistics accumulator:
+DELETION_STATS = {"files_deleted": 0, "dirs_deleted": 0, "bytes_deleted": 0}  # Tracks deleted files, directories and total bytes
+
 # Functions Definitions:
 
 
@@ -205,6 +208,21 @@ def convert_bytes_to_gb(size_bytes: int) -> float:
     return rounded_size_gb  # Return rounded gigabyte value
 
 
+def update_deletion_stats(deleted_files: int, deleted_bytes: int) -> dict:
+    """
+    Aggregate deletion statistics increments.
+
+    :param deleted_files: Number of files deleted in this operation.
+    :param deleted_bytes: Total bytes deleted in this operation.
+    :return: Current deletion statistics dictionary.
+    """
+
+    global DELETION_STATS  # Reference global deletion statistics accumulator
+    DELETION_STATS["files_deleted"] += int(deleted_files)  # Increment files deleted counter
+    DELETION_STATS["bytes_deleted"] += int(deleted_bytes)  # Increment bytes deleted accumulator
+    return DELETION_STATS  # Return the updated statistics dictionary
+
+
 def format_directory_name(dirname: str, size_gb: float) -> str:
     """
     Build the target directory name using the GB suffix format.
@@ -268,7 +286,13 @@ def delete_foto_directories(path: str) -> None:
 
         try:  # Protect recursive directory deletion
             import shutil  # Import here for deletion operation
+            files_count = 0  # Initialize file counter for this directory
+            for _root, _dirs, files in os.walk(entry_path):  # Traverse directory to count files
+                files_count += len(files)  # Increment file counter for each file found
+            dir_size = calculate_directory_size_bytes(entry_path)  # Compute directory size before deletion
             shutil.rmtree(entry_path)  # Delete directory and all nested contents
+            update_deletion_stats(files_count, dir_size)  # Aggregate deleted files and bytes metrics
+            DELETION_STATS["dirs_deleted"] += 1  # Increment deleted directories counter
         except (PermissionError, OSError):  # Handle deletion access and OS failures
             continue  # Skip failed deletion and continue processing
 
@@ -329,6 +353,7 @@ def move_video_contents_to_parent(path: str) -> None:
 
         try:  # Protect source directory removal after content move
             os.rmdir(source_directory_path)  # Remove now-empty Video target directory
+            DELETION_STATS["dirs_deleted"] += 1  # Increment deleted directories counter after successful removal
         except (PermissionError, OSError):  # Handle non-empty or inaccessible directory removal
             continue  # Skip failed directory removal and continue processing
 
@@ -429,9 +454,16 @@ def delete_empty_directories(paths: list) -> list:
                 try:  # Protect removal operation for empty directory
                     if not os.listdir(p):  # Verify if directory truly has no entries
                         os.rmdir(p)  # Remove empty directory
+                        DELETION_STATS["dirs_deleted"] += 1  # Increment deleted directories counter for empty dir
                     else:  # Handle case with no size but entries (defensive)
                         import shutil  # Import here for recursive removal
+                        files_count = 0  # Initialize file counter for recursive removal
+                        for _root, _dirs, files in os.walk(p):  # Traverse directory tree to count files
+                            files_count += len(files)  # Increment file counter for each file
+                        dir_size = calculate_directory_size_bytes(p)  # Compute total size before recursive removal
                         shutil.rmtree(p)  # Remove directory recursively
+                        update_deletion_stats(files_count, dir_size)  # Aggregate deleted files and bytes metrics for recursive removal
+                        DELETION_STATS["dirs_deleted"] += 1  # Increment deleted directories counter for recursive removal
                 except (PermissionError, OSError):  # Handle deletion access failures
                     remaining.append(p)  # Keep directory when deletion fails
             else:  # Keep non-empty directories
@@ -633,6 +665,10 @@ def main():
         f"{BackgroundColors.GREEN}Start time: {BackgroundColors.CYAN}{start_time.strftime('%d/%m/%Y - %H:%M:%S')}\n{BackgroundColors.GREEN}Finish time: {BackgroundColors.CYAN}{finish_time.strftime('%d/%m/%Y - %H:%M:%S')}\n{BackgroundColors.GREEN}Execution time: {BackgroundColors.CYAN}{calculate_execution_time(start_time, finish_time)}{Style.RESET_ALL}"
     )  # Output the start and finish times
 
+    print(
+        f"{BackgroundColors.GREEN}Total files deleted: {BackgroundColors.CYAN}{DELETION_STATS['files_deleted']}{BackgroundColors.GREEN} | Total dirs deleted: {BackgroundColors.CYAN}{DELETION_STATS['dirs_deleted']}{BackgroundColors.GREEN} | Total deleted size: {BackgroundColors.CYAN}{convert_bytes_to_gb(DELETION_STATS['bytes_deleted']):.2f}GB{Style.RESET_ALL}"
+    )  # Output aggregated deletion statistics
+    
     print(
         f"{BackgroundColors.BOLD}{BackgroundColors.GREEN}Program finished.{Style.RESET_ALL}"
     )  # Output the end of the program message