From cf9474a1fe87d4e812aab0eb5a6d92c4e224ee74 Mon Sep 17 00:00:00 2001
From: Max <m.giller.dev@gmail.com>
Date: Mon, 8 Sep 2025 08:08:06 +0200
Subject: [PATCH] Added files to PathClassification

---
 src/structures.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/structures.py b/src/structures.py
index a348f57..de4935e 100644
--- a/src/structures.py
+++ b/src/structures.py
@@ -59,3 +59,12 @@ class PathClassification:
 
     category: PathCategory = PathCategory.UNCLASSIFIED
     """Category of path media."""
+
+    files: list[FileClassification] = []
+    """List of all files in the path."""
+
+    def get_files_by_category(
+        self, category: FileCategory
+    ) -> filter[FileClassification]:
+        """Get all files of a specific category."""
+        return filter(lambda f: f.category == category, self.files)