mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-08-31 00:34:25 -04:00
165 lines
6.2 KiB
Python
165 lines
6.2 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import json
|
|
import re
|
|
|
|
from .common import InfoExtractor
|
|
from ..compat import (
|
|
# compat_str,
|
|
compat_HTTPError,
|
|
)
|
|
from ..utils import (
|
|
clean_html,
|
|
ExtractorError,
|
|
# remove_end,
|
|
str_or_none,
|
|
strip_or_none,
|
|
unified_timestamp,
|
|
# urljoin,
|
|
)
|
|
|
|
|
|
class PacktPubBaseIE(InfoExtractor):
|
|
# _PACKT_BASE = 'https://www.packtpub.com'
|
|
_STATIC_PRODUCTS_BASE = 'https://static.packt-cdn.com/products/'
|
|
|
|
|
|
class PacktPubIE(PacktPubBaseIE):
|
|
_VALID_URL = r'https?://(?:(?:www\.)?packtpub\.com/mapt|subscription\.packtpub\.com)/video/[^/]+/(?P<course_id>\d+)/(?P<chapter_id>[^/]+)/(?P<id>[^/]+)(?:/(?P<display_id>[^/?&#]+))?'
|
|
|
|
_TESTS = [{
|
|
'url': 'https://www.packtpub.com/mapt/video/web-development/9781787122215/20528/20530/Project+Intro',
|
|
'md5': '1e74bd6cfd45d7d07666f4684ef58f70',
|
|
'info_dict': {
|
|
'id': '20530',
|
|
'ext': 'mp4',
|
|
'title': 'Project Intro',
|
|
'thumbnail': r're:(?i)^https?://.*\.jpg',
|
|
'timestamp': 1490918400,
|
|
'upload_date': '20170331',
|
|
},
|
|
}, {
|
|
'url': 'https://subscription.packtpub.com/video/web_development/9781787122215/20528/20530/project-intro',
|
|
'only_matching': True,
|
|
}, {
|
|
'url': 'https://subscription.packtpub.com/video/programming/9781838988906/p1/video1_1/business-card-project',
|
|
'only_matching': True,
|
|
}]
|
|
_NETRC_MACHINE = 'packtpub'
|
|
_TOKEN = None
|
|
|
|
def _real_initialize(self):
|
|
username, password = self._get_login_info()
|
|
if username is None:
|
|
return
|
|
try:
|
|
self._TOKEN = self._download_json(
|
|
'https://services.packtpub.com/auth-v1/users/tokens', None,
|
|
'Downloading Authorization Token', data=json.dumps({
|
|
'username': username,
|
|
'password': password,
|
|
}).encode())['data']['access']
|
|
except ExtractorError as e:
|
|
if isinstance(e.cause, compat_HTTPError) and e.cause.code in (400, 401, 404):
|
|
message = self._parse_json(e.cause.read().decode(), None)['message']
|
|
raise ExtractorError(message, expected=True)
|
|
raise
|
|
|
|
def _real_extract(self, url):
|
|
course_id, chapter_id, video_id, display_id = re.match(self._VALID_URL, url).groups()
|
|
|
|
headers = {}
|
|
if self._TOKEN:
|
|
headers['Authorization'] = 'Bearer ' + self._TOKEN
|
|
try:
|
|
video_url = self._download_json(
|
|
'https://services.packtpub.com/products-v1/products/%s/%s/%s' % (course_id, chapter_id, video_id), video_id,
|
|
'Downloading JSON video', headers=headers)['data']
|
|
except ExtractorError as e:
|
|
if isinstance(e.cause, compat_HTTPError) and e.cause.code == 400:
|
|
self.raise_login_required('This video is locked')
|
|
raise
|
|
|
|
# TODO: find a better way to avoid duplicating course requests
|
|
# metadata = self._download_json(
|
|
# '%s/products/%s/chapters/%s/sections/%s/metadata'
|
|
# % (self._MAPT_REST, course_id, chapter_id, video_id),
|
|
# video_id)['data']
|
|
|
|
# title = metadata['pageTitle']
|
|
# course_title = metadata.get('title')
|
|
# if course_title:
|
|
# title = remove_end(title, ' - %s' % course_title)
|
|
# timestamp = unified_timestamp(metadata.get('publicationDate'))
|
|
# thumbnail = urljoin(self._PACKT_BASE, metadata.get('filepath'))
|
|
|
|
return {
|
|
'id': video_id,
|
|
'url': video_url,
|
|
'title': display_id or video_id, # title,
|
|
# 'thumbnail': thumbnail,
|
|
# 'timestamp': timestamp,
|
|
}
|
|
|
|
|
|
class PacktPubCourseIE(PacktPubBaseIE):
|
|
_VALID_URL = r'(?P<url>https?://(?:(?:www\.)?packtpub\.com/mapt|subscription\.packtpub\.com)/video/[^/]+/(?P<id>\d+))'
|
|
_TESTS = [{
|
|
'url': 'https://www.packtpub.com/mapt/video/web-development/9781787122215',
|
|
'info_dict': {
|
|
'id': '9781787122215',
|
|
'title': 'Learn Nodejs by building 12 projects [Video]',
|
|
'description': 'md5:489da8d953f416e51927b60a1c7db0aa',
|
|
},
|
|
'playlist_count': 90,
|
|
}, {
|
|
'url': 'https://subscription.packtpub.com/video/web_development/9781787122215',
|
|
'only_matching': True,
|
|
}]
|
|
|
|
@classmethod
|
|
def suitable(cls, url):
|
|
return False if PacktPubIE.suitable(url) else super(
|
|
PacktPubCourseIE, cls).suitable(url)
|
|
|
|
def _real_extract(self, url):
|
|
mobj = re.match(self._VALID_URL, url)
|
|
url, course_id = mobj.group('url', 'id')
|
|
|
|
course = self._download_json(
|
|
self._STATIC_PRODUCTS_BASE + '%s/toc' % course_id, course_id)
|
|
metadata = self._download_json(
|
|
self._STATIC_PRODUCTS_BASE + '%s/summary' % course_id,
|
|
course_id, fatal=False) or {}
|
|
|
|
entries = []
|
|
for chapter_num, chapter in enumerate(course['chapters'], 1):
|
|
chapter_id = str_or_none(chapter.get('id'))
|
|
sections = chapter.get('sections')
|
|
if not chapter_id or not isinstance(sections, list):
|
|
continue
|
|
chapter_info = {
|
|
'chapter': chapter.get('title'),
|
|
'chapter_number': chapter_num,
|
|
'chapter_id': chapter_id,
|
|
}
|
|
for section in sections:
|
|
section_id = str_or_none(section.get('id'))
|
|
if not section_id or section.get('contentType') != 'video':
|
|
continue
|
|
entry = {
|
|
'_type': 'url_transparent',
|
|
'url': '/'.join([url, chapter_id, section_id]),
|
|
'title': strip_or_none(section.get('title')),
|
|
'description': clean_html(section.get('summary')),
|
|
'thumbnail': metadata.get('coverImage'),
|
|
'timestamp': unified_timestamp(metadata.get('publicationDate')),
|
|
'ie_key': PacktPubIE.ie_key(),
|
|
}
|
|
entry.update(chapter_info)
|
|
entries.append(entry)
|
|
|
|
return self.playlist_result(
|
|
entries, course_id, metadata.get('title'),
|
|
clean_html(metadata.get('about')))
|