webapp.py 38 KB

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