Compare commits
2 Commits
3bf2ad38c2
...
47e8b0d11a
Author | SHA1 | Date | |
---|---|---|---|
47e8b0d11a | |||
ceae96422b |
0
src/__init__.py
Normal file
0
src/__init__.py
Normal file
|
@ -1 +0,0 @@
|
|||
from .track import Track
|
|
@ -1 +1,2 @@
|
|||
from .track import Track
|
||||
from .transcoder import Transcoder
|
||||
|
|
|
@ -5,6 +5,9 @@ class Track:
|
|||
def __init__(self, location: Path):
|
||||
self.path = location
|
||||
|
||||
def __str__(self):
|
||||
return str(self.path)
|
||||
|
||||
@property
|
||||
def filename(self) -> str:
|
||||
return self.path.name
|
|
@ -2,6 +2,7 @@ from pathlib import Path
|
|||
import shutil
|
||||
import subprocess
|
||||
from multiprocessing import Pool
|
||||
from . import Track
|
||||
|
||||
|
||||
class Transcoder:
|
||||
|
@ -29,17 +30,18 @@ class Transcoder:
|
|||
self.copy_album_art(album, album_out)
|
||||
for file in album.iterdir():
|
||||
if file.is_file() and file.suffix.lower() == '.flac':
|
||||
transcode_list.append((str(file), str(album_out / f"{file.stem}.{self.extension}")))
|
||||
transcode_list.append(Track(file))
|
||||
else:
|
||||
print(f"Warning, skipping non-dir '{album}' found in artist '{artist.parts[-1]}'")
|
||||
continue
|
||||
else:
|
||||
print(f"Warning, skipping non-dir '{artist}' found in root")
|
||||
continue
|
||||
self._transcode(transcode_list, self.encoder)
|
||||
self._transcode_single_thread(transcode_list, self.encoder)
|
||||
#self._transcode(transcode_list, self.encoder)
|
||||
|
||||
def _transcode(self, transcode_list: list, encoder: Path, workers=16):
|
||||
worker_args = [(row[0], row[1], encoder) for row in transcode_list]
|
||||
worker_args = [(track, encoder) for track in transcode_list]
|
||||
with Pool(workers) as p:
|
||||
results = p.starmap_async(self.transcode_worker, worker_args)
|
||||
p.close()
|
||||
|
@ -48,25 +50,26 @@ class Transcoder:
|
|||
print(result)
|
||||
|
||||
def _transcode_single_thread(self, transcode_list: list, encoder: Path):
|
||||
worker_args = [(row[0], row[1], encoder) for row in transcode_list]
|
||||
worker_args = [(track, encoder) for track in transcode_list]
|
||||
for args in worker_args:
|
||||
print(f"{args[0]}")
|
||||
self.transcode_worker(*args)
|
||||
|
||||
|
||||
def transcode_worker(self, in_track, out_track, encoder):
|
||||
def transcode_worker(self, track, encoder):
|
||||
enc_filename = encoder.parts[-1]
|
||||
track_output = track.output_file(self.output_root, self.extension)
|
||||
if enc_filename == "opusenc.exe":
|
||||
additional_args = ('--bitrate', '128', '--music')
|
||||
subprocess_args = (str(encoder),) + additional_args + (in_track, out_track)
|
||||
subprocess_args = (str(encoder),) + additional_args + (str(track.path), str(track_output))
|
||||
elif enc_filename == "qaac64.exe":
|
||||
additional_args = (in_track, '-o', out_track, '--threading')
|
||||
additional_args = (str(track), '-o', str(track_output), '--threading')
|
||||
subprocess_args = (str(encoder),) + additional_args
|
||||
try:
|
||||
subprocess.run(subprocess_args, capture_output=True, text=True, check=True)
|
||||
return f"Transcoded '{in_track}' successfully."
|
||||
return f"Transcoded '{track}' successfully."
|
||||
except Exception:
|
||||
return f"ERROR: Transcoding of '{in_track}' failed."
|
||||
return f"ERROR: Transcoding of '{track}' failed."
|
||||
|
||||
@staticmethod
|
||||
def copy_album_art(in_dir: Path, out_dir: Path, file_whitelist: list = None):
|
||||
|
|
Loading…
Reference in New Issue
Block a user