@@ -1,16 +1,17 @@
11## general mediawiki-engine (Web)
2−#
2+#
33# @website websites built on mediawiki (https://www.mediawiki.org)
44# @provide-api yes (http://www.mediawiki.org/wiki/API:Search)
5−#
5+#
66# @using-api yes
77# @results JSON
88# @stable yes
9−# @parse url, title
9+# @parse url, title
1010#
1111# @todo content
1212
1313from json import loads
14+from string import Formatter
1415from urllib import urlencode, quote
1516
1617# engine dependent config
@@ -18,28 +19,38 @@ categories = ['general']
1819language_support = True
1920paging = True
2021number_of_results = 1
21−
22+
2223# search-url
2324base_url = 'https://{language}.wikipedia.org/'
24−search_url = base_url + 'w/api.php?action=query&list=search&{query}&srprop=timestamp&format=json&sroffset={offset}&srlimit={limit}' # noqa
25+search_url = base_url + 'w/api.php?action=query\
26+ &list=search\
27+ &{query}\
28+ &srprop=timestamp\
29+ &format=json\
30+ &sroffset={offset}\
31+ &srlimit={limit}'
2532
2633
2734# do search-request
2835def request(query, params):
2936 offset = (params['pageno'] - 1) * number_of_results
37+ string_args = dict(query=urlencode({'srsearch': query}),
38+ offset=offset,
39+ limit=number_of_results)
40+ format_strings = Formatter().parse(base_url)
3041
3142 if params['language'] == 'all':
3243 language = 'en'
3344 else:
3445 language = params['language'].split('_')[0]
35−
46+
47+ if len(format_strings) > 1:
48+ string_args['language'] = language
49+
3650 # write search-language back to params, required in response
3751 params['language'] = language
3852
39− params['url'] = search_url.format(query=urlencode({'srsearch': query}),
40− offset=offset,
41− limit=number_of_results,
42− language=language)
53+ params['url'] = search_url.format(**string_args)
4354
4455 return params
4556
@@ -57,7 +68,7 @@ def response(resp):
5768 # parse results
5869 for result in search_results['query']['search']:
5970 url = base_url.format(language=resp.search_params['language']) + 'wiki/' + quote(result['title'].replace(' ', '_').encode('utf-8'))
60−
71+
6172 # append result
6273 results.append({'url': url,
6374 'title': result['title'],