from flask import Flask, render_template, g, redirect, url_for
import sqlite3

app = Flask(__name__)

DATABASE = 'workout.db'


def get_db():
    db = getattr(g, '_database', None)
    if db is None:
        db = g._database = sqlite3.connect(DATABASE)
    db.row_factory = sqlite3.Row
    return db


def query_db(query, args=(), one=False):
    cur = get_db().execute(query, args)
    rv = cur.fetchall()
    cur.close()
    return (rv[0] if rv else None) if one else rv


@app.route("/")
def dashboard():
    rows = query_db('select * from TopSet')
    print(rows[0]['TopSetId'])
    return render_template('index.html', name='peter')


@app.route("/person/<int:person_id>")
def display_workouts_for_person(person_id):
    return render_template('workouts.html', person_id=person_id)


@app.route("/person/<int:person_id>/new_workout")
def new_workout_for_person(person_id):
    # hardcoded workout_id, need to create record in Workout table and return workout_id
    return redirect(url_for('show_workout_for_person', person_id=person_id, workout_id=1))


@app.route("/person/<int:person_id>/workout/<int:workout_id>")
def show_workout_for_person(person_id, workout_id):
    return render_template('workout.html', person_id=person_id, workout_id=workout_id)


@app.route("/person/<int:person_id>/workout/<int:workout_id>/topset/<int:topset_id>")
def show_topset_from_workout_for_person(person_id, workout_id, topset_id):
    topset = query_db("""
        SELECT
            P.Name,
            W.StartDate,
            E.ExcerciseId,
            E.Name,
            T.Repetitions,
            T.Weight
        FROM
            Person P
            LEFT JOIN Workout W ON P.PersonId = W.PersonId
            INNER JOIN TopSet T ON W.WorkoutId = T.WorkoutId
            INNER JOIN Excercise E ON T.ExcerciseId = E.ExcerciseId
        WHERE
            P.PersonId = ?
            AND W.WorkoutId = ?
            AND T.TopSetId = ?""",
                      [person_id, workout_id, topset_id], one=True)

    if topset is None:
        return render_template('error.html', error='404', message=f'Unable to find TopSet({topset_id}) in Workout({workout_id}) completed by Person({person_id})')

    return render_template('topset.html', topset=topset, exercises=query_db('select * from Excercise'))


@app.teardown_appcontext
def close_connection(exception):
    db = getattr(g, '_database', None)
    if db is not None:
        db.close()