from flask import Flask, render_template, redirect, request, url_for, jsonify
from utils import get_closest, translate_to_attr, get_style_preds, translate_to_style
import json
import random
import re
from utils import get_drinks_like

app = Flask(__name__)

with open("./data/beer_info_small.json") as beers_json:
    beer = json.loads(beers_json.read())
    beer_ids = {}
    for long_id, val in beer.items():
        beer_ids[val[0]] = long_id.split('-')[1]

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 = {}
    for id, name in brewery_names.items():
        brewery_ids[name] = id
ATTRS = ['Barnyardy',
 'Bitter',
 'Boozy',
 'Bready',
 'Burnt',
 'Dank',
 'Dark',
 'Dry',
 'Floral',
 'Fruity',
 'Grassy',
 'Herbal',
 'Hoppy',
 'Juicy',
 'Light',
 'Malty',
 'Roasty',
 'Salty',
 'Smoky',
 'Sour',
 'Spicy',
 'Sweet',
 'Thin',
 'Warming',
 'Wheaty',
 'Woody',
 'Yeasty']


@app.route('/')
def index():
    return render_template('Homepage.jinja2')


@app.route('/random')
def random_beer():
    rando = random.choice(list(beers_for_random)).split('-')
    return redirect(url_for('get_beer_details', brewery_id=rando[0], beer_id=rando[1]))


@app.route('/brewery/<brewery_id>/<beer_id>', methods=['GET', 'POST'])
def get_beer_details(brewery_id, beer_id):
    result, brewery_name = beer_lookup(brewery_id, beer_id)
    full_id = brewery_id+'-'+beer_id
    beerinfo = BeerInfo(full_id, brewery_name, result)

    similar_beers = get_closest(full_id)

    style_map = get_style_preds(full_id)
    print('Style Map:{}'.format(style_map))
    similar_beer_objects = []
    for item in similar_beers:
        item_result, item_brewer = beer_lookup(item)
        similar_beer_objects.append(BeerInfo(item, item_brewer, item_result))

    kwargs = {
            "beer_data": beerinfo,
            "attrs": ATTRS,
            "similar_beers": similar_beer_objects,
            "style_preds": similar_beers,
            "style_map": style_map
            }

    if request.method == 'POST':
        attr = request.form.get('attr')
        direction = request.form.get('direction')
        style = request.form.get('style')
        if attr != None and direction != None:
            kwargs['attr'] = attr
            kwargs['direction'] = 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))
            kwargs['attr_beers'] = attr_beer_objects
        elif style != None:
            kwargs['style'] = style
            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))
            kwargs['style_beers'] = style_beer_objects

    return render_template('beerpage.jinja2', **kwargs)

@app.route('/all_beers.json')
def get_all_beers():
    return jsonify([ brewery_names[k.split('-')[0]]+' — '+v[0] for k, v in beer.items()])

@app.route('/search', methods=['POST'])
def search():
    q = request.form.get('q')
    if not q:
        return redirect('/')
    brewery, beer_name = q.split(' — ')
    brewery_id = brewery_ids[brewery]
    beer_id = beer_ids[beer_name]
    return redirect(url_for('get_beer_details', beer_id=beer_id, brewery_id=brewery_id))


def beer_lookup(brewery_id, beer_id=None):
    if beer_id:
        item_id = brewery_id + "-" + beer_id
        try:
            beer_result = beer[item_id]
            return beer_result, brewery_names[brewery_id]
        except:
            return None
    else:
        try:
            beer_result = beer[brewery_id]
            return beer_result, brewery_names[brewery_id.split("-")[0]]
        except:
            return None


def find_beers_from_brewery(brewery_id):
    pattern = re.compile('^' + brewery_id + '-.*')
    beers = [value for key, value in beer.items() if pattern.match(key)]
    return beers


class BeerInfo(object):

    def __init__(self, beer_id, brewery_name, item_json):
        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)
        self.link ="/brewery/" + beer_id.split('-')[0] + '/' + beer_id.split('-')[1]

class Brewery(object):

    def __init__(self, breweryid):
        self.id = breweryid
        self.brews = []

    def add_brew(self, brew):
        self.brews.append(brew)


if __name__ == '__main__':
    app.config.update(TEMPLATES_AUTO_RELOAD=True)
    app.run(host='0.0.0.0')