searxng/searx/engines/duckduckgo.py

138 lines
3.8 KiB
Python
Raw Normal View History

# SPDX-License-Identifier: AGPL-3.0-or-later
"""
DuckDuckGo (Web)
"""
2014-03-21 15:33:17 +00:00
from lxml.html import fromstring
from json import loads
from searx.utils import extract_text, match_language, eval_xpath, dict_subset
from searx.network import get
2013-10-14 21:09:13 +00:00
# about
about = {
"website": 'https://duckduckgo.com/',
"wikidata_id": 'Q12805',
"official_api_documentation": 'https://duckduckgo.com/api',
"use_official_api": False,
"require_api_key": False,
"results": 'HTML',
}
# engine dependent config
categories = ['general']
paging = False
supported_languages_url = 'https://duckduckgo.com/util/u172.js'
2016-07-18 14:15:37 +00:00
time_range_support = True
2013-10-14 21:09:13 +00:00
language_aliases = {
'ar-SA': 'ar-XA',
'es-419': 'es-XL',
'ja': 'jp-JP',
'ko': 'kr-KR',
'sl-SI': 'sl-SL',
'zh-TW': 'tzh-TW',
'zh-HK': 'tzh-HK'
}
# search-url
url = 'https://html.duckduckgo.com/html'
url_ping = 'https://duckduckgo.com/t/sl_h'
2016-07-18 14:15:37 +00:00
time_range_dict = {'day': 'd',
'week': 'w',
2021-03-24 23:25:36 +00:00
'month': 'm',
'year': 'y'}
# specific xpath variables
result_xpath = '//div[@class="result results_links results_links_deep web-result "]' # noqa
url_xpath = './/a[@class="result__a"]/@href'
title_xpath = './/a[@class="result__a"]'
content_xpath = './/a[@class="result__snippet"]'
2020-06-13 21:42:16 +00:00
correction_xpath = '//div[@id="did_you_mean"]//a'
2014-03-29 15:38:45 +00:00
2017-05-21 03:33:08 +00:00
# match query's language to a region code that duckduckgo will accept
def get_region_code(lang, lang_list=None):
if lang == 'all':
return None
lang_code = match_language(lang, lang_list or [], language_aliases, 'wt-WT')
lang_parts = lang_code.split('-')
# country code goes first
return lang_parts[1].lower() + '-' + lang_parts[0].lower()
2017-05-21 03:33:08 +00:00
def request(query, params):
if params['time_range'] is not None and params['time_range'] not in time_range_dict:
2017-05-21 03:33:08 +00:00
return params
params['url'] = url
params['method'] = 'POST'
params['data']['q'] = query
params['data']['b'] = ''
region_code = get_region_code(params['language'], supported_languages)
if region_code:
params['data']['kl'] = region_code
params['cookies']['kl'] = region_code
2016-07-18 14:15:37 +00:00
if params['time_range'] in time_range_dict:
params['data']['df'] = time_range_dict[params['time_range']]
params['allow_redirects'] = False
2013-10-14 21:09:13 +00:00
return params
# get response from search-request
2013-10-14 21:09:13 +00:00
def response(resp):
if resp.status_code == 303:
return []
# ping
headers_ping = dict_subset(resp.request.headers, ['User-Agent', 'Accept-Encoding', 'Accept', 'Cookie'])
get(url_ping, headers=headers_ping)
2014-03-21 15:33:17 +00:00
# parse the response
results = []
2014-03-21 15:33:17 +00:00
doc = fromstring(resp.text)
for i, r in enumerate(eval_xpath(doc, result_xpath)):
2019-10-14 11:52:15 +00:00
if i >= 30:
break
2014-03-21 17:17:13 +00:00
try:
res_url = eval_xpath(r, url_xpath)[-1]
2014-03-21 17:17:13 +00:00
except:
continue
2014-03-21 15:33:17 +00:00
if not res_url:
2013-10-15 17:11:43 +00:00
continue
title = extract_text(eval_xpath(r, title_xpath))
content = extract_text(eval_xpath(r, content_xpath))
# append result
2014-03-21 15:33:17 +00:00
results.append({'title': title,
'content': content,
2015-09-07 21:13:04 +00:00
'url': res_url})
2014-03-21 15:33:17 +00:00
2020-06-13 21:42:16 +00:00
# parse correction
for correction in eval_xpath(doc, correction_xpath):
# append correction
results.append({'correction': extract_text(correction)})
# return results
2013-10-15 17:11:43 +00:00
return results
# get supported languages from their site
def _fetch_supported_languages(resp):
# response is a js file with regions as an embedded object
response_page = resp.text
response_page = response_page[response_page.find('regions:{') + 8:]
response_page = response_page[:response_page.find('}') + 1]
regions_json = loads(response_page)
supported_languages = map((lambda x: x[3:] + '-' + x[:2].upper()), regions_json.keys())
return list(supported_languages)