Handle 404HEADmaster

1 files changed, 10 insertions, 7 deletions

diff --git a/app.py b/app.py
index de2a5af..0b3ea36 100644
--- a/app.py
+++ b/app.py
@@ -147,14 +147,17 @@ def get_all_beers():
 
 @app.route('/search', methods=['POST'])
 def search():
-    q = request.form.get('q')
-    if not q:
+    try:
+        q = request.form.get('q')
+        if not q:
+            return redirect('/')
+        q = html.escape(q)
+        brewery, beer_name = q.split(' — ')
+        print('SEARCHING: ', q)
+        beer_id, brewery_id = name_list[q].split('-')
+        return redirect(url_for('get_beer_details', beer_id=beer_id, brewery_id=brewery_id))
+    except:
         return redirect('/')
-    q = html.escape(q)
-    brewery, beer_name = q.split(' — ')
-    print('SEARCHING: ', q)
-    beer_id, brewery_id = name_list[q].split('-')
-    return redirect(url_for('get_beer_details', beer_id=beer_id, brewery_id=brewery_id))