Create seperate page view for workout (WIP)

2024-11-02 23:09:26 +11:00
parent 1384eb007b
commit 817a6016e2
4 changed files with 193 additions and 0 deletions
--- a/features/workout.py
+++ b/features/workout.py
@@ -0,0 +1,78 @@
+from flask import jsonify
+
+
+class Workout:
+    def __init__(self, db_connection_method):
+        self.execute = db_connection_method
+
+    def get(self, person_id, workout_id):
+        query = """
+        SELECT 
+            w.workout_id,
+            w.person_id,
+            w.start_date,
+            w.note,
+            t.topset_id,
+            e.exercise_id,
+            e.name AS exercise_name,
+            t.repetitions,
+            t.weight,
+            tag.tag_id,
+            tag.name AS tag_name
+        FROM 
+            workout w
+        JOIN 
+            topset t ON w.workout_id = t.workout_id
+        JOIN 
+            exercise e ON t.exercise_id = e.exercise_id
+        LEFT JOIN 
+            workout_tag wt ON w.workout_id = wt.workout_id
+        LEFT JOIN 
+            tag ON wt.tag_id = tag.tag_id
+        WHERE 
+            w.person_id = %s
+            AND w.workout_id = %s;
+        """
+        data = self.execute(query, [person_id, workout_id])
+
+        if not data:
+            return {"error": "Workout not found"}, 404
+        
+        exercises = self.execute("SELECT exercise_id, name FROM exercise;")
+
+        # Initialize workout dictionary
+        workout_data = {
+            "workout_id": data[0]["workout_id"],
+            "person_id": data[0]["person_id"],
+            "start_date": data[0]["start_date"],
+            "note": data[0]["note"],
+            "tags": [],
+            "top_sets": [],
+            "exercises": exercises
+        }
+        
+        # Initialize helpers for tracking unique tags and topsets
+        tag_set = set()
+        topsets_seen = set()
+
+        # Process each row and add tags and topsets to workout_data
+        for row in data:
+            # Add unique tags
+            tag_id = row["tag_id"]
+            if tag_id and tag_id not in tag_set:
+                workout_data["tags"].append({"tag_id": tag_id, "tag_name": row["tag_name"]})
+                tag_set.add(tag_id)
+
+            # Add unique topsets based on topset_id
+            topset_id = row["topset_id"]
+            if topset_id not in topsets_seen:
+                workout_data["top_sets"].append({
+                    "topset_id": topset_id,
+                    "exercise_id": row["exercise_id"],
+                    "exercise_name": row["exercise_name"],
+                    "repetitions": row["repetitions"],
+                    "weight": row["weight"]
+                })
+                topsets_seen.add(topset_id)
+
+        return workout_data