From b7a8c1bcfaccb797609ea763a1aa3140148bcddc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sergey=20M=E2=80=A4?= Date: Mon, 6 Feb 2017 23:23:40 +0700 Subject: [PATCH] [extractor/generic] Improve rtmp support (closes #11993) --- youtube_dl/extractor/generic.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/youtube_dl/extractor/generic.py b/youtube_dl/extractor/generic.py index 4156cf27db..1fb0e06737 100644 --- a/youtube_dl/extractor/generic.py +++ b/youtube_dl/extractor/generic.py @@ -29,6 +29,7 @@ UnsupportedError, xpath_text, ) +from .commonprotocols import RtmpIE from .brightcove import ( BrightcoveLegacyIE, BrightcoveNewIE, @@ -2487,6 +2488,8 @@ def _playlist_from_matches(matches, getter=None, ie=None): def check_video(vurl): if YoutubeIE.suitable(vurl): return True + if RtmpIE.suitable(vurl): + return True vpath = compat_urlparse.urlparse(vurl).path vext = determine_ext(vpath) return '.' in vpath and vext not in ('swf', 'png', 'jpg', 'srt', 'sbv', 'sub', 'vtt', 'ttml', 'js') @@ -2584,6 +2587,10 @@ def filter_video(urls): entries.append(self.url_result(video_url, 'Youtube')) continue + if RtmpIE.suitable(video_url): + entries.append(self.url_result(video_url, RtmpIE.ie_key())) + continue + # here's a fun little line of code for you: video_id = os.path.splitext(video_id)[0]