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