From 2dc93f6bcfa14dec4eac7c6b1a3f0d83e43b48ac Mon Sep 17 00:00:00 2001
From: stilbruch <stilbruch@protonmail.com>
Date: Mon, 9 May 2022 11:46:00 -0500
Subject: Fix "sets completed" on homescreen

---
 app/routes/basic.py     | 13 +++++++++++--
 app/templates/home.html |  2 +-
 2 files changed, 12 insertions(+), 3 deletions(-)

(limited to 'app')

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)
diff --git a/app/templates/home.html b/app/templates/home.html
index 07d6269..7bf951f 100644
--- a/app/templates/home.html
+++ b/app/templates/home.html
@@ -31,7 +31,7 @@
 			</div>
 			<div class="tile is-parent">
 				<article class="tile is-child box">
-					<p class="title has-text-black">0</p>
+					<p class="title has-text-black">{{ sets_completed }}</p>
 					<p class="subtitle has-text-black">Sets Completed</p>
 				</article>
 			</div>
-- 
cgit v1.2.3