def registration(): if request.method == 'POST': if request.form['submit'] == "Sign Me Up!": phone = db_phone(request.form['phone']) if len(phone) != 9: flash( "Please enter a valid phone number in the format provided") return redirect(url_for('registration')) user_phone = db_session.query( User.phone).filter_by(phone=phone).scalar() if user_phone: flash( "You are already registered, please use the RSVP form or RSVP via Text" ) return redirect(url_for('registration')) else: newUser = User(name=request.form['name'], phone=phone, email=request.form['email']) if request.form.get('tomorrow'): newUser.tomorrow = 1 newUser.attendance = 1 db_session.add(newUser) db_session.commit() return redirect(url_for('registration')) elif request.form['submit'] == "I'll be there!": phone = db_phone(request.form['rsvp_phone']) current_user = db_session.query(User).filter_by(phone=phone).one() current_user.tomorrow = 1 current_user.attendance = current_user.attendance + 1 db_session.add(current_user) db_session.commit() return redirect(url_for('registration')) return redirect(url_for('registration')) else: count = db_session.query(User).filter_by(tomorrow=1).count() return render_template('homepage.html', registration=count)