cleaned up csv parsing

This commit is contained in:
Alyssa Bigley 2021-06-03 15:08:47 -04:00
parent 2bc68707b5
commit 6ff5a1db42

View File

@ -666,7 +666,7 @@ class BulkImportView(GetReturnURLMixin, ObjectPermissionRequiredMixin, View):
from_form=self.model_form, from_form=self.model_form,
widget=Textarea(attrs=self.widget_attrs) widget=Textarea(attrs=self.widget_attrs)
) )
Upload_CSV = FileField( upload_csv = FileField(
required=False required=False
) )
return ImportForm(*args, **kwargs) return ImportForm(*args, **kwargs)
@ -692,7 +692,7 @@ class BulkImportView(GetReturnURLMixin, ObjectPermissionRequiredMixin, View):
def post(self, request): def post(self, request):
logger = logging.getLogger('netbox.views.BulkImportView') logger = logging.getLogger('netbox.views.BulkImportView')
new_objs = [] new_objs = []
form = self._import_form(request.POST) form = self._import_form(request.POST, request.FILES)
if form.is_valid(): if form.is_valid():
logger.debug("Form validation was successful") logger.debug("Form validation was successful")
@ -700,8 +700,8 @@ class BulkImportView(GetReturnURLMixin, ObjectPermissionRequiredMixin, View):
try: try:
# Iterate through CSV data and bind each row to a new model form instance. # Iterate through CSV data and bind each row to a new model form instance.
with transaction.atomic(): with transaction.atomic():
if len(request.FILES) != 0: if request.FILES:
csv_file = request.FILES["Upload_CSV"] csv_file = request.FILES["upload_csv"]
csv_file.seek(0) csv_file.seek(0)
csv_str = csv_file.read().decode('utf-8') csv_str = csv_file.read().decode('utf-8')
reader = csv.reader(csv_str.splitlines()) reader = csv.reader(csv_str.splitlines())