diff options
Diffstat (limited to 'app.py')
| -rw-r--r-- | app.py | 34 |
1 files changed, 23 insertions, 11 deletions
@@ -3,7 +3,7 @@ from flask import render_template from flask import redirect from flask import request from flask import url_for -from utils import get_closest, translate_to_attr, get_style_preds +from utils import get_closest, translate_to_attr, get_style_preds, translate_to_style import json import random import re @@ -77,16 +77,28 @@ def get_beer_details(brewery_id, beer_id): similar_beer_objects.append(BeerInfo(item, item_brewer, item_result)) if request.method == 'POST': - attr = request.form['attr'] - direction = request.form['direction'] - attr_beers = translate_to_attr(full_id, attr, int(direction)) - attr_beer_objects = [] - for item in attr_beers: - item_result, item_brewer = beer_lookup(item[0]) - attr_beer_objects.append(BeerInfo(item[0], item_brewer, item_result)) - return render_template('beerpage.jinja2', beer_data=beerinfo, - attrs=ATTRS, similar_beers=similar_beer_objects, attr=attr, - direction=direction, attr_beers=attr_beer_objects, style_preds=similar_beers) + + attr = request.form.get('attr') + direction = request.form.get('direction') + style = request.form.get('style') + if attr != None and direction != None: + attr_beers = translate_to_attr(full_id, attr, int(direction)) + attr_beer_objects = [] + for item in attr_beers: + item_result, item_brewer = beer_lookup(item[0]) + attr_beer_objects.append(BeerInfo(item[0], item_brewer, item_result)) + return render_template('beerpage.jinja2', beer_data=beerinfo, + attrs=attrs, similar_beers=similar_beer_objects, attr=attr, + direction=direction, attr_beers=attr_beer_objects, style_preds=similar_beers) + if style != None: + style_beers = translate_to_style(full_id, style) + style_beer_objects = [] + for item in style_beers: + item_result, item_brewer = beer_lookup(item[0]) + style_beer_objects.append(BeerInfo(item[0], item_brewer, item_result)) + return render_template('beerpage.jinja2', beer_data=beerinfo, + attrs=attrs, similar_beers=similar_beer_objects, + style=style, style_beers=style_beer_objects, style_preds=similar_beers) return render_template('beerpage.jinja2', beer_data=beerinfo, attrs=ATTRS, similar_beers=similar_beer_objects, style_preds=similar_beers) |