"""Markdown → Word（.docx）/ 简易 PDF；供任务报告与策略稿导出。""" from __future__ import annotations import os import re from io import BytesIO from pathlib import Path from typing import Any from xml.sax.saxutils import escape as xml_escape def _strip_inline_md(s: str) -> str: s = re.sub(r"\*\*(.+?)\*\*", r"\1", s) s = re.sub(r"`([^`]+)`", r"\1", s) return s def _is_table_sep(line: str) -> bool: t = line.strip() if not t.startswith("|"): return False inner = t.strip("|").replace(" ", "") return bool(inner) and all(p in ("", "---", ":---", "---:", ":---:") for p in t.split("|")) _img_line = re.compile(r"^!\[([^\]]*)\]\(([^)]+)\)\s*$") def markdown_to_docx_bytes(md: str, *, asset_root: Path | None = None) -> bytes: from docx import Document from docx.enum.text import WD_PARAGRAPH_ALIGNMENT from docx.shared import Inches, Pt doc = Document() try: style = doc.styles["Normal"] style.font.name = "Microsoft YaHei" style.font.size = Pt(10.5) except Exception: pass lines = (md or "").replace("\r\n", "\n").split("\n") i = 0 in_fence = False while i < len(lines): raw = lines[i] if raw.strip().startswith("```"): in_fence = not in_fence i += 1 continue if in_fence: p = doc.add_paragraph(xml_escape(raw) or " ") p.style = doc.styles["Normal"] for run in p.runs: run.font.name = "Consolas" run.font.size = Pt(9) i += 1 continue line = raw.rstrip() if not line.strip(): doc.add_paragraph("") i += 1 continue if line.startswith("# "): doc.add_heading(_strip_inline_md(line[2:].strip()), level=0) i += 1 continue if line.startswith("## "): doc.add_heading(_strip_inline_md(line[3:].strip()), level=1) i += 1 continue if line.startswith("### "): doc.add_heading(_strip_inline_md(line[4:].strip()), level=2) i += 1 continue if line.startswith("#### "): doc.add_heading(_strip_inline_md(line[5:].strip()), level=3) i += 1 continue mimg = _img_line.match(line.strip()) if mimg and asset_root is not None: rel = mimg.group(2).strip() if not (rel.startswith("http://") or rel.startswith("https://")): img_path = (asset_root / rel).resolve() try: img_path.relative_to(asset_root.resolve()) except ValueError: i += 1 continue if img_path.is_file(): doc.add_picture(str(img_path), width=Inches(5.9)) i += 1 continue if line.strip().startswith("|"): rows: list[list[str]] = [] while i < len(lines) and lines[i].strip().startswith("|"): row_line = lines[i].strip() if _is_table_sep(row_line): i += 1 continue cells = [c.strip() for c in row_line.strip("|").split("|")] rows.append([_strip_inline_md(c) for c in cells]) i += 1 if rows: max_cols = max(len(r) for r in rows) pad_rows = [r + [""] * (max_cols - len(r)) for r in rows] tbl = doc.add_table(rows=len(pad_rows), cols=max_cols) tbl.style = "Table Grid" for ri, row in enumerate(pad_rows): for ci, cell in enumerate(row): tbl.rows[ri].cells[ci].text = cell continue p = doc.add_paragraph() p.alignment = WD_PARAGRAPH_ALIGNMENT.LEFT text = _strip_inline_md(line) p.add_run(text) bio = BytesIO() doc.save(bio) return bio.getvalue() def _pdf_font_candidates() -> list[Path]: raw = (os.environ.get("MA_PDF_FONT") or "").strip() out: list[Path] = [] if raw: out.append(Path(raw)) windir = os.environ.get("WINDIR", r"C:\Windows") out.extend( [ Path(windir) / "Fonts" / "simhei.ttf", Path(windir) / "Fonts" / "simsun.ttc", Path(windir) / "Fonts" / "msyh.ttf", ] ) return out def markdown_to_pdf_bytes(md: str, *, asset_root: Path | None = None) -> bytes: """简易纯文本流式 PDF；需本机 .ttf 中文字体或环境变量 MA_PDF_FONT。""" from reportlab.lib.pagesizes import A4 from reportlab.lib.styles import ParagraphStyle, getSampleStyleSheet from reportlab.lib.units import cm from reportlab.pdfbase import pdfmetrics from reportlab.pdfbase.ttfonts import TTFont from reportlab.platypus import Image as RLImage from reportlab.platypus import Paragraph, SimpleDocTemplate, Spacer font_name = "MaExportCJK" registered = False for p in _pdf_font_candidates(): if not p.is_file(): continue try: if p.suffix.lower() == ".ttc": try: pdfmetrics.registerFont( TTFont(font_name, str(p), subfontIndex=0) ) except TypeError: pdfmetrics.registerFont(TTFont(font_name, str(p))) else: pdfmetrics.registerFont(TTFont(font_name, str(p))) registered = True break except Exception: continue if not registered: raise ValueError( "未找到可用的中文字体文件。请在 Windows 上安装黑体/宋体，" "或设置环境变量 MA_PDF_FONT 指向 .ttf 文件路径。" ) styles = getSampleStyleSheet() body = ParagraphStyle( name="BodyCJK", parent=styles["Normal"], fontName=font_name, fontSize=10, leading=14, ) h1s = ParagraphStyle( name="H1CJK", parent=body, fontSize=16, leading=20, spaceAfter=8, ) h2s = ParagraphStyle( name="H2CJK", parent=body, fontSize=13, leading=17, spaceAfter=6, ) story: list[Any] = [] lines = (md or "").replace("\r\n", "\n").split("\n") in_fence = False for raw in lines: if raw.strip().startswith("```"): in_fence = not in_fence continue s = raw.rstrip() if in_fence: story.append(Paragraph(xml_escape(s or " "), body)) story.append(Spacer(1, 0.1 * cm)) continue if not s.strip(): story.append(Spacer(1, 0.15 * cm)) continue mimg = _img_line.match(s.strip()) if mimg and asset_root is not None: rel = mimg.group(2).strip() if not (rel.startswith("http://") or rel.startswith("https://")): img_path = (asset_root / rel).resolve() try: img_path.relative_to(asset_root.resolve()) except ValueError: continue if img_path.is_file(): story.append(RLImage(str(img_path), width=13 * cm)) story.append(Spacer(1, 0.2 * cm)) continue plain = _strip_inline_md(s) text = xml_escape(plain) if s.startswith("# "): story.append(Paragraph(xml_escape(plain[2:]), h1s)) elif s.startswith("## "): story.append(Paragraph(xml_escape(plain[3:]), h2s)) elif s.startswith("### "): story.append(Paragraph(xml_escape(plain[4:]), body)) elif s.strip().startswith("|"): story.append(Paragraph(text.replace("|", " │ "), body)) else: story.append(Paragraph(text, body)) buf = BytesIO() doc = SimpleDocTemplate( buf, pagesize=A4, leftMargin=2 * cm, rightMargin=2 * cm, topMargin=2 * cm, bottomMargin=2 * cm, ) doc.build(story) return buf.getvalue()