Merge branch 'master' of gitlab.com:uncel_dolan/similar-beer

1 files changed, 10 insertions, 1 deletions

diff --git a/app.py b/app.py
index 26f3447..eb6d9c3 100644
--- a/app.py
+++ b/app.py
@@ -109,7 +109,7 @@ def get_beer_details(brewery_id, beer_id):
         if attr != None and direction != None:
             kwargs['attr'] = attr
             kwargs['direction'] = direction
-            attr_beers = translate_to_attr(full_id, attr, int(direction))
+            attr_beers = translate_to_attr(full_id, attr, directionValue(direction))
             attr_beer_objects = []
             for item in attr_beers:
                 item_result, item_brewer = beer_lookup(item[0])
@@ -126,6 +126,15 @@ def get_beer_details(brewery_id, beer_id):
 
     return render_template('beerpage.jinja2', **kwargs)
 
+def directionValue(name):
+    if name == 'Much more':
+        return 3
+    if name == 'More':
+        return 2
+    if name == 'Slightly more':
+        return 1
+    return 0
+
 @app.route('/all_beers.json')
 def get_all_beers():
     q = request.args.get('q')