colin/nix-files
1
0
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

sane-tag-music: handle multiple CLI paths by iterating a layer above

Browse Source
This commit is contained in:
Colin
2024-07-12 04:27:12 +00:00
parent d85ffa8539
commit c16f2473e5
1 changed files with 14 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
pkgs/additional/sane-scripts/src/sane-tag-music
View File

@@ -854,7 +854,11 @@ class Gatherer:
self.derive_tags = derive_tags
def files(self) -> list[tuple[MediaFile, Tags]]:
files = self.walk_files(self.roots)
for root in self.roots:
yield from self.files_from(root)
def files_from(self, root: str) -> list[tuple[MediaFile, Tags]]:
files = self.walk_files(root)
if self.derive_tags:
files = list(files)
@@ -872,17 +876,16 @@ class Gatherer:
if self.media_type is None or f.is_type(self.media_type):
yield f, tags
def walk_paths(self, *roots: str) -> None:
for root in roots:
if os.path.isdir(root):
for dir_, subdirs, files_ in os.walk(root):
for f in files_:
yield os.path.join(dir_, f)
else:
yield root
def walk_paths(self, root: str) -> None:
if os.path.isdir(root):
for dir_, subdirs, files_ in os.walk(root):
for f in files_:
yield os.path.join(dir_, f)
else:
yield root
def walk_files(self, paths: list[str]):
for path_ in self.walk_paths(*paths):
def walk_files(self, root: str):
for path_ in self.walk_paths(root):
file_ = MediaFile.new(path_)
yield file_, Tags()