webapp.py 42 KB

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