def create_party():
if not g.user:
flash('You must log in for that', category='danger')
return redirect('/login')
form = PartyForm()
if form.validate_on_submit():
p = Party.create(name=form.name.data,
address=form.address.data,
city=form.city.data,
state=form.state.data,
zip_code=form.zip_code.data,
leader_id=g.user.id)
if type(p) == str:
flash(p, category='info')
else:
flash('Successfully added party', category='success')
resturaunts = Resturaunt.get_resturaunts(party_id=p.id)
if type(resturaunts) == str:
flash(resturaunts, category='warning')
return redirect('/')
return render_template('/parties/create.html', form=form)
from models import User, db, Party, Vote, Resturaunt
import random
from app import app
db.drop_all()
db.create_all()
me = User.create(email='*****@*****.**', name='david', password='******')
kynsi = User.create(email='*****@*****.**', name='kynsi', password='******')
chris = User.create(email='*****@*****.**', name='chris', password='******')
tori = User.create(email='*****@*****.**', name='tori', password='******')
valentine = Party.create(address="12248 Hunter's Knoll Dr",
city="Burleson",
state="Texas",
zip_code=76028,
leader_id=me.id,
name="Valentine's day dinner")
Resturaunt.get_resturaunts(party_id=valentine.id)
valentine.add_member(kynsi.id)
valentine.add_member(chris.id)
valentine.add_member(tori.id)
for member in valentine.members:
for resturaunt in valentine.resturaunts:
vote = True if random.randint(0, 1) else False
new_vote = Vote.vote(member=member,
