radio/transcoder.py
2021-09-08 19:32:46 -04:00

32 lines
993 B
Python

import ffmpeg
from logger import logger
from stream_listener import StreamListener
from stream import StreamSource
import os
# converts the stream to mp3 before sending to the long-lasting mp3 connection
# write to a fifo file instead of directly via stdin to fool ffmpeg into thinking it has seekable input
# https://lists.libav.org/pipermail/ffmpeg-user/2007-May/008917.html
class Transcoder(StreamSource):
def __init__(self, upstream):
self.process = ( ffmpeg
.input('pipe:')
.output('pipe:', format='mp3')
.run_async(pipe_stdin=True, pipe_stdout=True, pipe_stderr=True)
)
logger.add(self.process.stderr, "transcoder.log")
self.listener = StreamListener(upstream, self)
self.listener.start()
def stop(self):
self.listener.stop()
self.process.stdin.close()
self.process.stdout.close()
self.process.stderr.close()
def getStream(self):
return self.process.stdout
def write(self, chunk):
self.process.stdin.write(chunk)