add tracks_only param to get folder tracks

This commit is contained in:
mungai-njoroge
2023-12-02 11:11:24 +03:00
parent b03e721ee5
commit 9dff629e1f
3 changed files with 85 additions and 34 deletions
+6 -2
View File
@@ -44,8 +44,9 @@ class GetFilesAndDirs:
Get files and folders from a directory.
"""
def __init__(self, path: str) -> None:
def __init__(self, path: str, tracks_only=False) -> None:
self.path = path
self.tracks_only = tracks_only
def __call__(self) -> tuple[list[Track], list[Folder]]:
try:
@@ -80,6 +81,9 @@ class GetFilesAndDirs:
files = [f["path"] for f in files_]
tracks = TrackStore.get_tracks_by_filepaths(files)
folders = get_folders(dirs)
folders = []
if not self.tracks_only:
folders = get_folders(dirs)
return tracks, folders