146 lines
6.3 KiB
Python
146 lines
6.3 KiB
Python
import customtkinter as ctk
|
|
from database import Database
|
|
from datetime import date
|
|
from tkinter import messagebox
|
|
from utils import parse_float
|
|
|
|
class WorkoutFrame(ctk.CTkFrame):
|
|
def __init__(self, master, db: Database, date_str: str):
|
|
super().__init__(master)
|
|
self.db = db
|
|
self.date_str = date_str
|
|
self.session_id = None
|
|
|
|
self.setup_ui()
|
|
self.refresh_workouts()
|
|
|
|
def setup_ui(self):
|
|
self.grid_columnconfigure(0, weight=1)
|
|
self.grid_rowconfigure(2, weight=1)
|
|
|
|
# Header
|
|
self.header_frame = ctk.CTkFrame(self, fg_color="transparent")
|
|
self.header_frame.grid(row=0, column=0, sticky="ew", pady=(0, 20))
|
|
|
|
self.header_label = ctk.CTkLabel(self.header_frame, text=f"Workouts: {self.date_str}", font=ctk.CTkFont(size=20, weight="bold"))
|
|
self.header_label.pack(side="left")
|
|
|
|
# Log New Set Section
|
|
self.input_frame = ctk.CTkFrame(self)
|
|
self.input_frame.grid(row=1, column=0, sticky="ew", pady=10, padx=5)
|
|
|
|
# Exercise Selection
|
|
ctk.CTkLabel(self.input_frame, text="Exercise:").grid(row=0, column=0, padx=10, pady=10)
|
|
self.exercises = self.db.get_all_exercises()
|
|
self.exercise_names = [e['name'] for e in self.exercises]
|
|
self.exercise_var = ctk.StringVar(value=self.exercise_names[0] if self.exercise_names else "")
|
|
self.exercise_menu = ctk.CTkOptionMenu(self.input_frame, values=self.exercise_names, variable=self.exercise_var, command=self.update_input_fields)
|
|
self.exercise_menu.grid(row=0, column=1, padx=10, pady=10)
|
|
|
|
# Variation
|
|
ctk.CTkLabel(self.input_frame, text="Variation:").grid(row=0, column=2, padx=10, pady=10)
|
|
self.variation_entry = ctk.CTkEntry(self.input_frame, placeholder_text="e.g. Knee, Incline")
|
|
self.variation_entry.grid(row=0, column=3, padx=10, pady=10)
|
|
|
|
# Reps/Time
|
|
self.reps_label = ctk.CTkLabel(self.input_frame, text="Reps:")
|
|
self.reps_label.grid(row=1, column=0, padx=10, pady=10)
|
|
self.reps_entry = ctk.CTkEntry(self.input_frame, placeholder_text="0")
|
|
self.reps_entry.grid(row=1, column=1, padx=10, pady=10)
|
|
|
|
# RPE
|
|
ctk.CTkLabel(self.input_frame, text="RPE (1-10):").grid(row=1, column=2, padx=10, pady=10)
|
|
self.rpe_slider = ctk.CTkSlider(self.input_frame, from_=1, to=10, number_of_steps=9)
|
|
self.rpe_slider.grid(row=1, column=3, padx=10, pady=10)
|
|
|
|
self.add_btn = ctk.CTkButton(self.input_frame, text="Add Set", command=self.add_set)
|
|
self.add_btn.grid(row=1, column=4, padx=20, pady=10)
|
|
|
|
# History of today's sets
|
|
self.history_label = ctk.CTkLabel(self, text="Today's Sets", font=ctk.CTkFont(size=16, weight="bold"))
|
|
self.history_label.grid(row=2, column=0, sticky="w", pady=(20, 5))
|
|
|
|
self.scrollable_frame = ctk.CTkScrollableFrame(self)
|
|
self.scrollable_frame.grid(row=3, column=0, sticky="nsew", pady=5)
|
|
self.scrollable_frame.grid_columnconfigure(0, weight=1)
|
|
|
|
def update_input_fields(self, choice):
|
|
exercise = next((e for e in self.exercises if e['name'] == choice), None)
|
|
if exercise and exercise['type'] == 'time':
|
|
self.reps_label.configure(text="Seconds:")
|
|
else:
|
|
self.reps_label.configure(text="Reps:")
|
|
|
|
def refresh_workouts(self):
|
|
# Clear scrollable frame
|
|
for widget in self.scrollable_frame.winfo_children():
|
|
widget.destroy()
|
|
|
|
workouts = self.db.get_workouts_by_date(self.date_str)
|
|
if not workouts:
|
|
ctk.CTkLabel(self.scrollable_frame, text="No sets logged for today.").pack(pady=20)
|
|
return
|
|
|
|
for session in workouts:
|
|
for s in session.get('sets', []):
|
|
set_str = f"{s['exercise_name']}"
|
|
if s['variation']: set_str += f" ({s['variation']})"
|
|
|
|
val_type = "reps" if s['reps'] > 0 else "sec"
|
|
val = s['reps'] if s['reps'] > 0 else s['duration_seconds']
|
|
|
|
set_detail = f"{val} {val_type} | RPE: {s['rpe']}"
|
|
|
|
row = ctk.CTkFrame(self.scrollable_frame)
|
|
row.pack(fill="x", pady=2, padx=5)
|
|
|
|
# Delete Button
|
|
del_btn = ctk.CTkButton(row, text="X", width=30, fg_color="#c0392b", hover_color="#e74c3c",
|
|
command=lambda sid=s['id']: self.delete_set(sid))
|
|
del_btn.pack(side="right", padx=5, pady=5)
|
|
|
|
ctk.CTkLabel(row, text=set_detail).pack(side="right", padx=10)
|
|
ctk.CTkLabel(row, text=set_str, font=ctk.CTkFont(weight="bold")).pack(side="left", padx=10)
|
|
|
|
def delete_set(self, set_id):
|
|
self.db.delete_set(set_id)
|
|
self.refresh_workouts()
|
|
|
|
def add_set(self):
|
|
if not self.session_id:
|
|
# Create session for today if it doesn't exist
|
|
# Note: For simplicity, we create one session per day in this view
|
|
existing = self.db.get_workouts_by_date(self.date_str)
|
|
if existing:
|
|
self.session_id = existing[0]['id']
|
|
else:
|
|
self.session_id = self.db.create_workout_session(self.date_str)
|
|
|
|
exercise = next((e for e in self.exercises if e['name'] == self.exercise_var.get()), None)
|
|
if not exercise: return
|
|
|
|
val_raw = self.reps_entry.get()
|
|
val_parsed = parse_float(val_raw)
|
|
|
|
if val_parsed is None and val_raw:
|
|
messagebox.showerror("Error", "Please enter a valid number.")
|
|
return
|
|
|
|
try:
|
|
val = int(val_parsed) if val_parsed is not None else 0
|
|
reps = val if exercise['type'] == 'reps' else 0
|
|
duration = val if exercise['type'] == 'time' else 0
|
|
|
|
self.db.add_set(
|
|
session_id=self.session_id,
|
|
exercise_id=exercise['id'],
|
|
reps=reps,
|
|
duration=duration,
|
|
rpe=int(self.rpe_slider.get()),
|
|
variation=self.variation_entry.get()
|
|
)
|
|
self.reps_entry.delete(0, 'end')
|
|
self.refresh_workouts()
|
|
except (ValueError, TypeError):
|
|
messagebox.showerror("Error", "Please enter a valid number for Reps/Seconds.")
|