85 lines
2.5 KiB
Python
Executable File
85 lines
2.5 KiB
Python
Executable File
#! /usr/bin/env python3
|
|
|
|
"""
|
|
We aren't using either the apt or the pip repositories for the youtube_dl
|
|
as there is a known bug affecting those two versions. The youtube API has changed
|
|
since their release, causing downloads to fail.
|
|
Make sure you use the ./setup.sh script to obtain the latest github release of
|
|
yt_dlp, as this version carries the latest fixes.
|
|
"""
|
|
#import youtube_dl
|
|
import yt_dlp as youtube_dl
|
|
from streamer import Streamer, is_alive
|
|
import subprocess
|
|
import time
|
|
import sys
|
|
import os
|
|
|
|
|
|
class Tuuube(Streamer):
|
|
REST_PATH = 'tuuube'
|
|
|
|
def __init__(self, name):
|
|
super().__init__()
|
|
self.name = name
|
|
self.playback = None
|
|
|
|
def source_path(self):
|
|
return f"/tmp/{self.name}.mp3"
|
|
|
|
def _stream_thread(self):
|
|
if not os.path.exists(self.source_path()) or not os.path.isfile(self.source_path()):
|
|
ydl_opts = {
|
|
'format': 'bestaudio/best',
|
|
'outtmpl': f'/tmp/{self.name}.%(ext)s', # yt_dlp will append %(ext) if not specified,
|
|
'postprocessors': [{ # resulting in `/tmp/file.mp3.mp3` :/
|
|
'key': 'FFmpegExtractAudio',
|
|
'preferredcodec': 'mp3',
|
|
'preferredquality': '192',
|
|
}],
|
|
}
|
|
|
|
try:
|
|
with youtube_dl.YoutubeDL(ydl_opts) as ydl:
|
|
ydl.download([f'https://www.youtube.com/watch?v={self.name}'])
|
|
except Exception as e:
|
|
print(f'File sourcing failed, aborting stream. {e}', file=sys.stderr)
|
|
self.run = False
|
|
return
|
|
|
|
self.playback = subprocess.Popen(
|
|
[
|
|
'/usr/bin/ffmpeg',
|
|
'-re',
|
|
'-stream_loop', '-1',
|
|
'-i', self.source_path(),
|
|
'-c', 'copy',
|
|
'-f', 'rtsp',
|
|
self.stream_address('localhost')
|
|
],
|
|
stdin=subprocess.PIPE,
|
|
stdout=subprocess.DEVNULL,
|
|
stderr=subprocess.DEVNULL
|
|
)
|
|
|
|
while self.run:
|
|
if not is_alive(self.playback):
|
|
print('Playback failed, aborting stream.', file=sys.stderr)
|
|
break
|
|
time.sleep(0.1)
|
|
|
|
self.run = False
|
|
|
|
self.playback.kill()
|
|
self.playback.wait()
|
|
self.playback = None
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
tube = Tuuube('BaW_jenozKc')
|
|
tube.start_stream()
|
|
|
|
while True:
|
|
print(tube.is_streaming())
|
|
time.sleep(1) |