From 6574182bfeb7eb2f8e5082ce3c6e92cdb4cc2233 Mon Sep 17 00:00:00 2001 From: grglcy Date: Fri, 10 May 2024 16:07:44 +0100 Subject: [PATCH] Copy album art and log files to output dir --- src/main.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/main.py b/src/main.py index 527c3eb..692dfbd 100644 --- a/src/main.py +++ b/src/main.py @@ -1,5 +1,6 @@ import argparse import subprocess +import shutil from subprocess import CalledProcessError from pathlib import Path @@ -42,6 +43,7 @@ def transcode(transcode_list: list, encoder: Path): def main(input_dir: Path, output_dir: Path, encoder: Path): transcode_list = [] + file_whitelist = ['.jpg', '.jpeg', '.png', '.log'] for artist in input_dir.iterdir(): if artist.is_dir(): artist_out = Path(output_dir) / artist.name @@ -57,11 +59,11 @@ def main(input_dir: Path, output_dir: Path, encoder: Path): print(f"Skipping '{artist.parts[-1]} / {album.parts[-1]}'") else: album_out.mkdir() - # todo: copy files not on blacklist - # todo: create blacklist for track in album.iterdir(): if track.is_file() and track.suffix.lower() == '.flac': transcode_list.append((str(track), str(album_out / f"{track.stem}.opus"))) + elif track.is_file() and track.suffix.lower() in file_whitelist: + shutil.copy(track, album_out / track.name) else: print(f"Warning, skipping non-dir '{album}' found in artist '{artist.parts[-1]}'") continue