Example #1
0
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)