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,