Add body weight logging with diff tracking

1 files changed, 9 insertions, 1 deletions

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