From 73b5b848a07744bf9cf1969b4c072e256545136b Mon Sep 17 00:00:00 2001 From: Ben Cohen Date: Sun, 28 Jul 2019 20:12:26 -0400 Subject: search! --- app.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/app.py b/app.py index 5c4fdde..3d87995 100644 --- a/app.py +++ b/app.py @@ -11,13 +11,18 @@ app = Flask(__name__) with open("./data/beer_info_small.json") as beers_json: beer = json.loads(beers_json.read()) + +valid_names = set() # beer_ids = {} -# for long_id, val in beer.items(): -# beer_ids[val[0]] = long_id.split('-')[1] +for long_id, val in beer.items(): + valid_names.add(val[0]) + +print(valid_names) + with open('./data/name_list.json') as f: name_list = json.load(f) -orig_terms = list(name_list.keys()) +orig_terms = list([x for x in name_list.keys() if x.split(' — ')[1] in valid_names]) search_words = [''.join(f for f in x.lower() if f not in string.punctuation) for x in orig_terms] @@ -25,7 +30,7 @@ orig_term_map = dict(zip(search_words, orig_terms)) beers_for_random = [b for b in beer if beer[b][3] > 75] -print(len(beers_for_random)) + with open('./data/brewery_names.json') as f: brewery_names = json.load(f) # brewery_ids = {} -- cgit v1.2.3