social network with flask

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
HOST = '0.0.0.0'
PORT = 8000

app=Flask(__name__)
app.secret_key='youdontknowmysecretkey'

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('/')
def index():
    try:
        tacos=models.Taco.select()
    except models.DoesNotExist:    
        return render_template('índex.html')
    else:
        return render_template('index.html', tacos=tacos)

@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("Email is incompatible", "error")

        if check_password_hash(user.password, form.password.data):
            login_user(user)
            flash("You are logged in", "success")
            return redirect(url_for('index'))
        else:
            flash("Password does not 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'))
@app.route('/taco', methods=['POST'])
@login_required
def taco():
    form = forms.CreateTacoForm()
    if form.validate_on_submit():
        flash("You Successfully Created Your Taco", "success")
        models.Taco.create_taco(user=g.user._get_current_object(),
                           protein=form.protein.data,
                           shell=form.shell.data,
                           cheese=form.cheese.data,
                           extras=form.extras.data.strip()
                          )
        return redirect(url_for('index'))
    return render_template('taco.html', form=form)


#@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

import datetime

from flask.ext.bcrypt import Bcrypt, generate_password_hash, check_password_hash
from flask.ext.login import UserMixin
from peewee import *

DATABASE = SqliteDatabase('taco.db')

class User(UserMixin, Model):
    email=CharField(unique=True)
    password=CharField(max_length=100)

    class Meta:
         database=DATABASE

    @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):
    protein=CharField()
    shell=CharField()
    cheese=BooleanField(default=True)
    extras=TextField()
    user=ForeignKeyField(
        rel_model=User,
        related_name='tacos'
    )

    class Meta:
        database = DATABASE

    @classmethod
    def create_taco(cls, protein, shell, cheese, extras, user):
        with DATABASE.transaction():#this is something I had in my notes
            cls.create(
                protein=protein,
                shell=shell,
                cheese=cheese,
                extras=extras,
                user=user)

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

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

from models import User

def email_exists(form, field):
    if User.select().where(User.email==field.data).exists():
        raise ValidatationError('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', validators = [DataRequired()])
    shell =StringField('Shell', validators = [DataRequired()])
    cheese = BooleanField('Cheese')
    extras = TextField("Any extras?")