mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-11-16 01:25:06 +00:00
67 lines
2.6 KiB
Python
67 lines
2.6 KiB
Python
|
from .common import InfoExtractor
|
||
|
from ..utils import (
|
||
|
ExtractorError,
|
||
|
join_nonempty,
|
||
|
traverse_obj,
|
||
|
unified_timestamp,
|
||
|
update_url_query,
|
||
|
)
|
||
|
|
||
|
|
||
|
class Kanal2IE(InfoExtractor):
|
||
|
_VALID_URL = r'https?://kanal2\.postimees\.ee/[^?#]+\?([^#]+&)?id=(?P<id>\d+)'
|
||
|
_TESTS = [{
|
||
|
'note': 'Test standard url (#5575)',
|
||
|
'url': 'https://kanal2.postimees.ee/pluss/video/?id=40792',
|
||
|
'md5': '7ea7b16266ec1798743777df241883dd',
|
||
|
'info_dict': {
|
||
|
'id': '40792',
|
||
|
'ext': 'mp4',
|
||
|
'title': 'Aedniku aabits / Osa 53 (05.08.2016 20:00)',
|
||
|
'thumbnail': r're:https?://.*\.jpg$',
|
||
|
'description': 'md5:53cabf3c5d73150d594747f727431248',
|
||
|
'upload_date': '20160805',
|
||
|
'timestamp': 1470420000,
|
||
|
},
|
||
|
}]
|
||
|
|
||
|
def _real_extract(self, url):
|
||
|
video_id = self._match_id(url)
|
||
|
playlist = self._download_json(
|
||
|
f'https://kanal2.postimees.ee/player/playlist/{video_id}',
|
||
|
video_id, query={'type': 'episodes'},
|
||
|
headers={'X-Requested-With': 'XMLHttpRequest'})
|
||
|
|
||
|
return {
|
||
|
'id': video_id,
|
||
|
'title': join_nonempty(*traverse_obj(playlist, ('info', ('title', 'subtitle'))), delim=' / '),
|
||
|
'description': traverse_obj(playlist, ('info', 'description')),
|
||
|
'thumbnail': traverse_obj(playlist, ('data', 'image')),
|
||
|
'formats': self.get_formats(playlist, video_id),
|
||
|
'timestamp': unified_timestamp(self._search_regex(
|
||
|
r'\((\d{2}\.\d{2}\.\d{4}\s\d{2}:\d{2})\)$',
|
||
|
traverse_obj(playlist, ('info', 'subtitle')), 'timestamp', default='') + ' +0200'),
|
||
|
}
|
||
|
|
||
|
def get_formats(self, playlist, video_id):
|
||
|
path = traverse_obj(playlist, ('data', 'path'))
|
||
|
if not path:
|
||
|
raise ExtractorError('Path value not found in playlist JSON response')
|
||
|
session = self._download_json(
|
||
|
'https://sts.postimees.ee/session/register',
|
||
|
video_id, note='Creating session', errnote='Error creating session',
|
||
|
headers={
|
||
|
'X-Original-URI': path,
|
||
|
'Accept': 'application/json',
|
||
|
})
|
||
|
if session.get('reason') != 'OK' or not session.get('session'):
|
||
|
reason = session.get('reason', 'unknown error')
|
||
|
raise ExtractorError(f'Unable to obtain session: {reason}')
|
||
|
|
||
|
formats = []
|
||
|
for stream in traverse_obj(playlist, ('data', 'streams', ..., 'file')):
|
||
|
formats.extend(self._extract_m3u8_formats(
|
||
|
update_url_query(stream, {'s': session['session']}), video_id, 'mp4'))
|
||
|
|
||
|
return formats
|