from files.db import get_db from flask import Blueprint, request, render_template, redirect, url_for, flash 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'] db = get_db() 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: db.execute( 'INSERT INTO messages' ' (name, email, message)' ' VALUES (?, ?, ?)', (name, email, message) ) db.commit() return redirect(url_for('about.thanks')) return render_template('about/about.html') @bp.route('/thanks') def thanks(): return render_template('about/thanks.html')