diff --git a/src/dir/album.py b/src/dir/album.py index 20fb30f..2ccc801 100644 --- a/src/dir/album.py +++ b/src/dir/album.py @@ -7,6 +7,11 @@ class Album(Directory): def __init__(self, path: Path, log: Log): super().__init__(path, log, 'ALB') + @property + def all_files(self) -> list: + # todo: handle unexpected dirs + return [file.path for file in self.contents] + def populate(self, log: Log) -> list: contents = list() for e in self.path.iterdir(): diff --git a/src/dir/directory.py b/src/dir/directory.py index 8e0f37e..4aea62d 100644 --- a/src/dir/directory.py +++ b/src/dir/directory.py @@ -30,6 +30,14 @@ class Directory(ABC): def __str__(self): return self.name + @property + def all_files(self) -> list: + files = list() + for c in self: + # todo: handle unexpected files + files += c.all_files() + return files + @property def name(self): return self.path.name