def save(self, user, event):
try:
rsvp = RSVP.objects.get(event=event, user=user)
if rsvp.attending != self.cleaned_data['attend']:
rsvp.attending = self.cleaned_data['attend']
rsvp.save()
except RSVP.DoesNotExist:
rsvp = RSVP(event=event,
user=user,
attending=self.cleaned_data['attend'])
rsvp.save()
return rsvp
def new_rsvp(event_id):
if session.get('user'):
rsvp_form = RSVPForm()
# validate_on_submit only validates using POST
if rsvp_form.validate_on_submit():
new_rsvp = RSVP(event_id, session['user_id'])
db.session.add(new_rsvp)
db.session.commit()
return redirect(url_for('get_event', event_id=event_id))
else:
return redirect(url_for('login'))
def rsvp_for_meal(MealId):
try:
rsvp = RSVP(MealId=MealId,
Email=current_user.Email,
Timestamp=datetime.datetime.now())
# Check if user already is RSVP'd
if session.query(RSVP).filter(RSVP.MealId == rsvp.MealId, RSVP.Email
== rsvp.Email).first() is None:
session.add(rsvp)
session.commit()
flash("Success! You RSVP'd.")
else:
flash("You were already RSVP'd for that meal.")
return redirect(url_for("get_meal", mealId=MealId))
except Exception as e:
print(
"RSVP for Meal with ID of {} was unsuccessful. Please try again. {}"
.format(MealId, e))
return redirect(url_for("get_meal", mealId=MealId))
