mediawiki.py 899B

1234567891011121314151617181920212223242526
  1. from json import loads
  2. from urllib import urlencode, quote
  3. url = 'https://en.wikipedia.org/'
  4. number_of_results = 10
  5. def request(query, params):
  6. search_url = url + 'w/api.php?action=query&list=search&{query}&srprop=timestamp&format=json'
  7. params['url'] = search_url.format(query=urlencode({'srsearch': query}))
  8. return params
  9. def response(resp):
  10. search_results = loads(resp.text)
  11. res = search_results.get('query', {}).get('search', [])
  12. return [{'url': url + 'wiki/' + quote(result['title'].replace(' ', '_').encode('utf-8')),
  13. 'title': result['title']} for result in res[:int(number_of_results)]]
  14. if not len(res):
  15. return results
  16. for result in res[:int(number_of_results)]:
  17. results.append({'url': url + 'wiki/' + quote(result['title'].replace(' ', '_').encode('utf-8')), 'title': result['title'], 'favicon':'wikipedia'})
  18. return results