1
0
mirror of https://github.com/l1ving/youtube-dl synced 2025-02-03 22:52:57 +08:00

[Aparat] Apply Requested Notes

- Used `_sort_formats` instead of `sorted`
- Removed code dupliction of extracting video URLs
- Put back video URL checking
This commit is contained in:
mxamin 2016-09-25 12:00:25 +03:30
parent 60e555b642
commit 964c304062

View File

@ -2,7 +2,7 @@
from __future__ import unicode_literals from __future__ import unicode_literals
from .common import InfoExtractor from .common import InfoExtractor
from ..utils import ExtractorError from ..utils import ExtractorError, HEADRequest
class AparatIE(InfoExtractor): class AparatIE(InfoExtractor):
@ -29,36 +29,20 @@ class AparatIE(InfoExtractor):
embed_url = 'http://www.aparat.com/video/video/embed/vt/frame/showvideo/yes/videohash/' + video_id embed_url = 'http://www.aparat.com/video/video/embed/vt/frame/showvideo/yes/videohash/' + video_id
webpage = self._download_webpage(embed_url, video_id) webpage = self._download_webpage(embed_url, video_id)
file_list = self._parse_json( patterns = [
self._search_regex( r'fileList\s*=\s*JSON\.parse\(\'([^\']+)\'\)',
r'fileList\s*=\s*JSON\.parse\(\'([^\']+)\'\)', r'fileListPseudo\s*=\s*JSON\.parse\(\'([^\']+)\'\)'
webpage, ]
'file list', file_list = []
default='[]' for p in patterns:
), res = self._parse_json(
video_id self._search_regex(p, webpage, 'file list', default='[]'),
) video_id
file_list_pseudo = self._parse_json( )
self._search_regex( if res:
r'fileListPseudo\s*=\s*JSON\.parse\(\'([^\']+)\'\)', file_list.extend(res[0])
webpage,
'file list pseudo',
default='[]'
),
video_id
)
total_file_list = [] prefs = {
if file_list:
total_file_list.extend(file_list[0])
if file_list_pseudo:
total_file_list.extend(file_list_pseudo[0])
if not total_file_list:
raise ExtractorError('No working video URLs found')
labels = {
'unknown': 0, 'unknown': 0,
'270p': 1, '270p': 1,
'360p': 2, '360p': 2,
@ -66,17 +50,32 @@ class AparatIE(InfoExtractor):
'1080p': 4 '1080p': 4
} }
formats = [] formats = []
for item in total_file_list: for i, item in enumerate(file_list):
# check for video availability
video_url = item['file']
req = HEADRequest(video_url)
res = self._request_webpage(
req,
video_id,
note='Testing video URL %d' % (i + 1),
errnote=False
)
if not res:
continue
video = {} video = {}
video['url'] = item['file'] video['url'] = item['file']
video['format'] = item['type'] video['format'] = item['type']
video['ext'] = 'mp4' video['ext'] = 'mp4'
video_label = item.get('label', 'unknown') video_pref = item.get('label', 'unknown')
video['label'] = labels.get(video_label, 0) video['preference'] = prefs.get(video_pref, -1)
formats.append(video) formats.append(video)
formats = sorted(formats, key=lambda x: x['label']) if not formats:
raise ExtractorError('No working video URLs found')
self._sort_formats(formats)
title = self._search_regex( title = self._search_regex(
r'\s+title:\s*"([^"]+)"', r'\s+title:\s*"([^"]+)"',
webpage, webpage,