startpage.py 1.5KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. from urllib import urlencode
  2. from lxml import html
  3. base_url = None
  4. search_url = None
  5. # TODO paging
  6. paging = False
  7. # TODO complete list of country mapping
  8. country_map = {'en_US': 'eng',
  9. 'en_UK': 'uk',
  10. 'nl_NL': 'ned'}
  11. def request(query, params):
  12. query = urlencode({'q': query})[2:]
  13. params['url'] = search_url
  14. params['method'] = 'POST'
  15. params['data'] = {'query': query,
  16. 'startat': (params['pageno'] - 1) * 10} # offset
  17. country = country_map.get(params['language'], 'eng')
  18. params['cookies']['preferences'] = \
  19. 'lang_homepageEEEs/air/{country}/N1NsslEEE1N1Nfont_sizeEEEmediumN1Nrecent_results_filterEEE1N1Nlanguage_uiEEEenglishN1Ndisable_open_in_new_windowEEE0N1Ncolor_schemeEEEnewN1Nnum_of_resultsEEE10N1N'.format(country=country) # noqa
  20. return params
  21. def response(resp):
  22. results = []
  23. dom = html.fromstring(resp.content)
  24. # ads xpath //div[@id="results"]/div[@id="sponsored"]//div[@class="result"]
  25. # not ads: div[@class="result"] are the direct childs of div[@id="results"]
  26. for result in dom.xpath('//div[@id="results"]/div[@class="result"]'):
  27. link = result.xpath('.//h3/a')[0]
  28. url = link.attrib.get('href')
  29. title = link.text_content()
  30. content = ''
  31. if result.xpath('./p[@class="desc"]'):
  32. content = result.xpath('./p[@class="desc"]')[0].text_content()
  33. results.append({'url': url, 'title': title, 'content': content})
  34. return results