]> Raphaël G. Git Repositories - youtubedl/blob - youtube_dl/extractor/bilibili.py
Prepare to release.
[youtubedl] / youtube_dl / extractor / bilibili.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 import hashlib
5 import re
6
7 from .common import InfoExtractor
8 from ..compat import (
9 compat_parse_qs,
10 compat_urlparse,
11 )
12 from ..utils import (
13 ExtractorError,
14 int_or_none,
15 float_or_none,
16 parse_iso8601,
17 smuggle_url,
18 strip_jsonp,
19 unified_timestamp,
20 unsmuggle_url,
21 urlencode_postdata,
22 )
23
24
25 class BiliBiliIE(InfoExtractor):
26 _VALID_URL = r'https?://(?:www\.|bangumi\.|)bilibili\.(?:tv|com)/(?:video/av|anime/(?P<anime_id>\d+)/play#)(?P<id>\d+)'
27
28 _TESTS = [{
29 'url': 'http://www.bilibili.tv/video/av1074402/',
30 'md5': '9fa226fe2b8a9a4d5a69b4c6a183417e',
31 'info_dict': {
32 'id': '1074402',
33 'ext': 'mp4',
34 'title': '【金坷垃】金泡沫',
35 'description': 'md5:ce18c2a2d2193f0df2917d270f2e5923',
36 'duration': 308.315,
37 'timestamp': 1398012660,
38 'upload_date': '20140420',
39 'thumbnail': r're:^https?://.+\.jpg',
40 'uploader': '菊子桑',
41 'uploader_id': '156160',
42 },
43 }, {
44 # Tested in BiliBiliBangumiIE
45 'url': 'http://bangumi.bilibili.com/anime/1869/play#40062',
46 'only_matching': True,
47 }, {
48 'url': 'http://bangumi.bilibili.com/anime/5802/play#100643',
49 'md5': '3f721ad1e75030cc06faf73587cfec57',
50 'info_dict': {
51 'id': '100643',
52 'ext': 'mp4',
53 'title': 'CHAOS;CHILD',
54 'description': '如果你是神明,并且能够让妄想成为现实。那你会进行怎么样的妄想?是淫靡的世界?独裁社会?毁灭性的制裁?还是……2015年,涩谷。从6年前发生的大灾害“涩谷地震”之后复兴了的这个街区里新设立的私立高中...',
55 },
56 'skip': 'Geo-restricted to China',
57 }]
58
59 _APP_KEY = '84956560bc028eb7'
60 _BILIBILI_KEY = '94aba54af9065f71de72f5508f1cd42e'
61
62 def _report_error(self, result):
63 if 'message' in result:
64 raise ExtractorError('%s said: %s' % (self.IE_NAME, result['message']), expected=True)
65 elif 'code' in result:
66 raise ExtractorError('%s returns error %d' % (self.IE_NAME, result['code']), expected=True)
67 else:
68 raise ExtractorError('Can\'t extract Bangumi episode ID')
69
70 def _real_extract(self, url):
71 url, smuggled_data = unsmuggle_url(url, {})
72
73 mobj = re.match(self._VALID_URL, url)
74 video_id = mobj.group('id')
75 anime_id = mobj.group('anime_id')
76 webpage = self._download_webpage(url, video_id)
77
78 if 'anime/' not in url:
79 cid = compat_parse_qs(self._search_regex(
80 [r'EmbedPlayer\([^)]+,\s*"([^"]+)"\)',
81 r'<iframe[^>]+src="https://secure\.bilibili\.com/secure,([^"]+)"'],
82 webpage, 'player parameters'))['cid'][0]
83 else:
84 if 'no_bangumi_tip' not in smuggled_data:
85 self.to_screen('Downloading episode %s. To download all videos in anime %s, re-run youtube-dl with %s' % (
86 video_id, anime_id, compat_urlparse.urljoin(url, '//bangumi.bilibili.com/anime/%s' % anime_id)))
87 headers = {
88 'Content-Type': 'application/x-www-form-urlencoded; charset=UTF-8',
89 }
90 headers.update(self.geo_verification_headers())
91
92 js = self._download_json(
93 'http://bangumi.bilibili.com/web_api/get_source', video_id,
94 data=urlencode_postdata({'episode_id': video_id}),
95 headers=headers)
96 if 'result' not in js:
97 self._report_error(js)
98 cid = js['result']['cid']
99
100 payload = 'appkey=%s&cid=%s&otype=json&quality=2&type=mp4' % (self._APP_KEY, cid)
101 sign = hashlib.md5((payload + self._BILIBILI_KEY).encode('utf-8')).hexdigest()
102
103 video_info = self._download_json(
104 'http://interface.bilibili.com/playurl?%s&sign=%s' % (payload, sign),
105 video_id, note='Downloading video info page',
106 headers=self.geo_verification_headers())
107
108 if 'durl' not in video_info:
109 self._report_error(video_info)
110
111 entries = []
112
113 for idx, durl in enumerate(video_info['durl']):
114 formats = [{
115 'url': durl['url'],
116 'filesize': int_or_none(durl['size']),
117 }]
118 for backup_url in durl.get('backup_url', []):
119 formats.append({
120 'url': backup_url,
121 # backup URLs have lower priorities
122 'preference': -2 if 'hd.mp4' in backup_url else -3,
123 })
124
125 self._sort_formats(formats)
126
127 entries.append({
128 'id': '%s_part%s' % (video_id, idx),
129 'duration': float_or_none(durl.get('length'), 1000),
130 'formats': formats,
131 })
132
133 title = self._html_search_regex('<h1[^>]+title="([^"]+)">', webpage, 'title')
134 description = self._html_search_meta('description', webpage)
135 timestamp = unified_timestamp(self._html_search_regex(
136 r'<time[^>]+datetime="([^"]+)"', webpage, 'upload time', default=None))
137 thumbnail = self._html_search_meta(['og:image', 'thumbnailUrl'], webpage)
138
139 # TODO 'view_count' requires deobfuscating Javascript
140 info = {
141 'id': video_id,
142 'title': title,
143 'description': description,
144 'timestamp': timestamp,
145 'thumbnail': thumbnail,
146 'duration': float_or_none(video_info.get('timelength'), scale=1000),
147 }
148
149 uploader_mobj = re.search(
150 r'<a[^>]+href="(?:https?:)?//space\.bilibili\.com/(?P<id>\d+)"[^>]+title="(?P<name>[^"]+)"',
151 webpage)
152 if uploader_mobj:
153 info.update({
154 'uploader': uploader_mobj.group('name'),
155 'uploader_id': uploader_mobj.group('id'),
156 })
157
158 for entry in entries:
159 entry.update(info)
160
161 if len(entries) == 1:
162 return entries[0]
163 else:
164 for idx, entry in enumerate(entries):
165 entry['id'] = '%s_part%d' % (video_id, (idx + 1))
166
167 return {
168 '_type': 'multi_video',
169 'id': video_id,
170 'title': title,
171 'description': description,
172 'entries': entries,
173 }
174
175
176 class BiliBiliBangumiIE(InfoExtractor):
177 _VALID_URL = r'https?://bangumi\.bilibili\.com/anime/(?P<id>\d+)'
178
179 IE_NAME = 'bangumi.bilibili.com'
180 IE_DESC = 'BiliBili番剧'
181
182 _TESTS = [{
183 'url': 'http://bangumi.bilibili.com/anime/1869',
184 'info_dict': {
185 'id': '1869',
186 'title': '混沌武士',
187 'description': 'md5:6a9622b911565794c11f25f81d6a97d2',
188 },
189 'playlist_count': 26,
190 }, {
191 'url': 'http://bangumi.bilibili.com/anime/1869',
192 'info_dict': {
193 'id': '1869',
194 'title': '混沌武士',
195 'description': 'md5:6a9622b911565794c11f25f81d6a97d2',
196 },
197 'playlist': [{
198 'md5': '91da8621454dd58316851c27c68b0c13',
199 'info_dict': {
200 'id': '40062',
201 'ext': 'mp4',
202 'title': '混沌武士',
203 'description': '故事发生在日本的江户时代。风是一个小酒馆的打工女。一日,酒馆里来了一群恶霸,虽然他们的举动令风十分不满,但是毕竟风只是一届女流,无法对他们采取什么行动,只能在心里嘟哝。这时,酒家里又进来了个“不良份子...',
204 'timestamp': 1414538739,
205 'upload_date': '20141028',
206 'episode': '疾风怒涛 Tempestuous Temperaments',
207 'episode_number': 1,
208 },
209 }],
210 'params': {
211 'playlist_items': '1',
212 },
213 }]
214
215 @classmethod
216 def suitable(cls, url):
217 return False if BiliBiliIE.suitable(url) else super(BiliBiliBangumiIE, cls).suitable(url)
218
219 def _real_extract(self, url):
220 bangumi_id = self._match_id(url)
221
222 # Sometimes this API returns a JSONP response
223 season_info = self._download_json(
224 'http://bangumi.bilibili.com/jsonp/seasoninfo/%s.ver' % bangumi_id,
225 bangumi_id, transform_source=strip_jsonp)['result']
226
227 entries = [{
228 '_type': 'url_transparent',
229 'url': smuggle_url(episode['webplay_url'], {'no_bangumi_tip': 1}),
230 'ie_key': BiliBiliIE.ie_key(),
231 'timestamp': parse_iso8601(episode.get('update_time'), delimiter=' '),
232 'episode': episode.get('index_title'),
233 'episode_number': int_or_none(episode.get('index')),
234 } for episode in season_info['episodes']]
235
236 entries = sorted(entries, key=lambda entry: entry.get('episode_number'))
237
238 return self.playlist_result(
239 entries, bangumi_id,
240 season_info.get('bangumi_title'), season_info.get('evaluate'))