tv2hu.py (2411B)
1 # encoding: utf-8 2 from __future__ import unicode_literals 3 4 from .common import InfoExtractor 5 from ..utils import int_or_none 6 7 8 class TV2HuIE(InfoExtractor): 9 IE_NAME = 'tv2.hu' 10 _VALID_URL = r'https?://(?:www\.)?tv2\.hu/(?:[^/]+/)+(?P<id>\d+)_[^/?#]+?\.html' 11 _TESTS = [{ 12 'url': 'http://tv2.hu/ezek_megorultek/217679_ezek-megorultek---1.-adas-1.-resz.html', 13 'md5': '585e58e2e090f34603804bb2c48e98d8', 14 'info_dict': { 15 'id': '217679', 16 'ext': 'mp4', 17 'title': 'Ezek megőrültek! - 1. adás 1. rész', 18 'upload_date': '20160826', 19 'thumbnail': r're:^https?://.*\.jpg$' 20 } 21 }, { 22 'url': 'http://tv2.hu/ezek_megorultek/teljes_adasok/217677_ezek-megorultek---1.-adas-2.-resz.html', 23 'only_matching': True 24 }, { 25 'url': 'http://tv2.hu/musoraink/aktiv/aktiv_teljes_adas/217963_aktiv-teljes-adas---2016.08.30..html', 26 'only_matching': True 27 }] 28 29 def _real_extract(self, url): 30 video_id = self._match_id(url) 31 webpage = self._download_webpage(url, video_id) 32 json_url = self._search_regex( 33 r'jsonUrl\s*=\s*"([^"]+)"', webpage, 'json url') 34 json_data = self._download_json(json_url, video_id) 35 36 formats = [] 37 for b in ('bitrates', 'backupBitrates'): 38 bitrates = json_data.get(b, {}) 39 m3u8_url = bitrates.get('hls') 40 if m3u8_url: 41 formats.extend(self._extract_wowza_formats( 42 m3u8_url, video_id, skip_protocols=['rtmp', 'rtsp'])) 43 44 for mp4_url in bitrates.get('mp4', []): 45 height = int_or_none(self._search_regex( 46 r'\.(\d+)p\.mp4', mp4_url, 'height', default=None)) 47 formats.append({ 48 'format_id': 'http' + ('-%d' % height if height else ''), 49 'url': mp4_url, 50 'height': height, 51 'width': int_or_none(height / 9.0 * 16.0 if height else None), 52 }) 53 self._sort_formats(formats) 54 55 return { 56 'id': video_id, 57 'title': self._og_search_title(webpage).strip(), 58 'thumbnail': self._og_search_thumbnail(webpage), 59 'upload_date': self._search_regex( 60 r'/vod/(\d{8})/', json_url, 'upload_date', default=None), 61 'formats': formats, 62 }