Browse Source

Update data upload method from one by one create to bulk create

pull/10/head
Hironsan 6 years ago
parent
commit
95f5c79a1d
2 changed files with 3 additions and 4 deletions
  1. BIN
      app/db.sqlite3
  2. 7
      app/server/views.py

BIN
app/db.sqlite3

7
app/server/views.py

@ -54,10 +54,9 @@ class DataUpload(SuperUserMixin, LoginRequiredMixin, TemplateView):
project = get_object_or_404(Project, pk=kwargs.get('project_id'))
try:
form_data = TextIOWrapper(request.FILES['csv_file'].file, encoding='utf-8')
reader = csv.reader(form_data)
for line in reader:
text = line[0]
Document(text=text, project=project).save()
Document.objects.bulk_create([Document(
text=line.strip(),
project=project) for line in form_data])
return HttpResponseRedirect(reverse('dataset', args=[project.id]))
except:
return HttpResponseRedirect(reverse('dataset-upload', args=[project.id]))

Loading…
Cancel
Save