diff --git a/datasources/connectors/csv.py b/datasources/connectors/csv.py index 4de3b4d67365309211a00397cafff205d4687648..f218b07fbb2b9ceda64e991e6b05c4cd8f1a795d 100644 --- a/datasources/connectors/csv.py +++ b/datasources/connectors/csv.py @@ -3,7 +3,6 @@ Connectors for handling CSV data. """ import csv -import json import typing from django.http import JsonResponse @@ -168,9 +167,12 @@ class CsvToMongoConnector(InternalDataConnector, DataSetConnector): # Couldn't store field 'id' in document - recover it for item in data: - if self.id_field_alias in item: + try: item['id'] = item.pop(self.id_field_alias) + except KeyError: + pass + return JsonResponse({ 'status': 'success', 'data': data,