diff options
author | pukkandan <pukkandan.ytdlp@gmail.com> | 2022-01-30 00:49:55 +0530 |
---|---|---|
committer | pukkandan <pukkandan.ytdlp@gmail.com> | 2022-01-30 00:49:55 +0530 |
commit | bfe72723d8318f8bfcb35dee69a40758df5fa3c0 (patch) | |
tree | 71091d1cd83e5e4784c6abe10fceea00675d0807 /youtube_dl | |
parent | ed99d68bdddfba0440dc81c105d5c0ea7cee7d1c (diff) | |
download | youtube-dl-bfe72723d8318f8bfcb35dee69a40758df5fa3c0.tar.xz |
Use `itertools.islice`
Diffstat (limited to 'youtube_dl')
-rw-r--r-- | youtube_dl/extractor/youtube.py | 11 |
1 files changed, 2 insertions, 9 deletions
diff --git a/youtube_dl/extractor/youtube.py b/youtube_dl/extractor/youtube.py index 578cfcf90..017837e10 100644 --- a/youtube_dl/extractor/youtube.py +++ b/youtube_dl/extractor/youtube.py @@ -3455,17 +3455,10 @@ class YoutubeSearchIE(SearchInfoExtractor, YoutubeBaseInfoExtractor): _SEARCH_PARAMS = 'EgIQAQ%3D%3D' # Videos only _TESTS = [] - def _entries(self, query, n): - total = 0 - for entry in self._search_results(query, self._SEARCH_PARAMS): - yield entry - total += 1 - if total >= n: - return - def _get_n_results(self, query, n): """Get a specified number of results for a query""" - return self.playlist_result(self._entries(query, n), query) + entries = itertools.islice(self._search_results(query, self._SEARCH_PARAMS), 0, None if n == float('inf') else n) + return self.playlist_result(entries, query, query) class YoutubeSearchDateIE(YoutubeSearchIE): |