jriou/yt-mp3-ui
1
0
Fork 0
Code Activity

[extractor] Simplify search extractors

Browse source
This commit is contained in:
pukkandan 2021-10-09 02:09:55 +05:30
commit cc16383ff3
No known key found for this signature in database
GPG key ID: 0F00D95A001F4698
6 changed files with 27 additions and 72 deletions
Download patch file Download diff file Expand all files Collapse all files

10
yt_dlp/extractor/common.py
View file

@ -4,6 +4,7 @@ from __future__ import unicode_literals
import base64
import datetime
import hashlib
import itertools
import json
import netrc
import os
@ -3617,7 +3618,14 @@ class SearchInfoExtractor(InfoExtractor):
return self._get_n_results(query, n)
def _get_n_results(self, query, n):
"""Get a specified number of results for a query"""
"""Get a specified number of results for a query.
Either this function or _search_results must be overridden by subclasses """
return self.playlist_result(
itertools.islice(self._search_results(query), 0, None if n == float('inf') else n),
query, query)
def _search_results(self, query):
"""Returns an iterator of search results"""
raise NotImplementedError('This method must be implemented by subclasses')
@property