From ce986db27ba1a31e6c76a9cfcfaaaaa5dd050df3 Mon Sep 17 00:00:00 2001
From: William Valentin <william.valentin.info@gmail.com>
Date: Tue, 29 Jul 2025 14:00:33 -0700
Subject: [PATCH] feat: Update DataManager to support new quetiapine medication
 format and adjust VSCode task command

---
 .gitignore          |  2 +-
 .vscode/tasks.json  |  8 +++++++-
 src/data_manager.py | 31 ++++++++++++++++++++++++++++---
 3 files changed, 36 insertions(+), 5 deletions(-)

diff --git a/.gitignore b/.gitignore
index 181d0a5..5884542 100644
--- a/.gitignore
+++ b/.gitignore
@@ -47,7 +47,7 @@ htmlcov/
 .pylint.d/
 
 # IDEs and editors
-.vscode/
+#.vscode/
 !.vscode/tasks.json
 !.vscode/launch.json
 .idea/
diff --git a/.vscode/tasks.json b/.vscode/tasks.json
index 590cecd..a121f5d 100644
--- a/.vscode/tasks.json
+++ b/.vscode/tasks.json
@@ -4,7 +4,13 @@
 		{
 			"label": "Run TheChart App",
 			"type": "shell",
-			"command": "cd /home/will/Code/thechart && python -m src.main",
+			"command": "/home/will/Code/thechart/.venv/bin/python",
+			"args": [
+				"src/main.py"
+			],
+			"options": {
+				"cwd": "/home/will/Code/thechart"
+			},
 			"group": "build",
 			"isBackground": false,
 			"problemMatcher": []
diff --git a/src/data_manager.py b/src/data_manager.py
index 6e970e8..ad96969 100644
--- a/src/data_manager.py
+++ b/src/data_manager.py
@@ -108,9 +108,32 @@ class DataManager:
                 return False
 
             # Find the row to update using original_date as a unique identifier
-            # Handle both old format (10 columns) and new format (14 columns)
-            if len(values) == 14:
-                # New format with dose columns
+            # Handle both old format (10 columns) and new format (16 columns)
+            if len(values) == 16:
+                # New format with all dose columns including quetiapine
+                df.loc[
+                    df["date"] == original_date,
+                    [
+                        "date",
+                        "depression",
+                        "anxiety",
+                        "sleep",
+                        "appetite",
+                        "bupropion",
+                        "bupropion_doses",
+                        "hydroxyzine",
+                        "hydroxyzine_doses",
+                        "gabapentin",
+                        "gabapentin_doses",
+                        "propranolol",
+                        "propranolol_doses",
+                        "quetiapine",
+                        "quetiapine_doses",
+                        "note",
+                    ],
+                ] = values
+            elif len(values) == 14:
+                # Format without quetiapine
                 df.loc[
                     df["date"] == original_date,
                     [
@@ -192,6 +215,8 @@ class DataManager:
                     "gabapentin_doses": "",
                     "propranolol": 0,
                     "propranolol_doses": "",
+                    "quetiapine": 0,
+                    "quetiapine_doses": "",
                     "note": "",
                 }
                 df = pd.concat([df, pd.DataFrame([new_entry])], ignore_index=True)