Fix "sets completed" on homescreen

1 files changed, 11 insertions, 2 deletions

diff --git a/app/routes/basic.py b/app/routes/basic.py
index 684daf2..0ef66f9 100644
--- a/app/routes/basic.py
+++ b/app/routes/basic.py
@@ -1,7 +1,7 @@
 from app import app, db
 from flask import render_template
 from flask_login import current_user, login_required
-from tables import Workout, WorkoutRecord
+from tables import Workout, WorkoutRecord, SetRecord
 
 
 @app.route("/", methods=["GET"])
@@ -19,4 +19,13 @@ def home():
         .order_by(WorkoutRecord.finished.desc())
     )
 
-    return render_template("home.html", records=records)
+    sets_completed = (
+        db.session.query(SetRecord, WorkoutRecord)
+        .filter(
+            WorkoutRecord.id == SetRecord.workout_record_id,
+            WorkoutRecord.user_id == current_user.id,
+        )
+        .count()
+    )
+
+    return render_template("home.html", records=records, sets_completed=sets_completed)