commit edits for new featuredata_download

1 files changed, 83 insertions, 1 deletions

diff --git a/server/views.py b/server/views.py
index ef13e45..b9e2dc7 100644
--- a/server/views.py
+++ b/server/views.py
@@ -28,6 +28,7 @@ from PIL import Image, UnidentifiedImageError
 import zipfile

 import json

 import threading

+from pypdf import PdfMerger

 

 # custom staff check function for non primary group staff members

 def isStaff(user):

@@ -416,7 +417,7 @@ def ulist(request):
                 # get template and build context

                 template = get_template('server/download_doc.html')

                 doc = [document, KeyVal.objects.filter(

-                    container=document), document.personal_data, document.medical_data, document.user.groups.values_list('name', flat=True)[0]]

+                    container=document), document.personal_data, document.medical_data]

                 context = {'doc': doc, 'vac': vac_file,

                            'health': health_file, 'sign_doc_file': sign_doc_file}

                 # render context

@@ -1899,6 +1900,87 @@ def debug_uc(request):
     return render(request, 'server/debug_usercode.html', context)

 

 @user_passes_test(isStaff)

+def download_data(request):

+    groups = getGroups(request)

+

+    # get all doctypes the user has access to

+    doctypes = DocumentType.objects.filter(group__in=groups)

+    # sort by name

+    doctypes = doctypes.order_by("name")

+

+    context = {

+        "docs": doctypes

+    }

+

+    return render(request, 'server/data_download.html', context)

+

+def download_data_progress(request):

+    # if user wants to download result

+    if 'download' in request.GET:

+        # if job is completed

+        if request.session['status']:

+            data = BytesIO(base64.b64decode(request.session['result']))

+            data.seek(0)

+            return FileResponse(data, as_attachment=True, filename="documents_" + datetime.now().strftime("%H_%M-%d_%m_%y") + ".pdf")

+

+    # otherwise return status

+    data = [request.session['progress'], request.session['total'], request.session['status']]

+    return HttpResponse(json.dumps(data))

+

+def prepare_documents(docs, session_key):

+    pdf_list = []

+

+    # get session

+    session = SessionStore(session_key=session_key)

+    for i in docs:

+        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]

+        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)

+        # append file

+        pdf_list.append(pdf)

+

+        session['progress'] += 1

+        session.save()

+

+    merger = PdfMerger()

+    for pdf in pdf_list:

+        merger.append(BytesIO(pdf))

+

+    mem = BytesIO()

+    merger.write(mem)

+    mem.seek(0)

+    # save result

+    session['result'] = base64.b64encode(mem.getvalue()).decode()

+    session['status'] = True

+    session.save()

+

+@user_passes_test(isStaff)

 def data_request(request):

     context = {}

     parent_group = getGroups(request)[0]