|
|
@ -63,7 +63,8 @@ def index():
|
|
|
|
request_data = request.form
|
|
|
|
request_data = request.form
|
|
|
|
else:
|
|
|
|
else:
|
|
|
|
request_data = request.args
|
|
|
|
request_data = request.args
|
|
|
|
if request_data.get('q'):
|
|
|
|
if not request_data.get('q'):
|
|
|
|
|
|
|
|
return render('index.html')
|
|
|
|
selected_engines = []
|
|
|
|
selected_engines = []
|
|
|
|
selected_categories = []
|
|
|
|
selected_categories = []
|
|
|
|
for pd_name,pd in request_data.items():
|
|
|
|
for pd_name,pd in request_data.items():
|
|
|
@ -72,18 +73,13 @@ def index():
|
|
|
|
if not category in categories:
|
|
|
|
if not category in categories:
|
|
|
|
continue
|
|
|
|
continue
|
|
|
|
selected_categories.append(category)
|
|
|
|
selected_categories.append(category)
|
|
|
|
selected_engines.extend(x.name for x in categories[category])
|
|
|
|
if not len(selected_categories):
|
|
|
|
if not len(selected_engines):
|
|
|
|
|
|
|
|
cookie_categories = request.cookies.get('categories', '').split(',')
|
|
|
|
cookie_categories = request.cookies.get('categories', '').split(',')
|
|
|
|
for ccateg in cookie_categories:
|
|
|
|
for ccateg in cookie_categories:
|
|
|
|
if ccateg in categories:
|
|
|
|
if ccateg in categories:
|
|
|
|
selected_categories.append(ccateg)
|
|
|
|
selected_categories.append(ccateg)
|
|
|
|
selected_engines.extend(x.name for x in categories[ccateg])
|
|
|
|
|
|
|
|
if not len(selected_engines):
|
|
|
|
|
|
|
|
selected_categories.append('general')
|
|
|
|
|
|
|
|
selected_engines.extend(x.name for x in categories['general'])
|
|
|
|
|
|
|
|
query = request_data['q'].encode('utf-8')
|
|
|
|
query = request_data['q'].encode('utf-8')
|
|
|
|
results = search(query, request, selected_engines)
|
|
|
|
results = search(query, request, selected_categories)
|
|
|
|
if request_data.get('format') == 'json':
|
|
|
|
if request_data.get('format') == 'json':
|
|
|
|
# TODO HTTP headers
|
|
|
|
# TODO HTTP headers
|
|
|
|
return json.dumps({'query': query, 'results': results})
|
|
|
|
return json.dumps({'query': query, 'results': results})
|
|
|
@ -96,7 +92,6 @@ def index():
|
|
|
|
resp = make_response(template)
|
|
|
|
resp = make_response(template)
|
|
|
|
resp.set_cookie('categories', ','.join(selected_categories))
|
|
|
|
resp.set_cookie('categories', ','.join(selected_categories))
|
|
|
|
return resp
|
|
|
|
return resp
|
|
|
|
return render('index.html')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@app.route('/favicon.ico', methods=['GET'])
|
|
|
|
@app.route('/favicon.ico', methods=['GET'])
|
|
|
|
def fav():
|
|
|
|
def fav():
|
|
|
|