[vice] Fix extraction for non en_us videos (closes #12967)
[youtube-dl] / youtube_dl / extractor / vice.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 import re
5 import time
6 import hashlib
7 import json
8
9 from .adobepass import AdobePassIE
10 from .common import InfoExtractor
11 from ..compat import compat_HTTPError
12 from ..utils import (
13     int_or_none,
14     parse_age_limit,
15     str_or_none,
16     parse_duration,
17     ExtractorError,
18     extract_attributes,
19 )
20
21
22 class ViceBaseIE(AdobePassIE):
23     def _extract_preplay_video(self, url, locale, webpage):
24         watch_hub_data = extract_attributes(self._search_regex(
25             r'(?s)(<watch-hub\s*.+?</watch-hub>)', webpage, 'watch hub'))
26         video_id = watch_hub_data['vms-id']
27         title = watch_hub_data['video-title']
28
29         query = {}
30         is_locked = watch_hub_data.get('video-locked') == '1'
31         if is_locked:
32             resource = self._get_mvpd_resource(
33                 'VICELAND', title, video_id,
34                 watch_hub_data.get('video-rating'))
35             query['tvetoken'] = self._extract_mvpd_auth(url, video_id, 'VICELAND', resource)
36
37         # signature generation algorithm is reverse engineered from signatureGenerator in
38         # webpack:///../shared/~/vice-player/dist/js/vice-player.js in
39         # https://www.viceland.com/assets/common/js/web.vendor.bundle.js
40         exp = int(time.time()) + 14400
41         query.update({
42             'exp': exp,
43             'sign': hashlib.sha512(('%s:GET:%d' % (video_id, exp)).encode()).hexdigest(),
44         })
45
46         try:
47             host = 'www.viceland' if is_locked else self._PREPLAY_HOST
48             preplay = self._download_json('https://%s.com/%s/preplay/%s' % (host, locale, video_id), video_id, query=query)
49         except ExtractorError as e:
50             if isinstance(e.cause, compat_HTTPError) and e.cause.code == 400:
51                 error = json.loads(e.cause.read().decode())
52                 raise ExtractorError('%s said: %s' % (self.IE_NAME, error['details']), expected=True)
53             raise
54
55         video_data = preplay['video']
56         base = video_data['base']
57         uplynk_preplay_url = preplay['preplayURL']
58         episode = video_data.get('episode', {})
59         channel = video_data.get('channel', {})
60
61         subtitles = {}
62         cc_url = preplay.get('ccURL')
63         if cc_url:
64             subtitles['en'] = [{
65                 'url': cc_url,
66             }]
67
68         return {
69             '_type': 'url_transparent',
70             'url': uplynk_preplay_url,
71             'id': video_id,
72             'title': title,
73             'description': base.get('body') or base.get('display_body'),
74             'thumbnail': watch_hub_data.get('cover-image') or watch_hub_data.get('thumbnail'),
75             'duration': int_or_none(video_data.get('video_duration')) or parse_duration(watch_hub_data.get('video-duration')),
76             'timestamp': int_or_none(video_data.get('created_at'), 1000),
77             'age_limit': parse_age_limit(video_data.get('video_rating')),
78             'series': video_data.get('show_title') or watch_hub_data.get('show-title'),
79             'episode_number': int_or_none(episode.get('episode_number') or watch_hub_data.get('episode')),
80             'episode_id': str_or_none(episode.get('id') or video_data.get('episode_id')),
81             'season_number': int_or_none(watch_hub_data.get('season')),
82             'season_id': str_or_none(episode.get('season_id')),
83             'uploader': channel.get('base', {}).get('title') or watch_hub_data.get('channel-title'),
84             'uploader_id': str_or_none(channel.get('id')),
85             'subtitles': subtitles,
86             'ie_key': 'UplynkPreplay',
87         }
88
89
90 class ViceIE(ViceBaseIE):
91     _VALID_URL = r'https?://(?:.+?\.)?vice\.com/(?P<locale>[^/]+/)(?:[^/]+/)?videos?/(?P<id>[^/?#&]+)'
92
93     _TESTS = [{
94         'url': 'http://www.vice.com/video/cowboy-capitalists-part-1',
95         'md5': 'e9d77741f9e42ba583e683cd170660f7',
96         'info_dict': {
97             'id': '43cW1mYzpia9IlestBjVpd23Yu3afAfp',
98             'ext': 'flv',
99             'title': 'VICE_COWBOYCAPITALISTS_PART01_v1_VICE_WM_1080p.mov',
100             'duration': 725.983,
101         },
102         'add_ie': ['Ooyala'],
103     }, {
104         'url': 'http://www.vice.com/video/how-to-hack-a-car',
105         'md5': 'a7ecf64ee4fa19b916c16f4b56184ae2',
106         'info_dict': {
107             'id': '3jstaBeXgAs',
108             'ext': 'mp4',
109             'title': 'How to Hack a Car: Phreaked Out (Episode 2)',
110             'description': 'md5:ee95453f7ff495db8efe14ae8bf56f30',
111             'uploader_id': 'MotherboardTV',
112             'uploader': 'Motherboard',
113             'upload_date': '20140529',
114         },
115         'add_ie': ['Youtube'],
116     }, {
117         'url': 'https://video.vice.com/en_us/video/the-signal-from-tolva/5816510690b70e6c5fd39a56',
118         'info_dict': {
119             'id': '5816510690b70e6c5fd39a56',
120             'ext': 'mp4',
121             'uploader': 'Waypoint',
122             'title': 'The Signal From Tölva',
123             'description': 'md5:3927e3c79f9e8094606a2b3c5b5e55d5',
124             'uploader_id': '57f7d621e05ca860fa9ccaf9',
125             'timestamp': 1477941983,
126             'upload_date': '20161031',
127         },
128         'params': {
129             # m3u8 download
130             'skip_download': True,
131         },
132         'add_ie': ['UplynkPreplay'],
133     }, {
134         'url': 'https://video.vice.com/alps/video/ulfs-wien-beruchtigste-grafitti-crew-part-1/581b12b60a0e1f4c0fb6ea2f',
135         'info_dict': {
136             'id': '581b12b60a0e1f4c0fb6ea2f',
137             'ext': 'mp4',
138             'title': 'ULFs - Wien berüchtigste Grafitti Crew - Part 1',
139             'description': '<p>Zwischen Hinterzimmer-Tattoos und U-Bahnschächten erzählen uns die Ulfs, wie es ist, "süchtig nach Sachbeschädigung" zu sein.</p>',
140             'uploader': 'VICE',
141             'uploader_id': '57a204088cb727dec794c67b',
142             'timestamp': 1485368119,
143             'upload_date': '20170125',
144             'age_limit': 14,
145         },
146         'params': {
147             # AES-encrypted m3u8
148             'skip_download': True,
149         },
150         'add_ie': ['UplynkPreplay'],
151     }, {
152         'url': 'https://news.vice.com/video/experimenting-on-animals-inside-the-monkey-lab',
153         'only_matching': True,
154     }, {
155         'url': 'http://www.vice.com/ru/video/big-night-out-ibiza-clive-martin-229',
156         'only_matching': True,
157     }, {
158         'url': 'https://munchies.vice.com/en/videos/watch-the-trailer-for-our-new-series-the-pizza-show',
159         'only_matching': True,
160     }]
161     _PREPLAY_HOST = 'video.vice'
162
163     def _real_extract(self, url):
164         mobj = re.match(self._VALID_URL, url)
165         video_id = mobj.group('id')
166         locale = mobj.group('locale')
167         video_id = self._match_id(url)
168         webpage, urlh = self._download_webpage_handle(url, video_id)
169         embed_code = self._search_regex(
170             r'embedCode=([^&\'"]+)', webpage,
171             'ooyala embed code', default=None)
172         if embed_code:
173             return self.url_result('ooyala:%s' % embed_code, 'Ooyala')
174         youtube_id = self._search_regex(
175             r'data-youtube-id="([^"]+)"', webpage, 'youtube id', default=None)
176         if youtube_id:
177             return self.url_result(youtube_id, 'Youtube')
178         return self._extract_preplay_video(urlh.geturl(), locale, webpage)
179
180
181 class ViceShowIE(InfoExtractor):
182     _VALID_URL = r'https?://(?:.+?\.)?vice\.com/(?:[^/]+/)?show/(?P<id>[^/?#&]+)'
183
184     _TEST = {
185         'url': 'https://munchies.vice.com/en/show/fuck-thats-delicious-2',
186         'info_dict': {
187             'id': 'fuck-thats-delicious-2',
188             'title': "Fuck, That's Delicious",
189             'description': 'Follow the culinary adventures of rapper Action Bronson during his ongoing world tour.',
190         },
191         'playlist_count': 17,
192     }
193
194     def _real_extract(self, url):
195         show_id = self._match_id(url)
196         webpage = self._download_webpage(url, show_id)
197
198         entries = [
199             self.url_result(video_url, ViceIE.ie_key())
200             for video_url, _ in re.findall(
201                 r'<h2[^>]+class="article-title"[^>]+data-id="\d+"[^>]*>\s*<a[^>]+href="(%s.*?)"'
202                 % ViceIE._VALID_URL, webpage)]
203
204         title = self._search_regex(
205             r'<title>(.+?)</title>', webpage, 'title', default=None)
206         if title:
207             title = re.sub(r'(.+)\s*\|\s*.+$', r'\1', title).strip()
208         description = self._html_search_meta('description', webpage, 'description')
209
210         return self.playlist_result(entries, show_id, title, description)