Improve performance of calendar view

routes/calendar.py

@@ -32,38 +32,52 @@ def _get_date_range_and_links(date_obj, view):
         raise ValueError('Invalid view type specified.')
     return start_date, end_date, prev_date, next_date
 
-def _fetch_raw_workout_data(db_executor, person_id, start_date, end_date):
+def _fetch_workout_data(db_executor, person_id, start_date, end_date, include_details=True):
     """Fetches workout data for a person within a date range."""
-    query = """
-    SELECT
-        w.workout_id,
-        w.start_date,
-        t.topset_id,
-        t.repetitions,
-        t.weight,
-        e.name AS exercise_name,
-        p.name AS person_name
-    FROM
-        person p
-        LEFT JOIN workout w ON p.person_id = w.person_id AND w.start_date BETWEEN %s AND %s
-        LEFT JOIN topset t ON w.workout_id = t.workout_id
-        LEFT JOIN exercise e ON t.exercise_id = e.exercise_id
-    WHERE
-        p.person_id = %s
-    ORDER BY
-        w.start_date,
-        t.topset_id;
-    """
+    if include_details:
+        query = """
+        SELECT
+            w.workout_id,
+            w.start_date,
+            t.topset_id,
+            t.repetitions,
+            t.weight,
+            e.name AS exercise_name,
+            p.name AS person_name
+        FROM
+            person p
+            LEFT JOIN workout w ON p.person_id = w.person_id AND w.start_date BETWEEN %s AND %s
+            LEFT JOIN topset t ON w.workout_id = t.workout_id
+            LEFT JOIN exercise e ON t.exercise_id = e.exercise_id
+        WHERE
+            p.person_id = %s
+        ORDER BY
+            w.start_date,
+            t.topset_id;
+        """
+    else:
+        query = """
+        SELECT
+            w.workout_id,
+            w.start_date,
+            p.name AS person_name
+        FROM
+            person p
+            LEFT JOIN workout w ON p.person_id = w.person_id AND w.start_date BETWEEN %s AND %s
+        WHERE
+            p.person_id = %s
+        ORDER BY
+            w.start_date;
+        """
     # Ensure dates are passed in a format the DB understands (e.g., YYYY-MM-DD strings)
     return db_executor(query, [start_date.strftime('%Y-%m-%d'), end_date.strftime('%Y-%m-%d'), person_id])
 
 def _group_workouts_by_date(workouts_data):
     """Groups workout data by date and workout ID."""
-    # Structure: { 'YYYY-MM-DD': { workout_id: { workout_details..., 'sets': [...] } } }
+    # Structure: { date_object: { workout_id: { workout_details..., 'sets': [...] } } }
     workouts_by_date = defaultdict(lambda: defaultdict(lambda: {'sets': []}))
     person_name = 'Unknown'
     if workouts_data:
-        # Use .get() for safer access in case the key doesn't exist
         person_name = workouts_data[0].get('person_name', 'Unknown')
 
     for row in workouts_data:
@@ -72,29 +86,27 @@ def _group_workouts_by_date(workouts_data):
              continue
 
         workout_date = row['start_date']
-        workout_date_str = workout_date.strftime("%Y-%m-%d")
         workout_id = row['workout_id']
 
         # Initialize workout details if this workout_id hasn't been seen for this date
-        if workout_id not in workouts_by_date[workout_date_str]:
-             workouts_by_date[workout_date_str][workout_id].update({
+        if workout_id not in workouts_by_date[workout_date]:
+             workouts_by_date[workout_date][workout_id].update({
                  'workout_id': workout_id,
                  'start_date': workout_date,
-                 # 'sets' is already initialized by defaultdict
              })
 
         # Add set details if topset_id exists
         if row.get('topset_id'):
-            workouts_by_date[workout_date_str][workout_id]['sets'].append({
+            workouts_by_date[workout_date][workout_id]['sets'].append({
                 'repetitions': row.get('repetitions'),
                 'weight': row.get('weight'),
                 'exercise_name': row.get('exercise_name')
             })
 
-    # Convert nested defaultdict to regular dict for easier handling/JSON serialization
+    # Convert nested defaultdict to regular dict
     processed_workouts = {
-        date_str: dict(workouts)
-        for date_str, workouts in workouts_by_date.items()
+        d: dict(w)
+        for d, w in workouts_by_date.items()
     }
     return processed_workouts, person_name
 
@@ -104,8 +116,7 @@ def _process_workouts_for_month_view(grouped_workouts, start_date, end_date, sel
     today = datetime.today().date()
     current_date = start_date
     while current_date <= end_date:
-        date_str = current_date.strftime("%Y-%m-%d")
-        day_workouts_dict = grouped_workouts.get(date_str, {})
+        day_workouts_dict = grouped_workouts.get(current_date, {})
         day_workouts_list = list(day_workouts_dict.values()) # Convert workout dicts to list
 
         days_data.append({
@@ -134,8 +145,7 @@ def _process_workouts_for_year_view(grouped_workouts, year_date):
         month_days_data = []
         current_day = start_date_month
         while current_day <= end_date_month:
-            date_str = current_day.strftime('%Y-%m-%d')
-            day_workouts_dict = grouped_workouts.get(date_str, {})
+            day_workouts_dict = grouped_workouts.get(current_day, {})
             day_workouts_list = list(day_workouts_dict.values())
             has_workouts = len(day_workouts_list) > 0
             # Get first workout ID if workouts exist
@@ -184,8 +194,9 @@ def get_calendar(person_id):
         # For now, returning a simple error response
         return f"Error: Invalid view type '{selected_view}'.", 400
 
-    # Fetch and process data
-    raw_workouts = _fetch_raw_workout_data(db.execute, person_id, start_date, end_date)
+    # Fetch and process data (only fetch details if in month view)
+    include_details = (selected_view == 'month')
+    raw_workouts = _fetch_workout_data(db.execute, person_id, start_date, end_date, include_details=include_details)
     grouped_workouts, person_name = _group_workouts_by_date(raw_workouts)
 
     # Prepare base context for the template