def index():
id = request.cookies.get('User_id')
my_rsvp = RSVP.find_by_id(id)
if my_rsvp:
form = RSVPEditForm(obj=my_rsvp)
else:
form = RSVPForm()
return render_template('wedding.html',form=form, my_rsvp=my_rsvp,user=current_user)
def rsvp_edit(id):
rsvp = RSVP.find_by_id(id)
if request.method == 'GET': # Get rsvp for editing
form = RSVPEditForm(obj=rsvp)
return render_template('rsvp_edit.html', form=form)
if request.method == 'POST':
form = RSVPEditForm()
if form.validate_on_submit(): # Save rsvp
form.populate_obj(rsvp)
rsvp.updated = datetime.now()
db.session.add(rsvp)
db.session.commit()
return redirect(url_for('wedding.rsvp_list'))
