def create_rsvp():
form = RSVPForm()
response= make_response(redirect(url_for('wedding.index')))
# modify header to redirect to location on page of rsvp
response.headers['location'] = response.headers['location'] +'#rsvp'
if form.validate_on_submit():
my_rsvp = RSVP()
form.populate_obj(my_rsvp)
my_rsvp.updated = datetime.now()
db.session.add(my_rsvp)
db.session.commit()
expire_date=datetime.strptime('01-1-20','%m-%d-%y')
response.set_cookie('User_id',value=str(my_rsvp.id), expires=expire_date)
return response
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'))
def rsvp_list():
rsvp = RSVP.all()
attending =RSVP.number_attending()
return render_template('rsvp_list.html', rsvp=rsvp, attending=attending)
