Xlsx jinja2 template implementations

Author: HenryVisuri

poetry.lock

@@ -20,6 +20,7 @@
 from django.core.cache import cache
 from docx.shared import Mm
 from docxtpl import DocxTemplate, InlineImage, Listing, RichText
+from xlsxtpl.writerx import BookWriter
 from PIL import Image as PImage, UnidentifiedImageError
 from ..models import Attribute, ProjectPhase, ProjectAttributeFile, ProjectPhaseSectionAttribute
 from ..models.utils import create_identifier
@@ -262,10 +263,12 @@ def fetch_relevant_attributes(doc):
 
     doc_type = get_file_type(document_template.file.path)
 
+    doc = None
+    writer = None
     if doc_type == 'docx':
         doc = DocxTemplate(document_template.file)
-    else:
-        doc = None
+    elif doc_type == 'xlsx':
+        writer = BookWriter(document_template.file)
 
     attribute_data_display = {}
     attribute_element_data = {}
@@ -519,6 +522,14 @@ def process_attribute(attr):
             doc.save(output)
         except Exception as exc:
             log.error('Error while rendering document', exc)
+    elif doc_type == 'xlsx':
+        try:
+            writer.add_filter('distinct', distinct)
+            writer.render_book([attribute_data_display])
+            output = io.BytesIO()
+            writer.save(output)
+        except Exception as exc:
+            log.error('Error while rendering document', exc)
     else:
         data = {
             'data': attribute_data_display,

projects/exporting/document.py

@@ -999,7 +999,8 @@ def sanitize_attribute_data_filter_result(attributes, attribute_data):
 
 DOCUMENT_CONTENT_TYPES = {
     'docx': "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
-    'pptx': "application/vnd.openxmlformats-officedocument.presentationml.presentation.main+xml"
+    'pptx': "application/vnd.openxmlformats-officedocument.presentationml.presentation.main+xml",
+    'xlsx': "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
 }
 
 

projects/helpers.py

@@ -43,6 +43,7 @@ drf-extensions = "0.8.0"
 drf-oidc-auth = "3.0.0"
 psycopg2 = "2.9.9"
 docxtpl = "0.16.7"
+xltpl = "0.21"
 openpyxl = "3.1.2"
 python-pptx = "1.0.2"
 django-allauth = "65.11.2"