webapp.py 51 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428
  1. #!/usr/bin/env python
  2. # SPDX-License-Identifier: AGPL-3.0-or-later
  3. # lint: pylint
  4. # pyright: basic
  5. """WebbApp
  6. """
  7. import hashlib
  8. import hmac
  9. import json
  10. import os
  11. import sys
  12. import base64
  13. from datetime import datetime, timedelta
  14. from timeit import default_timer
  15. from html import escape
  16. from io import StringIO
  17. import typing
  18. from typing import List, Dict, Iterable
  19. import urllib
  20. import urllib.parse
  21. from urllib.parse import urlencode, unquote
  22. import httpx
  23. from pygments import highlight
  24. from pygments.lexers import get_lexer_by_name
  25. from pygments.formatters import HtmlFormatter # pylint: disable=no-name-in-module
  26. import flask
  27. from flask import (
  28. Flask,
  29. render_template,
  30. url_for,
  31. make_response,
  32. redirect,
  33. send_from_directory,
  34. )
  35. from flask.wrappers import Response
  36. from flask.ctx import has_request_context
  37. from flask.json import jsonify
  38. from babel.support import Translations
  39. import flask_babel
  40. from flask_babel import (
  41. Babel,
  42. gettext,
  43. format_date,
  44. format_decimal,
  45. )
  46. from searx import (
  47. logger,
  48. get_setting,
  49. settings,
  50. searx_debug,
  51. )
  52. from searx import infopage
  53. from searx.data import ENGINE_DESCRIPTIONS
  54. from searx.results import Timing, UnresponsiveEngine
  55. from searx.settings_defaults import OUTPUT_FORMATS
  56. from searx.settings_loader import get_default_settings_path
  57. from searx.exceptions import SearxParameterException
  58. from searx.engines import (
  59. OTHER_CATEGORY,
  60. categories,
  61. engines,
  62. engine_shortcuts,
  63. )
  64. from searx.webutils import (
  65. UnicodeWriter,
  66. highlight_content,
  67. get_static_files,
  68. get_result_templates,
  69. get_themes,
  70. prettify_url,
  71. new_hmac,
  72. is_hmac_of,
  73. is_flask_run_cmdline,
  74. group_engines_in_tab,
  75. )
  76. from searx.webadapter import (
  77. get_search_query_from_webapp,
  78. get_selected_categories,
  79. )
  80. from searx.utils import (
  81. html_to_text,
  82. gen_useragent,
  83. dict_subset,
  84. match_language,
  85. )
  86. from searx.version import VERSION_STRING, GIT_URL, GIT_BRANCH
  87. from searx.query import RawTextQuery
  88. from searx.plugins import Plugin, plugins, initialize as plugin_initialize
  89. from searx.plugins.oa_doi_rewrite import get_doi_resolver
  90. from searx.preferences import (
  91. Preferences,
  92. ValidationException,
  93. )
  94. from searx.answerers import (
  95. answerers,
  96. ask,
  97. )
  98. from searx.metrics import (
  99. get_engines_stats,
  100. get_engine_errors,
  101. get_reliabilities,
  102. histogram,
  103. counter,
  104. )
  105. from searx.flaskfix import patch_application
  106. # renaming names from searx imports ...
  107. from searx.autocomplete import search_autocomplete, backends as autocomplete_backends
  108. from searx.languages import language_codes as languages
  109. from searx.locales import LOCALE_NAMES, RTL_LOCALES
  110. from searx.search import SearchWithPlugins, initialize as search_initialize
  111. from searx.network import stream as http_stream, set_context_network_name
  112. from searx.search.checker import get_result as checker_get_result
  113. logger = logger.getChild('webapp')
  114. # check secret_key
  115. if not searx_debug and settings['server']['secret_key'] == 'ultrasecretkey':
  116. logger.error('server.secret_key is not changed. Please use something else instead of ultrasecretkey.')
  117. sys.exit(1)
  118. # about static
  119. logger.debug('static directory is %s', settings['ui']['static_path'])
  120. static_files = get_static_files(settings['ui']['static_path'])
  121. # about templates
  122. logger.debug('templates directory is %s', settings['ui']['templates_path'])
  123. default_theme = settings['ui']['default_theme']
  124. templates_path = settings['ui']['templates_path']
  125. themes = get_themes(templates_path)
  126. result_templates = get_result_templates(templates_path)
  127. STATS_SORT_PARAMETERS = {
  128. 'name': (False, 'name', ''),
  129. 'score': (True, 'score', 0),
  130. 'result_count': (True, 'result_count', 0),
  131. 'time': (False, 'total', 0),
  132. 'reliability': (False, 'reliability', 100),
  133. }
  134. _INFO_PAGES = infopage.InfoPageSet()
  135. # Flask app
  136. app = Flask(__name__, static_folder=settings['ui']['static_path'], template_folder=templates_path)
  137. app.jinja_env.trim_blocks = True
  138. app.jinja_env.lstrip_blocks = True
  139. app.jinja_env.add_extension('jinja2.ext.loopcontrols') # pylint: disable=no-member
  140. app.jinja_env.filters['group_engines_in_tab'] = group_engines_in_tab # pylint: disable=no-member
  141. app.secret_key = settings['server']['secret_key']
  142. babel = Babel(app)
  143. timeout_text = gettext('timeout')
  144. parsing_error_text = gettext('parsing error')
  145. http_protocol_error_text = gettext('HTTP protocol error')
  146. network_error_text = gettext('network error')
  147. exception_classname_to_text = {
  148. None: gettext('unexpected crash'),
  149. 'timeout': timeout_text,
  150. 'asyncio.TimeoutError': timeout_text,
  151. 'httpx.TimeoutException': timeout_text,
  152. 'httpx.ConnectTimeout': timeout_text,
  153. 'httpx.ReadTimeout': timeout_text,
  154. 'httpx.WriteTimeout': timeout_text,
  155. 'httpx.HTTPStatusError': gettext('HTTP error'),
  156. 'httpx.ConnectError': gettext("HTTP connection error"),
  157. 'httpx.RemoteProtocolError': http_protocol_error_text,
  158. 'httpx.LocalProtocolError': http_protocol_error_text,
  159. 'httpx.ProtocolError': http_protocol_error_text,
  160. 'httpx.ReadError': network_error_text,
  161. 'httpx.WriteError': network_error_text,
  162. 'httpx.ProxyError': gettext("proxy error"),
  163. 'searx.exceptions.SearxEngineCaptchaException': gettext("CAPTCHA"),
  164. 'searx.exceptions.SearxEngineTooManyRequestsException': gettext("too many requests"),
  165. 'searx.exceptions.SearxEngineAccessDeniedException': gettext("access denied"),
  166. 'searx.exceptions.SearxEngineAPIException': gettext("server API error"),
  167. 'searx.exceptions.SearxEngineXPathException': parsing_error_text,
  168. 'KeyError': parsing_error_text,
  169. 'json.decoder.JSONDecodeError': parsing_error_text,
  170. 'lxml.etree.ParserError': parsing_error_text,
  171. }
  172. # monkey patch for flask_babel.get_translations
  173. _flask_babel_get_translations = flask_babel.get_translations
  174. class ExtendedRequest(flask.Request):
  175. """This class is never initialized and only used for type checking."""
  176. preferences: Preferences
  177. errors: List[str]
  178. user_plugins: List[Plugin]
  179. form: Dict[str, str]
  180. start_time: float
  181. render_time: float
  182. timings: List[Timing]
  183. request = typing.cast(ExtendedRequest, flask.request)
  184. def _get_translations():
  185. if has_request_context() and request.form.get('use-translation') == 'oc':
  186. babel_ext = flask_babel.current_app.extensions['babel']
  187. return Translations.load(next(babel_ext.translation_directories), 'oc')
  188. if has_request_context() and request.form.get('use-translation') == 'szl':
  189. babel_ext = flask_babel.current_app.extensions['babel']
  190. return Translations.load(next(babel_ext.translation_directories), 'szl')
  191. return _flask_babel_get_translations()
  192. flask_babel.get_translations = _get_translations
  193. @babel.localeselector
  194. def get_locale():
  195. locale = 'en'
  196. if has_request_context():
  197. value = request.preferences.get_value('locale')
  198. if value:
  199. locale = value
  200. if locale == 'oc':
  201. request.form['use-translation'] = 'oc'
  202. locale = 'fr_FR'
  203. if locale == 'szl':
  204. request.form['use-translation'] = 'szl'
  205. locale = 'pl'
  206. if locale == '':
  207. # if there is an error loading the preferences
  208. # the locale is going to be ''
  209. locale = 'en'
  210. # babel uses underscore instead of hyphen.
  211. locale = locale.replace('-', '_')
  212. logger.debug("%s uses locale `%s`", urllib.parse.quote(request.url), locale)
  213. return locale
  214. def _get_browser_language(req, lang_list):
  215. for lang in req.headers.get("Accept-Language", "en").split(","):
  216. if ';' in lang:
  217. lang = lang.split(';')[0]
  218. if '-' in lang:
  219. lang_parts = lang.split('-')
  220. lang = "{}-{}".format(lang_parts[0], lang_parts[-1].upper())
  221. locale = match_language(lang, lang_list, fallback=None)
  222. if locale is not None:
  223. return locale
  224. return 'en'
  225. def _get_locale_rfc5646(locale):
  226. """Get locale name for <html lang="...">
  227. Chrom* browsers don't detect the language when there is a subtag (ie a territory).
  228. For example "zh-TW" is detected but not "zh-Hant-TW".
  229. This function returns a locale without the subtag.
  230. """
  231. parts = locale.split('-')
  232. return parts[0].lower() + '-' + parts[-1].upper()
  233. # code-highlighter
  234. @app.template_filter('code_highlighter')
  235. def code_highlighter(codelines, language=None):
  236. if not language:
  237. language = 'text'
  238. try:
  239. # find lexer by programing language
  240. lexer = get_lexer_by_name(language, stripall=True)
  241. except Exception as e: # pylint: disable=broad-except
  242. logger.exception(e, exc_info=True)
  243. # if lexer is not found, using default one
  244. lexer = get_lexer_by_name('text', stripall=True)
  245. html_code = ''
  246. tmp_code = ''
  247. last_line = None
  248. line_code_start = None
  249. # parse lines
  250. for line, code in codelines:
  251. if not last_line:
  252. line_code_start = line
  253. # new codeblock is detected
  254. if last_line is not None and last_line + 1 != line:
  255. # highlight last codepart
  256. formatter = HtmlFormatter(linenos='inline', linenostart=line_code_start, cssclass="code-highlight")
  257. html_code = html_code + highlight(tmp_code, lexer, formatter)
  258. # reset conditions for next codepart
  259. tmp_code = ''
  260. line_code_start = line
  261. # add codepart
  262. tmp_code += code + '\n'
  263. # update line
  264. last_line = line
  265. # highlight last codepart
  266. formatter = HtmlFormatter(linenos='inline', linenostart=line_code_start, cssclass="code-highlight")
  267. html_code = html_code + highlight(tmp_code, lexer, formatter)
  268. return html_code
  269. def get_result_template(theme_name: str, template_name: str):
  270. themed_path = theme_name + '/result_templates/' + template_name
  271. if themed_path in result_templates:
  272. return themed_path
  273. return 'result_templates/' + template_name
  274. def custom_url_for(endpoint: str, **values):
  275. suffix = ""
  276. if endpoint == 'static' and values.get('filename'):
  277. file_hash = static_files.get(values['filename'])
  278. if not file_hash:
  279. # try file in the current theme
  280. theme_name = request.preferences.get_value('theme')
  281. filename_with_theme = "themes/{}/{}".format(theme_name, values['filename'])
  282. file_hash = static_files.get(filename_with_theme)
  283. if file_hash:
  284. values['filename'] = filename_with_theme
  285. if get_setting('ui.static_use_hash') and file_hash:
  286. suffix = "?" + file_hash
  287. if endpoint == 'info' and 'locale' not in values:
  288. locale = request.preferences.get_value('locale')
  289. if _INFO_PAGES.get_page(values['pagename'], locale) is None:
  290. locale = _INFO_PAGES.locale_default
  291. values['locale'] = locale
  292. return url_for(endpoint, **values) + suffix
  293. def proxify(url: str):
  294. if url.startswith('//'):
  295. url = 'https:' + url
  296. if not settings.get('result_proxy'):
  297. return url
  298. url_params = dict(mortyurl=url)
  299. if settings['result_proxy'].get('key'):
  300. url_params['mortyhash'] = hmac.new(settings['result_proxy']['key'], url.encode(), hashlib.sha256).hexdigest()
  301. return '{0}?{1}'.format(settings['result_proxy']['url'], urlencode(url_params))
  302. def image_proxify(url: str):
  303. if url.startswith('//'):
  304. url = 'https:' + url
  305. if not request.preferences.get_value('image_proxy'):
  306. return url
  307. if url.startswith('data:image/'):
  308. # 50 is an arbitrary number to get only the beginning of the image.
  309. partial_base64 = url[len('data:image/') : 50].split(';')
  310. if (
  311. len(partial_base64) == 2
  312. and partial_base64[0] in ['gif', 'png', 'jpeg', 'pjpeg', 'webp', 'tiff', 'bmp']
  313. and partial_base64[1].startswith('base64,')
  314. ):
  315. return url
  316. return None
  317. if settings.get('result_proxy'):
  318. return proxify(url)
  319. h = new_hmac(settings['server']['secret_key'], url.encode())
  320. return '{0}?{1}'.format(url_for('image_proxy'), urlencode(dict(url=url.encode(), h=h)))
  321. def get_translations():
  322. return {
  323. # when there is autocompletion
  324. 'no_item_found': gettext('No item found'),
  325. # /preferences: the source of the engine description (wikipedata, wikidata, website)
  326. 'Source': gettext('Source'),
  327. # infinite scroll
  328. 'error_loading_next_page': gettext('Error loading the next page'),
  329. }
  330. def _get_enable_categories(all_categories: Iterable[str]):
  331. disabled_engines = request.preferences.engines.get_disabled()
  332. enabled_categories = set(
  333. # pylint: disable=consider-using-dict-items
  334. category
  335. for engine_name in engines
  336. for category in engines[engine_name].categories
  337. if (engine_name, category) not in disabled_engines
  338. )
  339. return [x for x in all_categories if x in enabled_categories]
  340. def get_pretty_url(parsed_url: urllib.parse.ParseResult):
  341. path = parsed_url.path
  342. path = path[:-1] if len(path) > 0 and path[-1] == '/' else path
  343. path = unquote(path.replace("/", " › "))
  344. return [parsed_url.scheme + "://" + parsed_url.netloc, path]
  345. def get_client_settings():
  346. req_pref = request.preferences
  347. return {
  348. 'autocomplete_provider': req_pref.get_value('autocomplete'),
  349. 'autocomplete_min': get_setting('search.autocomplete_min'),
  350. 'http_method': req_pref.get_value('method'),
  351. 'infinite_scroll': req_pref.get_value('infinite_scroll'),
  352. 'translations': get_translations(),
  353. 'search_on_category_select': req_pref.plugins.choices['searx.plugins.search_on_category_select'],
  354. 'hotkeys': req_pref.plugins.choices['searx.plugins.vim_hotkeys'],
  355. 'theme_static_path': custom_url_for('static', filename='themes/simple'),
  356. }
  357. def render(template_name: str, **kwargs):
  358. kwargs['client_settings'] = str(
  359. base64.b64encode(
  360. bytes(
  361. json.dumps(get_client_settings()),
  362. encoding='utf-8',
  363. )
  364. ),
  365. encoding='utf-8',
  366. )
  367. # values from the HTTP requests
  368. kwargs['endpoint'] = 'results' if 'q' in kwargs else request.endpoint
  369. kwargs['cookies'] = request.cookies
  370. kwargs['errors'] = request.errors
  371. # values from the preferences
  372. kwargs['preferences'] = request.preferences
  373. kwargs['results_on_new_tab'] = request.preferences.get_value('results_on_new_tab')
  374. kwargs['advanced_search'] = request.preferences.get_value('advanced_search')
  375. kwargs['query_in_title'] = request.preferences.get_value('query_in_title')
  376. kwargs['safesearch'] = str(request.preferences.get_value('safesearch'))
  377. kwargs['theme'] = request.preferences.get_value('theme')
  378. kwargs['method'] = request.preferences.get_value('method')
  379. kwargs['categories_as_tabs'] = list(settings['categories_as_tabs'].keys())
  380. kwargs['categories'] = _get_enable_categories(categories.keys())
  381. kwargs['OTHER_CATEGORY'] = OTHER_CATEGORY
  382. # i18n
  383. kwargs['language_codes'] = [l for l in languages if l[0] in settings['search']['languages']]
  384. locale = request.preferences.get_value('locale')
  385. kwargs['locale_rfc5646'] = _get_locale_rfc5646(locale)
  386. if locale in RTL_LOCALES and 'rtl' not in kwargs:
  387. kwargs['rtl'] = True
  388. if 'current_language' not in kwargs:
  389. kwargs['current_language'] = match_language(
  390. request.preferences.get_value('language'), settings['search']['languages']
  391. )
  392. # values from settings
  393. kwargs['search_formats'] = [x for x in settings['search']['formats'] if x != 'html']
  394. kwargs['instance_name'] = get_setting('general.instance_name')
  395. kwargs['searx_version'] = VERSION_STRING
  396. kwargs['searx_git_url'] = GIT_URL
  397. kwargs['get_setting'] = get_setting
  398. kwargs['get_pretty_url'] = get_pretty_url
  399. # helpers to create links to other pages
  400. kwargs['url_for'] = custom_url_for # override url_for function in templates
  401. kwargs['image_proxify'] = image_proxify
  402. kwargs['proxify'] = proxify if settings.get('result_proxy', {}).get('url') else None
  403. kwargs['proxify_results'] = settings.get('result_proxy', {}).get('proxify_results', True)
  404. kwargs['get_result_template'] = get_result_template
  405. kwargs['opensearch_url'] = (
  406. url_for('opensearch')
  407. + '?'
  408. + urlencode(
  409. {
  410. 'method': request.preferences.get_value('method'),
  411. 'autocomplete': request.preferences.get_value('autocomplete'),
  412. }
  413. )
  414. )
  415. # scripts from plugins
  416. kwargs['scripts'] = set()
  417. for plugin in request.user_plugins:
  418. for script in plugin.js_dependencies:
  419. kwargs['scripts'].add(script)
  420. # styles from plugins
  421. kwargs['styles'] = set()
  422. for plugin in request.user_plugins:
  423. for css in plugin.css_dependencies:
  424. kwargs['styles'].add(css)
  425. start_time = default_timer()
  426. result = render_template('{}/{}'.format(kwargs['theme'], template_name), **kwargs)
  427. request.render_time += default_timer() - start_time # pylint: disable=assigning-non-slot
  428. return result
  429. @app.before_request
  430. def pre_request():
  431. request.start_time = default_timer() # pylint: disable=assigning-non-slot
  432. request.render_time = 0 # pylint: disable=assigning-non-slot
  433. request.timings = [] # pylint: disable=assigning-non-slot
  434. request.errors = [] # pylint: disable=assigning-non-slot
  435. preferences = Preferences(themes, list(categories.keys()), engines, plugins) # pylint: disable=redefined-outer-name
  436. user_agent = request.headers.get('User-Agent', '').lower()
  437. if 'webkit' in user_agent and 'android' in user_agent:
  438. preferences.key_value_settings['method'].value = 'GET'
  439. request.preferences = preferences # pylint: disable=assigning-non-slot
  440. try:
  441. preferences.parse_dict(request.cookies)
  442. except Exception as e: # pylint: disable=broad-except
  443. logger.exception(e, exc_info=True)
  444. request.errors.append(gettext('Invalid settings, please edit your preferences'))
  445. # merge GET, POST vars
  446. # request.form
  447. request.form = dict(request.form.items()) # pylint: disable=assigning-non-slot
  448. for k, v in request.args.items():
  449. if k not in request.form:
  450. request.form[k] = v
  451. if request.form.get('preferences'):
  452. preferences.parse_encoded_data(request.form['preferences'])
  453. else:
  454. try:
  455. preferences.parse_dict(request.form)
  456. except Exception as e: # pylint: disable=broad-except
  457. logger.exception(e, exc_info=True)
  458. request.errors.append(gettext('Invalid settings'))
  459. # language is defined neither in settings nor in preferences
  460. # use browser headers
  461. if not preferences.get_value("language"):
  462. language = _get_browser_language(request, settings['search']['languages'])
  463. preferences.parse_dict({"language": language})
  464. # locale is defined neither in settings nor in preferences
  465. # use browser headers
  466. if not preferences.get_value("locale"):
  467. locale = _get_browser_language(request, LOCALE_NAMES.keys())
  468. preferences.parse_dict({"locale": locale})
  469. # request.user_plugins
  470. request.user_plugins = [] # pylint: disable=assigning-non-slot
  471. allowed_plugins = preferences.plugins.get_enabled()
  472. disabled_plugins = preferences.plugins.get_disabled()
  473. for plugin in plugins:
  474. if (plugin.default_on and plugin.id not in disabled_plugins) or plugin.id in allowed_plugins:
  475. request.user_plugins.append(plugin)
  476. @app.after_request
  477. def add_default_headers(response: flask.Response):
  478. # set default http headers
  479. for header, value in settings['server']['default_http_headers'].items():
  480. if header in response.headers:
  481. continue
  482. response.headers[header] = value
  483. return response
  484. @app.after_request
  485. def post_request(response: flask.Response):
  486. total_time = default_timer() - request.start_time
  487. timings_all = [
  488. 'total;dur=' + str(round(total_time * 1000, 3)),
  489. 'render;dur=' + str(round(request.render_time * 1000, 3)),
  490. ]
  491. if len(request.timings) > 0:
  492. timings = sorted(request.timings, key=lambda t: t.total)
  493. timings_total = [
  494. 'total_' + str(i) + '_' + t.engine + ';dur=' + str(round(t.total * 1000, 3)) for i, t in enumerate(timings)
  495. ]
  496. timings_load = [
  497. 'load_' + str(i) + '_' + t.engine + ';dur=' + str(round(t.load * 1000, 3))
  498. for i, t in enumerate(timings)
  499. if t.load
  500. ]
  501. timings_all = timings_all + timings_total + timings_load
  502. response.headers.add('Server-Timing', ', '.join(timings_all))
  503. return response
  504. def index_error(output_format: str, error_message: str):
  505. if output_format == 'json':
  506. return Response(json.dumps({'error': error_message}), mimetype='application/json')
  507. if output_format == 'csv':
  508. response = Response('', mimetype='application/csv')
  509. cont_disp = 'attachment;Filename=searx.csv'
  510. response.headers.add('Content-Disposition', cont_disp)
  511. return response
  512. if output_format == 'rss':
  513. response_rss = render(
  514. 'opensearch_response_rss.xml',
  515. results=[],
  516. q=request.form['q'] if 'q' in request.form else '',
  517. number_of_results=0,
  518. error_message=error_message,
  519. )
  520. return Response(response_rss, mimetype='text/xml')
  521. # html
  522. request.errors.append(gettext('search error'))
  523. return render(
  524. # fmt: off
  525. 'index.html',
  526. selected_categories=get_selected_categories(request.preferences, request.form),
  527. # fmt: on
  528. )
  529. @app.route('/', methods=['GET', 'POST'])
  530. def index():
  531. """Render index page."""
  532. # redirect to search if there's a query in the request
  533. if request.form.get('q'):
  534. query = ('?' + request.query_string.decode()) if request.query_string else ''
  535. return redirect(url_for('search') + query, 308)
  536. return render(
  537. # fmt: off
  538. 'index.html',
  539. selected_categories=get_selected_categories(request.preferences, request.form),
  540. current_locale = request.preferences.get_value("locale"),
  541. # fmt: on
  542. )
  543. @app.route('/healthz', methods=['GET'])
  544. def health():
  545. return Response('OK', mimetype='text/plain')
  546. @app.route('/search', methods=['GET', 'POST'])
  547. def search():
  548. """Search query in q and return results.
  549. Supported outputs: html, json, csv, rss.
  550. """
  551. # pylint: disable=too-many-locals, too-many-return-statements, too-many-branches
  552. # pylint: disable=too-many-statements
  553. # output_format
  554. output_format = request.form.get('format', 'html')
  555. if output_format not in OUTPUT_FORMATS:
  556. output_format = 'html'
  557. if output_format not in settings['search']['formats']:
  558. flask.abort(403)
  559. # check if there is query (not None and not an empty string)
  560. if not request.form.get('q'):
  561. if output_format == 'html':
  562. return render(
  563. # fmt: off
  564. 'index.html',
  565. selected_categories=get_selected_categories(request.preferences, request.form),
  566. # fmt: on
  567. )
  568. return index_error(output_format, 'No query'), 400
  569. # search
  570. search_query = None
  571. raw_text_query = None
  572. result_container = None
  573. try:
  574. search_query, raw_text_query, _, _ = get_search_query_from_webapp(request.preferences, request.form)
  575. # search = Search(search_query) # without plugins
  576. search = SearchWithPlugins(search_query, request.user_plugins, request) # pylint: disable=redefined-outer-name
  577. result_container = search.search()
  578. except SearxParameterException as e:
  579. logger.exception('search error: SearxParameterException')
  580. return index_error(output_format, e.message), 400
  581. except Exception as e: # pylint: disable=broad-except
  582. logger.exception(e, exc_info=True)
  583. return index_error(output_format, gettext('search error')), 500
  584. # results
  585. results = result_container.get_ordered_results()
  586. number_of_results = result_container.results_number()
  587. if number_of_results < result_container.results_length():
  588. number_of_results = 0
  589. # checkin for a external bang
  590. if result_container.redirect_url:
  591. return redirect(result_container.redirect_url)
  592. # Server-Timing header
  593. request.timings = result_container.get_timings() # pylint: disable=assigning-non-slot
  594. current_template = None
  595. previous_result = None
  596. # output
  597. for result in results:
  598. if output_format == 'html':
  599. if 'content' in result and result['content']:
  600. result['content'] = highlight_content(escape(result['content'][:1024]), search_query.query)
  601. if 'title' in result and result['title']:
  602. result['title'] = highlight_content(escape(result['title'] or ''), search_query.query)
  603. else:
  604. if result.get('content'):
  605. result['content'] = html_to_text(result['content']).strip()
  606. # removing html content and whitespace duplications
  607. result['title'] = ' '.join(html_to_text(result['title']).strip().split())
  608. if 'url' in result:
  609. result['pretty_url'] = prettify_url(result['url'])
  610. # TODO, check if timezone is calculated right # pylint: disable=fixme
  611. if result.get('publishedDate'): # do not try to get a date from an empty string or a None type
  612. try: # test if publishedDate >= 1900 (datetime module bug)
  613. result['pubdate'] = result['publishedDate'].strftime('%Y-%m-%d %H:%M:%S%z')
  614. except ValueError:
  615. result['publishedDate'] = None
  616. else:
  617. if result['publishedDate'].replace(tzinfo=None) >= datetime.now() - timedelta(days=1):
  618. timedifference = datetime.now() - result['publishedDate'].replace(tzinfo=None)
  619. minutes = int((timedifference.seconds / 60) % 60)
  620. hours = int(timedifference.seconds / 60 / 60)
  621. if hours == 0:
  622. result['publishedDate'] = gettext('{minutes} minute(s) ago').format(minutes=minutes)
  623. else:
  624. result['publishedDate'] = gettext('{hours} hour(s), {minutes} minute(s) ago').format(
  625. hours=hours, minutes=minutes
  626. )
  627. else:
  628. result['publishedDate'] = format_date(result['publishedDate'])
  629. # set result['open_group'] = True when the template changes from the previous result
  630. # set result['close_group'] = True when the template changes on the next result
  631. if current_template != result.get('template'):
  632. result['open_group'] = True
  633. if previous_result:
  634. previous_result['close_group'] = True # pylint: disable=unsupported-assignment-operation
  635. current_template = result.get('template')
  636. previous_result = result
  637. if previous_result:
  638. previous_result['close_group'] = True
  639. if output_format == 'json':
  640. x = {
  641. 'query': search_query.query,
  642. 'number_of_results': number_of_results,
  643. 'results': results,
  644. 'answers': list(result_container.answers),
  645. 'corrections': list(result_container.corrections),
  646. 'infoboxes': result_container.infoboxes,
  647. 'suggestions': list(result_container.suggestions),
  648. 'unresponsive_engines': __get_translated_errors(result_container.unresponsive_engines),
  649. }
  650. response = json.dumps(x, default=lambda item: list(item) if isinstance(item, set) else item)
  651. return Response(response, mimetype='application/json')
  652. if output_format == 'csv':
  653. csv = UnicodeWriter(StringIO())
  654. keys = ('title', 'url', 'content', 'host', 'engine', 'score', 'type')
  655. csv.writerow(keys)
  656. for row in results:
  657. row['host'] = row['parsed_url'].netloc
  658. row['type'] = 'result'
  659. csv.writerow([row.get(key, '') for key in keys])
  660. for a in result_container.answers:
  661. row = {'title': a, 'type': 'answer'}
  662. csv.writerow([row.get(key, '') for key in keys])
  663. for a in result_container.suggestions:
  664. row = {'title': a, 'type': 'suggestion'}
  665. csv.writerow([row.get(key, '') for key in keys])
  666. for a in result_container.corrections:
  667. row = {'title': a, 'type': 'correction'}
  668. csv.writerow([row.get(key, '') for key in keys])
  669. csv.stream.seek(0)
  670. response = Response(csv.stream.read(), mimetype='application/csv')
  671. cont_disp = 'attachment;Filename=searx_-_{0}.csv'.format(search_query.query)
  672. response.headers.add('Content-Disposition', cont_disp)
  673. return response
  674. if output_format == 'rss':
  675. response_rss = render(
  676. 'opensearch_response_rss.xml',
  677. results=results,
  678. answers=result_container.answers,
  679. corrections=result_container.corrections,
  680. suggestions=result_container.suggestions,
  681. q=request.form['q'],
  682. number_of_results=number_of_results,
  683. )
  684. return Response(response_rss, mimetype='text/xml')
  685. # HTML output format
  686. # suggestions: use RawTextQuery to get the suggestion URLs with the same bang
  687. suggestion_urls = list(
  688. map(
  689. lambda suggestion: {'url': raw_text_query.changeQuery(suggestion).getFullQuery(), 'title': suggestion},
  690. result_container.suggestions,
  691. )
  692. )
  693. correction_urls = list(
  694. map(
  695. lambda correction: {'url': raw_text_query.changeQuery(correction).getFullQuery(), 'title': correction},
  696. result_container.corrections,
  697. )
  698. )
  699. return render(
  700. # fmt: off
  701. 'results.html',
  702. results = results,
  703. q=request.form['q'],
  704. selected_categories = search_query.categories,
  705. pageno = search_query.pageno,
  706. time_range = search_query.time_range,
  707. number_of_results = format_decimal(number_of_results),
  708. suggestions = suggestion_urls,
  709. answers = result_container.answers,
  710. corrections = correction_urls,
  711. infoboxes = result_container.infoboxes,
  712. engine_data = result_container.engine_data,
  713. paging = result_container.paging,
  714. unresponsive_engines = __get_translated_errors(
  715. result_container.unresponsive_engines
  716. ),
  717. current_locale = request.preferences.get_value("locale"),
  718. current_language = match_language(
  719. search_query.lang,
  720. settings['search']['languages'],
  721. fallback=request.preferences.get_value("language")
  722. ),
  723. timeout_limit = request.form.get('timeout_limit', None)
  724. # fmt: on
  725. )
  726. def __get_translated_errors(unresponsive_engines: Iterable[UnresponsiveEngine]):
  727. translated_errors = []
  728. # make a copy unresponsive_engines to avoid "RuntimeError: Set changed size
  729. # during iteration" it happens when an engine modifies the ResultContainer
  730. # after the search_multiple_requests method has stopped waiting
  731. for unresponsive_engine in unresponsive_engines:
  732. error_user_text = exception_classname_to_text.get(unresponsive_engine.error_type)
  733. if not error_user_text:
  734. error_user_text = exception_classname_to_text[None]
  735. error_msg = gettext(error_user_text)
  736. if unresponsive_engine.suspended:
  737. error_msg = gettext('Suspended') + ': ' + error_msg
  738. translated_errors.append((unresponsive_engine.engine, error_msg))
  739. return sorted(translated_errors, key=lambda e: e[0])
  740. @app.route('/about', methods=['GET'])
  741. def about():
  742. """Redirect to about page"""
  743. # custom_url_for is going to add the locale
  744. return redirect(custom_url_for('info', pagename='about'))
  745. @app.route('/info/<locale>/<pagename>', methods=['GET'])
  746. def info(pagename, locale):
  747. """Render page of online user documentation"""
  748. page = _INFO_PAGES.get_page(pagename, locale)
  749. if page is None:
  750. flask.abort(404)
  751. user_locale = request.preferences.get_value('locale')
  752. return render(
  753. 'info.html',
  754. all_pages=_INFO_PAGES.iter_pages(user_locale, fallback_to_default=True),
  755. active_page=page,
  756. active_pagename=pagename,
  757. )
  758. @app.route('/autocompleter', methods=['GET', 'POST'])
  759. def autocompleter():
  760. """Return autocompleter results"""
  761. # run autocompleter
  762. results = []
  763. # set blocked engines
  764. disabled_engines = request.preferences.engines.get_disabled()
  765. # parse query
  766. raw_text_query = RawTextQuery(request.form.get('q', ''), disabled_engines)
  767. sug_prefix = raw_text_query.getQuery()
  768. # normal autocompletion results only appear if no inner results returned
  769. # and there is a query part
  770. if len(raw_text_query.autocomplete_list) == 0 and len(sug_prefix) > 0:
  771. # get language from cookie
  772. language = request.preferences.get_value('language')
  773. if not language or language == 'all':
  774. language = 'en'
  775. else:
  776. language = language.split('-')[0]
  777. # run autocompletion
  778. raw_results = search_autocomplete(request.preferences.get_value('autocomplete'), sug_prefix, language)
  779. for result in raw_results:
  780. # attention: this loop will change raw_text_query object and this is
  781. # the reason why the sug_prefix was stored before (see above)
  782. results.append(raw_text_query.changeQuery(result).getFullQuery())
  783. if len(raw_text_query.autocomplete_list) > 0:
  784. for autocomplete_text in raw_text_query.autocomplete_list:
  785. results.append(raw_text_query.get_autocomplete_full_query(autocomplete_text))
  786. for answers in ask(raw_text_query):
  787. for answer in answers:
  788. results.append(str(answer['answer']))
  789. if request.headers.get('X-Requested-With') == 'XMLHttpRequest':
  790. # the suggestion request comes from the searx search form
  791. suggestions = json.dumps(results)
  792. mimetype = 'application/json'
  793. else:
  794. # the suggestion request comes from browser's URL bar
  795. suggestions = json.dumps([sug_prefix, results])
  796. mimetype = 'application/x-suggestions+json'
  797. suggestions = escape(suggestions, False)
  798. return Response(suggestions, mimetype=mimetype)
  799. @app.route('/preferences', methods=['GET', 'POST'])
  800. def preferences():
  801. """Render preferences page && save user preferences"""
  802. # pylint: disable=too-many-locals, too-many-return-statements, too-many-branches
  803. # pylint: disable=too-many-statements
  804. # save preferences
  805. if request.method == 'POST':
  806. resp = make_response(redirect(url_for('index', _external=True)))
  807. try:
  808. request.preferences.parse_form(request.form)
  809. except ValidationException:
  810. request.errors.append(gettext('Invalid settings, please edit your preferences'))
  811. return resp
  812. return request.preferences.save(resp)
  813. # render preferences
  814. image_proxy = request.preferences.get_value('image_proxy') # pylint: disable=redefined-outer-name
  815. disabled_engines = request.preferences.engines.get_disabled()
  816. allowed_plugins = request.preferences.plugins.get_enabled()
  817. # stats for preferences page
  818. filtered_engines = dict(filter(lambda kv: request.preferences.validate_token(kv[1]), engines.items()))
  819. engines_by_category = {}
  820. for c in categories: # pylint: disable=consider-using-dict-items
  821. engines_by_category[c] = [e for e in categories[c] if e.name in filtered_engines]
  822. # sort the engines alphabetically since the order in settings.yml is meaningless.
  823. list.sort(engines_by_category[c], key=lambda e: e.name)
  824. # get first element [0], the engine time,
  825. # and then the second element [1] : the time (the first one is the label)
  826. stats = {} # pylint: disable=redefined-outer-name
  827. max_rate95 = 0
  828. for _, e in filtered_engines.items():
  829. h = histogram('engine', e.name, 'time', 'total')
  830. median = round(h.percentage(50), 1) if h.count > 0 else None
  831. rate80 = round(h.percentage(80), 1) if h.count > 0 else None
  832. rate95 = round(h.percentage(95), 1) if h.count > 0 else None
  833. max_rate95 = max(max_rate95, rate95 or 0)
  834. result_count_sum = histogram('engine', e.name, 'result', 'count').sum
  835. successful_count = counter('engine', e.name, 'search', 'count', 'successful')
  836. result_count = int(result_count_sum / float(successful_count)) if successful_count else 0
  837. stats[e.name] = {
  838. 'time': median,
  839. 'rate80': rate80,
  840. 'rate95': rate95,
  841. 'warn_timeout': e.timeout > settings['outgoing']['request_timeout'],
  842. 'supports_selected_language': _is_selected_language_supported(e, request.preferences),
  843. 'result_count': result_count,
  844. }
  845. # end of stats
  846. # reliabilities
  847. reliabilities = {}
  848. engine_errors = get_engine_errors(filtered_engines)
  849. checker_results = checker_get_result()
  850. checker_results = (
  851. checker_results['engines'] if checker_results['status'] == 'ok' and 'engines' in checker_results else {}
  852. )
  853. for _, e in filtered_engines.items():
  854. checker_result = checker_results.get(e.name, {})
  855. checker_success = checker_result.get('success', True)
  856. errors = engine_errors.get(e.name) or []
  857. if counter('engine', e.name, 'search', 'count', 'sent') == 0:
  858. # no request
  859. reliablity = None
  860. elif checker_success and not errors:
  861. reliablity = 100
  862. elif 'simple' in checker_result.get('errors', {}):
  863. # the basic (simple) test doesn't work: the engine is broken accoding to the checker
  864. # even if there is no exception
  865. reliablity = 0
  866. else:
  867. reliablity = 100 - sum([error['percentage'] for error in errors if not error.get('secondary')])
  868. reliabilities[e.name] = {
  869. 'reliablity': reliablity,
  870. 'errors': [],
  871. 'checker': checker_results.get(e.name, {}).get('errors', {}).keys(),
  872. }
  873. # keep the order of the list checker_results[e.name]['errors'] and deduplicate.
  874. # the first element has the highest percentage rate.
  875. reliabilities_errors = []
  876. for error in errors:
  877. error_user_text = None
  878. if error.get('secondary') or 'exception_classname' not in error:
  879. continue
  880. error_user_text = exception_classname_to_text.get(error.get('exception_classname'))
  881. if not error:
  882. error_user_text = exception_classname_to_text[None]
  883. if error_user_text not in reliabilities_errors:
  884. reliabilities_errors.append(error_user_text)
  885. reliabilities[e.name]['errors'] = reliabilities_errors
  886. # supports
  887. supports = {}
  888. for _, e in filtered_engines.items():
  889. supports_selected_language = _is_selected_language_supported(e, request.preferences)
  890. safesearch = e.safesearch
  891. time_range_support = e.time_range_support
  892. for checker_test_name in checker_results.get(e.name, {}).get('errors', {}):
  893. if supports_selected_language and checker_test_name.startswith('lang_'):
  894. supports_selected_language = '?'
  895. elif safesearch and checker_test_name == 'safesearch':
  896. safesearch = '?'
  897. elif time_range_support and checker_test_name == 'time_range':
  898. time_range_support = '?'
  899. supports[e.name] = {
  900. 'supports_selected_language': supports_selected_language,
  901. 'safesearch': safesearch,
  902. 'time_range_support': time_range_support,
  903. }
  904. return render(
  905. # fmt: off
  906. 'preferences.html',
  907. selected_categories = get_selected_categories(request.preferences, request.form),
  908. locales = LOCALE_NAMES,
  909. current_locale = request.preferences.get_value("locale"),
  910. image_proxy = image_proxy,
  911. engines_by_category = engines_by_category,
  912. stats = stats,
  913. max_rate95 = max_rate95,
  914. reliabilities = reliabilities,
  915. supports = supports,
  916. answerers = [
  917. {'info': a.self_info(), 'keywords': a.keywords}
  918. for a in answerers
  919. ],
  920. disabled_engines = disabled_engines,
  921. autocomplete_backends = autocomplete_backends,
  922. shortcuts = {y: x for x, y in engine_shortcuts.items()},
  923. themes = themes,
  924. plugins = plugins,
  925. doi_resolvers = settings['doi_resolvers'],
  926. current_doi_resolver = get_doi_resolver(request.preferences),
  927. allowed_plugins = allowed_plugins,
  928. preferences_url_params = request.preferences.get_as_url_params(),
  929. locked_preferences = settings['preferences']['lock'],
  930. preferences = True
  931. # fmt: on
  932. )
  933. def _is_selected_language_supported(engine, preferences: Preferences): # pylint: disable=redefined-outer-name
  934. language = preferences.get_value('language')
  935. if language == 'all':
  936. return True
  937. x = match_language(
  938. language, getattr(engine, 'supported_languages', []), getattr(engine, 'language_aliases', {}), None
  939. )
  940. return bool(x)
  941. @app.route('/image_proxy', methods=['GET'])
  942. def image_proxy():
  943. # pylint: disable=too-many-return-statements, too-many-branches
  944. url = request.args.get('url')
  945. if not url:
  946. return '', 400
  947. if not is_hmac_of(settings['server']['secret_key'], url.encode(), request.args.get('h', '')):
  948. return '', 400
  949. maximum_size = 5 * 1024 * 1024
  950. forward_resp = False
  951. resp = None
  952. try:
  953. request_headers = {
  954. 'User-Agent': gen_useragent(),
  955. 'Accept': 'image/webp,*/*',
  956. 'Accept-Encoding': 'gzip, deflate',
  957. 'Sec-GPC': '1',
  958. 'DNT': '1',
  959. }
  960. set_context_network_name('image_proxy')
  961. resp, stream = http_stream(method='GET', url=url, headers=request_headers, allow_redirects=True)
  962. content_length = resp.headers.get('Content-Length')
  963. if content_length and content_length.isdigit() and int(content_length) > maximum_size:
  964. return 'Max size', 400
  965. if resp.status_code != 200:
  966. logger.debug('image-proxy: wrong response code: %i', resp.status_code)
  967. if resp.status_code >= 400:
  968. return '', resp.status_code
  969. return '', 400
  970. if not resp.headers.get('Content-Type', '').startswith('image/'):
  971. logger.debug('image-proxy: wrong content-type: %s', resp.headers.get('Content-Type', ''))
  972. return '', 400
  973. forward_resp = True
  974. except httpx.HTTPError:
  975. logger.exception('HTTP error')
  976. return '', 400
  977. finally:
  978. if resp and not forward_resp:
  979. # the code is about to return an HTTP 400 error to the browser
  980. # we make sure to close the response between searxng and the HTTP server
  981. try:
  982. resp.close()
  983. except httpx.HTTPError:
  984. logger.exception('HTTP error on closing')
  985. def close_stream():
  986. nonlocal resp, stream
  987. try:
  988. if resp:
  989. resp.close()
  990. del resp
  991. del stream
  992. except httpx.HTTPError as e:
  993. logger.debug('Exception while closing response', e)
  994. try:
  995. headers = dict_subset(resp.headers, {'Content-Type', 'Content-Encoding', 'Content-Length', 'Length'})
  996. response = Response(stream, mimetype=resp.headers['Content-Type'], headers=headers, direct_passthrough=True)
  997. response.call_on_close(close_stream)
  998. return response
  999. except httpx.HTTPError:
  1000. close_stream()
  1001. return '', 400
  1002. @app.route('/engine_descriptions.json', methods=['GET'])
  1003. def engine_descriptions():
  1004. locale = get_locale().split('_')[0]
  1005. result = ENGINE_DESCRIPTIONS['en'].copy()
  1006. if locale != 'en':
  1007. for engine, description in ENGINE_DESCRIPTIONS.get(locale, {}).items():
  1008. result[engine] = description
  1009. for engine, description in result.items():
  1010. if len(description) == 2 and description[1] == 'ref':
  1011. ref_engine, ref_lang = description[0].split(':')
  1012. description = ENGINE_DESCRIPTIONS[ref_lang][ref_engine]
  1013. if isinstance(description, str):
  1014. description = [description, 'wikipedia']
  1015. result[engine] = description
  1016. # overwrite by about:description (from settings)
  1017. for engine_name, engine_mod in engines.items():
  1018. descr = getattr(engine_mod, 'about', {}).get('description', None)
  1019. if descr is not None:
  1020. result[engine_name] = [descr, "SearXNG config"]
  1021. return jsonify(result)
  1022. @app.route('/stats', methods=['GET'])
  1023. def stats():
  1024. """Render engine statistics page."""
  1025. sort_order = request.args.get('sort', default='name', type=str)
  1026. selected_engine_name = request.args.get('engine', default=None, type=str)
  1027. filtered_engines = dict(filter(lambda kv: request.preferences.validate_token(kv[1]), engines.items()))
  1028. if selected_engine_name:
  1029. if selected_engine_name not in filtered_engines:
  1030. selected_engine_name = None
  1031. else:
  1032. filtered_engines = [selected_engine_name]
  1033. checker_results = checker_get_result()
  1034. checker_results = (
  1035. checker_results['engines'] if checker_results['status'] == 'ok' and 'engines' in checker_results else {}
  1036. )
  1037. engine_stats = get_engines_stats(filtered_engines)
  1038. engine_reliabilities = get_reliabilities(filtered_engines, checker_results)
  1039. if sort_order not in STATS_SORT_PARAMETERS:
  1040. sort_order = 'name'
  1041. reverse, key_name, default_value = STATS_SORT_PARAMETERS[sort_order]
  1042. def get_key(engine_stat):
  1043. reliability = engine_reliabilities.get(engine_stat['name'], {}).get('reliablity', 0)
  1044. reliability_order = 0 if reliability else 1
  1045. if key_name == 'reliability':
  1046. key = reliability
  1047. reliability_order = 0
  1048. else:
  1049. key = engine_stat.get(key_name) or default_value
  1050. if reverse:
  1051. reliability_order = 1 - reliability_order
  1052. return (reliability_order, key, engine_stat['name'])
  1053. engine_stats['time'] = sorted(engine_stats['time'], reverse=reverse, key=get_key)
  1054. return render(
  1055. # fmt: off
  1056. 'stats.html',
  1057. sort_order = sort_order,
  1058. engine_stats = engine_stats,
  1059. engine_reliabilities = engine_reliabilities,
  1060. selected_engine_name = selected_engine_name,
  1061. searx_git_branch = GIT_BRANCH,
  1062. # fmt: on
  1063. )
  1064. @app.route('/stats/errors', methods=['GET'])
  1065. def stats_errors():
  1066. filtered_engines = dict(filter(lambda kv: request.preferences.validate_token(kv[1]), engines.items()))
  1067. result = get_engine_errors(filtered_engines)
  1068. return jsonify(result)
  1069. @app.route('/stats/checker', methods=['GET'])
  1070. def stats_checker():
  1071. result = checker_get_result()
  1072. return jsonify(result)
  1073. @app.route('/robots.txt', methods=['GET'])
  1074. def robots():
  1075. return Response(
  1076. """User-agent: *
  1077. Allow: /info/en/about
  1078. Disallow: /stats
  1079. Disallow: /image_proxy
  1080. Disallow: /preferences
  1081. Disallow: /*?*q=*
  1082. """,
  1083. mimetype='text/plain',
  1084. )
  1085. @app.route('/opensearch.xml', methods=['GET'])
  1086. def opensearch():
  1087. method = 'post'
  1088. if request.preferences.get_value('method') == 'GET':
  1089. method = 'get'
  1090. # chrome/chromium only supports HTTP GET....
  1091. if request.headers.get('User-Agent', '').lower().find('webkit') >= 0:
  1092. method = 'get'
  1093. autocomplete = request.preferences.get_value('autocomplete')
  1094. ret = render('opensearch.xml', opensearch_method=method, autocomplete=autocomplete)
  1095. resp = Response(response=ret, status=200, mimetype="application/opensearchdescription+xml")
  1096. return resp
  1097. @app.route('/favicon.ico')
  1098. def favicon():
  1099. theme = request.preferences.get_value("theme")
  1100. return send_from_directory(
  1101. os.path.join(app.root_path, settings['ui']['static_path'], 'themes', theme, 'img'),
  1102. 'favicon.png',
  1103. mimetype='image/vnd.microsoft.icon',
  1104. )
  1105. @app.route('/clear_cookies')
  1106. def clear_cookies():
  1107. resp = make_response(redirect(url_for('index', _external=True)))
  1108. for cookie_name in request.cookies:
  1109. resp.delete_cookie(cookie_name)
  1110. return resp
  1111. @app.route('/config')
  1112. def config():
  1113. """Return configuration in JSON format."""
  1114. _engines = []
  1115. for name, engine in engines.items():
  1116. if not request.preferences.validate_token(engine):
  1117. continue
  1118. supported_languages = engine.supported_languages
  1119. if isinstance(engine.supported_languages, dict):
  1120. supported_languages = list(engine.supported_languages.keys())
  1121. _engines.append(
  1122. {
  1123. 'name': name,
  1124. 'categories': engine.categories,
  1125. 'shortcut': engine.shortcut,
  1126. 'enabled': not engine.disabled,
  1127. 'paging': engine.paging,
  1128. 'language_support': engine.language_support,
  1129. 'supported_languages': supported_languages,
  1130. 'safesearch': engine.safesearch,
  1131. 'time_range_support': engine.time_range_support,
  1132. 'timeout': engine.timeout,
  1133. }
  1134. )
  1135. _plugins = []
  1136. for _ in plugins:
  1137. _plugins.append({'name': _.name, 'enabled': _.default_on})
  1138. return jsonify(
  1139. {
  1140. 'categories': list(categories.keys()),
  1141. 'engines': _engines,
  1142. 'plugins': _plugins,
  1143. 'instance_name': settings['general']['instance_name'],
  1144. 'locales': LOCALE_NAMES,
  1145. 'default_locale': settings['ui']['default_locale'],
  1146. 'autocomplete': settings['search']['autocomplete'],
  1147. 'safe_search': settings['search']['safe_search'],
  1148. 'default_theme': settings['ui']['default_theme'],
  1149. 'version': VERSION_STRING,
  1150. 'brand': {
  1151. 'CONTACT_URL': get_setting('general.contact_url'),
  1152. 'GIT_URL': GIT_URL,
  1153. 'GIT_BRANCH': GIT_BRANCH,
  1154. 'DOCS_URL': get_setting('brand.docs_url'),
  1155. },
  1156. 'doi_resolvers': list(settings['doi_resolvers'].keys()),
  1157. 'default_doi_resolver': settings['default_doi_resolver'],
  1158. }
  1159. )
  1160. @app.errorhandler(404)
  1161. def page_not_found(_e):
  1162. return render('404.html'), 404
  1163. # see https://flask.palletsprojects.com/en/1.1.x/cli/
  1164. # True if "FLASK_APP=searx/webapp.py FLASK_ENV=development flask run"
  1165. flask_run_development = (
  1166. os.environ.get("FLASK_APP") is not None and os.environ.get("FLASK_ENV") == 'development' and is_flask_run_cmdline()
  1167. )
  1168. # True if reload feature is activated of werkzeug, False otherwise (including uwsgi, etc..)
  1169. # __name__ != "__main__" if searx.webapp is imported (make test, make docs, uwsgi...)
  1170. # see run() at the end of this file : searx_debug activates the reload feature.
  1171. werkzeug_reloader = flask_run_development or (searx_debug and __name__ == "__main__")
  1172. # initialize the engines except on the first run of the werkzeug server.
  1173. if not werkzeug_reloader or (werkzeug_reloader and os.environ.get("WERKZEUG_RUN_MAIN") == "true"):
  1174. plugin_initialize(app)
  1175. search_initialize(enable_checker=True, check_network=True, enable_metrics=settings['general']['enable_metrics'])
  1176. def run():
  1177. logger.debug('starting webserver on %s:%s', settings['server']['bind_address'], settings['server']['port'])
  1178. app.run(
  1179. debug=searx_debug,
  1180. use_debugger=searx_debug,
  1181. port=settings['server']['port'],
  1182. host=settings['server']['bind_address'],
  1183. threaded=True,
  1184. extra_files=[get_default_settings_path()],
  1185. )
  1186. application = app
  1187. patch_application(app)
  1188. if __name__ == "__main__":
  1189. run()