[youtube] Fix extraction.
[youtube-dl] / youtube_dl / extractor / linkedin.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 import re
5
6 from .common import InfoExtractor
7 from ..utils import (
8     ExtractorError,
9     float_or_none,
10     int_or_none,
11     urlencode_postdata,
12     urljoin,
13 )
14
15
16 class LinkedInLearningBaseIE(InfoExtractor):
17     _NETRC_MACHINE = 'linkedin'
18     _LOGIN_URL = 'https://www.linkedin.com/uas/login?trk=learning'
19
20     def _call_api(self, course_slug, fields, video_slug=None, resolution=None):
21         query = {
22             'courseSlug': course_slug,
23             'fields': fields,
24             'q': 'slugs',
25         }
26         sub = ''
27         if video_slug:
28             query.update({
29                 'videoSlug': video_slug,
30                 'resolution': '_%s' % resolution,
31             })
32             sub = ' %dp' % resolution
33         api_url = 'https://www.linkedin.com/learning-api/detailedCourses'
34         return self._download_json(
35             api_url, video_slug, 'Downloading%s JSON metadata' % sub, headers={
36                 'Csrf-Token': self._get_cookies(api_url)['JSESSIONID'].value,
37             }, query=query)['elements'][0]
38
39     def _get_urn_id(self, video_data):
40         urn = video_data.get('urn')
41         if urn:
42             mobj = re.search(r'urn:li:lyndaCourse:\d+,(\d+)', urn)
43             if mobj:
44                 return mobj.group(1)
45
46     def _get_video_id(self, video_data, course_slug, video_slug):
47         return self._get_urn_id(video_data) or '%s/%s' % (course_slug, video_slug)
48
49     def _real_initialize(self):
50         email, password = self._get_login_info()
51         if email is None:
52             return
53
54         login_page = self._download_webpage(
55             self._LOGIN_URL, None, 'Downloading login page')
56         action_url = urljoin(self._LOGIN_URL, self._search_regex(
57             r'<form[^>]+action=(["\'])(?P<url>.+?)\1', login_page, 'post url',
58             default='https://www.linkedin.com/uas/login-submit', group='url'))
59         data = self._hidden_inputs(login_page)
60         data.update({
61             'session_key': email,
62             'session_password': password,
63         })
64         login_submit_page = self._download_webpage(
65             action_url, None, 'Logging in',
66             data=urlencode_postdata(data))
67         error = self._search_regex(
68             r'<span[^>]+class="error"[^>]*>\s*(.+?)\s*</span>',
69             login_submit_page, 'error', default=None)
70         if error:
71             raise ExtractorError(error, expected=True)
72
73
74 class LinkedInLearningIE(LinkedInLearningBaseIE):
75     IE_NAME = 'linkedin:learning'
76     _VALID_URL = r'https?://(?:www\.)?linkedin\.com/learning/(?P<course_slug>[^/]+)/(?P<id>[^/?#]+)'
77     _TEST = {
78         'url': 'https://www.linkedin.com/learning/programming-foundations-fundamentals/welcome?autoplay=true',
79         'md5': 'a1d74422ff0d5e66a792deb996693167',
80         'info_dict': {
81             'id': '90426',
82             'ext': 'mp4',
83             'title': 'Welcome',
84             'timestamp': 1430396150.82,
85             'upload_date': '20150430',
86         },
87     }
88
89     def _real_extract(self, url):
90         course_slug, video_slug = re.match(self._VALID_URL, url).groups()
91
92         video_data = None
93         formats = []
94         for width, height in ((640, 360), (960, 540), (1280, 720)):
95             video_data = self._call_api(
96                 course_slug, 'selectedVideo', video_slug, height)['selectedVideo']
97
98             video_url_data = video_data.get('url') or {}
99             progressive_url = video_url_data.get('progressiveUrl')
100             if progressive_url:
101                 formats.append({
102                     'format_id': 'progressive-%dp' % height,
103                     'url': progressive_url,
104                     'height': height,
105                     'width': width,
106                     'source_preference': 1,
107                 })
108
109         title = video_data['title']
110
111         audio_url = video_data.get('audio', {}).get('progressiveUrl')
112         if audio_url:
113             formats.append({
114                 'abr': 64,
115                 'ext': 'm4a',
116                 'format_id': 'audio',
117                 'url': audio_url,
118                 'vcodec': 'none',
119             })
120
121         streaming_url = video_url_data.get('streamingUrl')
122         if streaming_url:
123             formats.extend(self._extract_m3u8_formats(
124                 streaming_url, video_slug, 'mp4',
125                 'm3u8_native', m3u8_id='hls', fatal=False))
126
127         self._sort_formats(formats, ('width', 'height', 'source_preference', 'tbr', 'abr'))
128
129         return {
130             'id': self._get_video_id(video_data, course_slug, video_slug),
131             'title': title,
132             'formats': formats,
133             'thumbnail': video_data.get('defaultThumbnail'),
134             'timestamp': float_or_none(video_data.get('publishedOn'), 1000),
135             'duration': int_or_none(video_data.get('durationInSeconds')),
136         }
137
138
139 class LinkedInLearningCourseIE(LinkedInLearningBaseIE):
140     IE_NAME = 'linkedin:learning:course'
141     _VALID_URL = r'https?://(?:www\.)?linkedin\.com/learning/(?P<id>[^/?#]+)'
142     _TEST = {
143         'url': 'https://www.linkedin.com/learning/programming-foundations-fundamentals',
144         'info_dict': {
145             'id': 'programming-foundations-fundamentals',
146             'title': 'Programming Foundations: Fundamentals',
147             'description': 'md5:76e580b017694eb89dc8e8923fff5c86',
148         },
149         'playlist_mincount': 61,
150     }
151
152     @classmethod
153     def suitable(cls, url):
154         return False if LinkedInLearningIE.suitable(url) else super(LinkedInLearningCourseIE, cls).suitable(url)
155
156     def _real_extract(self, url):
157         course_slug = self._match_id(url)
158         course_data = self._call_api(course_slug, 'chapters,description,title')
159
160         entries = []
161         for chapter_number, chapter in enumerate(course_data.get('chapters', []), 1):
162             chapter_title = chapter.get('title')
163             chapter_id = self._get_urn_id(chapter)
164             for video in chapter.get('videos', []):
165                 video_slug = video.get('slug')
166                 if not video_slug:
167                     continue
168                 entries.append({
169                     '_type': 'url_transparent',
170                     'id': self._get_video_id(video, course_slug, video_slug),
171                     'title': video.get('title'),
172                     'url': 'https://www.linkedin.com/learning/%s/%s' % (course_slug, video_slug),
173                     'chapter': chapter_title,
174                     'chapter_number': chapter_number,
175                     'chapter_id': chapter_id,
176                     'ie_key': LinkedInLearningIE.ie_key(),
177                 })
178
179         return self.playlist_result(
180             entries, course_slug,
181             course_data.get('title'),
182             course_data.get('description'))