[YoutubeDL] format spec: Do not fail when a filter gives an empty result

For example with 'best[height<40]' we ended getting a 'IndexError: list index out of range'.
This commit is contained in:
Jaime Marquínez Ferrándiz 2015-07-04 21:41:09 +02:00
parent f5f4a27a96
commit bb8e553662
2 changed files with 11 additions and 1 deletions

View File

@ -15,7 +15,7 @@
from youtube_dl.compat import compat_str from youtube_dl.compat import compat_str
from youtube_dl.extractor import YoutubeIE from youtube_dl.extractor import YoutubeIE
from youtube_dl.postprocessor.common import PostProcessor from youtube_dl.postprocessor.common import PostProcessor
from youtube_dl.utils import match_filter_func from youtube_dl.utils import ExtractorError, match_filter_func
TEST_URL = 'http://localhost/sample.mp4' TEST_URL = 'http://localhost/sample.mp4'
@ -362,6 +362,13 @@ def test_format_filtering(self):
downloaded_ids = [info['format_id'] for info in ydl.downloaded_info_dicts] downloaded_ids = [info['format_id'] for info in ydl.downloaded_info_dicts]
self.assertEqual(downloaded_ids, ['B', 'C', 'D']) self.assertEqual(downloaded_ids, ['B', 'C', 'D'])
ydl = YDL({'format': 'best[height<40]'})
try:
ydl.process_ie_result(info_dict)
except ExtractorError:
pass
self.assertEqual(ydl.downloaded_info_dicts, [])
class TestYoutubeDL(unittest.TestCase): class TestYoutubeDL(unittest.TestCase):
def test_subtitles(self): def test_subtitles(self):

View File

@ -1005,6 +1005,9 @@ def selector_function(formats):
format_spec = selector.selector format_spec = selector.selector
def selector_function(formats): def selector_function(formats):
formats = list(formats)
if not formats:
return
if format_spec == 'all': if format_spec == 'all':
for f in formats: for f in formats:
yield f yield f