server.py

#!/usr/bin/env python

from flask import Flask, render_template, redirect, request, flash, session, url_for, send_from_directory
from model import User, Recipe, Brew, Style, Extract, Hop, Misc, Yeast, Fermentable, YeastIns, HopIns, FermIns, MiscIns, ExtIns, connect_to_db, db
from feeder import load_recipes, load_hops_ins, load_ferm_ins, load_ext_ins, load_misc_ins, load_yeast_ins
# , calc_color
from builder import feed_recipe_form, get_selectlists, show_brew_recipe, color_conversion, normalize_batch_size, get_recipe_instructions
import xml.etree.ElementTree as ET
import os
from werkzeug import secure_filename
import json
import datetime

app = Flask(__name__)

# Required to use Flask sessions and the debug toolbar
app.secret_key = "ABC"

app.config['UPLOAD_FOLDER'] = 'tmp/uploads/'
app.config['ALLOWED_EXTENSIONS'] = set(['xml'])


def allowed_file(filename):
    return '.' in filename and \
           filename.rsplit('.', 1)[1] in app.config['ALLOWED_EXTENSIONS']


# ***************************************************************************************
# Home page

@app.route('/')
def index():
    """ Display home page """
    return render_template("homepage.html")


# ***************************************************************************************
# Explore recipes selections

@app.route('/explore', methods=['GET', 'POST'])
def show_explore():
    """ Display explore page """
    if session:
        selectlist_recipes, selectlist_styles, selectlist_user, sel_user_styles = get_selectlists(session["user_id"])
        new = True
    else:
        selectlist_recipes, selectlist_styles, selectlist_user, sel_user_styles = get_selectlists(9999)
        new = True

    # For recipe selected, show recipe.
    if request.method == "POST":
        # Render either recipe list for style selection or recipe
        if request.form.get("style"):
            style = request.form.get("style")
            list_recipes = [recipe.name for recipe in Recipe.query.filter_by(style_name=style).all()]
            return render_template("explore_brews.html", list_recipes=list_recipes,
                                   selectlist_recipes=selectlist_recipes,
                                   selectlist_styles=selectlist_styles)
        elif request.form.get("recipe"):
            recipe_name = request.form.get("recipe")
            recipe, color = get_recipe_instructions(recipe_name)

            print "SERVER SRM ", color
            color = color or recipe.calc_color
            color = color_conversion(color)
            deleteable = False
            if "user_id" in session and Recipe.query.filter_by(name=recipe_name).one().user_id == session["user_id"]:
                deleteable = True
            return render_template("explore_brews.html", 
                                   selectlist_recipes=selectlist_recipes,
                                   selectlist_styles=selectlist_styles,
                                   recipe=recipe, color=color,
                                   deleteable=deleteable)

    return render_template("explore_brews.html", new=new, selectlist_recipes=selectlist_recipes,
                           selectlist_styles=selectlist_styles, selectlisrt_user=selectlist_user,
                           sel_user_styles=sel_user_styles, session=session)


# ***************************************************************************************
# Explore Recipes right panel displays


@app.route('/myrecipes')
def get_my_recipes():
    """ List user's recipes """

    selectlist_recipes, selectlist_styles, selectlist_user, sel_user_styles = get_selectlists(session.get("user_id"))
    recipe_list = Recipe.query.filter_by(user_id=session.get("user_id")).all()
    recipes = []
    for recipe in recipe_list:
        recipes.append(recipe.name)

    return render_template("explore_brews.html", recipes=recipes, selectlist_recipes=selectlist_recipes,
                           selectlist_styles=selectlist_styles, selectlisrt_user=selectlist_user,
                           sel_user_styles=sel_user_styles)


@app.route('/recipe/<string:recipe>')
def get_recipes(recipe):
    """Show details of a single recipe """

    selectlist_recipes, selectlist_styles, selectlist_user, sel_user_styles = get_selectlists(session.get("user_id"))
    deleteable = False
    if Recipe.query.filter_by(name=recipe).one().user_id == session.get("user_id"):
        deleteable = True
    recipe, color = get_recipe_instructions(recipe)
    color = color_conversion(color)
    return render_template("explore_brews.html", selectlist_recipes=selectlist_recipes,
                           selectlist_styles=selectlist_styles, recipe=recipe,
                           color=color, deleteable=deleteable)


@app.route('/check_brew', methods=['GET', 'POST'])
def check_brew():
    """ Ajax call - Check for duplicate brew """

    recipe = request.form.get("name")
    user_id = session["user_id"]
    date = datetime.date.today()

    recipe_id = Recipe.query.filter_by(name=recipe).one().recipe_id
    if Brew.query.filter_by(user_id=user_id, recipe_id=recipe_id, date=date).first():
        return "duplicate"
    else:
        return "not duplicate"


@app.route('/addbrew/<string:recipe>')
def add_brew(recipe):
    """ Add new brew with today's date and route to brew page """

    user_id = session["user_id"]
    date = datetime.date.today()
    recipe_id = Recipe.query.filter_by(name=recipe).one().recipe_id
    new_brew = Brew(user_id=user_id, recipe_id=recipe_id, date=date)
    db.session.add(new_brew)
    db.session.commit()

    brew_id = str(new_brew.id)

    return redirect('/brew/' + brew_id)


@app.route('/deleterecipe/<string:recipe>')
def delete_recipe(recipe):
    """ Delete recipe from database and delete associated instructions """

    recipe_id = Recipe.query.filter_by(name=recipe).one().recipe_id
    HopIns.query.filter_by(recipe_id=recipe_id).delete()
    FermIns.query.filter_by(recipe_id=recipe_id).delete()
    ExtIns.query.filter_by(recipe_id=recipe_id).delete()
    MiscIns.query.filter_by(recipe_id=recipe_id).delete()
    YeastIns.query.filter_by(recipe_id=recipe_id).delete()
    Recipe.query.filter_by(name=recipe).delete()
    db.session.commit()

    flash("Recipe deleted")
    return redirect('/explore')


# ***************************************************************************************
# My Brews display and response to selections

@app.route('/mybrews', methods=['GET', 'POST'])
def show_mybrews():
    """ Display users brewed recipes, all or by filter. """

    selectlist_recipes, selectlist_styles, selectlist_user, sel_user_styles = get_selectlists(session["user_id"])
    filtered = False

    # Filtered lists
    if request.method == "POST":
        filtered_brews = []
        # Recipe search:
        if request.form.get("recipe"):
            recipe_name = request.form.get("recipe")
            recipe_id = Recipe.query.filter_by(name=recipe_name).one().recipe_id
            all_brews = Brew.query.filter_by(user_id=session["user_id"], recipe_id=recipe_id).all()
            filtered = "Brews of " + recipe_name + ":"

        # Style search:
        if request.form.get("style"):
            style = request.form.get("style")
            all_brews = db.session.query(Brew).join(Brew.recipe).filter_by(style_name=style, user_id=session["user_id"]).all()
            filtered = style + " Style Brews:"

    # Unfiltered list of all brews
    else:
        all_brews = Brew.query.filter_by(user_id=session["user_id"]).all()

    # Add hex color attribute for view
    for brew in all_brews:
        brew.color = color_conversion(brew.recipe.calc_color())

    return render_template("mybrews.html", brewlist=all_brews,
                           filtered=filtered,
                           selectlist_user=selectlist_user,
                           sel_user_styles=sel_user_styles)


@app.route('/delete_brew/<int:brew_id>')
def delete_brew(brew_id):
    """ Delete brew from database """

    Brew.query.filter_by(id=brew_id).delete()
    db.session.commit()
    flash("Your brew was deleted")
    return redirect("/mybrews")


# ***************************************************************************************
# Brew page

@app.route('/brew/<int:brew_id>', methods=['GET', 'POST'])
def brew_process(brew_id):
    brew = Brew.query.filter_by(id=brew_id).one()
    recipe = Recipe.query.filter_by(recipe_id=brew.recipe_id).one().name
    if request.method == "POST":
        brew_update = update_brew(brew)
        return redirect("/mybrews")
    else:
        recipe, times, timerset, boiltime, steep, yeast, secondary, extracts = show_brew_recipe(recipe)
        rating = brew.rating
        c_gravity = brew.cg
        c_gravity_date = brew.cg_date
        color = color_conversion(recipe.srm_color)
    print yeast
    print secondary

    return render_template("brew.html", brew=brew, recipe=recipe, 
                   times=times, timerset=timerset, boiltime=boiltime, steep=steep, yeast=yeast, secondary=secondary,
                   extracts=extracts, c_gravity=c_gravity, c_gravity_date=c_gravity_date,
                   rating=rating)


@app.route('/boil', methods=["POST"])
def note_boil_time():
    """ Ajax call -Store boil start time on change """

    boil_start = request.form.get("boil_start")
    brew_id = request.form.get("brew_id")
    startformat = '%H:%M'

    boil_start = datetime.datetime.strptime(boil_start, startformat).time()
    Brew.query.filter_by(id=brew_id).one().boil_start = boil_start
    db.session.commit()


def update_brew(brew):
    """ Save brew data to database """

    brew.user_id = session["user_id"]
    date_get = request.form.get('brew_date')
    brew.date = datetime.datetime.strptime(date_get, "%Y-%m-%d").date()
    time_get = request.form.get("boil_start")
    if request.form.get("curr_gravity"):
        brew.cg = request.form.get("curr_gravity")
    if request.form.get("cg_date"):
        cg_date_get = request.form.get("cg_date")
        brew.cg_date = datetime.datetime.strptime(cg_date_get, "%Y-%m-%d").date()

    if (request.form.get('orig_gravity')):
        brew.og = float(request.form.get('orig_gravity'))
    if request.form.get('final_gravity'):
        brew.fg = float(request.form.get('final_gravity'))
        brew.og = float(brew.og)
        brew.abv = (brew.og - brew.fg) * 131

    if request.form.get("finished"):
        end_date_get = request.form.get('finished')
        brew.end_date = datetime.datetime.strptime(end_date_get, "%Y-%m-%d").date()
    brew.notes = request.form.get('brew_notes')
    brew.results_notes = request.form.get('results_notes')
    brew.rating = request.form.get('rating')

    db.session.commit()

    brew_update = "Brew successfully updated"

    return brew_update


# ***************************************************************************************
# Add new recipes

@app.route('/addrecipe', methods=['GET', 'POST'])
def enter_recipe():
    """ Get  - display add recipe.
        Post - Add recipe and edit recipe pages info routed here for processing and uploading to db
    """

    if request.method == "POST":
        data = request.get_json()
        grains = data['grains']
        extracts = data['extracts']
        hops = data['hops']
        miscs = data['miscs']
        yeasts = data['yeasts']

        # Recipe values
        name = data["name"]
        user_id = session["user_id"]
        source = data["source"]
        public = data["share"]
        style_name = data["style"]
        notes = data["notes"]
        batch_size = data["batch_size"]
        batch_units = data["units"]
        new_recipe = Recipe(name=name, source=source, user_id=user_id, public=public,
                            notes=notes, style_name=style_name, batch_size=batch_size,
                            batch_units=batch_units)

        db.session.add(new_recipe)
        db.session.commit()

        # Recipe_id will be used in all instructions uploads
        recipe_id = Recipe.query.filter_by(name=name, user_id=user_id)[0].recipe_id

        # Grain Instructions
        for i in range(0, len(data['grains']), 3):
            grain_name = grains[i]["value"]
            ferm_id = Fermentable.query.filter_by(name=grain_name)[0].id
            grain_amount = grains[i+1]["value"]
            grain_units = grains[i+2]["value"]
            new_fermins = FermIns(recipe_id=recipe_id, ferm_id=ferm_id, amount=grain_amount,
                                  units=grain_units, )
            db.session.add(new_fermins)
            db.session.commit()

        # Extract Instructions
        if extracts:
            for i in range(0, len(data['extracts']), 3):
                extract_name = extracts[i]["value"]
                extract_id = Extract.query.filter_by(name=extract_name)[0].id
                extract_amount = extracts[i+1]["value"]
                extract_units = extracts[i+2]["value"]

                new_extins = ExtIns(recipe_id=recipe_id, extract_id=extract_id, amount=extract_amount,
                                    units=extract_units)
                db.session.add(new_extins)
                db.session.commit()

        # Hops Instructions
        for i in range(0, len(data['hops']), 6):
            hop_name = hops[i]["value"]
            hop_id = Hop.query.filter_by(name=hop_name)[0].hop_id
            hop_amount = hops[i+1]["value"]
            hop_units = hops[i+2]["value"]
            hop_phase = hops[i+3]["value"]
            time = hops[i+4]["value"]
            kind = hops[i+5]["value"]

            new_hopsins = HopIns(recipe_id=recipe_id, hop_id=hop_id, amount=hop_amount, phase=hop_phase,
                                 time=time, kind=kind)
            db.session.add(new_hopsins)
            db.session.commit()

        # Misc Instructions
        if miscs != []:
            for i in range(0, len(data['miscs']), 5):
                misc_name = miscs[i]["value"]
                misc_id = Misc.query.filter_by(name=misc_name)[0].misc_id
                misc_amount = miscs[i+1]["value"]
                misc_units = miscs[i+2]["value"]
                misc_phase = miscs[i+3]["value"]
                misc_time = miscs[i+4]["value"]
                new_miscins = MiscIns(recipe_id=recipe_id, misc_id=misc_id,
                                      phase=misc_phase, amount=misc_amount,
                                      time=misc_time, units=misc_units)
                db.session.add(new_miscins)
                db.session.commit()

        # Yeast Instructions
        for i in range(0, len(data['yeasts']), 4):
            yeast_name = yeasts[i]["value"]
            yeast_id = Yeast.query.filter_by(name=yeast_name)[0].yeast_id
            yeast_amount = yeasts[i+1]["value"]
            yeast_units = yeasts[i+2]["value"]
            yeast_phase = yeasts[i+3]["value"]
            new_yeastins = YeastIns(recipe_id=recipe_id, yeast_id=yeast_id,
                                    amount=yeast_amount, units=yeast_units,
                                    phase=yeast_phase)
            db.session.add(new_yeastins)
            db.session.commit

        message = "Your recipe has been added."
        return message

    grain_choice, extract_choice, hop_choice, misc_choice, yeast_choice, selectlist_styles = feed_recipe_form()

    return render_template("recipeform.html", selectlist_styles=selectlist_styles,
                           grain_choice=grain_choice, hop_choice=hop_choice,
                           extract_choice=extract_choice, misc_choice=misc_choice,
                           yeast_choice=yeast_choice)


@app.route('/editrecipe/<string:recipe>')
def editrecipe(recipe):
    """ Display edit recipe page """
    recipe, color = get_recipe_instructions(recipe)

    grain_choice, extract_choice, hop_choice, misc_choice, yeast_choice, selectlist_styles = feed_recipe_form()

    public = Recipe.query.filter_by(name=recipe.name).one().public
    return render_template("edit_recipe.html", recipe=recipe, public=public, grain_choice=grain_choice, 
                           extract_choice=extract_choice, hop_choice=hop_choice, misc_choice=misc_choice,
                           yeast_choice=yeast_choice, selectlist_styles=selectlist_styles)


@app.route('/uploadrecipe', methods=['GET', 'POST'])
def upload_file():
    """ Display page for recipe upload and process uploaded recipe. Return either error page or recipe page. """

    if request.method == 'POST':
        share = request.form.get('share')
        files = []
        recipes = []
        files.append(request.files['file-input1'])
        files.append(request.files['file-input2'])
        files.append(request.files['file-input3'])
        files.append(request.files['file-input4'])
        files.append(request.files['file-input5'])

        def save_files(newfile):
            if newfile and allowed_file(newfile.filename):
                filename = secure_filename(newfile.filename)
                newfile.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
                filepath = "tmp/uploads/" + filename

                status, name = load_recipes(filepath, session["user_id"], share)
                if status == "error":
                    error_names.append(name)
                else:
                    recipes.append(name)

        error_names = []
        session['error_list'] = []

        for newfile in files:
            print newfile.filename
            save_files(newfile)

        if error_names != []:
            session['error_list'] = error_names
            return render_template("uploaderror.html")
        else:
            notice = "Recipe(s) successfully added!"
            return render_template("uploadrecipe.html", notice=notice,
                                   recipes=recipes)

    return render_template("uploadrecipe.html")


@app.route('/check_recipe_name', methods=["POST"])
def check_name():
    """ Ajax call - Check for duplicate recipe names """

    test_name = request.form.get("name")
    if Recipe.query.filter_by(name=test_name).all():
        return "nope"
    return "okay"


@app.route('/colorcalc', methods=['GET', 'POST'])
def calculate_color():
    """ Ajax call -Calculate color for recipe """

    data = request.get_json()
    batch_size, batch_units = normalize_batch_size(data["batch_size"], data["units"])

    srm = Fermentable.get_srm_from_ingredient_list(data["grains"], batch_size, batch_units) \
          + Extract.get_srm_from_ingredient_list(data["extracts"], batch_size, batch_units)

    color = color_conversion(srm)
    return color


# ***************************************************************************************
# Register and Log In

@app.route('/register', methods=['GET', 'POST'])
def register():
    """ Register a new user. """

    if request.method == "POST":
        email = request.form.get("email")
        password = request.form.get("password")
        username = request.form.get("username")
        first_name = request.form.get("fname")
        last_name = request.form.get("lname")

        # Check to see if email address is taken
        if (User.query.filter_by(email=email).all()) != []:
            flash("Sorry, that email address already has an account")
            return redirect('/register')
        elif (User.query.filter_by(username=username).all()) != []:
            flash("Sorry, that username is already in use")
            return redirect('/register')
        else:
            # Creating instance of User class with associated info
            current_person = User(first_name=first_name, last_name=last_name, email=email,
                                  username=username, password=password)

            # adding the user's info to the DB
            db.session.add(current_person)
            db.session.commit()

            flash("You were successfully registered!")
            return redirect("/login")

    return render_template("registration.html")


@app.route('/login', methods=['GET', 'POST'])
def login():
    """ Check and login new user """

    if request.method == "POST":
        username = request.form.get("username")
        password = request.form.get("password")

        pw_in_db = User.query.filter_by(username=username).all()

        if pw_in_db == []:
            flash("This username is not registered - please create an account.")
            return redirect("/register")
        else:
            if pw_in_db[0].password == password:
                session["username"] = username
                session["password"] = password
                user_id = User.query.filter_by(username=username).one().user_id
                session["user_id"] = user_id

                flash("You were successfully logged in!")
                return redirect("/")
            else:
                flash("User name and password do not match.")
                return redirect("/login")

    return render_template("login.html")


@app.route('/logout', methods=["GET"])
def logout():
    """ Log Out User """
    session["username"] = None
    session["password"] = None
    session["user_id"] = None

    flash("Logged you out!")
    return redirect("/")

# ***************************************************************************************

if __name__ == "__main__":

    app.debug = True

    connect_to_db(app)

    app.run()