PrivacyRetrieval/static/software/1/2023-03-12 09:10:01.089359/TestProject/18-views.py

"""
Flask Module Docs:  http://flask.pocoo.org/docs/api/#flask.Module

This file is used for both the routing and logic of your
application.
"""

from google.appengine.api import mail

from flask import Blueprint, url_for, render_template, request, redirect
from models import Todo
from forms import TodoForm, EmailForm

views = Blueprint('views', __name__)


@views.route('/')
def index():
    """Render website's index page."""
    return render_template('index.html')


@views.route('/todo/')
def todo_list():
    """Simple todo page."""
    form = TodoForm()
    todos = Todo.all().order('-created_at')
    return render_template('todo.html', form=form, todos=todos)


@views.route('/todo/add', methods=["POST"])
def add_todo():
    """Add a todo."""
    form = TodoForm()
    if request.method == 'POST' and form.validate_on_submit():
        todo = Todo(text=form.todo.data)
        todo.save()
    return redirect(url_for('todo_list'))


@views.route('/email/')
def email():
    """Render a form for sending email."""
    form = EmailForm()
    return render_template('email.html', form=form)


@views.route('/email/someone/', methods=['POST'])
def email_someone():
    """
    This function actually emails the message.

    Make sure you change the from_address variable if you want to use this
    functionality -- otherwise it won't work.
    """
    form = EmailForm()
    if request.method == 'POST' and form.validate_on_submit():
        from_address = form.name.data + '@<YOURAPPID>.appspotmail.com'
        to_address = form.recipient.data
        subject = "%s <%s>" % (form.name.data, form.email.data)
        message = ("From: %s\n\n"
                   "Email: %s\n\n"
                   "Message: %s") % (form.name.data, form.email.data,
                                     form.message.data)
        mail.send_mail(sender=from_address, to=to_address,
                       subject=subject, body=message)
        status = 'success'
    else:
        status = 'failed'
    return redirect(url_for('email_status', status=status))


@views.route('/email/<status>/')
def email_status(status):
    """Render a success or failed status for the email."""
    return render_template('email_status.html', status=status)


@views.route('/qunit/')
def qunit():
    """Render a QUnit page for JavaScript tests."""
    return render_template('test_js.html')


@views.after_request
def add_header(response):
    """Add header to force latest IE rendering engine and Chrome Frame."""
    response.headers['X-UA-Compatible'] = 'IE=Edge,chrome=1'
    return response


@views.app_errorhandler(404)
def page_not_found(error):
    """Custom 404 page."""
    return render_template('404.html'), 404