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)