Compare commits
2
Commits
c14b43f9ab
...
26dda12fff
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
26dda12fff | ||
|
|
2dc2b62d7b |
+48
-37
@@ -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
|
||||
|
||||
+3
-3
@@ -62,7 +62,7 @@
|
||||
A2.25 2.25 0 006 21h7.5
|
||||
a2.25 2.25 0 002.25-2.25V15m-3-3h10.5m0 0l-3-3m3 3l-3 3" />
|
||||
</svg>
|
||||
Logout
|
||||
<span class="hidden sm:inline">Logout</span>
|
||||
</a>
|
||||
{% else %}
|
||||
<!-- Show Login and Sign Up links -->
|
||||
@@ -77,7 +77,7 @@
|
||||
a2.25 2.25 0 002.25-2.25V15m-3-3h10.5
|
||||
m0 0l-3-3m3 3l-3 3" />
|
||||
</svg>
|
||||
Login
|
||||
<span class="hidden sm:inline">Login</span>
|
||||
</a>
|
||||
<a href="{{ url_for('auth.signup') }}"
|
||||
class="text-slate-400 hover:text-slate-500 flex items-center gap-1">
|
||||
@@ -89,7 +89,7 @@
|
||||
-3.315-2.685-6-6-6H6c
|
||||
-3.315 0-6 2.685-6 6m17.25-11.25h3m0 0v3m0-3l-3 0" />
|
||||
</svg>
|
||||
Sign Up
|
||||
<span class="hidden sm:inline">Sign Up</span>
|
||||
</a>
|
||||
{% endif %}
|
||||
<a href="https://github.com/GabePope/WorkoutTracker"
|
||||
|
||||
Reference in New Issue
Block a user