mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-11-04 23:43:20 +00:00
[GoogleSearch] Fix extractor
This commit is contained in:
parent
b8b3f4562a
commit
c533c89ce1
|
@ -8,36 +8,33 @@
|
||||||
|
|
||||||
class GoogleSearchIE(SearchInfoExtractor):
|
class GoogleSearchIE(SearchInfoExtractor):
|
||||||
IE_DESC = 'Google Video search'
|
IE_DESC = 'Google Video search'
|
||||||
_MAX_RESULTS = 1000
|
|
||||||
IE_NAME = 'video.google:search'
|
IE_NAME = 'video.google:search'
|
||||||
_SEARCH_KEY = 'gvsearch'
|
_SEARCH_KEY = 'gvsearch'
|
||||||
_WORKING = False
|
_TESTS = [{
|
||||||
_TEST = {
|
|
||||||
'url': 'gvsearch15:python language',
|
'url': 'gvsearch15:python language',
|
||||||
'info_dict': {
|
'info_dict': {
|
||||||
'id': 'python language',
|
'id': 'python language',
|
||||||
'title': 'python language',
|
'title': 'python language',
|
||||||
},
|
},
|
||||||
'playlist_count': 15,
|
'playlist_count': 15,
|
||||||
}
|
}]
|
||||||
|
_PAGE_SIZE = 100
|
||||||
|
|
||||||
def _search_results(self, query):
|
def _search_results(self, query):
|
||||||
for pagenum in itertools.count():
|
for pagenum in itertools.count():
|
||||||
webpage = self._download_webpage(
|
webpage = self._download_webpage(
|
||||||
'http://www.google.com/search',
|
'http://www.google.com/search', f'gvsearch:{query}',
|
||||||
'gvsearch:' + query,
|
note=f'Downloading result page {pagenum + 1}',
|
||||||
note='Downloading result page %s' % (pagenum + 1),
|
|
||||||
query={
|
query={
|
||||||
'tbm': 'vid',
|
'tbm': 'vid',
|
||||||
'q': query,
|
'q': query,
|
||||||
'start': pagenum * 10,
|
'start': pagenum * self._PAGE_SIZE,
|
||||||
|
'num': self._PAGE_SIZE,
|
||||||
'hl': 'en',
|
'hl': 'en',
|
||||||
})
|
})
|
||||||
|
|
||||||
for hit_idx, mobj in enumerate(re.finditer(
|
for url in re.findall(r'<div[^>]* class="dXiKIc"[^>]*><a href="([^"]+)"', webpage):
|
||||||
r'<h3 class="r"><a href="([^"]+)"', webpage)):
|
yield self.url_result(url)
|
||||||
if re.search(f'id="vidthumb{hit_idx + 1}"', webpage):
|
|
||||||
yield self.url_result(mobj.group(1))
|
|
||||||
|
|
||||||
if not re.search(r'id="pnnext"', webpage):
|
if not re.search(r'id="pnnext"', webpage):
|
||||||
return
|
return
|
||||||
|
|
Loading…
Reference in a new issue