Raphaël G. Git Repositories - youtubedl/blob - youtube_dl/extractor/adultswim.py

   1 # coding: utf-8
   2 from __future__ import unicode_literals
   3
   4 import re
   5
   6 from .common import InfoExtractor
   7 from ..utils import (
   8     determine_ext,
   9     ExtractorError,
  10     float_or_none,
  11     xpath_text,
  12 )
  13
  14
  15 class AdultSwimIE(InfoExtractor):
  16     _VALID_URL = r'https?://(?:www\.)?adultswim\.com/videos/(?P<is_playlist>playlists/)?(?P<show_path>[^/]+)/(?P<episode_path>[^/?#]+)/?'
  17
  18     _TESTS = [{
  19         'url': 'http://adultswim.com/videos/rick-and-morty/pilot',
  20         'playlist': [
  21             {
  22                 'md5': '247572debc75c7652f253c8daa51a14d',
  23                 'info_dict': {
  24                     'id': 'rQxZvXQ4ROaSOqq-or2Mow-0',
  25                     'ext': 'flv',
  26                     'title': 'Rick and Morty - Pilot Part 1',
  27                     'description': "Rick moves in with his daughter's family and establishes himself as a bad influence on his grandson, Morty. "
  28                 },
  29             },
  30             {
  31                 'md5': '77b0e037a4b20ec6b98671c4c379f48d',
  32                 'info_dict': {
  33                     'id': 'rQxZvXQ4ROaSOqq-or2Mow-3',
  34                     'ext': 'flv',
  35                     'title': 'Rick and Morty - Pilot Part 4',
  36                     'description': "Rick moves in with his daughter's family and establishes himself as a bad influence on his grandson, Morty. "
  37                 },
  38             },
  39         ],
  40         'info_dict': {
  41             'id': 'rQxZvXQ4ROaSOqq-or2Mow',
  42             'title': 'Rick and Morty - Pilot',
  43             'description': "Rick moves in with his daughter's family and establishes himself as a bad influence on his grandson, Morty. "
  44         },
  45         'skip': 'This video is only available for registered users',
  46     }, {
  47         'url': 'http://www.adultswim.com/videos/playlists/american-parenting/putting-francine-out-of-business/',
  48         'playlist': [
  49             {
  50                 'md5': '2eb5c06d0f9a1539da3718d897f13ec5',
  51                 'info_dict': {
  52                     'id': '-t8CamQlQ2aYZ49ItZCFog-0',
  53                     'ext': 'flv',
  54                     'title': 'American Dad - Putting Francine Out of Business',
  55                     'description': 'Stan hatches a plan to get Francine out of the real estate business.Watch more American Dad on [adult swim].'
  56                 },
  57             }
  58         ],
  59         'info_dict': {
  60             'id': '-t8CamQlQ2aYZ49ItZCFog',
  61             'title': 'American Dad - Putting Francine Out of Business',
  62             'description': 'Stan hatches a plan to get Francine out of the real estate business.Watch more American Dad on [adult swim].'
  63         },
  64     }, {
  65         'url': 'http://www.adultswim.com/videos/tim-and-eric-awesome-show-great-job/dr-steve-brule-for-your-wine/',
  66         'playlist': [
  67             {
  68                 'md5': '3e346a2ab0087d687a05e1e7f3b3e529',
  69                 'info_dict': {
  70                     'id': 'sY3cMUR_TbuE4YmdjzbIcQ-0',
  71                     'ext': 'mp4',
  72                     'title': 'Tim and Eric Awesome Show Great Job! - Dr. Steve Brule, For Your Wine',
  73                     'description': 'Dr. Brule reports live from Wine Country with a special report on wines.  \r\nWatch Tim and Eric Awesome Show Great Job! episode #20, "Embarrassed" on Adult Swim.\r\n\r\n',
  74                 },
  75             }
  76         ],
  77         'info_dict': {
  78             'id': 'sY3cMUR_TbuE4YmdjzbIcQ',
  79             'title': 'Tim and Eric Awesome Show Great Job! - Dr. Steve Brule, For Your Wine',
  80             'description': 'Dr. Brule reports live from Wine Country with a special report on wines.  \r\nWatch Tim and Eric Awesome Show Great Job! episode #20, "Embarrassed" on Adult Swim.\r\n\r\n',
  81         },
  82         'params': {
  83             # m3u8 download
  84             'skip_download': True,
  85         }
  86     }, {
  87         # heroMetadata.trailer
  88         'url': 'http://www.adultswim.com/videos/decker/inside-decker-a-new-hero/',
  89         'info_dict': {
  90             'id': 'I0LQFQkaSUaFp8PnAWHhoQ',
  91             'ext': 'mp4',
  92             'title': 'Decker - Inside Decker: A New Hero',
  93             'description': 'md5:c916df071d425d62d70c86d4399d3ee0',
  94             'duration': 249.008,
  95         },
  96         'params': {
  97             # m3u8 download
  98             'skip_download': True,
  99         }
 100     }]
 101
 102     @staticmethod
 103     def find_video_info(collection, slug):
 104         for video in collection.get('videos'):
 105             if video.get('slug') == slug:
 106                 return video
 107
 108     @staticmethod
 109     def find_collection_by_linkURL(collections, linkURL):
 110         for collection in collections:
 111             if collection.get('linkURL') == linkURL:
 112                 return collection
 113
 114     @staticmethod
 115     def find_collection_containing_video(collections, slug):
 116         for collection in collections:
 117             for video in collection.get('videos'):
 118                 if video.get('slug') == slug:
 119                     return collection, video
 120         return None, None
 121
 122     def _real_extract(self, url):
 123         mobj = re.match(self._VALID_URL, url)
 124         show_path = mobj.group('show_path')
 125         episode_path = mobj.group('episode_path')
 126         is_playlist = True if mobj.group('is_playlist') else False
 127
 128         webpage = self._download_webpage(url, episode_path)
 129
 130         # Extract the value of `bootstrappedData` from the Javascript in the page.
 131         bootstrapped_data = self._parse_json(self._search_regex(
 132             r'var bootstrappedData = ({.*});', webpage, 'bootstraped data'), episode_path)
 133
 134         # Downloading videos from a /videos/playlist/ URL needs to be handled differently.
 135         # NOTE: We are only downloading one video (the current one) not the playlist
 136         if is_playlist:
 137             collections = bootstrapped_data['playlists']['collections']
 138             collection = self.find_collection_by_linkURL(collections, show_path)
 139             video_info = self.find_video_info(collection, episode_path)
 140
 141             show_title = video_info['showTitle']
 142             segment_ids = [video_info['videoPlaybackID']]
 143         else:
 144             collections = bootstrapped_data['show']['collections']
 145             collection, video_info = self.find_collection_containing_video(collections, episode_path)
 146             # Video wasn't found in the collections, let's try `slugged_video`.
 147             if video_info is None:
 148                 if bootstrapped_data.get('slugged_video', {}).get('slug') == episode_path:
 149                     video_info = bootstrapped_data['slugged_video']
 150             if not video_info:
 151                 video_info = bootstrapped_data.get('heroMetadata', {}).get('trailer').get('video')
 152             if not video_info:
 153                 raise ExtractorError('Unable to find video info')
 154
 155             show = bootstrapped_data['show']
 156             show_title = show['title']
 157             stream = video_info.get('stream')
 158             if stream and stream.get('videoPlaybackID'):
 159                 segment_ids = [stream['videoPlaybackID']]
 160             elif video_info.get('clips'):
 161                 segment_ids = [clip['videoPlaybackID'] for clip in video_info['clips']]
 162             elif video_info.get('videoPlaybackID'):
 163                 segment_ids = [video_info['videoPlaybackID']]
 164             else:
 165                 raise ExtractorError(
 166                     'This video is only available via cable service provider subscription that'
 167                     ' is not currently supported. You may want to use --cookies.'
 168                     if video_info.get('auth') is True else 'Unable to find stream or clips',
 169                     expected=True)
 170
 171         episode_id = video_info['id']
 172         episode_title = video_info['title']
 173         episode_description = video_info['description']
 174         episode_duration = video_info.get('duration')
 175
 176         entries = []
 177         for part_num, segment_id in enumerate(segment_ids):
 178             segment_url = 'http://www.adultswim.com/videos/api/v0/assets?id=%s&platform=desktop' % segment_id
 179
 180             segment_title = '%s - %s' % (show_title, episode_title)
 181             if len(segment_ids) > 1:
 182                 segment_title += ' Part %d' % (part_num + 1)
 183
 184             idoc = self._download_xml(
 185                 segment_url, segment_title,
 186                 'Downloading segment information', 'Unable to download segment information')
 187
 188             segment_duration = float_or_none(
 189                 xpath_text(idoc, './/trt', 'segment duration').strip())
 190
 191             formats = []
 192             file_els = idoc.findall('.//files/file') or idoc.findall('./files/file')
 193
 194             unique_urls = []
 195             unique_file_els = []
 196             for file_el in file_els:
 197                 media_url = file_el.text
 198                 if not media_url or determine_ext(media_url) == 'f4m':
 199                     continue
 200                 if file_el.text not in unique_urls:
 201                     unique_urls.append(file_el.text)
 202                     unique_file_els.append(file_el)
 203
 204             for file_el in unique_file_els:
 205                 bitrate = file_el.attrib.get('bitrate')
 206                 ftype = file_el.attrib.get('type')
 207                 media_url = file_el.text
 208                 if determine_ext(media_url) == 'm3u8':
 209                     formats.extend(self._extract_m3u8_formats(
 210                         media_url, segment_title, 'mp4', preference=0,
 211                         m3u8_id='hls', fatal=False))
 212                 else:
 213                     formats.append({
 214                         'format_id': '%s_%s' % (bitrate, ftype),
 215                         'url': file_el.text.strip(),
 216                         # The bitrate may not be a number (for example: 'iphone')
 217                         'tbr': int(bitrate) if bitrate.isdigit() else None,
 218                     })
 219
 220             self._sort_formats(formats)
 221
 222             entries.append({
 223                 'id': segment_id,
 224                 'title': segment_title,
 225                 'formats': formats,
 226                 'duration': segment_duration,
 227                 'description': episode_description
 228             })
 229
 230         return {
 231             '_type': 'playlist',
 232             'id': episode_id,
 233             'display_id': episode_path,
 234             'entries': entries,
 235             'title': '%s - %s' % (show_title, episode_title),
 236             'description': episode_description,
 237             'duration': episode_duration
 238         }