Validate message for 0 tacos

from flask import (Flask, g, render_template, flash, redirect, url_for, abort)
from flask.ext.bcrypt import check_password_hash
from flask.ext.login import (LoginManager, login_user, logout_user,
                             login_required, current_user)

import forms
import models

DEBUG = True
PORT = 8000
HOST = '0.0.0.0'

app = Flask(__name__)
app.secret_key = 'adfaldv mz,æxc x,cxvzæxfldvmadæmk!!!!12+1249)"¤)"¤?="¤?='

login_manager = LoginManager()
login_manager.init_app(app)
login_manager.login_view = 'login'

@login_manager.user_loader
def load_user(userid):
    try:
        return models.User.get(models.User.id == userid)
    except models.DoesNotExist:
        return None

@app.before_request
def before_request():
    """Connect to the database before each request."""
    g.db = models.DATABASE
    g.db.connect()
    g.user = current_user


@app.after_request
def after_request(response):
    """Close the database connection after each request."""
    g.db.close()
    return response


@app.route('/register', methods=('GET', 'POST'))
def register():
    form = forms.RegisterForm()
    if form.validate_on_submit():
        flash("Yay, you registered!", "success")
        models.User.create_user(
            email=form.email.data,
            password=form.password.data,
        )
        return redirect(url_for('index'))
    return render_template('register.html', form=form)


@app.route('/login', methods=('GET', 'POST'))
def login():
    form = forms.LoginForm()
    if form.validate_on_submit():
        try:
            user = models.User.get(models.User.email == form.email.data)
        except models.DoesNotExist:
            flash("Your email or password doesn't match!", "error")
        else:
            if check_password_hash(user.password, form.password.data):
                login_user(user)
                flash("You've been logged in!", "success")
                return redirect(url_for('index'))
            else:
                flash("Your email or password doesn't match!", "error")
    return render_template('login.html', form=form)

@app.route('/logout')
@login_required
def logout():
    logout_user()
    flash("You've been logged out! Come back soon!", "success")
    return redirect(url_for('index'))


@app.route('/taco', methods=("GET", "POST"))
@login_required
def taco():
    form = forms.TacoForm()
    if form.validate_on_submit():
        flash("A taco has been born.", "success")
        models.Taco.create(
            protein=form.protein.data,
            cheese=form.cheese.data,
            shell=form.shell.data,
            extras=form.extras.data,
            user=g.user.id
        )
        return redirect(url_for('index'))
    return redirect(url_for('index'))
    taco_protein = type1, type2, type3, None
    cheese = True, False
    shell = wheat, corn, mysterious


@app.route('/')
def index():
    tacos = models.Taco.select().limit(10)
    return render_template('index.html', tacos=tacos)

if __name__== '__main__':
    models.initialize()
    try:
        with models.DATABASE.transaction():
            models.User.create_user(
              email = 'billy@example.com',
              password = 'password'
            )
    except ValueError:
        pass
    app.run(debug=DEBUG, host=HOST, port=PORT)

import datetime

from flask.ext.bcrypt import generate_password_hash
from flask.ext.login import UserMixin
from peewee import *

DATABASE = SqliteDatabase('tacocat.db', threadlocals=True)

class User(UserMixin, Model):
    email = CharField(unique=True)
    password = CharField(max_length=100)
    joined_at = DateTimeField(default=datetime.datetime.now)
    is_admin = BooleanField(default=False)

    class Meta:
        database = DATABASE
        order_by = ('-joined_at',)

    def get_tacos(self):
        return Taco.select().where(Taco.user == self)

    def get_stream(self):
        return Taco.select().where(
            (Taco.user == self)
        )

    @classmethod
    def create_user(cls, email, password):
        try:
            with DATABASE.transaction():
                cls.create(
                    email = email,
                    password = generate_password_hash(password))
        except IntegrityError:
            raise ValueError("User already exists.")



class Taco(Model):
    timestamp = DateTimeField(default=datetime.datetime.now)
    user = ForeignKeyField(
        rel_model=User,
        related_name='tacos'
    )
    protein = CharField()
    shell = CharField()
    cheese = BooleanField()
    extras = CharField()

    class Meta:
        database = DATABASE
        ordering = ('-timestamp',)

def initialize():
    DATABASE.connect()
    DATABASE.create_tables([User, Taco], safe=True)
    DATABASE.close()

from flask_wtf import Form
from wtforms import StringField, PasswordField, BooleanField, TextAreaField
from wtforms.validators import (DataRequired, Regexp, ValidationError, Email, Length, EqualTo
                    )

from models import User

def email_exists(form, field):
    if User.select().where(User.email == field.data).exists():
        raise ValidationError('User with that email already exists')

class RegisterForm(Form):
    email = StringField(
        'Email',
        validators = [
            DataRequired(),
            Email(),
            email_exists,
        ])
    password = PasswordField(
        'Password',
        validators = [
            DataRequired(),
            Length(min=2),
            EqualTo('password2', message='Passwords must match')
        ])
    password2 = PasswordField(
        'Confirm Password',
        validators = [DataRequired()]
    )

class LoginForm(Form):
    email = StringField('Email', validators = [DataRequired(), Email()])
    password = PasswordField('Password', validators = [DataRequired()])

class TacoForm(Form):
    protein = StringField('Protein')
    shell = StringField('Shell', validators = [DataRequired()])
    cheese = BooleanField('Cheese')
    extras = TextAreaField('Any extras?')