From: Philipp Hagemeister Date: Wed, 17 Dec 2014 22:18:06 +0000 (+0100) Subject: [dvtv] PEP8 and correct format sorting (#4502) X-Git-Url: http://git.bitcoin.ninja/?a=commitdiff_plain;h=b9465395cbc222fc9609b7881098c7246ee93989;p=youtube-dl [dvtv] PEP8 and correct format sorting (#4502) --- diff --git a/youtube_dl/extractor/dvtv.py b/youtube_dl/extractor/dvtv.py index a52bb3469..af552831c 100644 --- a/youtube_dl/extractor/dvtv.py +++ b/youtube_dl/extractor/dvtv.py @@ -2,62 +2,62 @@ from __future__ import unicode_literals -import re -import json from .common import InfoExtractor from ..utils import ( - ExtractorError, - js_to_json, - unescapeHTML + js_to_json, + unescapeHTML ) class DVTVIE(InfoExtractor): - IE_NAME = 'dvtv' - IE_DESC = 'http://video.aktualne.cz/dvtv/' - - _VALID_URL = r'http://video\.aktualne\.cz/dvtv/(?P[a-z0-9-]+/r~[0-9a-f]{32})/?' - - _TESTS = [{ - 'url': 'http://video.aktualne.cz/dvtv/vondra-o-ceskem-stoleti-pri-pohledu-na-havla-mi-bylo-trapne/r~e5efe9ca855511e4833a0025900fea04/', - 'md5': '75800f964fa0f82939a2914563301f72', - 'info_dict': { - 'id': 'e5efe9ca855511e4833a0025900fea04', - 'ext': 'webm', - 'title': 'Vondra o Českém století: Při pohledu na Havla mi bylo trapně' - } - }, { - 'url': 'http://video.aktualne.cz/dvtv/stropnicky-policie-vrbetice-preventivne-nekontrolovala/r~82ed4322849211e4a10c0025900fea04/', - 'md5': 'd50455195a67a94c57f931360cc68a1b', - 'info_dict': { - 'id': '82ed4322849211e4a10c0025900fea04', - 'ext': 'webm', - 'title': 'Stropnický: Policie Vrbětice preventivně nekontrolovala' - } - }] - - def _real_extract(self, url): - video_id = self._match_id(url) - webpage = self._download_webpage(url, video_id) - - code = self._search_regex(r'embedData[0-9a-f]{32}\[\'asset\'\] = (\{.+?\});', webpage, 'video JSON', flags=re.DOTALL) - payload = self._parse_json(code, video_id, transform_source=js_to_json) - formats = [] - for source in payload['sources']: - formats.append({ - 'url': source['file'], - 'ext': source['type'][6:], - 'format': '%s %s' % (source['type'][6:], source['label']), - 'format_id': '%s-%s' % (source['type'][6:], source['label']), - 'resolution': source['label'], - 'fps': 25, - 'preference': -1 if source['type'][6:] == 'mp4' and source['label'] == '720p' else -2 - }) - - return { - 'id': video_id[-32:], - 'display_id': video_id[:-35], - 'title': unescapeHTML(payload['title']), - 'thumbnail': 'http:%s' % payload['image'], - 'formats': formats - } + IE_NAME = 'dvtv' + IE_DESC = 'http://video.aktualne.cz/dvtv/' + + _VALID_URL = r'http://video\.aktualne\.cz/dvtv/(?P[a-z0-9-]+/r~[0-9a-f]{32})/?' + + _TESTS = [{ + 'url': 'http://video.aktualne.cz/dvtv/vondra-o-ceskem-stoleti-pri-pohledu-na-havla-mi-bylo-trapne/r~e5efe9ca855511e4833a0025900fea04/', + 'md5': '75800f964fa0f82939a2914563301f72', + 'info_dict': { + 'id': 'e5efe9ca855511e4833a0025900fea04', + 'ext': 'webm', + 'title': 'Vondra o Českém století: Při pohledu na Havla mi bylo trapně' + } + }, { + 'url': 'http://video.aktualne.cz/dvtv/stropnicky-policie-vrbetice-preventivne-nekontrolovala/r~82ed4322849211e4a10c0025900fea04/', + 'md5': '6388f1941b48537dbd28791f712af8bf', + 'info_dict': { + 'id': '82ed4322849211e4a10c0025900fea04', + 'ext': 'mp4', + 'title': 'Stropnický: Policie Vrbětice preventivně nekontrolovala' + } + }] + + def _real_extract(self, url): + video_id = self._match_id(url) + webpage = self._download_webpage(url, video_id) + + code = self._search_regex( + r'(?s)embedData[0-9a-f]{32}\[\'asset\'\] = (\{.+?\});', + webpage, 'video JSON') + payload = self._parse_json(code, video_id, transform_source=js_to_json) + formats = [] + for source in payload['sources']: + ext = source['type'][6:] + formats.append({ + 'url': source['file'], + 'ext': ext, + 'format': '%s %s' % (ext, source['label']), + 'format_id': '%s-%s' % (ext, source['label']), + 'height': int(source['label'].rstrip('p')), + 'fps': 25, + }) + self._sort_formats(formats) + + return { + 'id': video_id[-32:], + 'display_id': video_id[:-35], + 'title': unescapeHTML(payload['title']), + 'thumbnail': 'http:%s' % payload['image'], + 'formats': formats + }