From edf55d7e2fa6cb66e8fd56f4c755e1c0a62ad755 Mon Sep 17 00:00:00 2001
From: zsloan
Date: Fri, 13 Jun 2025 16:37:16 +0000
Subject: Include n_cases in batch update queries

---
 gn3/db/sample_data.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

(limited to 'gn3/db/sample_data.py')

diff --git a/gn3/db/sample_data.py b/gn3/db/sample_data.py
index af979fe..f4ce64a 100644
--- a/gn3/db/sample_data.py
+++ b/gn3/db/sample_data.py
@@ -660,6 +660,14 @@ def batch_update_sample_data(
                         f"WHERE DataId = %s AND StrainId = %s"
                     ), (diffs['error']['Current'], data_id, strain_id)
                 )
+            if 'n_cases' in diffs:
+                cursor.execute(
+                    (
+                        f"UPDATE NStrain "
+                        f"SET count = %s "
+                        f"WHERE DataId = %s AND StrainId = %s"
+                    ), (diffs['n_cases']['Current'], data_id, strain_id)
+                )
 
         conn.commit()
 
@@ -679,6 +687,13 @@ def batch_update_sample_data(
                         f"VALUES (%s, %s, %s)"
                     ), (data_id, strain_id, diffs['error'])
                 )
+            if 'n_cases' in diffs:
+                cursor.execute(
+                    (
+                        f"INSERT INTO NStrain (DataId, StrainId, count)"
+                        f"VALUES (%s, %s, %s)"
+                    ), (data_id, strain_id, diffs['n_cases'])
+                )
 
         conn.commit()
 
@@ -698,6 +713,13 @@ def batch_update_sample_data(
                         f"WHERE DataId = %s AND StrainId = %s"
                     ), (data_id, strain_id)
                 )
+            if 'n_cases' in diffs:
+                cursor.execute(
+                    (
+                        f"DELETE FROM NStrain "
+                        f"WHERE DataId = %s AND StrainId = %s"
+                    ), (data_id, strain_id)
+                )
 
         conn.commit()
 
-- 
cgit 1.4.1