mirror of
https://github.com/primedigitaltech/market-assistant.git
synced 2026-07-22 08:01:34 +08:00
43b72cb32e
- 流水线：评价关注词与使用场景的大模型延伸（jd_runner、竞品报告、测试） - 界面：报告生成与查看的通俗化文案；策略稿集中度按店铺/品牌区分 - 导出：PDF 插图限制最大尺寸避免 LayoutError；报告 Word/PDF 改为 blob 下载；策略稿导出错误解析；PDF 补充 Linux 中文字体路径 Made-with: Cursor
286 lines
9.5 KiB
Python
286 lines
9.5 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",
|
||
]
|
||
)
|
||
# Linux / 容器常见中文字体（路径不存在则跳过）
|
||
out.extend(
|
||
[
|
||
Path("/usr/share/fonts/truetype/wqy/wqy-microhei.ttc"),
|
||
Path("/usr/share/fonts/truetype/wqy/wqy-zenhei.ttc"),
|
||
Path("/usr/share/fonts/truetype/noto/NotoSansCJK-Regular.ttc"),
|
||
Path("/usr/share/fonts/truetype/noto/NotoSansCJKsc-Regular.otf"),
|
||
Path("/usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc"),
|
||
]
|
||
)
|
||
return out
|
||
|
||
|
||
def _pdf_flowable_image(img_path: Path, *, max_w: float, max_h: float) -> Any:
|
||
"""将插图缩放到不超过 max_w×max_h（ReportLab 单位，与 cm 一致），保持宽高比，避免矩阵长图撑爆版面。"""
|
||
from reportlab.lib.utils import ImageReader
|
||
from reportlab.platypus import Image as RLImage
|
||
|
||
p = str(img_path)
|
||
try:
|
||
ir = ImageReader(p)
|
||
iw, ih = ir.getSize()
|
||
except Exception:
|
||
return RLImage(p, width=max_w * 0.9, height=max_h * 0.9)
|
||
if iw <= 0 or ih <= 0:
|
||
return RLImage(p, width=max_w * 0.9, height=max_h * 0.9)
|
||
w = float(max_w)
|
||
h = w * (float(ih) / float(iw))
|
||
if h > float(max_h):
|
||
h = float(max_h)
|
||
w = h * (float(iw) / float(ih))
|
||
return RLImage(p, width=w, height=h)
|
||
|
||
|
||
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 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():
|
||
# 版面可用高度需小于正文框（A4 减边距后约 24.6cm），否则 ReportLab 报 LayoutError
|
||
story.append(
|
||
_pdf_flowable_image(
|
||
img_path, max_w=13 * cm, max_h=24 * 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()
|