Merge pull request #634 from pierotofy/importfix

Fixed task import from file upload
pull/636/head
Piero Toffanin 2019-03-04 11:57:43 -05:00 zatwierdzone przez GitHub
commit 5d269516a3
Nie znaleziono w bazie danych klucza dla tego podpisu
ID klucza GPG: 4AEE18F83AFDEB23
1 zmienionych plików z 9 dodań i 3 usunięć

Wyświetl plik

@ -3,8 +3,9 @@ from wsgiref.util import FileWrapper
import mimetypes
import datetime
from shutil import copyfileobj
from django.core.exceptions import ObjectDoesNotExist, SuspiciousFileOperation, ValidationError
from django.core.files.uploadedfile import InMemoryUploadedFile
from django.db import transaction
from django.http import FileResponse
from django.http import HttpResponse
@ -357,9 +358,14 @@ class TaskAssetsImport(APIView):
if len(files) > 0:
destination_file = task.assets_path("all.zip")
with open(destination_file, 'wb+') as fd:
for chunk in files[0].chunks():
fd.write(chunk)
if isinstance(files[0], InMemoryUploadedFile):
for chunk in files[0].chunks():
fd.write(chunk)
else:
with open(files[0].temporary_file_path(), 'rb') as file:
copyfileobj(file, fd)
worker_tasks.process_task.delay(task.id)