mirror of
				https://github.com/yt-dlp/yt-dlp.git
				synced 2025-10-31 14:45:14 +00:00 
			
		
		
		
	[orf:tvthek] Fix extraction and modernize (closes #10898)
This commit is contained in:
		| @@ -1,28 +1,28 @@ | |||||||
| # coding: utf-8 | # coding: utf-8 | ||||||
| from __future__ import unicode_literals | from __future__ import unicode_literals | ||||||
|  |  | ||||||
| import json |  | ||||||
| import re | import re | ||||||
| import calendar | import calendar | ||||||
| import datetime | import datetime | ||||||
|  |  | ||||||
| from .common import InfoExtractor | from .common import InfoExtractor | ||||||
|  | from ..compat import compat_str | ||||||
| from ..utils import ( | from ..utils import ( | ||||||
|     HEADRequest, |     HEADRequest, | ||||||
|     unified_strdate, |     unified_strdate, | ||||||
|     ExtractorError, |  | ||||||
|     strip_jsonp, |     strip_jsonp, | ||||||
|     int_or_none, |     int_or_none, | ||||||
|     float_or_none, |     float_or_none, | ||||||
|     determine_ext, |     determine_ext, | ||||||
|     remove_end, |     remove_end, | ||||||
|  |     unescapeHTML, | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  |  | ||||||
| class ORFTVthekIE(InfoExtractor): | class ORFTVthekIE(InfoExtractor): | ||||||
|     IE_NAME = 'orf:tvthek' |     IE_NAME = 'orf:tvthek' | ||||||
|     IE_DESC = 'ORF TVthek' |     IE_DESC = 'ORF TVthek' | ||||||
|     _VALID_URL = r'https?://tvthek\.orf\.at/(?:programs/.+?/episodes|topics?/.+?|program/[^/]+)/(?P<id>\d+)' |     _VALID_URL = r'https?://tvthek\.orf\.at/(?:[^/]+/)+(?P<id>\d+)' | ||||||
|  |  | ||||||
|     _TESTS = [{ |     _TESTS = [{ | ||||||
|         'url': 'http://tvthek.orf.at/program/Aufgetischt/2745173/Aufgetischt-Mit-der-Steirischen-Tafelrunde/8891389', |         'url': 'http://tvthek.orf.at/program/Aufgetischt/2745173/Aufgetischt-Mit-der-Steirischen-Tafelrunde/8891389', | ||||||
| @@ -51,26 +51,23 @@ class ORFTVthekIE(InfoExtractor): | |||||||
|             'skip_download': True,  # rtsp downloads |             'skip_download': True,  # rtsp downloads | ||||||
|         }, |         }, | ||||||
|         '_skip': 'Blocked outside of Austria / Germany', |         '_skip': 'Blocked outside of Austria / Germany', | ||||||
|  |     }, { | ||||||
|  |         'url': 'http://tvthek.orf.at/topic/Fluechtlingskrise/10463081/Heimat-Fremde-Heimat/13879132/Senioren-betreuen-Migrantenkinder/13879141', | ||||||
|  |         'skip_download': True, | ||||||
|  |     }, { | ||||||
|  |         'url': 'http://tvthek.orf.at/profile/Universum/35429', | ||||||
|  |         'skip_download': True, | ||||||
|     }] |     }] | ||||||
|  |  | ||||||
|     def _real_extract(self, url): |     def _real_extract(self, url): | ||||||
|         playlist_id = self._match_id(url) |         playlist_id = self._match_id(url) | ||||||
|         webpage = self._download_webpage(url, playlist_id) |         webpage = self._download_webpage(url, playlist_id) | ||||||
|  |  | ||||||
|         data_json = self._search_regex( |         data_jsb = self._parse_json( | ||||||
|             r'initializeAdworx\((.+?)\);\n', webpage, 'video info') |             self._search_regex( | ||||||
|         all_data = json.loads(data_json) |                 r'<div[^>]+class=(["\']).*?VideoPlaylist.*?\1[^>]+data-jsb=(["\'])(?P<json>.+?)\2', | ||||||
|  |                 webpage, 'playlist', group='json'), | ||||||
|         def get_segments(all_data): |             playlist_id, transform_source=unescapeHTML)['playlist']['videos'] | ||||||
|             for data in all_data: |  | ||||||
|                 if data['name'] in ( |  | ||||||
|                         'Tracker::EPISODE_DETAIL_PAGE_OVER_PROGRAM', |  | ||||||
|                         'Tracker::EPISODE_DETAIL_PAGE_OVER_TOPIC'): |  | ||||||
|                     return data['values']['segments'] |  | ||||||
|  |  | ||||||
|         sdata = get_segments(all_data) |  | ||||||
|         if not sdata: |  | ||||||
|             raise ExtractorError('Unable to extract segments') |  | ||||||
|  |  | ||||||
|         def quality_to_int(s): |         def quality_to_int(s): | ||||||
|             m = re.search('([0-9]+)', s) |             m = re.search('([0-9]+)', s) | ||||||
| @@ -79,8 +76,11 @@ class ORFTVthekIE(InfoExtractor): | |||||||
|             return int(m.group(1)) |             return int(m.group(1)) | ||||||
|  |  | ||||||
|         entries = [] |         entries = [] | ||||||
|         for sd in sdata: |         for sd in data_jsb: | ||||||
|             video_id = sd['id'] |             video_id, title = sd.get('id'), sd.get('title') | ||||||
|  |             if not video_id or not title: | ||||||
|  |                 continue | ||||||
|  |             video_id = compat_str(video_id) | ||||||
|             formats = [{ |             formats = [{ | ||||||
|                 'preference': -10 if fd['delivery'] == 'hls' else None, |                 'preference': -10 if fd['delivery'] == 'hls' else None, | ||||||
|                 'format_id': '%s-%s-%s' % ( |                 'format_id': '%s-%s-%s' % ( | ||||||
| @@ -88,7 +88,7 @@ class ORFTVthekIE(InfoExtractor): | |||||||
|                 'url': fd['src'], |                 'url': fd['src'], | ||||||
|                 'protocol': fd['protocol'], |                 'protocol': fd['protocol'], | ||||||
|                 'quality': quality_to_int(fd['quality']), |                 'quality': quality_to_int(fd['quality']), | ||||||
|             } for fd in sd['playlist_item_array']['sources']] |             } for fd in sd['sources']] | ||||||
|  |  | ||||||
|             # Check for geoblocking. |             # Check for geoblocking. | ||||||
|             # There is a property is_geoprotection, but that's always false |             # There is a property is_geoprotection, but that's always false | ||||||
| @@ -115,14 +115,14 @@ class ORFTVthekIE(InfoExtractor): | |||||||
|             self._check_formats(formats, video_id) |             self._check_formats(formats, video_id) | ||||||
|             self._sort_formats(formats) |             self._sort_formats(formats) | ||||||
|  |  | ||||||
|             upload_date = unified_strdate(sd['created_date']) |             upload_date = unified_strdate(sd.get('created_date')) | ||||||
|             entries.append({ |             entries.append({ | ||||||
|                 '_type': 'video', |                 '_type': 'video', | ||||||
|                 'id': video_id, |                 'id': video_id, | ||||||
|                 'title': sd['header'], |                 'title': title, | ||||||
|                 'formats': formats, |                 'formats': formats, | ||||||
|                 'description': sd.get('description'), |                 'description': sd.get('description'), | ||||||
|                 'duration': int(sd['duration_in_seconds']), |                 'duration': int_or_none(sd.get('duration_in_seconds')), | ||||||
|                 'upload_date': upload_date, |                 'upload_date': upload_date, | ||||||
|                 'thumbnail': sd.get('image_full_url'), |                 'thumbnail': sd.get('image_full_url'), | ||||||
|             }) |             }) | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user
	 Sergey M․
					Sergey M․