mediawiki.py 632 B

1234567891011121314151617181920
  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)]]