100 lines
3.6 KiB
Python
100 lines
3.6 KiB
Python
import customtkinter as ctk
|
|
from database import Database
|
|
import matplotlib.pyplot as plt
|
|
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
|
|
import datetime
|
|
|
|
class StatsFrame(ctk.CTkFrame):
|
|
def __init__(self, master, db: Database):
|
|
super().__init__(master)
|
|
self.db = db
|
|
self.setup_ui()
|
|
|
|
def setup_ui(self):
|
|
self.grid_columnconfigure(0, weight=1)
|
|
|
|
self.header = ctk.CTkLabel(self, text="Progress & Motivation", font=ctk.CTkFont(size=20, weight="bold"))
|
|
self.header.grid(row=0, column=0, pady=10, sticky="w")
|
|
|
|
# Stats Summary (Streaks, etc.)
|
|
self.summary_frame = ctk.CTkFrame(self)
|
|
self.summary_frame.grid(row=1, column=0, sticky="ew", pady=10)
|
|
|
|
logs = self.db.get_log_history(30)
|
|
streak = self.calculate_streak(logs)
|
|
|
|
self.streak_label = ctk.CTkLabel(self.summary_frame, text=f"🔥 Current Streak: {streak} Days", font=ctk.CTkFont(size=16))
|
|
self.streak_label.pack(side="left", padx=20, pady=10)
|
|
|
|
# Chart Section
|
|
self.chart_frame = ctk.CTkFrame(self)
|
|
self.chart_frame.grid(row=2, column=0, sticky="nsew", pady=10)
|
|
self.chart_frame.grid_columnconfigure(0, weight=1)
|
|
|
|
self.plot_weight_waist(logs)
|
|
|
|
def calculate_streak(self, logs):
|
|
if not logs: return 0
|
|
streak = 0
|
|
today = datetime.date.today()
|
|
|
|
# Simple streak calculation based on existing logs
|
|
# Checks if logs are consecutive
|
|
log_dates = [datetime.datetime.strptime(l['date'], "%Y-%m-%d").date() for l in logs]
|
|
log_dates.sort(reverse=True)
|
|
|
|
current = today
|
|
if log_dates and log_dates[0] < today - datetime.timedelta(days=1):
|
|
return 0 # Missed yesterday and today
|
|
|
|
for d in log_dates:
|
|
if d == current or d == current - datetime.timedelta(days=1):
|
|
streak += 1
|
|
current = d
|
|
else:
|
|
break
|
|
return streak
|
|
|
|
def plot_weight_waist(self, logs):
|
|
if not logs or len(logs) < 2:
|
|
ctk.CTkLabel(self.chart_frame, text="Log data for at least 2 days to see charts.").pack(pady=50)
|
|
return
|
|
|
|
# Prepare data
|
|
logs.reverse() # Chronological
|
|
dates = [l['date'][5:] for l in logs if l['weight'] or l['waist']] # MM-DD
|
|
weights = [l['weight'] for l in logs if l['weight'] or l['waist']]
|
|
waists = [l['waist'] for l in logs if l['weight'] or l['waist']]
|
|
|
|
if not weights: return
|
|
|
|
# Create plot
|
|
fig, ax1 = plt.subplots(figsize=(6, 4), dpi=100)
|
|
fig.patch.set_facecolor('#2b2b2b') # Matches dark mode
|
|
ax1.set_facecolor('#2b2b2b')
|
|
|
|
color = 'tab:blue'
|
|
ax1.set_xlabel('Date')
|
|
ax1.set_ylabel('Weight (kg)', color=color)
|
|
ax1.plot(dates, weights, color=color, marker='o', label='Weight')
|
|
ax1.tick_params(axis='y', labelcolor=color)
|
|
ax1.tick_params(axis='x', rotation=45, colors='white')
|
|
ax1.spines['bottom'].set_color('white')
|
|
ax1.spines['top'].set_color('none')
|
|
ax1.spines['left'].set_color('white')
|
|
ax1.spines['right'].set_color('none')
|
|
|
|
# Second axis for Waist
|
|
ax2 = ax1.twinx()
|
|
color = 'tab:red'
|
|
ax2.set_ylabel('Waist (cm)', color=color)
|
|
ax2.plot(dates, waists, color=color, marker='x', linestyle='--', label='Waist')
|
|
ax2.tick_params(axis='y', labelcolor=color)
|
|
ax2.spines['right'].set_color('white')
|
|
|
|
fig.tight_layout()
|
|
|
|
canvas = FigureCanvasTkAgg(fig, master=self.chart_frame)
|
|
canvas.draw()
|
|
canvas.get_tk_widget().pack(fill="both", expand=True)
|