@@ -22,7 +22,7 @@ if __name__ == "__main__":
22 from sys import path
22 from sys import path
23 path.append(os.path.realpath(os.path.dirname(os.path.realpath(__file__))+'/../'))
23 path.append(os.path.realpath(os.path.dirname(os.path.realpath(__file__))+'/../'))
25−from flask import Flask, request, render_template, url_for, Response, make_response
25+from flask import Flask, request, render_template, url_for, Response, make_response, redirect
26from searx.engines import search, categories, engines, get_engines_stats
26from searx.engines import search, categories, engines, get_engines_stats
27from searx import settings
27from searx import settings
28import json
28import json
@@ -124,29 +124,46 @@ def index():
124 response.headers.add('Content-Disposition', 'attachment;Filename=searx_-_{0}.csv'.format('_'.join(query.split())))
124 response.headers.add('Content-Disposition', 'attachment;Filename=searx_-_{0}.csv'.format('_'.join(query.split())))
125 return response
125 return response
127− template = render('results.html'
127+ return render('results.html'
128− ,results=results
128+ ,results=results
129− ,q=request_data['q']
129+ ,q=request_data['q']
130− ,selected_categories=selected_categories
130+ ,selected_categories=selected_categories
131− ,number_of_results=len(results)
131+ ,number_of_results=len(results)
132− ,suggestions=suggestions
132+ ,suggestions=suggestions
134− resp = make_response(template)
135− resp.set_cookie('categories', ','.join(selected_categories))
139@app.route('/about', methods=['GET'])
136@app.route('/about', methods=['GET'])
140def about():
137def about():
141 global categories
138 global categories
142 return render('about.html', categs=categories.items())
139 return render('about.html', categs=categories.items())
142+@app.route('/preferences', methods=['GET', 'POST'])
145+ if request.method=='POST':
146+ selected_categories = []
147+ for pd_name,pd in request.form.items():
148+ if pd_name.startswith('category_'):
149+ category = pd_name[9:]
150+ if not category in categories:
152+ selected_categories.append(category)
153+ if selected_categories:
154+ resp = make_response(redirect('/'))
155+ resp.set_cookie('categories', ','.join(selected_categories))
157+ return render('preferences.html')
144@app.route('/stats', methods=['GET'])
160@app.route('/stats', methods=['GET'])
145def stats():
161def stats():
146 global categories
162 global categories
147 stats = get_engines_stats()
163 stats = get_engines_stats()
148 return render('stats.html', stats=stats)
164 return render('stats.html', stats=stats)
150@app.route('/robots.txt', methods=['GET'])
167@app.route('/robots.txt', methods=['GET'])
151def robots():
168def robots():
152 return Response("""User-agent: *
169 return Response("""User-agent: *
@@ -155,6 +172,7 @@ Allow: /about
155Disallow: /stats
172Disallow: /stats
156""", mimetype='text/plain')
173""", mimetype='text/plain')
158@app.route('/opensearch.xml', methods=['GET'])
176@app.route('/opensearch.xml', methods=['GET'])
159def opensearch():
177def opensearch():
160 global opensearch_xml
178 global opensearch_xml
@@ -165,8 +183,8 @@ def opensearch():
165 method = 'get'
183 method = 'get'
166 if request.is_secure:
184 if request.is_secure:
167 scheme = 'https'
185 scheme = 'https'
168− if settings.hostname:
186+ if settings.base_url:
169− hostname = '{0}://{1}/'.format(scheme,settings.hostname)
187+ hostname = settings.base_url
171 hostname = url_for('index', _external=True, _scheme=scheme)
189 hostname = url_for('index', _external=True, _scheme=scheme)
172 ret = opensearch_xml.format(method=method, host=hostname)
190 ret = opensearch_xml.format(method=method, host=hostname)