36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
from pathlib import Path
|
|
import subprocess
|
|
import logging
|
|
|
|
import ffmpeg
|
|
|
|
logger = logging.getLogger('transcoder')
|
|
|
|
|
|
def open(infile: Path, bufsize: int = 4096):
|
|
"""
|
|
Return a stream of mp3 data for the given path on disk.
|
|
|
|
If the requested path is an mp3, return a filehandle on the file. Otherwise,
|
|
invoke ffmpeg to tranascode whatever was requested to mp3 format and return
|
|
a pipe to ffmpeg's STDOUT.
|
|
"""
|
|
suffix = infile.suffix.lower()
|
|
if suffix == '.mp3':
|
|
logger.debug(f"Not transcoding mp3 {infile = }")
|
|
return infile.open('rb', buffering=bufsize)
|
|
|
|
ffmpeg_args = (
|
|
ffmpeg
|
|
.input(str(infile))
|
|
.output('-', format='mp3', q=2)
|
|
.global_args('-hide_banner', '-loglevel', 'quiet')
|
|
.compile()
|
|
)
|
|
|
|
# Force close STDIN to prevent ffmpeg from trying to read from it. silly ffmpeg.
|
|
proc = subprocess.Popen(ffmpeg_args, bufsize=bufsize, stdout=subprocess.PIPE, stdin=subprocess.PIPE)
|
|
proc.stdin.close()
|
|
logger.debug(f"Spawned ffmpeg (PID {proc.pid}) with args {ffmpeg_args = }")
|
|
return proc.stdout
|