forked from Ponysearch/Ponysearch
32 lines
1 KiB
Python
32 lines
1 KiB
Python
from urllib import quote
|
|
from lxml import html
|
|
from urlparse import urljoin, urlparse
|
|
from cgi import escape
|
|
|
|
base_url = 'https://startpage.com/'
|
|
search_url = base_url+'do/search'
|
|
|
|
def request(query, params):
|
|
global search_url
|
|
query = quote(query.replace(' ', '+'), safe='+')
|
|
params['url'] = search_url
|
|
params['method'] = 'POST'
|
|
params['data'] = {'query': query}
|
|
return params
|
|
|
|
|
|
def response(resp):
|
|
global base_url
|
|
results = []
|
|
dom = html.fromstring(resp.text)
|
|
for result in dom.xpath('//div[@class="result"]'):
|
|
link = result.xpath('.//h3/a')[0]
|
|
url = urljoin(base_url, link.attrib.get('href'))
|
|
parsed_url = urlparse(url)
|
|
# TODO better google link detection
|
|
if parsed_url.netloc.find('www.google.com') >= 0:
|
|
continue
|
|
title = ' '.join(link.xpath('.//text()'))
|
|
content = escape(' '.join(result.xpath('.//p[@class="desc"]//text()')))
|
|
results.append({'url': url, 'title': title, 'content': content})
|
|
return results
|