kumi/yt-dlp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

[tudou] Use InAdvancePagedList (closes #8884)

Browse source
This commit is contained in:
Yen Chi Hsuan 2016-03-20 15:45:31 +08:00
parent 3c20208eff
commit 664bcd80b9
1 changed files with 9 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
youtube_dl/extractor/tudou.py
View file

@ -6,6 +6,7 @@ from .common import InfoExtractor
from ..compat import compat_str from ..compat import compat_str
from ..utils import ( from ..utils import (
int_or_none, int_or_none,
InAdvancePagedList,
float_or_none, float_or_none,
unescapeHTML, unescapeHTML,
) )
@ -75,15 +76,16 @@ class TudouIE(InfoExtractor):
quality = sorted(filter(lambda k: k.isdigit(), segments.keys()), quality = sorted(filter(lambda k: k.isdigit(), segments.keys()),
key=lambda k: int(k))[-1] key=lambda k: int(k))[-1]
parts = segments[quality] parts = segments[quality]
result = []
len_parts = len(parts) len_parts = len(parts)
if len_parts > 1: if len_parts > 1:
self.to_screen('%s: found %s parts' % (video_id, len_parts)) self.to_screen('%s: found %s parts' % (video_id, len_parts))
for part in parts:
def part_func(partnum):
part = parts[partnum]
part_id = part['k'] part_id = part['k']
final_url = self._url_for_id(part_id, quality) final_url = self._url_for_id(part_id, quality)
ext = (final_url.split('?')[0]).split('.')[-1] ext = (final_url.split('?')[0]).split('.')[-1]
part_info = { return [{
'id': '%s' % part_id, 'id': '%s' % part_id,
'url': final_url, 'url': final_url,
'ext': ext, 'ext': ext,
@ -97,12 +99,13 @@ class TudouIE(InfoExtractor):
'http_headers': { 'http_headers': {
'Referer': self._PLAYER_URL, 'Referer': self._PLAYER_URL,
}, },
} }]
result.append(part_info)
entries = InAdvancePagedList(part_func, len_parts, 1)
return { return {
'_type': 'multi_video', '_type': 'multi_video',
'entries': result, 'entries': entries,
'id': video_id, 'id': video_id,
'title': title, 'title': title,
} }