support other types of streams such as rtmp

This commit is contained in:
zuckerberg 2021-09-08 21:21:46 -04:00
parent 2a95c597cc
commit 462d06e84a
2 changed files with 2 additions and 2 deletions

View File

@ -4,5 +4,5 @@ pkgs.python3Packages.buildPythonApplication {
pname = "radio";
src = self;
version = "0.1";
propagatedBuildInputs = with pkgs.python3Packages; [ pip ffmpeg-python flask requests pkgs.ffmpeg pkgs.rtmpdump ];
propagatedBuildInputs = with pkgs.python3Packages; [ pip ffmpeg-python flask requests pkgs.ffmpeg ];
}

View File

@ -29,7 +29,7 @@ class Radio(object):
info = downloader.getVideoInfo(self.playingUrl)
if info is None:
return self.play()
elif "direct" in info and info["direct"] == True:
elif ("direct" in info and info["direct"] == True) or ("format_id" in info and info["format_id"] == "rtmp"): # stdout for rtmp in ytdl is broken
self.downloader = downloader.DirectDownloader(self.playingUrl, self.downloadFinished)
else:
self.downloader = downloader.YtdlpDownloader(self.playingUrl, self.downloadFinished)