From 7641343e90f9a7f03cf0baf30818675efac6d7f3 Mon Sep 17 00:00:00 2001
From: BonfaceKilz
Date: Thu, 10 Mar 2022 13:27:06 +0300
Subject: Create action dict that's created when updating data

* gn3/db/sample_data.py (__extract_actions): An update on a vector of data can
contain: inserts, deletes and updates. This functions extracts these actions
during an update.
* tests/unit/db/test_sample_data.py (test_extract_actions): Add test-case for
the above.
---
 gn3/db/sample_data.py | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

(limited to 'gn3')

diff --git a/gn3/db/sample_data.py b/gn3/db/sample_data.py
index ccb03f2..06c3cc5 100644
--- a/gn3/db/sample_data.py
+++ b/gn3/db/sample_data.py
@@ -11,6 +11,38 @@ _MAP = {
 }
 
 
+def __extract_actions(original_data: str,
+                      updated_data: str,
+                      csv_header: str) -> dict:
+    original_data = original_data.strip().split(",")
+    updated_data = updated_data.strip().split(",")
+    csv_header = csv_header.strip().split(",")
+    result = {
+        "delete": {"data": [], "csv_header": []},
+        "insert": {"data": [], "csv_header": []},
+        "update": {"data": [], "csv_header": []},
+    }
+    for _o, _u, _h in zip(original_data, updated_data, csv_header):
+        if _o == _u:  # No change
+            continue
+        elif _o and _u == "x":  # Deletion
+            result["delete"]["data"].append(_o)
+            result["delete"]["csv_header"].append(_h)
+        elif _o == "x" and _u:  # Insert
+            result["insert"]["data"].append(_u)
+            result["insert"]["csv_header"].append(_h)
+        elif _o and _u:  # Update
+            result["update"]["data"].append(_u)
+            result["update"]["csv_header"].append(_h)
+    for key, val in result.items():
+        if not val["data"]:
+            result[key] = None
+        else:
+            result[key]["data"] = ",".join(result[key]["data"])
+            result[key]["csv_header"] = ",".join(result[key]["csv_header"])
+    return result
+
+
 def get_trait_csv_sample_data(conn: Any,
                               trait_name: int, phenotype_id: int) -> str:
     """Fetch a trait and return it as a csv string"""
-- 
cgit v1.2.3