diff options
Diffstat (limited to '')
| -rw-r--r-- | server/views.py | 47 |
1 files changed, 46 insertions, 1 deletions
diff --git a/server/views.py b/server/views.py index cdc1b74..250feec 100644 --- a/server/views.py +++ b/server/views.py @@ -19,6 +19,7 @@ from io import BytesIO import os import base64 from PIL import Image, UnidentifiedImageError +import zipfile # custom staff check function for non primary group staff members @@ -604,7 +605,7 @@ def doclist(request): pdf = pdfkit.from_string(html, False) result = BytesIO(pdf) result.seek(0) - return FileResponse(result, filename=document.user.username+"_"+document.document_type.name+".pdf") + return FileResponse(result, as_attachment=True, filename=document.user.username+"_"+document.document_type.name+".pdf") # get selected documents and check if user has permission to view selected = [] @@ -614,6 +615,8 @@ def doclist(request): if docc.group.name in parent_groups: selected.append(docc) + # create list of pdfs + files = [] # execute action on selected documents for i in selected: if request.POST["action"] == 'delete' and settings.DEBUG: @@ -638,6 +641,48 @@ def doclist(request): else: error = True error_text = "Non puoi dearchiviare un documento non archiviato" + elif request.POST["action"] == "download": + vac_file = "" + health_file = "" + sign_doc_file = "" + + # prepare pictures in base64 + if i.medical_data: + if i.medical_data.vac_certificate.name: + with open(i.medical_data.vac_certificate.name, 'rb') as image_file: + vac_file = base64.b64encode( + image_file.read()).decode() + + if i.medical_data.health_care_certificate.name: + with open(i.medical_data.health_care_certificate.name, 'rb') as image_file: + health_file = base64.b64encode( + image_file.read()).decode() + if i.signed_doc: + with open(i.signed_doc.name, 'rb') as image_file: + sign_doc_file = base64.b64encode( + image_file.read()).decode() + + template = get_template('server/download_doc.html') + doc = [i, KeyVal.objects.filter( + container=i), i.personal_data, i.medical_data, i.user.groups.values_list('name', flat=True)[0]] + context = {'doc': doc, 'vac': vac_file, + 'health': health_file, 'sign_doc_file': sign_doc_file} + # render context + html = template.render(context) + # render pdf using wkhtmltopdf + pdf = pdfkit.from_string(html, False) + filename = i.user.username+"_"+i.document_type.name+".pdf" + files.append((filename, pdf)) + + if request.POST["action"] == "download": + 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) + return FileResponse(mem_zip, as_attachment=True, filename="documents_" + datetime.now().strftime("%H_%M-%d_%m_%y") + ".zip") # get filter values hidden = "filter_hidden" in request.POST |