parent
9b122384e9
commit
3fc03845a1
2 changed files with 39 additions and 30 deletions
@ -0,0 +1,39 @@ |
|||||||
|
import itertools |
||||||
|
import re |
||||||
|
|
||||||
|
from .common import SearchInfoExtractor |
||||||
|
from ..utils import ( |
||||||
|
compat_urllib_parse, |
||||||
|
) |
||||||
|
|
||||||
|
|
||||||
|
class GoogleSearchIE(SearchInfoExtractor): |
||||||
|
"""Information Extractor for Google Video search queries.""" |
||||||
|
_MORE_PAGES_INDICATOR = r'id="pnnext" class="pn"' |
||||||
|
_MAX_RESULTS = 1000 |
||||||
|
IE_NAME = u'video.google:search' |
||||||
|
_SEARCH_KEY = 'gvsearch' |
||||||
|
|
||||||
|
def _get_n_results(self, query, n): |
||||||
|
"""Get a specified number of results for a query""" |
||||||
|
|
||||||
|
res = { |
||||||
|
'_type': 'playlist', |
||||||
|
'id': query, |
||||||
|
'entries': [] |
||||||
|
} |
||||||
|
|
||||||
|
for pagenum in itertools.count(1): |
||||||
|
result_url = u'http://www.google.com/search?tbm=vid&q=%s&start=%s&hl=en' % (compat_urllib_parse.quote_plus(query), pagenum*10) |
||||||
|
webpage = self._download_webpage(result_url, u'gvsearch:' + query, |
||||||
|
note='Downloading result page ' + str(pagenum)) |
||||||
|
|
||||||
|
for mobj in re.finditer(r'<h3 class="r"><a href="([^"]+)"', webpage): |
||||||
|
e = { |
||||||
|
'_type': 'url', |
||||||
|
'url': mobj.group(1) |
||||||
|
} |
||||||
|
res['entries'].append(e) |
||||||
|
|
||||||
|
if (pagenum * 10 > n) or not re.search(self._MORE_PAGES_INDICATOR, webpage): |
||||||
|
return res |
Loading…
Reference in new issue