feat: 实现批量 Excel 生成（文本、图片、表格区域）

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-05-05 13:05:23 +08:00
parent 504bd2f65f
commit eb82650c03
4 changed files with 140 additions and 2 deletions
@@ -1 +1,73 @@
-# placeholder
+import sys, os, tempfile
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), "../../python"))
+
+from generator import generate
+from openpyxl import load_workbook
+
+FIXTURE = os.path.join(os.path.dirname(__file__), "../fixtures/sample_template.xlsx")
+IMAGE = os.path.join(os.path.dirname(__file__), "../fixtures/sample_image.png")
+
+def _base_request(output_dir):
+    return {
+        "template_path": FIXTURE,
+        "fields": [
+            {"name": "编号", "type": "text", "sheet": "Sheet1", "cell": "B3"},
+            {"name": "姓名", "type": "text", "sheet": "Sheet1", "cell": "C5"},
+        ],
+        "rows": [
+            {"编号": "001", "姓名": "张三"},
+            {"编号": "002", "姓名": "李四"},
+        ],
+        "output_dir": output_dir,
+        "filename_pattern": "{{编号}}_报告.xlsx",
+    }
+
+def test_generates_correct_number_of_files():
+    with tempfile.TemporaryDirectory() as tmpdir:
+        result = generate(_base_request(tmpdir))
+        assert len(result["results"]) == 2
+
+def test_output_files_exist():
+    with tempfile.TemporaryDirectory() as tmpdir:
+        result = generate(_base_request(tmpdir))
+        for r in result["results"]:
+            assert r["status"] == "success"
+            assert os.path.exists(r["file"])
+
+def test_text_fields_are_filled():
+    with tempfile.TemporaryDirectory() as tmpdir:
+        result = generate(_base_request(tmpdir))
+        wb = load_workbook(result["results"][0]["file"])
+        ws = wb["Sheet1"]
+        assert ws["B3"].value == "001"
+        assert ws["C5"].value == "张三"
+
+def test_filename_pattern_applied():
+    with tempfile.TemporaryDirectory() as tmpdir:
+        result = generate(_base_request(tmpdir))
+        names = [os.path.basename(r["file"]) for r in result["results"]]
+        assert "001_报告.xlsx" in names
+        assert "002_报告.xlsx" in names
+
+def test_missing_field_is_skipped_not_crashed():
+    with tempfile.TemporaryDirectory() as tmpdir:
+        req = _base_request(tmpdir)
+        req["rows"] = [{"编号": "003"}]  # 缺 姓名
+        result = generate(req)
+        assert result["results"][0]["status"] == "success"
+
+def test_image_field_is_inserted():
+    with tempfile.TemporaryDirectory() as tmpdir:
+        req = {
+            "template_path": FIXTURE,
+            "fields": [
+                {"name": "图片1", "type": "image", "sheet": "Sheet1", "cell": "A1"},
+            ],
+            "rows": [{"图片1": IMAGE}],
+            "output_dir": tmpdir,
+            "filename_pattern": "img_test.xlsx",
+        }
+        result = generate(req)
+        assert result["results"][0]["status"] == "success"
+        wb = load_workbook(result["results"][0]["file"])
+        assert len(wb["Sheet1"]._images) == 1