]> Raphaƫl G. Git Repositories - youtubedl/blob - youtube_dl/extractor/safari.py
bd9ee1647d47d47bfc8d8341139c2bf953ecf158
[youtubedl] / youtube_dl / extractor / safari.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 import json
5 import re
6
7 from .common import InfoExtractor
8
9 from ..compat import (
10 compat_parse_qs,
11 compat_str,
12 compat_urlparse,
13 )
14 from ..utils import (
15 ExtractorError,
16 update_url_query,
17 )
18
19
20 class SafariBaseIE(InfoExtractor):
21 _LOGIN_URL = 'https://learning.oreilly.com/accounts/login/'
22 _NETRC_MACHINE = 'safari'
23
24 _API_BASE = 'https://learning.oreilly.com/api/v1'
25 _API_FORMAT = 'json'
26
27 LOGGED_IN = False
28
29 def _real_initialize(self):
30 self._login()
31
32 def _login(self):
33 username, password = self._get_login_info()
34 if username is None:
35 return
36
37 _, urlh = self._download_webpage_handle(
38 'https://learning.oreilly.com/accounts/login-check/', None,
39 'Downloading login page')
40
41 def is_logged(urlh):
42 return 'learning.oreilly.com/home/' in compat_str(urlh.geturl())
43
44 if is_logged(urlh):
45 self.LOGGED_IN = True
46 return
47
48 redirect_url = compat_str(urlh.geturl())
49 parsed_url = compat_urlparse.urlparse(redirect_url)
50 qs = compat_parse_qs(parsed_url.query)
51 next_uri = compat_urlparse.urljoin(
52 'https://api.oreilly.com', qs['next'][0])
53
54 auth, urlh = self._download_json_handle(
55 'https://www.oreilly.com/member/auth/login/', None, 'Logging in',
56 data=json.dumps({
57 'email': username,
58 'password': password,
59 'redirect_uri': next_uri,
60 }).encode(), headers={
61 'Content-Type': 'application/json',
62 'Referer': redirect_url,
63 }, expected_status=400)
64
65 credentials = auth.get('credentials')
66 if (not auth.get('logged_in') and not auth.get('redirect_uri')
67 and credentials):
68 raise ExtractorError(
69 'Unable to login: %s' % credentials, expected=True)
70
71 # oreilly serves two same instances of the following cookies
72 # in Set-Cookie header and expects first one to be actually set
73 for cookie in ('groot_sessionid', 'orm-jwt', 'orm-rt'):
74 self._apply_first_set_cookie_header(urlh, cookie)
75
76 _, urlh = self._download_webpage_handle(
77 auth.get('redirect_uri') or next_uri, None, 'Completing login',)
78
79 if is_logged(urlh):
80 self.LOGGED_IN = True
81 return
82
83 raise ExtractorError('Unable to log in')
84
85
86 class SafariIE(SafariBaseIE):
87 IE_NAME = 'safari'
88 IE_DESC = 'safaribooksonline.com online video'
89 _VALID_URL = r'''(?x)
90 https?://
91 (?:www\.)?(?:safaribooksonline|(?:learning\.)?oreilly)\.com/
92 (?:
93 library/view/[^/]+/(?P<course_id>[^/]+)/(?P<part>[^/?\#&]+)\.html|
94 videos/[^/]+/[^/]+/(?P<reference_id>[^-]+-[^/?\#&]+)
95 )
96 '''
97
98 _TESTS = [{
99 'url': 'https://www.safaribooksonline.com/library/view/hadoop-fundamentals-livelessons/9780133392838/part00.html',
100 'md5': 'dcc5a425e79f2564148652616af1f2a3',
101 'info_dict': {
102 'id': '0_qbqx90ic',
103 'ext': 'mp4',
104 'title': 'Introduction to Hadoop Fundamentals LiveLessons',
105 'timestamp': 1437758058,
106 'upload_date': '20150724',
107 'uploader_id': 'stork',
108 },
109 }, {
110 # non-digits in course id
111 'url': 'https://www.safaribooksonline.com/library/view/create-a-nodejs/100000006A0210/part00.html',
112 'only_matching': True,
113 }, {
114 'url': 'https://www.safaribooksonline.com/library/view/learning-path-red/9780134664057/RHCE_Introduction.html',
115 'only_matching': True,
116 }, {
117 'url': 'https://www.safaribooksonline.com/videos/python-programming-language/9780134217314/9780134217314-PYMC_13_00',
118 'only_matching': True,
119 }, {
120 'url': 'https://learning.oreilly.com/videos/hadoop-fundamentals-livelessons/9780133392838/9780133392838-00_SeriesIntro',
121 'only_matching': True,
122 }, {
123 'url': 'https://www.oreilly.com/library/view/hadoop-fundamentals-livelessons/9780133392838/00_SeriesIntro.html',
124 'only_matching': True,
125 }]
126
127 _PARTNER_ID = '1926081'
128 _UICONF_ID = '29375172'
129
130 def _real_extract(self, url):
131 mobj = re.match(self._VALID_URL, url)
132
133 reference_id = mobj.group('reference_id')
134 if reference_id:
135 video_id = reference_id
136 partner_id = self._PARTNER_ID
137 ui_id = self._UICONF_ID
138 else:
139 video_id = '%s-%s' % (mobj.group('course_id'), mobj.group('part'))
140
141 webpage, urlh = self._download_webpage_handle(url, video_id)
142
143 mobj = re.match(self._VALID_URL, urlh.geturl())
144 reference_id = mobj.group('reference_id')
145 if not reference_id:
146 reference_id = self._search_regex(
147 r'data-reference-id=(["\'])(?P<id>(?:(?!\1).)+)\1',
148 webpage, 'kaltura reference id', group='id')
149 partner_id = self._search_regex(
150 r'data-partner-id=(["\'])(?P<id>(?:(?!\1).)+)\1',
151 webpage, 'kaltura widget id', default=self._PARTNER_ID,
152 group='id')
153 ui_id = self._search_regex(
154 r'data-ui-id=(["\'])(?P<id>(?:(?!\1).)+)\1',
155 webpage, 'kaltura uiconf id', default=self._UICONF_ID,
156 group='id')
157
158 query = {
159 'wid': '_%s' % partner_id,
160 'uiconf_id': ui_id,
161 'flashvars[referenceId]': reference_id,
162 }
163
164 if self.LOGGED_IN:
165 kaltura_session = self._download_json(
166 '%s/player/kaltura_session/?reference_id=%s' % (self._API_BASE, reference_id),
167 video_id, 'Downloading kaltura session JSON',
168 'Unable to download kaltura session JSON', fatal=False)
169 if kaltura_session:
170 session = kaltura_session.get('session')
171 if session:
172 query['flashvars[ks]'] = session
173
174 return self.url_result(update_url_query(
175 'https://cdnapisec.kaltura.com/html5/html5lib/v2.37.1/mwEmbedFrame.php', query),
176 'Kaltura')
177
178
179 class SafariApiIE(SafariBaseIE):
180 IE_NAME = 'safari:api'
181 _VALID_URL = r'https?://(?:www\.)?(?:safaribooksonline|(?:learning\.)?oreilly)\.com/api/v1/book/(?P<course_id>[^/]+)/chapter(?:-content)?/(?P<part>[^/?#&]+)\.html'
182
183 _TESTS = [{
184 'url': 'https://www.safaribooksonline.com/api/v1/book/9780133392838/chapter/part00.html',
185 'only_matching': True,
186 }, {
187 'url': 'https://www.safaribooksonline.com/api/v1/book/9780134664057/chapter/RHCE_Introduction.html',
188 'only_matching': True,
189 }]
190
191 def _real_extract(self, url):
192 mobj = re.match(self._VALID_URL, url)
193 part = self._download_json(
194 url, '%s/%s' % (mobj.group('course_id'), mobj.group('part')),
195 'Downloading part JSON')
196 return self.url_result(part['web_url'], SafariIE.ie_key())
197
198
199 class SafariCourseIE(SafariBaseIE):
200 IE_NAME = 'safari:course'
201 IE_DESC = 'safaribooksonline.com online courses'
202
203 _VALID_URL = r'''(?x)
204 https?://
205 (?:
206 (?:www\.)?(?:safaribooksonline|(?:learning\.)?oreilly)\.com/
207 (?:
208 library/view/[^/]+|
209 api/v1/book|
210 videos/[^/]+
211 )|
212 techbus\.safaribooksonline\.com
213 )
214 /(?P<id>[^/]+)
215 '''
216
217 _TESTS = [{
218 'url': 'https://www.safaribooksonline.com/library/view/hadoop-fundamentals-livelessons/9780133392838/',
219 'info_dict': {
220 'id': '9780133392838',
221 'title': 'Hadoop Fundamentals LiveLessons',
222 },
223 'playlist_count': 22,
224 'skip': 'Requires safaribooksonline account credentials',
225 }, {
226 'url': 'https://www.safaribooksonline.com/api/v1/book/9781449396459/?override_format=json',
227 'only_matching': True,
228 }, {
229 'url': 'http://techbus.safaribooksonline.com/9780134426365',
230 'only_matching': True,
231 }, {
232 'url': 'https://www.safaribooksonline.com/videos/python-programming-language/9780134217314',
233 'only_matching': True,
234 }, {
235 'url': 'https://learning.oreilly.com/videos/hadoop-fundamentals-livelessons/9780133392838',
236 'only_matching': True,
237 }, {
238 'url': 'https://www.oreilly.com/library/view/hadoop-fundamentals-livelessons/9780133392838/',
239 'only_matching': True,
240 }]
241
242 @classmethod
243 def suitable(cls, url):
244 return (False if SafariIE.suitable(url) or SafariApiIE.suitable(url)
245 else super(SafariCourseIE, cls).suitable(url))
246
247 def _real_extract(self, url):
248 course_id = self._match_id(url)
249
250 course_json = self._download_json(
251 '%s/book/%s/?override_format=%s' % (self._API_BASE, course_id, self._API_FORMAT),
252 course_id, 'Downloading course JSON')
253
254 if 'chapters' not in course_json:
255 raise ExtractorError(
256 'No chapters found for course %s' % course_id, expected=True)
257
258 entries = [
259 self.url_result(chapter, SafariApiIE.ie_key())
260 for chapter in course_json['chapters']]
261
262 course_title = course_json['title']
263
264 return self.playlist_result(entries, course_id, course_title)