計算機科学のブログ

from flask import Flask, render_template, request, session

import data_utils

app = Flask(__name__)
app.secret_key = 'You will never guess...'


@app.get('/')
def index():
    return render_template(
        'index.html.j2',
        title='Welcome to Swimclub',
    )


@app.get('/swims')
def display_swim_sessions():
    data = data_utils.get_swim_sessions()
    dates = [session[0].split()[0] for session in data]
    return render_template(
        'select.html.j2',
        title='Select a swim session',
        select_id='chosen_date',
        url='/swimmers',
        data=dates,
    )


@app.post('/swimmers')
def display_swimmers():
    session['chosen_date'] = request.form['chosen_date']
    swimmers = [
        f'{name}-{age}'
        for name, age in data_utils.get_session_swimmers(
            session['chosen_date']
        )
    ]
    return render_template(
        'select.html.j2',
        title='Select a swimmer',
        url='/showevents',
        select_id='swimmer',
        data=swimmers,
    )


@app.post('/showevents')
def display_swimmer_events():
    session['swimmer'], session['age'] = request.form['swimmer'].split('-')
    data = data_utils.get_swimmers_events(
        session['swimmer'], session['age'], session['chosen_date']
    )
    events = [f'{distance}-{stroke}' for distance, stroke in data]
    return render_template(
        'select.html.j2',
        title='Select an event',
        url='/showbarchart',
        select_id='event',
        data=events,
    )


@app.post('/showbarchart')
def show_bar_chart():
    file_id = request.form['file']
    # location = swimclub.produce_bar_chart(file_id, 'templates/')
    # return render_template(location.split('/')[-1])
    return ''


if __name__ == '__main__':
    app.run(debug=True)