From 91d6f1baa0b9d6931ead5d846686843beb9c7d68 Mon Sep 17 00:00:00 2001 From: chrisr3d Date: Thu, 7 Nov 2019 11:50:16 +0100 Subject: [PATCH] fix: Fixed csv file parsing --- misp_modules/modules/import_mod/csvimport.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/misp_modules/modules/import_mod/csvimport.py b/misp_modules/modules/import_mod/csvimport.py index d5e2d59..96e42b1 100644 --- a/misp_modules/modules/import_mod/csvimport.py +++ b/misp_modules/modules/import_mod/csvimport.py @@ -206,11 +206,11 @@ def __any_mandatory_misp_field(header): def __special_parsing(data, delimiter): - return list(line.split(delimiter) for line in csv.reader(io.TextIOWrapper(io.BytesIO(data.encode()), encoding='utf-8')) if line and not line.startswith('#')) + return list(tuple(l.strip() for l in line[0].split(delimiter)) for line in csv.reader(io.TextIOWrapper(io.BytesIO(data.encode()), encoding='utf-8')) if line and not line[0].startswith('#')) def __standard_parsing(data): - return list(line for line in csv.reader(io.TextIOWrapper(io.BytesIO(data.encode()), encoding='utf-8')) if line and not line[0].startswith('#')) + return list(tuple(l.strip() for l in line) for line in csv.reader(io.TextIOWrapper(io.BytesIO(data.encode()), encoding='utf-8')) if line and not line[0].startswith('#')) def handler(q=False):