From ba006bdd5ae2053575e11539186d5c5041f070c1 Mon Sep 17 00:00:00 2001
From: BonfaceKilz
Date: Tue, 1 Mar 2022 16:49:54 +0300
Subject: Fill in empty values in csv text with: "x"

* gn3/csvcmp.py (fill_csv): Update this function to allow empty lists to be
filled with the default value(set in the args).
* tests/unit/test_csvcmp.py (test_fill_csv): Update test to capture above.
---
 gn3/csvcmp.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'gn3')

diff --git a/gn3/csvcmp.py b/gn3/csvcmp.py
index 360a101..eef64b5 100644
--- a/gn3/csvcmp.py
+++ b/gn3/csvcmp.py
@@ -92,6 +92,10 @@ def fill_csv(csv_text, width, value="x"):
         if line.startswith("Strain") or line.startswith("#"):
             data.append(line)
         elif line:
+            _n = line.split(",")
+            for i, val in enumerate(_n):
+                if not val.strip():
+                    _n[i] = value
             data.append(
-                ",".join((_n:=line.split(",")) + [value] * (width - len(_n))))
+                ",".join(_n + [value] * (width - len(_n))))
     return "\n".join(data)
-- 
cgit v1.2.3