]> Raphaƫl G. Git Repositories - youtubedl/blob - youtube_dl/extractor/adultswim.py
debian/changelog: Annotate bugs to close.
[youtubedl] / 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 ExtractorError,
9 float_or_none,
10 xpath_text,
11 )
12
13
14 class AdultSwimIE(InfoExtractor):
15 _VALID_URL = r'https?://(?:www\.)?adultswim\.com/videos/(?P<is_playlist>playlists/)?(?P<show_path>[^/]+)/(?P<episode_path>[^/?#]+)/?'
16
17 _TESTS = [{
18 'url': 'http://adultswim.com/videos/rick-and-morty/pilot',
19 'playlist': [
20 {
21 'md5': '247572debc75c7652f253c8daa51a14d',
22 'info_dict': {
23 'id': 'rQxZvXQ4ROaSOqq-or2Mow-0',
24 'ext': 'flv',
25 'title': 'Rick and Morty - Pilot Part 1',
26 'description': "Rick moves in with his daughter's family and establishes himself as a bad influence on his grandson, Morty. "
27 },
28 },
29 {
30 'md5': '77b0e037a4b20ec6b98671c4c379f48d',
31 'info_dict': {
32 'id': 'rQxZvXQ4ROaSOqq-or2Mow-3',
33 'ext': 'flv',
34 'title': 'Rick and Morty - Pilot Part 4',
35 'description': "Rick moves in with his daughter's family and establishes himself as a bad influence on his grandson, Morty. "
36 },
37 },
38 ],
39 'info_dict': {
40 'id': 'rQxZvXQ4ROaSOqq-or2Mow',
41 'title': 'Rick and Morty - Pilot',
42 'description': "Rick moves in with his daughter's family and establishes himself as a bad influence on his grandson, Morty. "
43 }
44 }, {
45 'url': 'http://www.adultswim.com/videos/playlists/american-parenting/putting-francine-out-of-business/',
46 'playlist': [
47 {
48 'md5': '2eb5c06d0f9a1539da3718d897f13ec5',
49 'info_dict': {
50 'id': '-t8CamQlQ2aYZ49ItZCFog-0',
51 'ext': 'flv',
52 'title': 'American Dad - Putting Francine Out of Business',
53 'description': 'Stan hatches a plan to get Francine out of the real estate business.Watch more American Dad on [adult swim].'
54 },
55 }
56 ],
57 'info_dict': {
58 'id': '-t8CamQlQ2aYZ49ItZCFog',
59 'title': 'American Dad - Putting Francine Out of Business',
60 'description': 'Stan hatches a plan to get Francine out of the real estate business.Watch more American Dad on [adult swim].'
61 },
62 }, {
63 'url': 'http://www.adultswim.com/videos/tim-and-eric-awesome-show-great-job/dr-steve-brule-for-your-wine/',
64 'playlist': [
65 {
66 'md5': '3e346a2ab0087d687a05e1e7f3b3e529',
67 'info_dict': {
68 'id': 'sY3cMUR_TbuE4YmdjzbIcQ-0',
69 'ext': 'flv',
70 'title': 'Tim and Eric Awesome Show Great Job! - Dr. Steve Brule, For Your Wine',
71 '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',
72 },
73 }
74 ],
75 'info_dict': {
76 'id': 'sY3cMUR_TbuE4YmdjzbIcQ',
77 'title': 'Tim and Eric Awesome Show Great Job! - Dr. Steve Brule, For Your Wine',
78 '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',
79 },
80 }]
81
82 @staticmethod
83 def find_video_info(collection, slug):
84 for video in collection.get('videos'):
85 if video.get('slug') == slug:
86 return video
87
88 @staticmethod
89 def find_collection_by_linkURL(collections, linkURL):
90 for collection in collections:
91 if collection.get('linkURL') == linkURL:
92 return collection
93
94 @staticmethod
95 def find_collection_containing_video(collections, slug):
96 for collection in collections:
97 for video in collection.get('videos'):
98 if video.get('slug') == slug:
99 return collection, video
100 return None, None
101
102 def _real_extract(self, url):
103 mobj = re.match(self._VALID_URL, url)
104 show_path = mobj.group('show_path')
105 episode_path = mobj.group('episode_path')
106 is_playlist = True if mobj.group('is_playlist') else False
107
108 webpage = self._download_webpage(url, episode_path)
109
110 # Extract the value of `bootstrappedData` from the Javascript in the page.
111 bootstrapped_data = self._parse_json(self._search_regex(
112 r'var bootstrappedData = ({.*});', webpage, 'bootstraped data'), episode_path)
113
114 # Downloading videos from a /videos/playlist/ URL needs to be handled differently.
115 # NOTE: We are only downloading one video (the current one) not the playlist
116 if is_playlist:
117 collections = bootstrapped_data['playlists']['collections']
118 collection = self.find_collection_by_linkURL(collections, show_path)
119 video_info = self.find_video_info(collection, episode_path)
120
121 show_title = video_info['showTitle']
122 segment_ids = [video_info['videoPlaybackID']]
123 else:
124 collections = bootstrapped_data['show']['collections']
125 collection, video_info = self.find_collection_containing_video(collections, episode_path)
126
127 # Video wasn't found in the collections, let's try `slugged_video`.
128 if video_info is None:
129 if bootstrapped_data.get('slugged_video', {}).get('slug') == episode_path:
130 video_info = bootstrapped_data['slugged_video']
131 else:
132 raise ExtractorError('Unable to find video info')
133
134 show = bootstrapped_data['show']
135 show_title = show['title']
136 segment_ids = [clip['videoPlaybackID'] for clip in video_info['clips']]
137
138 episode_id = video_info['id']
139 episode_title = video_info['title']
140 episode_description = video_info['description']
141 episode_duration = video_info.get('duration')
142
143 entries = []
144 for part_num, segment_id in enumerate(segment_ids):
145 segment_url = 'http://www.adultswim.com/videos/api/v0/assets?id=%s&platform=mobile' % segment_id
146
147 segment_title = '%s - %s' % (show_title, episode_title)
148 if len(segment_ids) > 1:
149 segment_title += ' Part %d' % (part_num + 1)
150
151 idoc = self._download_xml(
152 segment_url, segment_title,
153 'Downloading segment information', 'Unable to download segment information')
154
155 segment_duration = float_or_none(
156 xpath_text(idoc, './/trt', 'segment duration').strip())
157
158 formats = []
159 file_els = idoc.findall('.//files/file')
160
161 for file_el in file_els:
162 bitrate = file_el.attrib.get('bitrate')
163 ftype = file_el.attrib.get('type')
164
165 formats.append({
166 'format_id': '%s_%s' % (bitrate, ftype),
167 'url': file_el.text.strip(),
168 # The bitrate may not be a number (for example: 'iphone')
169 'tbr': int(bitrate) if bitrate.isdigit() else None,
170 'quality': 1 if ftype == 'hd' else -1
171 })
172
173 self._sort_formats(formats)
174
175 entries.append({
176 'id': segment_id,
177 'title': segment_title,
178 'formats': formats,
179 'duration': segment_duration,
180 'description': episode_description
181 })
182
183 return {
184 '_type': 'playlist',
185 'id': episode_id,
186 'display_id': episode_path,
187 'entries': entries,
188 'title': '%s - %s' % (show_title, episode_title),
189 'description': episode_description,
190 'duration': episode_duration
191 }