mirror of
https://github.com/primedigitaltech/market-assistant.git
synced 2026-07-21 23:41:39 +08:00
251 lines
8.0 KiB
Python
251 lines
8.0 KiB
Python
"""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()
|