webapp.py 38 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075
  1. #!/usr/bin/env python
  2. '''
  3. searx is free software: you can redistribute it and/or modify
  4. it under the terms of the GNU Affero General Public License as published by
  5. the Free Software Foundation, either version 3 of the License, or
  6. (at your option) any later version.
  7. searx is distributed in the hope that it will be useful,
  8. but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  10. GNU Affero General Public License for more details.
  11. You should have received a copy of the GNU Affero General Public License
  12. along with searx. If not, see < http://www.gnu.org/licenses/ >.
  13. (C) 2013- by Adam Tauber, <asciimoo@gmail.com>
  14. '''
  15. import sys
  16. if sys.version_info[0] < 3:
  17. print('\033[1;31m Python2 is no longer supported\033[0m')
  18. exit(1)
  19. if __name__ == '__main__':
  20. from os.path import realpath, dirname
  21. sys.path.append(realpath(dirname(realpath(__file__)) + '/../'))
  22. import hashlib
  23. import hmac
  24. import json
  25. import os
  26. import requests
  27. from searx import logger
  28. logger = logger.getChild('webapp')
  29. from datetime import datetime, timedelta
  30. from time import time
  31. from html import escape
  32. from io import StringIO
  33. from urllib.parse import urlencode, urlparse, urljoin
  34. from pygments import highlight
  35. from pygments.lexers import get_lexer_by_name
  36. from pygments.formatters import HtmlFormatter
  37. from werkzeug.middleware.proxy_fix import ProxyFix
  38. from flask import (
  39. Flask, request, render_template, url_for, Response, make_response,
  40. redirect, send_from_directory
  41. )
  42. from babel.support import Translations
  43. import flask_babel
  44. from flask_babel import Babel, gettext, format_date, format_decimal
  45. from flask.ctx import has_request_context
  46. from flask.json import jsonify
  47. from searx import brand, static_path
  48. from searx import settings, searx_dir, searx_debug
  49. from searx.exceptions import SearxParameterException
  50. from searx.engines import (
  51. categories, engines, engine_shortcuts, get_engines_stats, initialize_engines
  52. )
  53. from searx.webutils import (
  54. UnicodeWriter, highlight_content, get_resources_directory,
  55. get_static_files, get_result_templates, get_themes,
  56. prettify_url, new_hmac
  57. )
  58. from searx.webadapter import get_search_query_from_webapp, get_selected_categories
  59. from searx.utils import html_to_text, gen_useragent, dict_subset, match_language
  60. from searx.version import VERSION_STRING
  61. from searx.languages import language_codes as languages
  62. from searx.search import SearchWithPlugins
  63. from searx.query import RawTextQuery
  64. from searx.autocomplete import searx_bang, backends as autocomplete_backends
  65. from searx.plugins import plugins
  66. from searx.plugins.oa_doi_rewrite import get_doi_resolver
  67. from searx.preferences import Preferences, ValidationException, LANGUAGE_CODES
  68. from searx.answerers import answerers
  69. # check if the pyopenssl package is installed.
  70. # It is needed for SSL connection without trouble, see #298
  71. try:
  72. import OpenSSL.SSL # NOQA
  73. except ImportError:
  74. logger.critical("The pyopenssl package has to be installed.\n"
  75. "Some HTTPS connections will fail")
  76. # serve pages with HTTP/1.1
  77. from werkzeug.serving import WSGIRequestHandler
  78. WSGIRequestHandler.protocol_version = "HTTP/{}".format(settings['server'].get('http_protocol_version', '1.0'))
  79. # about static
  80. static_path = get_resources_directory(searx_dir, 'static', settings['ui']['static_path'])
  81. logger.debug('static directory is %s', static_path)
  82. static_files = get_static_files(static_path)
  83. # about templates
  84. default_theme = settings['ui']['default_theme']
  85. templates_path = get_resources_directory(searx_dir, 'templates', settings['ui']['templates_path'])
  86. logger.debug('templates directory is %s', templates_path)
  87. themes = get_themes(templates_path)
  88. result_templates = get_result_templates(templates_path)
  89. global_favicons = []
  90. for indice, theme in enumerate(themes):
  91. global_favicons.append([])
  92. theme_img_path = os.path.join(static_path, 'themes', theme, 'img', 'icons')
  93. for (dirpath, dirnames, filenames) in os.walk(theme_img_path):
  94. global_favicons[indice].extend(filenames)
  95. # Flask app
  96. app = Flask(
  97. __name__,
  98. static_folder=static_path,
  99. template_folder=templates_path
  100. )
  101. app.jinja_env.trim_blocks = True
  102. app.jinja_env.lstrip_blocks = True
  103. app.jinja_env.add_extension('jinja2.ext.loopcontrols')
  104. app.secret_key = settings['server']['secret_key']
  105. if not searx_debug \
  106. or os.environ.get("WERKZEUG_RUN_MAIN") == "true" \
  107. or os.environ.get('UWSGI_ORIGINAL_PROC_NAME') is not None:
  108. initialize_engines(settings['engines'])
  109. babel = Babel(app)
  110. rtl_locales = ['ar', 'arc', 'bcc', 'bqi', 'ckb', 'dv', 'fa', 'fa_IR', 'glk', 'he',
  111. 'ku', 'mzn', 'pnb', 'ps', 'sd', 'ug', 'ur', 'yi']
  112. # used when translating category names
  113. _category_names = (gettext('files'),
  114. gettext('general'),
  115. gettext('music'),
  116. gettext('social media'),
  117. gettext('images'),
  118. gettext('videos'),
  119. gettext('it'),
  120. gettext('news'),
  121. gettext('map'),
  122. gettext('science'))
  123. outgoing_proxies = settings['outgoing'].get('proxies') or None
  124. _flask_babel_get_translations = flask_babel.get_translations
  125. # monkey patch for flask_babel.get_translations
  126. def _get_translations():
  127. if has_request_context() and request.form.get('use-translation') == 'oc':
  128. babel_ext = flask_babel.current_app.extensions['babel']
  129. return Translations.load(next(babel_ext.translation_directories), 'oc')
  130. return _flask_babel_get_translations()
  131. flask_babel.get_translations = _get_translations
  132. def _get_browser_or_settings_language(request, lang_list):
  133. for lang in request.headers.get("Accept-Language", "en").split(","):
  134. if ';' in lang:
  135. lang = lang.split(';')[0]
  136. locale = match_language(lang, lang_list, fallback=None)
  137. if locale is not None:
  138. return locale
  139. return settings['search']['default_lang'] or 'en'
  140. @babel.localeselector
  141. def get_locale():
  142. if 'locale' in request.form\
  143. and request.form['locale'] in settings['locales']:
  144. # use locale from the form
  145. locale = request.form['locale']
  146. locale_source = 'form'
  147. elif request.preferences.get_value('locale') != '':
  148. # use locale from the preferences
  149. locale = request.preferences.get_value('locale')
  150. locale_source = 'preferences'
  151. else:
  152. # use local from the browser
  153. locale = _get_browser_or_settings_language(request, settings['locales'].keys())
  154. locale_source = 'browser'
  155. #
  156. if locale == 'zh_TW':
  157. locale = 'zh_Hant_TW'
  158. # see _get_translations function
  159. # and https://github.com/searx/searx/pull/1863
  160. if locale == 'oc':
  161. request.form['use-translation'] = 'oc'
  162. locale = 'fr_FR'
  163. logger.debug("%s uses locale `%s` from %s", request.url, locale, locale_source)
  164. return locale
  165. # code-highlighter
  166. @app.template_filter('code_highlighter')
  167. def code_highlighter(codelines, language=None):
  168. if not language:
  169. language = 'text'
  170. try:
  171. # find lexer by programing language
  172. lexer = get_lexer_by_name(language, stripall=True)
  173. except:
  174. # if lexer is not found, using default one
  175. logger.debug('highlighter cannot find lexer for {0}'.format(language))
  176. lexer = get_lexer_by_name('text', stripall=True)
  177. html_code = ''
  178. tmp_code = ''
  179. last_line = None
  180. # parse lines
  181. for line, code in codelines:
  182. if not last_line:
  183. line_code_start = line
  184. # new codeblock is detected
  185. if last_line is not None and\
  186. last_line + 1 != line:
  187. # highlight last codepart
  188. formatter = HtmlFormatter(linenos='inline',
  189. linenostart=line_code_start)
  190. html_code = html_code + highlight(tmp_code, lexer, formatter)
  191. # reset conditions for next codepart
  192. tmp_code = ''
  193. line_code_start = line
  194. # add codepart
  195. tmp_code += code + '\n'
  196. # update line
  197. last_line = line
  198. # highlight last codepart
  199. formatter = HtmlFormatter(linenos='inline', linenostart=line_code_start)
  200. html_code = html_code + highlight(tmp_code, lexer, formatter)
  201. return html_code
  202. # Extract domain from url
  203. @app.template_filter('extract_domain')
  204. def extract_domain(url):
  205. return urlparse(url)[1]
  206. def get_base_url():
  207. if settings['server']['base_url']:
  208. hostname = settings['server']['base_url']
  209. else:
  210. scheme = 'http'
  211. if request.is_secure:
  212. scheme = 'https'
  213. hostname = url_for('index', _external=True, _scheme=scheme)
  214. return hostname
  215. def get_current_theme_name(override=None):
  216. """Returns theme name.
  217. Checks in this order:
  218. 1. override
  219. 2. cookies
  220. 3. settings"""
  221. if override and (override in themes or override == '__common__'):
  222. return override
  223. theme_name = request.args.get('theme', request.preferences.get_value('theme'))
  224. if theme_name not in themes:
  225. theme_name = default_theme
  226. return theme_name
  227. def get_result_template(theme, template_name):
  228. themed_path = theme + '/result_templates/' + template_name
  229. if themed_path in result_templates:
  230. return themed_path
  231. return 'result_templates/' + template_name
  232. def url_for_theme(endpoint, override_theme=None, **values):
  233. if endpoint == 'static' and values.get('filename'):
  234. theme_name = get_current_theme_name(override=override_theme)
  235. filename_with_theme = "themes/{}/{}".format(theme_name, values['filename'])
  236. if filename_with_theme in static_files:
  237. values['filename'] = filename_with_theme
  238. return url_for(endpoint, **values)
  239. def proxify(url):
  240. if url.startswith('//'):
  241. url = 'https:' + url
  242. if not settings.get('result_proxy'):
  243. return url
  244. url_params = dict(mortyurl=url.encode())
  245. if settings['result_proxy'].get('key'):
  246. url_params['mortyhash'] = hmac.new(settings['result_proxy']['key'],
  247. url.encode(),
  248. hashlib.sha256).hexdigest()
  249. return '{0}?{1}'.format(settings['result_proxy']['url'],
  250. urlencode(url_params))
  251. def image_proxify(url):
  252. if url.startswith('//'):
  253. url = 'https:' + url
  254. if not request.preferences.get_value('image_proxy'):
  255. return url
  256. if url.startswith('data:image/'):
  257. # 50 is an arbitrary number to get only the beginning of the image.
  258. partial_base64 = url[len('data:image/'):50].split(';')
  259. if len(partial_base64) == 2 \
  260. and partial_base64[0] in ['gif', 'png', 'jpeg', 'pjpeg', 'webp', 'tiff', 'bmp']\
  261. and partial_base64[1].startswith('base64,'):
  262. return url
  263. else:
  264. return None
  265. if settings.get('result_proxy'):
  266. return proxify(url)
  267. h = new_hmac(settings['server']['secret_key'], url.encode())
  268. return '{0}?{1}'.format(url_for('image_proxy'),
  269. urlencode(dict(url=url.encode(), h=h)))
  270. def render(template_name, override_theme=None, **kwargs):
  271. disabled_engines = request.preferences.engines.get_disabled()
  272. enabled_categories = set(category for engine_name in engines
  273. for category in engines[engine_name].categories
  274. if (engine_name, category) not in disabled_engines)
  275. if 'categories' not in kwargs:
  276. kwargs['categories'] = [x for x in
  277. _get_ordered_categories()
  278. if x in enabled_categories]
  279. if 'autocomplete' not in kwargs:
  280. kwargs['autocomplete'] = request.preferences.get_value('autocomplete')
  281. locale = request.preferences.get_value('locale')
  282. if locale in rtl_locales and 'rtl' not in kwargs:
  283. kwargs['rtl'] = True
  284. kwargs['searx_version'] = VERSION_STRING
  285. kwargs['method'] = request.preferences.get_value('method')
  286. kwargs['safesearch'] = str(request.preferences.get_value('safesearch'))
  287. kwargs['language_codes'] = languages
  288. if 'current_language' not in kwargs:
  289. kwargs['current_language'] = match_language(request.preferences.get_value('language'),
  290. LANGUAGE_CODES)
  291. # override url_for function in templates
  292. kwargs['url_for'] = url_for_theme
  293. kwargs['image_proxify'] = image_proxify
  294. kwargs['proxify'] = proxify if settings.get('result_proxy', {}).get('url') else None
  295. kwargs['get_result_template'] = get_result_template
  296. kwargs['theme'] = get_current_theme_name(override=override_theme)
  297. kwargs['template_name'] = template_name
  298. kwargs['cookies'] = request.cookies
  299. kwargs['errors'] = request.errors
  300. kwargs['instance_name'] = settings['general']['instance_name']
  301. kwargs['results_on_new_tab'] = request.preferences.get_value('results_on_new_tab')
  302. kwargs['preferences'] = request.preferences
  303. kwargs['brand'] = brand
  304. kwargs['scripts'] = set()
  305. kwargs['endpoint'] = 'results' if 'q' in kwargs else request.endpoint
  306. for plugin in request.user_plugins:
  307. for script in plugin.js_dependencies:
  308. kwargs['scripts'].add(script)
  309. kwargs['styles'] = set()
  310. for plugin in request.user_plugins:
  311. for css in plugin.css_dependencies:
  312. kwargs['styles'].add(css)
  313. return render_template(
  314. '{}/{}'.format(kwargs['theme'], template_name), **kwargs)
  315. def _get_ordered_categories():
  316. ordered_categories = []
  317. if 'categories_order' not in settings['ui']:
  318. ordered_categories = ['general']
  319. ordered_categories.extend(x for x in sorted(categories.keys()) if x != 'general')
  320. return ordered_categories
  321. ordered_categories = settings['ui']['categories_order']
  322. ordered_categories.extend(x for x in sorted(categories.keys()) if x not in ordered_categories)
  323. return ordered_categories
  324. @app.before_request
  325. def pre_request():
  326. request.start_time = time()
  327. request.timings = []
  328. request.errors = []
  329. preferences = Preferences(themes, list(categories.keys()), engines, plugins)
  330. user_agent = request.headers.get('User-Agent', '').lower()
  331. if 'webkit' in user_agent and 'android' in user_agent:
  332. preferences.key_value_settings['method'].value = 'GET'
  333. request.preferences = preferences
  334. try:
  335. preferences.parse_dict(request.cookies)
  336. except:
  337. request.errors.append(gettext('Invalid settings, please edit your preferences'))
  338. # merge GET, POST vars
  339. # request.form
  340. request.form = dict(request.form.items())
  341. for k, v in request.args.items():
  342. if k not in request.form:
  343. request.form[k] = v
  344. if request.form.get('preferences'):
  345. preferences.parse_encoded_data(request.form['preferences'])
  346. else:
  347. try:
  348. preferences.parse_dict(request.form)
  349. except Exception as e:
  350. logger.exception('invalid settings')
  351. request.errors.append(gettext('Invalid settings'))
  352. # init search language and locale
  353. if not preferences.get_value("language"):
  354. preferences.parse_dict({"language": _get_browser_or_settings_language(request, LANGUAGE_CODES)})
  355. if not preferences.get_value("locale"):
  356. preferences.parse_dict({"locale": get_locale()})
  357. # request.user_plugins
  358. request.user_plugins = []
  359. allowed_plugins = preferences.plugins.get_enabled()
  360. disabled_plugins = preferences.plugins.get_disabled()
  361. for plugin in plugins:
  362. if ((plugin.default_on and plugin.id not in disabled_plugins)
  363. or plugin.id in allowed_plugins):
  364. request.user_plugins.append(plugin)
  365. @app.after_request
  366. def post_request(response):
  367. total_time = time() - request.start_time
  368. timings_all = ['total;dur=' + str(round(total_time * 1000, 3))]
  369. if len(request.timings) > 0:
  370. timings = sorted(request.timings, key=lambda v: v['total'])
  371. timings_total = ['total_' + str(i) + '_' + v['engine'] +
  372. ';dur=' + str(round(v['total'] * 1000, 3)) for i, v in enumerate(timings)]
  373. timings_load = ['load_' + str(i) + '_' + v['engine'] +
  374. ';dur=' + str(round(v['load'] * 1000, 3)) for i, v in enumerate(timings)]
  375. timings_all = timings_all + timings_total + timings_load
  376. response.headers.add('Server-Timing', ', '.join(timings_all))
  377. return response
  378. def index_error(output_format, error_message):
  379. if output_format == 'json':
  380. return Response(json.dumps({'error': error_message}),
  381. mimetype='application/json')
  382. elif output_format == 'csv':
  383. response = Response('', mimetype='application/csv')
  384. cont_disp = 'attachment;Filename=searx.csv'
  385. response.headers.add('Content-Disposition', cont_disp)
  386. return response
  387. elif output_format == 'rss':
  388. response_rss = render(
  389. 'opensearch_response_rss.xml',
  390. results=[],
  391. q=request.form['q'] if 'q' in request.form else '',
  392. number_of_results=0,
  393. base_url=get_base_url(),
  394. error_message=error_message,
  395. override_theme='__common__',
  396. )
  397. return Response(response_rss, mimetype='text/xml')
  398. else:
  399. # html
  400. request.errors.append(gettext('search error'))
  401. return render(
  402. 'index.html',
  403. selected_categories=get_selected_categories(request.form, request.preferences),
  404. )
  405. @app.route('/search', methods=['GET', 'POST'])
  406. @app.route('/', methods=['GET', 'POST'])
  407. def index():
  408. """Render index page.
  409. Supported outputs: html, json, csv, rss.
  410. """
  411. # output_format
  412. output_format = request.form.get('format', 'html')
  413. if output_format not in ['html', 'csv', 'json', 'rss']:
  414. output_format = 'html'
  415. # check if there is query
  416. if request.form.get('q') is None:
  417. if output_format == 'html':
  418. return render(
  419. 'index.html',
  420. selected_categories=get_selected_categories(request.form, request.preferences),
  421. )
  422. else:
  423. return index_error(output_format, 'No query'), 400
  424. # search
  425. search_query = None
  426. raw_text_query = None
  427. result_container = None
  428. try:
  429. search_query, raw_text_query, _, _ = get_search_query_from_webapp(request.preferences, request.form)
  430. # search = Search(search_query) # without plugins
  431. search = SearchWithPlugins(search_query, request.user_plugins, request)
  432. result_container = search.search()
  433. except Exception as e:
  434. # log exception
  435. logger.exception('search error')
  436. # is it an invalid input parameter or something else ?
  437. if (issubclass(e.__class__, SearxParameterException)):
  438. return index_error(output_format, e.message), 400
  439. else:
  440. return index_error(output_format, gettext('search error')), 500
  441. # results
  442. results = result_container.get_ordered_results()
  443. number_of_results = result_container.results_number()
  444. if number_of_results < result_container.results_length():
  445. number_of_results = 0
  446. # checkin for a external bang
  447. if result_container.redirect_url:
  448. return redirect(result_container.redirect_url)
  449. # UI
  450. advanced_search = request.form.get('advanced_search', None)
  451. # Server-Timing header
  452. request.timings = result_container.get_timings()
  453. # output
  454. for result in results:
  455. if output_format == 'html':
  456. if 'content' in result and result['content']:
  457. result['content'] = highlight_content(escape(result['content'][:1024]), search_query.query)
  458. if 'title' in result and result['title']:
  459. result['title'] = highlight_content(escape(result['title'] or ''), search_query.query)
  460. else:
  461. if result.get('content'):
  462. result['content'] = html_to_text(result['content']).strip()
  463. # removing html content and whitespace duplications
  464. result['title'] = ' '.join(html_to_text(result['title']).strip().split())
  465. if 'url' in result:
  466. result['pretty_url'] = prettify_url(result['url'])
  467. # TODO, check if timezone is calculated right
  468. if 'publishedDate' in result:
  469. try: # test if publishedDate >= 1900 (datetime module bug)
  470. result['pubdate'] = result['publishedDate'].strftime('%Y-%m-%d %H:%M:%S%z')
  471. except ValueError:
  472. result['publishedDate'] = None
  473. else:
  474. if result['publishedDate'].replace(tzinfo=None) >= datetime.now() - timedelta(days=1):
  475. timedifference = datetime.now() - result['publishedDate'].replace(tzinfo=None)
  476. minutes = int((timedifference.seconds / 60) % 60)
  477. hours = int(timedifference.seconds / 60 / 60)
  478. if hours == 0:
  479. result['publishedDate'] = gettext('{minutes} minute(s) ago').format(minutes=minutes)
  480. else:
  481. result['publishedDate'] = gettext('{hours} hour(s), {minutes} minute(s) ago').format(hours=hours, minutes=minutes) # noqa
  482. else:
  483. result['publishedDate'] = format_date(result['publishedDate'])
  484. if output_format == 'json':
  485. return Response(json.dumps({'query': search_query.query,
  486. 'number_of_results': number_of_results,
  487. 'results': results,
  488. 'answers': list(result_container.answers),
  489. 'corrections': list(result_container.corrections),
  490. 'infoboxes': result_container.infoboxes,
  491. 'suggestions': list(result_container.suggestions),
  492. 'unresponsive_engines': __get_translated_errors(result_container.unresponsive_engines)}, # noqa
  493. default=lambda item: list(item) if isinstance(item, set) else item),
  494. mimetype='application/json')
  495. elif output_format == 'csv':
  496. csv = UnicodeWriter(StringIO())
  497. keys = ('title', 'url', 'content', 'host', 'engine', 'score', 'type')
  498. csv.writerow(keys)
  499. for row in results:
  500. row['host'] = row['parsed_url'].netloc
  501. row['type'] = 'result'
  502. csv.writerow([row.get(key, '') for key in keys])
  503. for a in result_container.answers:
  504. row = {'title': a, 'type': 'answer'}
  505. csv.writerow([row.get(key, '') for key in keys])
  506. for a in result_container.suggestions:
  507. row = {'title': a, 'type': 'suggestion'}
  508. csv.writerow([row.get(key, '') for key in keys])
  509. for a in result_container.corrections:
  510. row = {'title': a, 'type': 'correction'}
  511. csv.writerow([row.get(key, '') for key in keys])
  512. csv.stream.seek(0)
  513. response = Response(csv.stream.read(), mimetype='application/csv')
  514. cont_disp = 'attachment;Filename=searx_-_{0}.csv'.format(search_query.query)
  515. response.headers.add('Content-Disposition', cont_disp)
  516. return response
  517. elif output_format == 'rss':
  518. response_rss = render(
  519. 'opensearch_response_rss.xml',
  520. results=results,
  521. answers=result_container.answers,
  522. corrections=result_container.corrections,
  523. suggestions=result_container.suggestions,
  524. q=request.form['q'],
  525. number_of_results=number_of_results,
  526. base_url=get_base_url(),
  527. override_theme='__common__',
  528. )
  529. return Response(response_rss, mimetype='text/xml')
  530. # HTML output format
  531. # suggestions: use RawTextQuery to get the suggestion URLs with the same bang
  532. suggestion_urls = list(map(lambda suggestion: {
  533. 'url': raw_text_query.changeQuery(suggestion).getFullQuery(),
  534. 'title': suggestion
  535. },
  536. result_container.suggestions))
  537. correction_urls = list(map(lambda correction: {
  538. 'url': raw_text_query.changeQuery(correction).getFullQuery(),
  539. 'title': correction
  540. },
  541. result_container.corrections))
  542. #
  543. return render(
  544. 'results.html',
  545. results=results,
  546. q=request.form['q'],
  547. selected_categories=search_query.categories,
  548. pageno=search_query.pageno,
  549. time_range=search_query.time_range,
  550. number_of_results=format_decimal(number_of_results),
  551. advanced_search=advanced_search,
  552. suggestions=suggestion_urls,
  553. answers=result_container.answers,
  554. corrections=correction_urls,
  555. infoboxes=result_container.infoboxes,
  556. paging=result_container.paging,
  557. unresponsive_engines=__get_translated_errors(result_container.unresponsive_engines),
  558. current_language=match_language(search_query.lang,
  559. LANGUAGE_CODES,
  560. fallback=request.preferences.get_value("language")),
  561. base_url=get_base_url(),
  562. theme=get_current_theme_name(),
  563. favicons=global_favicons[themes.index(get_current_theme_name())],
  564. timeout_limit=request.form.get('timeout_limit', None)
  565. )
  566. def __get_translated_errors(unresponsive_engines):
  567. translated_errors = []
  568. for unresponsive_engine in unresponsive_engines:
  569. error_msg = gettext(unresponsive_engine[1])
  570. if unresponsive_engine[2]:
  571. error_msg = "{} {}".format(error_msg, unresponsive_engine[2])
  572. translated_errors.append((unresponsive_engine[0], error_msg))
  573. return translated_errors
  574. @app.route('/about', methods=['GET'])
  575. def about():
  576. """Render about page"""
  577. return render(
  578. 'about.html',
  579. )
  580. @app.route('/autocompleter', methods=['GET', 'POST'])
  581. def autocompleter():
  582. """Return autocompleter results"""
  583. # set blocked engines
  584. disabled_engines = request.preferences.engines.get_disabled()
  585. # parse query
  586. raw_text_query = RawTextQuery(request.form.get('q', ''), disabled_engines)
  587. # check if search query is set
  588. if not raw_text_query.getQuery():
  589. return '', 400
  590. # run autocompleter
  591. completer = autocomplete_backends.get(request.preferences.get_value('autocomplete'))
  592. # parse searx specific autocompleter results like !bang
  593. raw_results = searx_bang(raw_text_query)
  594. # normal autocompletion results only appear if no inner results returned
  595. # and there is a query part besides the engine and language bangs
  596. if len(raw_results) == 0 and completer and (len(raw_text_query.query_parts) > 1 or
  597. (len(raw_text_query.languages) == 0 and
  598. not raw_text_query.specific)):
  599. # get language from cookie
  600. language = request.preferences.get_value('language')
  601. if not language or language == 'all':
  602. language = 'en'
  603. else:
  604. language = language.split('-')[0]
  605. # run autocompletion
  606. raw_results.extend(completer(raw_text_query.getQuery(), language))
  607. # parse results (write :language and !engine back to result string)
  608. results = []
  609. for result in raw_results:
  610. raw_text_query.changeQuery(result)
  611. # add parsed result
  612. results.append(raw_text_query.getFullQuery())
  613. # return autocompleter results
  614. if request.headers.get('X-Requested-With') == 'XMLHttpRequest':
  615. return Response(json.dumps(results),
  616. mimetype='application/json')
  617. return Response(json.dumps([raw_text_query.query, results]),
  618. mimetype='application/x-suggestions+json')
  619. @app.route('/preferences', methods=['GET', 'POST'])
  620. def preferences():
  621. """Render preferences page && save user preferences"""
  622. # save preferences
  623. if request.method == 'POST':
  624. resp = make_response(redirect(urljoin(settings['server']['base_url'], url_for('index'))))
  625. try:
  626. request.preferences.parse_form(request.form)
  627. except ValidationException:
  628. request.errors.append(gettext('Invalid settings, please edit your preferences'))
  629. return resp
  630. return request.preferences.save(resp)
  631. # render preferences
  632. image_proxy = request.preferences.get_value('image_proxy')
  633. lang = request.preferences.get_value('language')
  634. disabled_engines = request.preferences.engines.get_disabled()
  635. allowed_plugins = request.preferences.plugins.get_enabled()
  636. # stats for preferences page
  637. stats = {}
  638. engines_by_category = {}
  639. for c in categories:
  640. engines_by_category[c] = []
  641. for e in categories[c]:
  642. if not request.preferences.validate_token(e):
  643. continue
  644. stats[e.name] = {'time': None,
  645. 'warn_timeout': False,
  646. 'warn_time': False}
  647. if e.timeout > settings['outgoing']['request_timeout']:
  648. stats[e.name]['warn_timeout'] = True
  649. stats[e.name]['supports_selected_language'] = _is_selected_language_supported(e, request.preferences)
  650. engines_by_category[c].append(e)
  651. # get first element [0], the engine time,
  652. # and then the second element [1] : the time (the first one is the label)
  653. for engine_stat in get_engines_stats(request.preferences)[0][1]:
  654. stats[engine_stat.get('name')]['time'] = round(engine_stat.get('avg'), 3)
  655. if engine_stat.get('avg') > settings['outgoing']['request_timeout']:
  656. stats[engine_stat.get('name')]['warn_time'] = True
  657. # end of stats
  658. return render('preferences.html',
  659. selected_categories=get_selected_categories(request.form, request.preferences),
  660. all_categories=_get_ordered_categories(),
  661. locales=settings['locales'],
  662. current_locale=request.preferences.get_value("locale"),
  663. image_proxy=image_proxy,
  664. engines_by_category=engines_by_category,
  665. stats=stats,
  666. answerers=[{'info': a.self_info(), 'keywords': a.keywords} for a in answerers],
  667. disabled_engines=disabled_engines,
  668. autocomplete_backends=autocomplete_backends,
  669. shortcuts={y: x for x, y in engine_shortcuts.items()},
  670. themes=themes,
  671. plugins=plugins,
  672. doi_resolvers=settings['doi_resolvers'],
  673. current_doi_resolver=get_doi_resolver(request.args, request.preferences.get_value('doi_resolver')),
  674. allowed_plugins=allowed_plugins,
  675. theme=get_current_theme_name(),
  676. preferences_url_params=request.preferences.get_as_url_params(),
  677. base_url=get_base_url(),
  678. preferences=True)
  679. def _is_selected_language_supported(engine, preferences):
  680. language = preferences.get_value('language')
  681. return (language == 'all'
  682. or match_language(language,
  683. getattr(engine, 'supported_languages', []),
  684. getattr(engine, 'language_aliases', {}), None))
  685. @app.route('/image_proxy', methods=['GET'])
  686. def image_proxy():
  687. url = request.args.get('url').encode()
  688. if not url:
  689. return '', 400
  690. h = new_hmac(settings['server']['secret_key'], url)
  691. if h != request.args.get('h'):
  692. return '', 400
  693. headers = dict_subset(request.headers, {'If-Modified-Since', 'If-None-Match'})
  694. headers['User-Agent'] = gen_useragent()
  695. resp = requests.get(url,
  696. stream=True,
  697. timeout=settings['outgoing']['request_timeout'],
  698. headers=headers,
  699. proxies=outgoing_proxies)
  700. if resp.status_code == 304:
  701. return '', resp.status_code
  702. if resp.status_code != 200:
  703. logger.debug('image-proxy: wrong response code: {0}'.format(resp.status_code))
  704. if resp.status_code >= 400:
  705. return '', resp.status_code
  706. return '', 400
  707. if not resp.headers.get('content-type', '').startswith('image/'):
  708. logger.debug('image-proxy: wrong content-type: {0}'.format(resp.headers.get('content-type')))
  709. return '', 400
  710. img = b''
  711. chunk_counter = 0
  712. for chunk in resp.iter_content(1024 * 1024):
  713. chunk_counter += 1
  714. if chunk_counter > 5:
  715. return '', 502 # Bad gateway - file is too big (>5M)
  716. img += chunk
  717. headers = dict_subset(resp.headers, {'Content-Length', 'Length', 'Date', 'Last-Modified', 'Expires', 'Etag'})
  718. return Response(img, mimetype=resp.headers['content-type'], headers=headers)
  719. @app.route('/stats', methods=['GET'])
  720. def stats():
  721. """Render engine statistics page."""
  722. stats = get_engines_stats(request.preferences)
  723. return render(
  724. 'stats.html',
  725. stats=stats,
  726. )
  727. @app.route('/robots.txt', methods=['GET'])
  728. def robots():
  729. return Response("""User-agent: *
  730. Allow: /
  731. Allow: /about
  732. Disallow: /stats
  733. Disallow: /preferences
  734. Disallow: /*?*q=*
  735. """, mimetype='text/plain')
  736. @app.route('/opensearch.xml', methods=['GET'])
  737. def opensearch():
  738. method = 'post'
  739. if request.preferences.get_value('method') == 'GET':
  740. method = 'get'
  741. # chrome/chromium only supports HTTP GET....
  742. if request.headers.get('User-Agent', '').lower().find('webkit') >= 0:
  743. method = 'get'
  744. ret = render('opensearch.xml',
  745. opensearch_method=method,
  746. host=get_base_url(),
  747. urljoin=urljoin,
  748. override_theme='__common__')
  749. resp = Response(response=ret,
  750. status=200,
  751. mimetype="application/opensearchdescription+xml")
  752. return resp
  753. @app.route('/favicon.ico')
  754. def favicon():
  755. return send_from_directory(os.path.join(app.root_path,
  756. static_path,
  757. 'themes',
  758. get_current_theme_name(),
  759. 'img'),
  760. 'favicon.png',
  761. mimetype='image/vnd.microsoft.icon')
  762. @app.route('/clear_cookies')
  763. def clear_cookies():
  764. resp = make_response(redirect(urljoin(settings['server']['base_url'], url_for('index'))))
  765. for cookie_name in request.cookies:
  766. resp.delete_cookie(cookie_name)
  767. return resp
  768. @app.route('/config')
  769. def config():
  770. """Return configuration in JSON format."""
  771. _engines = []
  772. for name, engine in engines.items():
  773. if not request.preferences.validate_token(engine):
  774. continue
  775. supported_languages = engine.supported_languages
  776. if isinstance(engine.supported_languages, dict):
  777. supported_languages = list(engine.supported_languages.keys())
  778. _engines.append({
  779. 'name': name,
  780. 'categories': engine.categories,
  781. 'shortcut': engine.shortcut,
  782. 'enabled': not engine.disabled,
  783. 'paging': engine.paging,
  784. 'language_support': engine.language_support,
  785. 'supported_languages': supported_languages,
  786. 'safesearch': engine.safesearch,
  787. 'time_range_support': engine.time_range_support,
  788. 'timeout': engine.timeout
  789. })
  790. _plugins = []
  791. for _ in plugins:
  792. _plugins.append({'name': _.name, 'enabled': _.default_on})
  793. return jsonify({
  794. 'categories': list(categories.keys()),
  795. 'engines': _engines,
  796. 'plugins': _plugins,
  797. 'instance_name': settings['general']['instance_name'],
  798. 'locales': settings['locales'],
  799. 'default_locale': settings['ui']['default_locale'],
  800. 'autocomplete': settings['search']['autocomplete'],
  801. 'safe_search': settings['search']['safe_search'],
  802. 'default_theme': settings['ui']['default_theme'],
  803. 'version': VERSION_STRING,
  804. 'brand': {
  805. 'GIT_URL': brand.GIT_URL,
  806. 'DOCS_URL': brand.DOCS_URL
  807. },
  808. 'doi_resolvers': [r for r in settings['doi_resolvers']],
  809. 'default_doi_resolver': settings['default_doi_resolver'],
  810. })
  811. @app.route('/translations.js')
  812. def js_translations():
  813. return render(
  814. 'translations.js.tpl',
  815. override_theme='__common__',
  816. ), {'Content-Type': 'text/javascript; charset=UTF-8'}
  817. @app.errorhandler(404)
  818. def page_not_found(e):
  819. return render('404.html'), 404
  820. def run():
  821. logger.debug('starting webserver on %s:%s', settings['server']['bind_address'], settings['server']['port'])
  822. app.run(
  823. debug=searx_debug,
  824. use_debugger=searx_debug,
  825. port=settings['server']['port'],
  826. host=settings['server']['bind_address'],
  827. threaded=True
  828. )
  829. class ReverseProxyPathFix:
  830. '''Wrap the application in this middleware and configure the
  831. front-end server to add these headers, to let you quietly bind
  832. this to a URL other than / and to an HTTP scheme that is
  833. different than what is used locally.
  834. http://flask.pocoo.org/snippets/35/
  835. In nginx:
  836. location /myprefix {
  837. proxy_pass http://127.0.0.1:8000;
  838. proxy_set_header Host $host;
  839. proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
  840. proxy_set_header X-Scheme $scheme;
  841. proxy_set_header X-Script-Name /myprefix;
  842. }
  843. :param app: the WSGI application
  844. '''
  845. def __init__(self, app):
  846. self.app = app
  847. def __call__(self, environ, start_response):
  848. script_name = environ.get('HTTP_X_SCRIPT_NAME', '')
  849. if script_name:
  850. environ['SCRIPT_NAME'] = script_name
  851. path_info = environ['PATH_INFO']
  852. if path_info.startswith(script_name):
  853. environ['PATH_INFO'] = path_info[len(script_name):]
  854. scheme = environ.get('HTTP_X_SCHEME', '')
  855. if scheme:
  856. environ['wsgi.url_scheme'] = scheme
  857. return self.app(environ, start_response)
  858. application = app
  859. # patch app to handle non root url-s behind proxy & wsgi
  860. app.wsgi_app = ReverseProxyPathFix(ProxyFix(application.wsgi_app))
  861. if __name__ == "__main__":
  862. run()