mirror of
				https://github.com/yt-dlp/yt-dlp.git
				synced 2025-11-04 08:35:12 +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
 |