]> Raphaƫl G. Git Repositories - youtubedl/blob - youtube_dl/extractor/bliptv.py
Prepare for release.
[youtubedl] / youtube_dl / extractor / bliptv.py
1 import datetime
2 import json
3 import os
4 import re
5 import socket
6
7 from .common import InfoExtractor
8 from ..utils import (
9 compat_http_client,
10 compat_parse_qs,
11 compat_str,
12 compat_urllib_error,
13 compat_urllib_parse_urlparse,
14 compat_urllib_request,
15
16 ExtractorError,
17 unescapeHTML,
18 )
19
20
21 class BlipTVIE(InfoExtractor):
22 """Information extractor for blip.tv"""
23
24 _VALID_URL = r'^(?:https?://)?(?:\w+\.)?blip\.tv/((.+/)|(play/)|(api\.swf#))(.+)$'
25 _URL_EXT = r'^.*\.([a-z0-9]+)$'
26 IE_NAME = u'blip.tv'
27 _TEST = {
28 u'url': u'http://blip.tv/cbr/cbr-exclusive-gotham-city-imposters-bats-vs-jokerz-short-3-5796352',
29 u'file': u'5779306.m4v',
30 u'md5': u'80baf1ec5c3d2019037c1c707d676b9f',
31 u'info_dict': {
32 u"upload_date": u"20111205",
33 u"description": u"md5:9bc31f227219cde65e47eeec8d2dc596",
34 u"uploader": u"Comic Book Resources - CBR TV",
35 u"title": u"CBR EXCLUSIVE: \"Gotham City Imposters\" Bats VS Jokerz Short 3"
36 }
37 }
38
39 def report_direct_download(self, title):
40 """Report information extraction."""
41 self.to_screen(u'%s: Direct download detected' % title)
42
43 def _real_extract(self, url):
44 mobj = re.match(self._VALID_URL, url)
45 if mobj is None:
46 raise ExtractorError(u'Invalid URL: %s' % url)
47
48 # See https://github.com/rg3/youtube-dl/issues/857
49 api_mobj = re.match(r'http://a\.blip\.tv/api\.swf#(?P<video_id>[\d\w]+)', url)
50 if api_mobj is not None:
51 url = 'http://blip.tv/play/g_%s' % api_mobj.group('video_id')
52 urlp = compat_urllib_parse_urlparse(url)
53 if urlp.path.startswith('/play/'):
54 request = compat_urllib_request.Request(url)
55 response = compat_urllib_request.urlopen(request)
56 redirecturl = response.geturl()
57 rurlp = compat_urllib_parse_urlparse(redirecturl)
58 file_id = compat_parse_qs(rurlp.fragment)['file'][0].rpartition('/')[2]
59 url = 'http://blip.tv/a/a-' + file_id
60 return self._real_extract(url)
61
62
63 if '?' in url:
64 cchar = '&'
65 else:
66 cchar = '?'
67 json_url = url + cchar + 'skin=json&version=2&no_wrap=1'
68 request = compat_urllib_request.Request(json_url)
69 request.add_header('User-Agent', 'iTunes/10.6.1')
70 self.report_extraction(mobj.group(1))
71 info = None
72 try:
73 urlh = compat_urllib_request.urlopen(request)
74 if urlh.headers.get('Content-Type', '').startswith('video/'): # Direct download
75 basename = url.split('/')[-1]
76 title,ext = os.path.splitext(basename)
77 title = title.decode('UTF-8')
78 ext = ext.replace('.', '')
79 self.report_direct_download(title)
80 info = {
81 'id': title,
82 'url': url,
83 'uploader': None,
84 'upload_date': None,
85 'title': title,
86 'ext': ext,
87 'urlhandle': urlh
88 }
89 except (compat_urllib_error.URLError, compat_http_client.HTTPException, socket.error) as err:
90 raise ExtractorError(u'ERROR: unable to download video info webpage: %s' % compat_str(err))
91 if info is None: # Regular URL
92 try:
93 json_code_bytes = urlh.read()
94 json_code = json_code_bytes.decode('utf-8')
95 except (compat_urllib_error.URLError, compat_http_client.HTTPException, socket.error) as err:
96 raise ExtractorError(u'Unable to read video info webpage: %s' % compat_str(err))
97
98 try:
99 json_data = json.loads(json_code)
100 if 'Post' in json_data:
101 data = json_data['Post']
102 else:
103 data = json_data
104
105 upload_date = datetime.datetime.strptime(data['datestamp'], '%m-%d-%y %H:%M%p').strftime('%Y%m%d')
106 if 'additionalMedia' in data:
107 formats = sorted(data['additionalMedia'], key=lambda f: int(f['media_height']))
108 best_format = formats[-1]
109 video_url = best_format['url']
110 else:
111 video_url = data['media']['url']
112 umobj = re.match(self._URL_EXT, video_url)
113 if umobj is None:
114 raise ValueError('Can not determine filename extension')
115 ext = umobj.group(1)
116
117 info = {
118 'id': data['item_id'],
119 'url': video_url,
120 'uploader': data['display_name'],
121 'upload_date': upload_date,
122 'title': data['title'],
123 'ext': ext,
124 'format': data['media']['mimeType'],
125 'thumbnail': data['thumbnailUrl'],
126 'description': data['description'],
127 'player_url': data['embedUrl'],
128 'user_agent': 'iTunes/10.6.1',
129 }
130 except (ValueError,KeyError) as err:
131 raise ExtractorError(u'Unable to parse video information: %s' % repr(err))
132
133 return [info]
134
135
136 class BlipTVUserIE(InfoExtractor):
137 """Information Extractor for blip.tv users."""
138
139 _VALID_URL = r'(?:(?:(?:https?://)?(?:\w+\.)?blip\.tv/)|bliptvuser:)([^/]+)/*$'
140 _PAGE_SIZE = 12
141 IE_NAME = u'blip.tv:user'
142
143 def _real_extract(self, url):
144 # Extract username
145 mobj = re.match(self._VALID_URL, url)
146 if mobj is None:
147 raise ExtractorError(u'Invalid URL: %s' % url)
148
149 username = mobj.group(1)
150
151 page_base = 'http://m.blip.tv/pr/show_get_full_episode_list?users_id=%s&lite=0&esi=1'
152
153 page = self._download_webpage(url, username, u'Downloading user page')
154 mobj = re.search(r'data-users-id="([^"]+)"', page)
155 page_base = page_base % mobj.group(1)
156
157
158 # Download video ids using BlipTV Ajax calls. Result size per
159 # query is limited (currently to 12 videos) so we need to query
160 # page by page until there are no video ids - it means we got
161 # all of them.
162
163 video_ids = []
164 pagenum = 1
165
166 while True:
167 url = page_base + "&page=" + str(pagenum)
168 page = self._download_webpage(url, username,
169 u'Downloading video ids from page %d' % pagenum)
170
171 # Extract video identifiers
172 ids_in_page = []
173
174 for mobj in re.finditer(r'href="/([^"]+)"', page):
175 if mobj.group(1) not in ids_in_page:
176 ids_in_page.append(unescapeHTML(mobj.group(1)))
177
178 video_ids.extend(ids_in_page)
179
180 # A little optimization - if current page is not
181 # "full", ie. does not contain PAGE_SIZE video ids then
182 # we can assume that this page is the last one - there
183 # are no more ids on further pages - no need to query
184 # again.
185
186 if len(ids_in_page) < self._PAGE_SIZE:
187 break
188
189 pagenum += 1
190
191 urls = [u'http://blip.tv/%s' % video_id for video_id in video_ids]
192 url_entries = [self.url_result(vurl, 'BlipTV') for vurl in urls]
193 return [self.playlist_result(url_entries, playlist_title = username)]