79 lines
2.4 KiB
Python
79 lines
2.4 KiB
Python
#!/usr/bin/env python3
|
||
"""Offline runtime smoke test for the LocalSandbox Office toolchain."""
|
||
|
||
from __future__ import annotations
|
||
|
||
import shutil
|
||
import subprocess
|
||
import tempfile
|
||
from pathlib import Path
|
||
|
||
|
||
def require_command(name: str) -> str:
|
||
path = shutil.which(name)
|
||
if path is None:
|
||
raise RuntimeError(f"missing command: {name}")
|
||
return path
|
||
|
||
|
||
def assert_contains(text: str, expected: str, label: str) -> None:
|
||
normalized = text.replace("\\_", "_")
|
||
if expected not in normalized:
|
||
raise RuntimeError(f"{label} conversion lost test text: {text[:300]!r}")
|
||
|
||
|
||
def main() -> int:
|
||
from docx import Document
|
||
from markitdown import MarkItDown
|
||
from openpyxl import Workbook
|
||
from pptx import Presentation
|
||
|
||
soffice = require_command("soffice")
|
||
require_command("pandoc")
|
||
require_command("pdftotext")
|
||
require_command("file")
|
||
|
||
subprocess.run(
|
||
[soffice, "--headless", "--version"],
|
||
check=True,
|
||
capture_output=True,
|
||
text=True,
|
||
timeout=30,
|
||
)
|
||
|
||
with tempfile.TemporaryDirectory(prefix="deerflow-office-selfcheck-") as temporary:
|
||
root = Path(temporary)
|
||
|
||
word_path = root / "word-test.docx"
|
||
document = Document()
|
||
document.add_heading("Word 自检", level=1)
|
||
document.add_paragraph("DEERFLOW_WORD_OK")
|
||
document.save(word_path)
|
||
|
||
excel_path = root / "excel-test.xlsx"
|
||
workbook = Workbook()
|
||
worksheet = workbook.active
|
||
worksheet.title = "自检"
|
||
worksheet.append(["状态", "DEERFLOW_EXCEL_OK"])
|
||
workbook.save(excel_path)
|
||
|
||
ppt_path = root / "ppt-test.pptx"
|
||
presentation = Presentation()
|
||
slide = presentation.slides.add_slide(presentation.slide_layouts[5])
|
||
slide.shapes.title.text = "DEERFLOW_PPT_OK"
|
||
presentation.save(ppt_path)
|
||
|
||
converter = MarkItDown()
|
||
assert_contains(converter.convert(word_path).text_content, "DEERFLOW_WORD_OK", "Word")
|
||
assert_contains(converter.convert(excel_path).text_content, "DEERFLOW_EXCEL_OK", "Excel")
|
||
assert_contains(converter.convert(ppt_path).text_content, "DEERFLOW_PPT_OK", "PowerPoint")
|
||
|
||
print("OFFICE_RUNTIME_SELF_CHECK=PASS")
|
||
print("支持：DOC/DOCX、XLS/XLSX、PPT/PPTX、ODT/ODS/ODP、PDF")
|
||
print("组件：MarkItDown、python-docx、openpyxl、python-pptx、LibreOffice、Pandoc、中文字体")
|
||
return 0
|
||
|
||
|
||
if __name__ == "__main__":
|
||
raise SystemExit(main())
|