-> small

1 files changed, 6 insertions, 3 deletions

diff --git a/app.py b/app.py
index 6d510f1..aff30aa 100644
--- a/app.py
+++ b/app.py
@@ -6,10 +6,11 @@ from flask import url_for
 import json
 import random
 import re
+from utils import get_drinks_like
 
 app = Flask(__name__)
 
-with open("./data/beer_info.json") as beers_json:
+with open("./data/beer_info_small.json") as beers_json:
     beer = json.loads(beers_json.read())
 
 with open('./data/brewery_names.json') as f:
@@ -35,7 +36,7 @@ def get_brewery_list(brewery_id):
 @app.route('/brewery/<brewery_id>/<beer_id>', methods=['GET', 'POST'])
 def get_beer_details(brewery_id, beer_id):
     beer_result, brewery_name = beer_lookup(brewery_id, beer_id)
-    beerinfo = BeerInfo(beer_result, brewery_name)
+    beerinfo = BeerInfo(beer_result, brewery_name, brewery_id + '-' +beer_id)
     attrs = ['Burnt', 'Bready', 'Malty', 'Dark', 'Floral', 'Hoppy', 'Dank',
              'Grassy', 'Boozy', 'Warming', 'Bitter', 'Wheaty', 'Sour',
              'Barnyardy', 'Woody', 'Smoky', 'Roasty', 'Yeasty', 'Sweet',
@@ -67,13 +68,15 @@ def find_beers_from_brewery(brewery_id):
 
 class BeerInfo(object):
 
-    def __init__(self, item_json, brewery_name):
+    def __init__(self, item_json, brewery_name, beer_id):
+        self._id = beer_id
         self.parent_style = item_json[1][1]
         self.base_style = item_json[1][0]
         self.name = item_json[0]
         self.abv = item_json[2]
         self.brewer = brewery_name
         self.num_reviews = item_json[3]
+        self.drinks_like = get_drinks_like(self._id)
 
 
 class Brewery(object):