fixed spaces in header bug

animesh-chouhan · Jun 30, 2021 · cb6e424 · cb6e424
1 parent 36d9c65
commit cb6e424
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/vcf_creator/vcf.py b/vcf_creator/vcf.py
@@ -65,17 +65,18 @@ def vcard_generator(filename):
             # Read the csv file
             reader = csv.reader(csvfile, dialect)
             # Get header
-            header = reader.__next__()
+            header = [h.strip() for h in reader.__next__()]
             # Name is required
             if "Name" not in header:
                 print("Header not supported.")
                 return -1
 
-            # Iterate over the rows
+            # Set attributes_present
             for i, attr in enumerate(header):
                 attributes_present[attr] = i
 
             ret = []
+            # Iterate over the rows
             for row in reader:
                 ret.append(vcard_formatter(row))
             return "\n".join(ret)