feat(engines): render all excel-sheets with thee same data

adfinis · Jun 3, 2022 · ca54651 · ca54651
1 parent 4481f52
commit ca54651
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 7 deletions.
diff --git a/document_merge_service/api/data/xlsx-structure.xlsx b/document_merge_service/api/data/xlsx-structure.xlsx
diff --git a/document_merge_service/api/engines.py b/document_merge_service/api/engines.py
@@ -209,7 +209,8 @@ def merge(self, data, buf):
         writer.jinja_env.filters.update(get_jinja_filters())
         writer.jinja_env.globals.update(dir=dir, getattr=getattr)
 
-        writer.render_book(payloads=[data])
+        data = [data] * len(writer.sheet_resource_map.sheet_state_list)
+        writer.render_book(payloads=data)
         writer.save(buf)
         return buf
 

diff --git a/document_merge_service/api/tests/test_structure.py b/document_merge_service/api/tests/test_structure.py
@@ -25,9 +25,9 @@ def test_structure():
     engine.merge(_structure, buf)
     buf.seek(0)
     doc = openpyxl.load_workbook(buf)
-    ws = doc.active
-    assert ws["A1"].value == "xdata0"
-    assert ws["A2"].value == "xdata1"
-    assert ws["A5"].value == "Item: mixed"
-    assert ws["A6"].value == "Item: list"
-    assert ws["A7"].value == "Subitem: xdata2"
+    for ws in doc.worksheets:
+        assert ws["A1"].value == "xdata0"
+        assert ws["A2"].value == "xdata1"
+        assert ws["A5"].value == "Item: mixed"
+        assert ws["A6"].value == "Item: list"
+        assert ws["A7"].value == "Subitem: xdata2"