[mtvservices] Fix ext for RTMP streams
authorYen Chi Hsuan <yan12125@gmail.com>
Sat, 18 Jun 2016 04:19:06 +0000 (12:19 +0800)
committerYen Chi Hsuan <yan12125@gmail.com>
Sat, 18 Jun 2016 04:19:06 +0000 (12:19 +0800)
youtube_dl/extractor/mtv.py

index 8a638a47c56c272d57cc2bed99f04fdd71b789a2..dd06395891ac02ce81b40efd471776cb080d328d 100644 (file)
@@ -85,9 +85,10 @@ class MTVServicesInfoExtractor(InfoExtractor):
                 rtmp_video_url = rendition.find('./src').text
                 if rtmp_video_url.endswith('siteunavail.png'):
                     continue
+                new_url = self._transform_rtmp_url(rtmp_video_url)
                 formats.append({
-                    'ext': ext,
-                    'url': self._transform_rtmp_url(rtmp_video_url),
+                    'ext': 'flv' if new_url.startswith('rtmp') else ext,
+                    'url': new_url,
                     'format_id': rendition.get('bitrate'),
                     'width': int(rendition.get('width')),
                     'height': int(rendition.get('height')),