From 78f22407ee73c89eb7477e7e025c37c671037ab0 Mon Sep 17 00:00:00 2001
From: BonfaceKilz
Date: Fri, 11 Mar 2022 11:35:34 +0300
Subject: Append the strain name when extracting "actions"

* gn3/db/sample_data.py (__extract_actions): During updates, make sure that
the strain name is part of the returned string when extracting "actions".
* tests/unit/db/test_sample_data.py: Add test cases for the above.
---
 gn3/db/sample_data.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

(limited to 'gn3/db')

diff --git a/gn3/db/sample_data.py b/gn3/db/sample_data.py
index e5c5e48..c697de0 100644
--- a/gn3/db/sample_data.py
+++ b/gn3/db/sample_data.py
@@ -22,7 +22,10 @@ def __extract_actions(original_data: str,
         "insert": {"data": [], "csv_header": []},
         "update": {"data": [], "csv_header": []},
     }
+    strain_name = ""
     for _o, _u, _h in zip(original_data, updated_data, csv_header):
+        if _h == "Strain Name":
+            strain_name = _o
         if _o == _u:  # No change
             continue
         elif _o and _u == "x":  # Deletion
@@ -38,8 +41,10 @@ def __extract_actions(original_data: str,
         if not val["data"]:
             result[key] = None
         else:
-            result[key]["data"] = ",".join(result[key]["data"])
-            result[key]["csv_header"] = ",".join(result[key]["csv_header"])
+            result[key]["data"] = (f"{strain_name}," +
+                                   ",".join(result[key]["data"]))
+            result[key]["csv_header"] = ("Strain Name," +
+                                         ",".join(result[key]["csv_header"]))
     return result
 
 
-- 
cgit v1.2.3