mirror of
https://github.com/KevinMidboe/spotify-downloader.git
synced 2025-10-29 18:00:15 +00:00
113 lines
3.7 KiB
Python
113 lines
3.7 KiB
Python
import subprocess
|
|
import os
|
|
|
|
from spotdl.encode import EncoderBase
|
|
from spotdl.encode.exceptions import EncoderNotFoundError
|
|
from spotdl.encode.exceptions import FFmpegNotFoundError
|
|
|
|
import logging
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
# Key: from format
|
|
# Subkey: to format
|
|
RULES = {
|
|
"m4a": {
|
|
"mp3": "-codec:v copy -codec:a libmp3lame -ar 48000",
|
|
"opus": "-codec:a libopus -vbr on",
|
|
"m4a": "-acodec copy",
|
|
"flac": "-codec:a flac -ar 48000",
|
|
},
|
|
"opus": {
|
|
"mp3": "-codec:a libmp3lame -ar 48000",
|
|
"m4a": "-cutoff 20000 -codec:a aac -ar 48000",
|
|
"flac": "-codec:a flac -ar 48000",
|
|
},
|
|
}
|
|
|
|
|
|
class EncoderFFmpeg(EncoderBase):
|
|
def __init__(self, encoder_path="ffmpeg"):
|
|
_loglevel = "-hide_banner -nostats -v panic"
|
|
_additional_arguments = ["-b:a", "192k", "-vn"]
|
|
try:
|
|
super().__init__(encoder_path, _loglevel, _additional_arguments)
|
|
except EncoderNotFoundError as e:
|
|
raise FFmpegNotFoundError(e.args[0])
|
|
self._rules = RULES
|
|
|
|
def set_trim_silence(self):
|
|
self.set_argument("-af silenceremove=start_periods=1")
|
|
|
|
def get_encoding(self, path):
|
|
return super().get_encoding(path)
|
|
|
|
def _generate_encoding_arguments(self, input_encoding, target_encoding):
|
|
initial_arguments = self._rules.get(input_encoding)
|
|
if initial_arguments is None:
|
|
raise TypeError(
|
|
'The input format ("{}") is not supported.'.format(
|
|
input_encoding,
|
|
))
|
|
arguments = initial_arguments.get(target_encoding)
|
|
if arguments is None:
|
|
raise TypeError(
|
|
'The output format ("{}") is not supported.'.format(
|
|
target_encoding,
|
|
))
|
|
return arguments
|
|
|
|
def set_debuglog(self):
|
|
self._loglevel = "-loglevel debug"
|
|
|
|
def _generate_encode_command(self, input_path, target_file,
|
|
input_encoding=None, target_encoding=None):
|
|
if input_encoding is None:
|
|
input_encoding = self.get_encoding(input_path)
|
|
if target_encoding is None:
|
|
target_encoding = self.get_encoding(target_file)
|
|
arguments = self._generate_encoding_arguments(
|
|
input_encoding,
|
|
target_encoding
|
|
)
|
|
command = [self.encoder_path] \
|
|
+ ["-y", "-nostdin"] \
|
|
+ self._loglevel.split() \
|
|
+ ["-i", input_path] \
|
|
+ arguments.split() \
|
|
+ self._additional_arguments \
|
|
+ ["-f", self.target_format_from_encoding(target_encoding)] \
|
|
+ [target_file]
|
|
|
|
return command
|
|
|
|
def re_encode(self, input_path, target_file, target_encoding=None, delete_original=False):
|
|
encode_command = self._generate_encode_command(
|
|
input_path,
|
|
target_file,
|
|
target_encoding=target_encoding
|
|
)
|
|
logger.debug("Calling FFmpeg with:\n{command}".format(
|
|
command=encode_command,
|
|
))
|
|
process = subprocess.Popen(encode_command)
|
|
process.wait()
|
|
encode_successful = process.returncode == 0
|
|
if encode_successful and delete_original:
|
|
os.remove(input_path)
|
|
return process
|
|
|
|
def re_encode_from_stdin(self, input_encoding, target_file, target_encoding=None):
|
|
encode_command = self._generate_encode_command(
|
|
"-",
|
|
target_file,
|
|
input_encoding=input_encoding,
|
|
target_encoding=target_encoding,
|
|
)
|
|
logger.debug("Calling FFmpeg with:\n{command}".format(
|
|
command=encode_command,
|
|
))
|
|
process = subprocess.Popen(encode_command, stdin=subprocess.PIPE)
|
|
return process
|
|
|