Add body weight logging with diff tracking

5 files changed, 122 insertions, 1 deletions

diff --git a/db.py b/db.py
index a5f3250..a7be976 100644
--- a/db.py
+++ b/db.py
@@ -59,6 +59,16 @@ CREATE INDEX IF NOT EXISTS idx_workout_sessions_date
 
 CREATE INDEX IF NOT EXISTS idx_workout_template_exercises_template
     ON workout_template_exercises(template_id);
+
+CREATE TABLE IF NOT EXISTS weight_logs (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    date_time TEXT NOT NULL,
+    weight REAL NOT NULL,
+    note TEXT
+);
+
+CREATE INDEX IF NOT EXISTS idx_weight_logs_date
+    ON weight_logs(date_time);
 """
 
 
diff --git a/egometrics.py b/egometrics.py
index f45fa6d..5a8a556 100644
--- a/egometrics.py
+++ b/egometrics.py
@@ -2,9 +2,11 @@ import sqlite3
 
 from db import get_connection, init_db
 from screens import (
+    log_weight,
     log_workout,
     manage_workout_exercises,
     manage_workout_templates,
+    view_weight_logs,
     view_workout_sessions,
 )
 
@@ -19,7 +21,9 @@ def main_menu(conn: sqlite3.Connection) -> None:
         print("2. View Workout Sessions")
         print("3. Manage Workout Exercises")
         print("4. Manage Workout Templates")
-        print("5. Quit")
+        print("5. Log Weight")
+        print("6. View Weight Logs")
+        print("7. Quit")
         choice = input("\n> ").strip()
         if choice == "1":
             log_workout(conn)
@@ -30,6 +34,10 @@ def main_menu(conn: sqlite3.Connection) -> None:
         elif choice == "4":
             manage_workout_templates(conn)
         elif choice == "5":
+            log_weight(conn)
+        elif choice == "6":
+            view_weight_logs(conn)
+        elif choice == "7":
             break
 
 
diff --git a/models.py b/models.py
index a844271..6def19e 100644
--- a/models.py
+++ b/models.py
@@ -252,3 +252,33 @@ def save_workout_template_exercises(
             ),
         )
     conn.commit()
+
+
+# --- Weight Logs ---
+
+
+def add_weight_log(
+    conn: sqlite3.Connection,
+    date_time: str,
+    weight: float,
+    note: str | None = None,
+) -> int:
+    cur = conn.execute(
+        "INSERT INTO weight_logs (date_time, weight, note) VALUES (?, ?, ?)",
+        (date_time, weight, note),
+    )
+    conn.commit()
+    assert cur.lastrowid is not None
+    return cur.lastrowid
+
+
+def list_weight_logs(conn: sqlite3.Connection, limit: int = 20) -> list[sqlite3.Row]:
+    return conn.execute(
+        "SELECT id, date_time, weight, note FROM weight_logs ORDER BY date_time DESC LIMIT ?",
+        (limit,),
+    ).fetchall()
+
+
+def delete_weight_log(conn: sqlite3.Connection, log_id: int) -> None:
+    conn.execute("DELETE FROM weight_logs WHERE id = ?", (log_id,))
+    conn.commit()
diff --git a/screens/__init__.py b/screens/__init__.py
index 5bd64e6..72858c8 100644
--- a/screens/__init__.py
+++ b/screens/__init__.py
@@ -1,3 +1,4 @@
 from screens.exercises import manage_workout_exercises
 from screens.sessions import log_workout, view_workout_sessions
 from screens.templates import manage_workout_templates
+from screens.weight import log_weight, view_weight_logs
diff --git a/screens/weight.py b/screens/weight.py
new file mode 100644
index 0000000..5669f07
--- /dev/null
+++ b/screens/weight.py
@@ -0,0 +1,72 @@
+import sqlite3
+
+import models
+import ui
+
+
+def log_weight(conn: sqlite3.Connection) -> None:
+    ui.clear_screen()
+    ui.print_header("Log Weight")
+    date_time = ui.prompt_datetime("Date/Time")
+    weight = ui.prompt_float("Weight (kg): ", min_val=0.1)
+    assert weight is not None
+    note = ui.prompt_str("Note (optional): ", required=False)
+    models.add_weight_log(conn, date_time, weight, note)
+    print(f"Logged {weight:.1f}kg.")
+    ui.pause()
+
+
+def _fmt_diff(current: float, previous: float) -> str:
+    diff = round(current - previous, 1)
+    if diff > 0:
+        return f"+{diff:.1f}"
+    elif diff < 0:
+        return f"{diff:.1f}"
+    return "0.0"
+
+
+def view_weight_logs(conn: sqlite3.Connection) -> None:
+    while True:
+        ui.clear_screen()
+        logs = models.list_weight_logs(conn)
+        if not logs:
+            print("\nNo weight logs recorded yet.")
+            ui.pause()
+            return
+        ui.print_header("Weight Logs")
+
+        # Logs are newest-first; diff compares to the next row (previous in time)
+        rows = []
+        for i, log in enumerate(logs):
+            weight = log["weight"]
+            if i < len(logs) - 1:
+                diff = _fmt_diff(weight, logs[i + 1]["weight"])
+            else:
+                diff = ""
+            rows.append(
+                [
+                    str(i + 1),
+                    log["date_time"],
+                    f"{weight:.1f}",
+                    diff,
+                    log["note"] or "",
+                ]
+            )
+
+        ui.print_table(["#", "Date", "Weight (kg)", "Diff", "Note"], rows)
+
+        choice = input("\nSelect # to delete ('b' = back): ").strip()
+        if choice.lower() == "b":
+            break
+        try:
+            idx = int(choice) - 1
+            if idx < 0 or idx >= len(logs):
+                print("Invalid selection.")
+                continue
+        except ValueError:
+            print("Invalid input.")
+            continue
+        if ui.confirm(f"Delete log from {logs[idx]['date_time']}?"):
+            models.delete_weight_log(conn, logs[idx]["id"])
+            print("Log deleted.")
+            ui.pause()