from flask import Flask, render_template, request, url_for
from werkzeug.utils import redirect


app = Flask('france_rsvp')


@app.route('/france/', methods=['GET', 'POST'])
def form():
    error = None
    if request.method == 'POST':
        if (
            'code' in request.form
            and len(request.form['code']) == 4
        ):
            return redirect(url_for('rsvp', code=request.form['code']))
        else:
            error = "Code invalide"
    return render_template('form.html', error=error)


@app.route('/france/thanks')
def thanks():
    return render_template('thanks.html')


@app.route('/france/<code>', methods=['GET', 'POST'])
def rsvp(code):
    if request.method == 'POST':
        # TODO: Store response
        return redirect(url_for('thanks'))

    # TODO: Lookup guest from code
    adults = ['Remi', 'Vicky']
    kids = ['LB']

    return render_template(
        'rsvp.html',
        adults=adults,
        kids=kids,
    )