Prefix non-public screen functions with _ and split sessions.py into helpers

1 files changed, 18 insertions, 13 deletions

diff --git a/screens/templates.py b/screens/templates.py
index 7c8ff60..fe122ce 100644
--- a/screens/templates.py
+++ b/screens/templates.py
@@ -15,18 +15,18 @@ def manage_workout_templates(conn: sqlite3.Connection) -> None:
         print("5. Back")
         choice = input("\n> ").strip()
         if choice == "1":
-            list_workout_templates(conn)
+            _list_workout_templates(conn)
         elif choice == "2":
-            create_workout_template(conn)
+            _create_workout_template(conn)
         elif choice == "3":
-            edit_workout_template(conn)
+            _edit_workout_template(conn)
         elif choice == "4":
-            delete_workout_template(conn)
+            _delete_workout_template(conn)
         elif choice == "5":
             break
 
 
-def list_workout_templates(conn: sqlite3.Connection, pause: bool = True) -> None:
+def _list_workout_templates(conn: sqlite3.Connection, pause: bool = True) -> None:
     templates = models.list_workout_templates(conn)
     ui.print_header("All Workout Templates")
     ui.print_table(
@@ -37,7 +37,7 @@ def list_workout_templates(conn: sqlite3.Connection, pause: bool = True) -> None
         ui.pause()
 
 
-def create_workout_template(conn: sqlite3.Connection) -> None:
+def _create_workout_template(conn: sqlite3.Connection) -> None:
     name = ui.prompt_str("Template name: ")
     assert name is not None
     try:
@@ -50,25 +50,25 @@ def create_workout_template(conn: sqlite3.Connection) -> None:
     # Immediately enter exercise editor
     print("Now add exercises to the template.")
     ui.pause()
-    edit_template_exercises(conn, template_id)
+    _edit_template_exercises(conn, template_id)
 
 
-def edit_workout_template(conn: sqlite3.Connection) -> None:
+def _edit_workout_template(conn: sqlite3.Connection) -> None:
     templates = models.list_workout_templates(conn)
     if not templates:
         print("\nNo templates defined.")
         return
-    list_workout_templates(conn, pause=False)
+    _list_workout_templates(conn, pause=False)
     tid = ui.prompt_int("\nTemplate ID to edit: ")
     assert tid is not None
     template = models.get_workout_template(conn, tid)
     if not template:
         print("Template not found.")
         return
-    edit_template_exercises(conn, tid)
+    _edit_template_exercises(conn, tid)
 
 
-def edit_template_exercises(conn: sqlite3.Connection, template_id: int) -> None:
+def _edit_template_exercises(conn: sqlite3.Connection, template_id: int) -> None:
     """Interactive editor for a template's exercise list."""
     while True:
         ui.clear_screen()
@@ -118,6 +118,7 @@ def edit_template_exercises(conn: sqlite3.Connection, template_id: int) -> None:
         if action == "b":
             break
 
+        # Rename the template
         elif action == "n":
             new_name = ui.prompt_str("New template name: ")
             assert new_name is not None
@@ -128,6 +129,7 @@ def edit_template_exercises(conn: sqlite3.Connection, template_id: int) -> None:
                 print(f'Template "{new_name}" already exists.')
             ui.pause()
 
+        # Add an exercise: pick from available, prompt for defaults, append and save
         elif action == "a":
             available = models.list_workout_exercises(conn)
             if not available:
@@ -162,6 +164,7 @@ def edit_template_exercises(conn: sqlite3.Connection, template_id: int) -> None:
             print(f'"{ex["name"]}" added to template.')
             ui.pause()
 
+        # Remove an exercise by position
         elif action == "r":
             if not exercises_data:
                 print("No exercises to remove.")
@@ -176,6 +179,7 @@ def edit_template_exercises(conn: sqlite3.Connection, template_id: int) -> None:
             print(f'Removed "{removed["exercise_name"]}".')
             ui.pause()
 
+        # Reorder: pop from one position and insert at another
         elif action == "m":
             if len(exercises_data) < 2:
                 print("Need at least 2 exercises to move.")
@@ -193,6 +197,7 @@ def edit_template_exercises(conn: sqlite3.Connection, template_id: int) -> None:
             print("Exercise moved.")
             ui.pause()
 
+        # Edit default values for an existing exercise (Enter keeps current)
         elif action == "e":
             if not exercises_data:
                 print("No exercises to edit.")
@@ -231,12 +236,12 @@ def edit_template_exercises(conn: sqlite3.Connection, template_id: int) -> None:
             ui.pause()
 
 
-def delete_workout_template(conn: sqlite3.Connection) -> None:
+def _delete_workout_template(conn: sqlite3.Connection) -> None:
     templates = models.list_workout_templates(conn)
     if not templates:
         print("\nNo templates defined.")
         return
-    list_workout_templates(conn, pause=False)
+    _list_workout_templates(conn, pause=False)
     tid = ui.prompt_int("\nTemplate ID to delete: ")
     assert tid is not None
     template = models.get_workout_template(conn, tid)