from flask import Blueprint, request, render_template, redirect, url_for, flash
from files.models import db, Message

bp = Blueprint('about', __name__, url_prefix='/about')


@bp.route('/', methods=('GET', 'POST'))
def about():
    if request.method == 'POST':
        name = request.form['name']
        email = request.form['email']
        message = request.form['message']

        error = None

        if not name:
            error = 'Missing name'
        elif not email:
            error = 'Missing email'
        elif not message:
            error = 'Empty message'

        if error is not None:
            flash(error)
        else:
            save_message = Message(name, email, message)
            db.session.add(save_message)
            db.session.commit()
            return redirect(url_for('about.thanks'))

    return render_template('about/about.html')


@bp.route('/thanks')
def thanks():
    return render_template('about/thanks.html')