main.py

# Coffee & Wi-Fi v1.1

# Note: The course did not have an actual coding project for this day, so I've decided to experiment
# with different color & font combinations for the website from Day 62 instead, along with some other
# tweaks to the page templates.

from flask import Flask, render_template, redirect
from flask_wtf import FlaskForm
from wtforms import StringField, SelectField, SubmitField
from wtforms.validators import DataRequired, URL
from flask_bootstrap import Bootstrap

from cafe_data import CafeManager

app = Flask(__name__)
app.secret_key = "ThisIsASecretKey"
Bootstrap(app)


class CafeForm(FlaskForm):
    name = StringField("Cafe Name", validators=[DataRequired()])
    # make sure this is a proper URL
    location = StringField("Cafe Location on Google Maps (URL)", validators=[DataRequired(), URL()])
    time_open = StringField("Opening Time e.g. 8:00 AM", validators=[DataRequired()])
    time_close = StringField("Closing Time e.g. 5:30 PM", validators=[DataRequired()])
    coffee = SelectField("Coffee Rating", choices=["☕️", "☕☕", "☕☕☕", "☕☕☕☕", "☕☕☕☕☕"],
                         validators=[DataRequired()])
    wifi = SelectField("Wifi Strength Rating", choices=["✘", "💪", "💪💪", "💪💪💪", "💪💪💪💪", "💪💪💪💪💪"],
                       validators=[DataRequired()])
    power = SelectField("Power Socket Availability", choices=["✘", "🔌", "🔌🔌", "🔌🔌🔌", "🔌🔌🔌🔌", "🔌🔌🔌🔌🔌"],
                        validators=[DataRequired()])
    submit = SubmitField('Submit')


cafe_list = CafeManager()


@app.route("/")
def home():
    return render_template("index.html")


# "secret" page to add a new entry
@app.route("/add", methods=["GET", "POST"])
def add_cafe():
    form = CafeForm()
    if form.validate_on_submit():
        # using list comprehension to get just elements that are needed
        new_cafe = [form.data[item] for item in form.data][:7]
        cafe_list.add_cafe(new_cafe)
        # redirect to the cafes page
        return redirect("/cafes")
    return render_template("add.html", add_cafe_form=form)


@app.route("/cafes")
def cafes():
    # get the header and the list of cafes separately, to allow different formatting
    header, entries = cafe_list.get_cafes()
    return render_template("cafes.html", header=header, cafes=entries)


if __name__ == "__main__":
    app.run(debug=True)