From 490a9c2306d9b05e132b1fcef6cd65a985a14b71 Mon Sep 17 00:00:00 2001
From: BonfaceKilz
Date: Fri, 18 Mar 2022 15:39:19 +0300
Subject: Create new function for cleaning individual fields in csv text

* gn3/csvcmp.py (clean_csv_text): New function.
* tests/unit/test_csvcmp.py: Import "csv_text".
(test_clean_csv_text): Test case for the above.
---
 gn3/csvcmp.py | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'gn3')

diff --git a/gn3/csvcmp.py b/gn3/csvcmp.py
index 975814a..10c5d3e 100644
--- a/gn3/csvcmp.py
+++ b/gn3/csvcmp.py
@@ -48,6 +48,15 @@ def remove_insignificant_edits(diff_data, epsilon=0.001):
     return diff_data
 
 
+def clean_csv_text(csv_text: str) -> str:
+    """Remove extra white space elements in all elements of the CSV file"""
+    _csv_text = []
+    for line in csv_text.strip().split("\n"):
+        _csv_text.append(
+            ",".join([el.strip() for el in line.split(",")]))
+    return "\n".join(_csv_text)
+
+
 def csv_diff(base_csv, delta_csv, tmp_dir="/tmp") -> dict:
     """Diff 2 csv strings"""
     base_csv_list = base_csv.strip().split("\n")
-- 
cgit 1.4.1