]>
Commit | Line | Data |
---|---|---|
f226880c PH |
1 | # coding: utf-8 |
2 | from __future__ import unicode_literals | |
3 | ||
4 | import json | |
5 | ||
6 | from .common import InfoExtractor | |
7 | ||
8 | from ..utils import ( | |
9 | ExtractorError, | |
10 | unified_timestamp, | |
11 | ) | |
12 | ||
13 | ||
14 | class TennisTVIE(InfoExtractor): | |
15 | _VALID_URL = r'https?://(?:www\.)?tennistv\.com/videos/(?P<id>[-a-z0-9]+)' | |
16 | _TEST = { | |
17 | 'url': 'https://www.tennistv.com/videos/indian-wells-2018-verdasco-fritz', | |
18 | 'info_dict': { | |
19 | 'id': 'indian-wells-2018-verdasco-fritz', | |
20 | 'ext': 'mp4', | |
21 | 'title': 'Fernando Verdasco v Taylor Fritz', | |
22 | 'description': 're:^After his stunning victory.{174}$', | |
23 | 'thumbnail': 'https://atp-prod.akamaized.net/api/images/v1/images/112831/landscape/1242/0', | |
24 | 'timestamp': 1521017381, | |
25 | 'upload_date': '20180314', | |
26 | }, | |
27 | 'params': { | |
28 | 'skip_download': True, | |
29 | }, | |
30 | 'skip': 'Requires email and password of a subscribed account', | |
31 | } | |
32 | _NETRC_MACHINE = 'tennistv' | |
33 | ||
34 | def _login(self): | |
35 | (username, password) = self._get_login_info() | |
36 | if not username or not password: | |
37 | raise ExtractorError('No login info available, needed for using %s.' % self.IE_NAME, expected=True) | |
38 | ||
39 | login_form = { | |
40 | 'Email': username, | |
41 | 'Password': password, | |
42 | } | |
e6e68069 | 43 | login_json = json.dumps(login_form).encode('utf-8') |
f226880c PH |
44 | headers = { |
45 | 'content-type': 'application/json', | |
46 | 'Referer': 'https://www.tennistv.com/login', | |
47 | 'Origin': 'https://www.tennistv.com', | |
48 | } | |
49 | ||
50 | login_result = self._download_json( | |
51 | 'https://www.tennistv.com/api/users/v1/login', None, | |
52 | note='Logging in', | |
53 | errnote='Login failed (wrong password?)', | |
54 | headers=headers, | |
55 | data=login_json) | |
56 | ||
57 | if login_result['error']['errorCode']: | |
58 | raise ExtractorError('Login failed, %s said: %r' % (self.IE_NAME, login_result['error']['errorMessage'])) | |
59 | ||
60 | if login_result['entitlement'] != 'SUBSCRIBED': | |
61 | self.report_warning('%s may not be subscribed to %s.' % (username, self.IE_NAME)) | |
62 | ||
63 | self._session_token = login_result['sessionToken'] | |
64 | ||
65 | def _real_initialize(self): | |
66 | self._login() | |
67 | ||
68 | def _real_extract(self, url): | |
69 | video_id = self._match_id(url) | |
70 | webpage = self._download_webpage(url, video_id) | |
71 | ||
72 | internal_id = self._search_regex(r'video=([0-9]+)', webpage, 'internal video id') | |
73 | ||
74 | headers = { | |
75 | 'Origin': 'https://www.tennistv.com', | |
76 | 'authorization': 'ATP %s' % self._session_token, | |
77 | 'content-type': 'application/json', | |
78 | 'Referer': url, | |
79 | } | |
80 | check_data = { | |
81 | 'videoID': internal_id, | |
82 | 'VideoUrlType': 'HLSV3', | |
83 | } | |
e6e68069 | 84 | check_json = json.dumps(check_data).encode('utf-8') |
f226880c PH |
85 | check_result = self._download_json( |
86 | 'https://www.tennistv.com/api/users/v1/entitlementchecknondiva', | |
87 | video_id, note='Checking video authorization', headers=headers, data=check_json) | |
88 | formats = self._extract_m3u8_formats(check_result['contentUrl'], video_id, ext='mp4') | |
89 | ||
90 | vdata_url = 'https://www.tennistv.com/api/channels/v1/de/none/video/%s' % video_id | |
91 | vdata = self._download_json(vdata_url, video_id) | |
92 | ||
93 | timestamp = unified_timestamp(vdata['timestamp']) | |
94 | thumbnail = vdata['video']['thumbnailUrl'] | |
95 | description = vdata['displayText']['description'] | |
96 | title = vdata['video']['title'] | |
97 | ||
98 | series = vdata['tour'] | |
99 | venue = vdata['displayText']['venue'] | |
100 | round_str = vdata['seo']['round'] | |
101 | ||
102 | return { | |
103 | 'id': video_id, | |
104 | 'title': title, | |
105 | 'description': description, | |
106 | 'formats': formats, | |
f226880c PH |
107 | 'thumbnail': thumbnail, |
108 | 'timestamp': timestamp, | |
109 | 'series': series, | |
110 | 'season': venue, | |
111 | 'episode': round_str, | |
112 | } |