will/thechart
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Update DataManager to support new quetiapine medication format and adjust VSCode task command
Some checks failed
Build and Push Docker Image / build-and-push (push) Has been cancelled
Details

Browse Source
This commit is contained in:
William Valentin
2025-07-29 14:00:33 -07:00
parent 188fb542be
commit ce986db27b
3 changed files with 36 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
src/data_manager.py
View File

@@ -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)