feat: Enhance dose tracking functionality in edit window and add punch button support

test_dose_editing_functionality.py

@@ -0,0 +1,147 @@
+#!/usr/bin/env python3
+"""
+Test script to verify dose editing functionality in the edit window.
+"""
+
+import logging
+import os
+import shutil
+import sys
+
+# Add the src directory to the path so we can import our modules
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), "src"))
+
+from data_manager import DataManager
+
+
+def test_dose_editing_functionality():
+    """Test the dose editing functionality with the edit window."""
+    print("Testing dose editing functionality in edit window...")
+
+    # Create a backup of the current CSV
+    try:
+        shutil.copy("thechart_data.csv", "thechart_data_backup.csv")
+        print("✓ Created backup of current CSV")
+    except Exception as e:
+        print(f"✗ Failed to create backup: {e}")
+        return False
+
+    try:
+        # Create a logger for the DataManager
+        logger = logging.getLogger("test_logger")
+        logger.setLevel(logging.DEBUG)
+
+        # Initialize data manager
+        data_manager = DataManager("thechart_data.csv", logger)
+
+        # Load current data
+        df = data_manager.load_data()
+        print(f"✓ Loaded {len(df)} entries from CSV")
+
+        if df.empty:
+            print("✗ No data to test dose editing functionality")
+            return False
+
+        # Test 1: Check that we can retrieve full row data including doses
+        print("\n=== Testing Full Row Data Retrieval ===")
+        first_entry_date = df.iloc[0]["date"]
+        first_entry = df[df["date"] == first_entry_date].iloc[0]
+
+        print(f"Testing with date: {first_entry_date}")
+
+        # Check that all expected columns are present
+        expected_columns = [
+            "date",
+            "depression",
+            "anxiety",
+            "sleep",
+            "appetite",
+            "bupropion",
+            "bupropion_doses",
+            "hydroxyzine",
+            "hydroxyzine_doses",
+            "gabapentin",
+            "gabapentin_doses",
+            "propranolol",
+            "propranolol_doses",
+            "note",
+        ]
+
+        missing_columns = [col for col in expected_columns if col not in df.columns]
+        if missing_columns:
+            print(f"✗ Missing columns: {missing_columns}")
+            return False
+        else:
+            print("✓ All expected columns present in CSV")
+
+        # Test 2: Check dose data access
+        print("\n=== Testing Dose Data Access ===")
+        dose_columns = [
+            "bupropion_doses",
+            "hydroxyzine_doses",
+            "gabapentin_doses",
+            "propranolol_doses",
+        ]
+
+        for col in dose_columns:
+            dose_data = first_entry[col]
+            print(f"{col}: '{dose_data}'")
+
+        print("✓ Dose data accessible from CSV")
+
+        # Test 3: Test parsing dose text (simulate edit window input)
+        print("\n=== Testing Dose Text Parsing ===")
+
+        # Simulate some dose text that a user might enter
+        test_dose_text = "09:00: 150mg\n18:30: 150mg"
+        test_date = "07/28/2025"
+
+        # Test the parsing logic (we'll need to import this)
+        try:
+            import tkinter as tk
+
+            from ui_manager import UIManager
+
+            # Create a temporary UI manager to test the parsing
+            root = tk.Tk()
+            root.withdraw()  # Hide the window
+            ui_manager = UIManager(root, logger)
+
+            parsed_doses = ui_manager._parse_dose_text(test_dose_text, test_date)
+            print(f"Original text: '{test_dose_text}'")
+            print(f"Parsed doses: '{parsed_doses}'")
+
+            if "|" in parsed_doses and "2025-07-28" in parsed_doses:
+                print("✓ Dose text parsing working correctly")
+            else:
+                print("✗ Dose text parsing failed")
+                root.destroy()
+                return False
+
+            root.destroy()
+
+        except Exception as e:
+            print(f"✗ Error testing dose parsing: {e}")
+            return False
+
+        print("\n✓ All dose editing functionality tests passed!")
+        return True
+
+    except Exception as e:
+        print(f"✗ Error during test: {e}")
+        import traceback
+
+        traceback.print_exc()
+        return False
+
+    finally:
+        # Restore the backup
+        try:
+            shutil.move("thechart_data_backup.csv", "thechart_data.csv")
+            print("✓ Restored original CSV from backup")
+        except Exception as e:
+            print(f"✗ Failed to restore backup: {e}")
+
+
+if __name__ == "__main__":
+    test_dose_editing_functionality()