Handle leading spaces correctly #229

awslabs · Jun 30, 2022 · 060b387 · 060b387
1 parent b89a233
commit 060b387
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/csv-gremlin/csv-gremlin.py b/csv-gremlin/csv-gremlin.py
@@ -451,7 +451,7 @@ def process_csv_file(self,fname):
         self.property_count = 0
         try:
             with open(fname, newline='') as csvfile:
-                reader = csv.DictReader(csvfile,escapechar="\\")
+                reader = csv.DictReader(csvfile, skipinitialspace=True, escapechar="\\")
 
                 if not '~id' in reader.fieldnames:
                     self.print_error('The header row must include an ~id column')

diff --git a/csv-gremlin/test-files/header-with-spaces-edge.csv b/csv-gremlin/test-files/header-with-spaces-edge.csv
@@ -0,0 +1,3 @@
+ ~id, ~label, ~from,    ~to
+e1, likes, a1, a2
+e2, " likes ", a1, a2
diff --git a/csv-gremlin/test-files/header-with-spaces.csv b/csv-gremlin/test-files/header-with-spaces.csv
@@ -0,0 +1,4 @@
+ ~id, ~label,   type
+a1, animal, cat
+a2, " animal ", cat
+a3, animal , cat