diff options
Diffstat (limited to '')
| -rw-r--r-- | server/views.py | 42 |
1 files changed, 32 insertions, 10 deletions
diff --git a/server/views.py b/server/views.py index c585d8b..b7462bb 100644 --- a/server/views.py +++ b/server/views.py @@ -25,6 +25,7 @@ import threading import random progress = {} +lock = threading.Lock() # custom staff check function for non primary group staff members def isStaff(user): @@ -788,30 +789,47 @@ def doclist(request): 'settings': settings, } + # check if download multiple documents if request.method == "POST": if request.POST["action"] == "download" and len(selected) > 0: + # generate job id code = request.user.username + "_" + str(random.randint(100, 999)) - progress[code] = [0, len(selected), False, request.user, None] + while code in progress.keys(): + code = request.user.username + "_" + str(random.randint(100, 999)) + + # create job status object + with lock: + progress[code] = [0, len(selected), False, request.user, None] + # run job threading.Thread(target=zip_documents, args=(selected, code)).start() + # pass job id context["task_id"] = code return render(request, 'server/doc_list.html', context) def get_progress(request): + # check if asked for a job if 'job' in request.GET: job_id = request.GET['job'] else: return HttpResponse(json.dumps({"error": True})) + # check if job is valid if not job_id in progress.keys(): return HttpResponse(json.dumps({"error": "Lavoro non esistente. Provare a richiedere di nuovo il lavoro."})) + # if user wants to download result if 'download' in request.GET: - if progress[job_id][3] == request.user: - data = progress[job_id][4] - del progress[job_id] - return data - data = progress[job_id][:3] + with lock: + # if user is authorized and job is completed + if request.GET['download'] == "true" and progress[job_id][3] == request.user: + data = progress[job_id][4] + del progress[job_id] + return data + + # otherwise return status + with lock: + data = progress[job_id][:3] return HttpResponse(json.dumps(data)) def zip_documents(docs, code): @@ -847,18 +865,22 @@ def zip_documents(docs, code): # render pdf using wkhtmltopdf pdf = pdfkit.from_string(html, False) filename = i.user.username+"_"+i.document_type.name+".pdf" + # append file files.append((filename, pdf)) - progress[code][0] += 1 + with lock: + progress[code][0] += 1 + # zip documents mem_zip = BytesIO() - with zipfile.ZipFile(mem_zip, mode="w",compression=zipfile.ZIP_DEFLATED) as zf: for f in files: zf.writestr(f[0], f[1]) mem_zip.seek(0) - progress[code][4] = FileResponse(mem_zip, as_attachment=True, filename="documents_" + datetime.now().strftime("%H_%M-%d_%m_%y") + ".zip") - progress[code][2] = True + # save result + with lock: + progress[code][4] = FileResponse(mem_zip, as_attachment=True, filename="documents_" + datetime.now().strftime("%H_%M-%d_%m_%y") + ".zip") + progress[code][2] = True @user_passes_test(isStaff) def upload_doc(request): |