714 lines
26 KiB
Python
714 lines
26 KiB
Python
"""把 sections/*.md 渲染成科研方向简报 .docx(简报体例,区别于 paper 的投稿稿)。
|
||
|
||
相对 paper/render_docx.py 的简报专属增强:
|
||
- **商务红配色**(主色 #C00000):标题分级染色 + 标题下细色条;TL;DR / 「判断」行做浅红底纹 callout
|
||
- **引文上标 + 内部超链接**:正文 [1] / [W3] → 上标红色,点击锚到文末参考文献对应条目
|
||
- **参考文献可点击**:DOI → https://doi.org/... 蓝色超链接;web 条目里的域名/路径 → https:// 超链接
|
||
- **化学式下标(白名单)**:CO2 / C3S2 / Na2O / SO4 ... → 真实下标,**白名单精确匹配**,不误伤 LC3 / EN 197-5 / 8.5 Mt / 2026
|
||
|
||
字体规范同院内其它渲染:中文宋体小四 / 英文 Times New Roman 小四 / 行距 1.5 / 首行缩进 2 字符。
|
||
支持 **加粗** / *斜体* / `等宽` / 列表 / 表格 /  居中插图。
|
||
|
||
用法:
|
||
python render_docx.py <sections_dir> -o <out.docx>
|
||
python render_docx.py <sections_dir> --no-color -o <out.docx> # 关配色出纯黑白
|
||
"""
|
||
from __future__ import annotations
|
||
import argparse
|
||
import re
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
from docx import Document
|
||
from docx.enum.text import WD_ALIGN_PARAGRAPH
|
||
from docx.opc.constants import RELATIONSHIP_TYPE as RT
|
||
from docx.oxml import OxmlElement
|
||
from docx.oxml.ns import qn
|
||
from docx.shared import Cm, Pt, RGBColor
|
||
|
||
# ───────────────────────── 主题色 ─────────────────────────
|
||
|
||
PRIMARY = "C00000" # 商务红主色
|
||
PRIMARY_RGB = RGBColor(0xC0, 0x00, 0x00)
|
||
TLDR_FILL = "FBE9E9" # TL;DR 浅红底纹
|
||
CALLOUT_FILL = "F7DDDD" # 「判断」callout 底纹
|
||
LINK_BLUE = "1155CC" # 超链接蓝
|
||
TABLE_HEAD_FILL = "C00000"
|
||
|
||
|
||
# ───────────────────────── 字体 / 低层 OOXML 辅助 ─────────────────────────
|
||
|
||
def _set_run_fonts(run, *, cn_font="宋体", en_font="Times New Roman") -> None:
|
||
rPr = run._element.get_or_add_rPr()
|
||
rFonts = rPr.find(qn("w:rFonts"))
|
||
if rFonts is None:
|
||
rFonts = OxmlElement("w:rFonts")
|
||
rPr.append(rFonts)
|
||
rFonts.set(qn("w:eastAsia"), cn_font)
|
||
rFonts.set(qn("w:ascii"), en_font)
|
||
rFonts.set(qn("w:hAnsi"), en_font)
|
||
|
||
|
||
def _set_style_fonts(style, *, cn_font="宋体", en_font="Times New Roman") -> None:
|
||
el = style.element
|
||
rPr = el.find(qn("w:rPr"))
|
||
if rPr is None:
|
||
rPr = OxmlElement("w:rPr")
|
||
el.insert(0, rPr)
|
||
rFonts = rPr.find(qn("w:rFonts"))
|
||
if rFonts is None:
|
||
rFonts = OxmlElement("w:rFonts")
|
||
rPr.append(rFonts)
|
||
rFonts.set(qn("w:eastAsia"), cn_font)
|
||
rFonts.set(qn("w:ascii"), en_font)
|
||
rFonts.set(qn("w:hAnsi"), en_font)
|
||
|
||
|
||
def _set_subscript(run) -> None:
|
||
rPr = run._element.get_or_add_rPr()
|
||
va = OxmlElement("w:vertAlign")
|
||
va.set(qn("w:val"), "subscript")
|
||
rPr.append(va)
|
||
|
||
|
||
def _para_shading(paragraph, fill: str) -> None:
|
||
pPr = paragraph._p.get_or_add_pPr()
|
||
shd = OxmlElement("w:shd")
|
||
shd.set(qn("w:val"), "clear")
|
||
shd.set(qn("w:color"), "auto")
|
||
shd.set(qn("w:fill"), fill)
|
||
pPr.append(shd)
|
||
|
||
|
||
def _para_border(paragraph, *, sides=("bottom",), color=PRIMARY, size=8, space=3) -> None:
|
||
pPr = paragraph._p.get_or_add_pPr()
|
||
pBdr = pPr.find(qn("w:pBdr"))
|
||
if pBdr is None:
|
||
pBdr = OxmlElement("w:pBdr")
|
||
pPr.append(pBdr)
|
||
for side in sides:
|
||
el = OxmlElement(f"w:{side}")
|
||
el.set(qn("w:val"), "single")
|
||
el.set(qn("w:sz"), str(size))
|
||
el.set(qn("w:space"), str(space))
|
||
el.set(qn("w:color"), color)
|
||
pBdr.append(el)
|
||
|
||
|
||
def _add_bookmark(paragraph, name: str, bm_id: int) -> None:
|
||
start = OxmlElement("w:bookmarkStart")
|
||
start.set(qn("w:id"), str(bm_id))
|
||
start.set(qn("w:name"), name)
|
||
end = OxmlElement("w:bookmarkEnd")
|
||
end.set(qn("w:id"), str(bm_id))
|
||
paragraph._p.insert(0, start)
|
||
paragraph._p.append(end)
|
||
|
||
|
||
def _mk_run_xml(text: str, *, size_pt: float, color=None, superscript=False,
|
||
underline=False, bold=False, cn_font="宋体", en_font="Times New Roman"):
|
||
r = OxmlElement("w:r")
|
||
rPr = OxmlElement("w:rPr")
|
||
rFonts = OxmlElement("w:rFonts")
|
||
rFonts.set(qn("w:eastAsia"), cn_font)
|
||
rFonts.set(qn("w:ascii"), en_font)
|
||
rFonts.set(qn("w:hAnsi"), en_font)
|
||
rPr.append(rFonts)
|
||
if bold:
|
||
rPr.append(OxmlElement("w:b"))
|
||
if color:
|
||
c = OxmlElement("w:color")
|
||
c.set(qn("w:val"), color)
|
||
rPr.append(c)
|
||
if underline:
|
||
u = OxmlElement("w:u")
|
||
u.set(qn("w:val"), "single")
|
||
rPr.append(u)
|
||
if superscript:
|
||
va = OxmlElement("w:vertAlign")
|
||
va.set(qn("w:val"), "superscript")
|
||
rPr.append(va)
|
||
sz = OxmlElement("w:sz")
|
||
sz.set(qn("w:val"), str(int(size_pt * 2)))
|
||
rPr.append(sz)
|
||
r.append(rPr)
|
||
t = OxmlElement("w:t")
|
||
t.set(qn("xml:space"), "preserve")
|
||
t.text = text
|
||
r.append(t)
|
||
return r
|
||
|
||
|
||
def add_internal_link(paragraph, anchor: str, text: str, *, size_pt: float,
|
||
color=PRIMARY, superscript=False) -> None:
|
||
h = OxmlElement("w:hyperlink")
|
||
h.set(qn("w:anchor"), anchor)
|
||
h.append(_mk_run_xml(text, size_pt=size_pt, color=color, superscript=superscript))
|
||
paragraph._p.append(h)
|
||
|
||
|
||
def add_external_link(paragraph, url: str, text: str, *, size_pt: float) -> None:
|
||
part = paragraph.part
|
||
r_id = part.relate_to(url, RT.HYPERLINK, is_external=True)
|
||
h = OxmlElement("w:hyperlink")
|
||
h.set(qn("r:id"), r_id)
|
||
h.append(_mk_run_xml(text, size_pt=size_pt, color=LINK_BLUE, underline=True))
|
||
paragraph._p.append(h)
|
||
|
||
|
||
# ───────────────────────── 文档初始化 ─────────────────────────
|
||
|
||
def init_doc(color: bool) -> Document:
|
||
doc = Document()
|
||
section = doc.sections[0]
|
||
section.page_height = Cm(29.7)
|
||
section.page_width = Cm(21)
|
||
for m in ("top_margin", "bottom_margin", "left_margin", "right_margin"):
|
||
setattr(section, m, Cm(2.5))
|
||
|
||
normal = doc.styles["Normal"]
|
||
normal.font.name = "Times New Roman"
|
||
normal.font.size = Pt(12)
|
||
_set_style_fonts(normal, cn_font="宋体")
|
||
pf = normal.paragraph_format
|
||
pf.line_spacing = 1.5
|
||
pf.space_before = Pt(0)
|
||
pf.space_after = Pt(0)
|
||
|
||
head_color = PRIMARY_RGB if color else RGBColor(0, 0, 0)
|
||
for lvl, sz, cn in [(1, Pt(18), "黑体"), (2, Pt(14), "黑体"), (3, Pt(12), "黑体")]:
|
||
h = doc.styles[f"Heading {lvl}"]
|
||
h.font.name = "Times New Roman"
|
||
h.font.size = sz
|
||
h.font.bold = True
|
||
h.font.color.rgb = head_color
|
||
_set_style_fonts(h, cn_font=cn)
|
||
h.paragraph_format.line_spacing = 1.3
|
||
h.paragraph_format.space_before = Pt(10 if lvl <= 2 else 6)
|
||
h.paragraph_format.space_after = Pt(4)
|
||
h.paragraph_format.first_line_indent = None
|
||
return doc
|
||
|
||
|
||
# ───────────────────────── 内联:bold/italic/code 切分 ─────────────────────────
|
||
|
||
_INLINE_RE = re.compile(
|
||
r"(?P<bold>\*\*(?P<bold_t>[^*\n]+?)\*\*)"
|
||
r"|(?P<italic>(?<![\*\w])\*(?P<italic_t>[^*\n]+?)\*(?!\*))"
|
||
r"|(?P<code>`(?P<code_t>[^`\n]+?)`)"
|
||
)
|
||
|
||
# 引文标记 [12] / [W3]
|
||
_CITE_RE = re.compile(r"\[(W?\d+)\]")
|
||
|
||
# 化学式下标白名单(统一三处渲染器共用同一份;长的在前,\b 防误伤 LC3 / C595 / 2026;
|
||
# 不含 Ca2+ 这类带电荷的——它是上标不是下标,白名单不收即天然避开)
|
||
_CHEM_RE = re.compile(
|
||
r"Ca\(OH\)2|Mg\(OH\)2"
|
||
r"|\b(?:Al2O3|Fe2O3|Fe3O4|Mn2O3|Cr2O3|P2O5|Na2SO4|K2SO4|CaSO4|CaCO3|MgCO3|"
|
||
r"CaCl2|MgCl2|Na2O|K2O|SiO2|TiO2|ZrO2|SO4|SO3|SO2|CO3|CO2|NO3|NO2|PO4|"
|
||
r"H2O|NH3|CH4|C4AF|C3S2|C2AS|C3S|C2S|C3A|O2|N2|H2)\b"
|
||
)
|
||
|
||
|
||
def _emit_chem(paragraph, text: str, *, size_pt: float, cn_font: str) -> None:
|
||
"""把白名单化学式里的数字渲成下标,其余正常。"""
|
||
pos = 0
|
||
for m in _CHEM_RE.finditer(text):
|
||
if m.start() > pos:
|
||
_emit_plain_run(paragraph, text[pos:m.start()], size_pt=size_pt, cn_font=cn_font)
|
||
formula = m.group(0)
|
||
buf = ""
|
||
for ch in formula:
|
||
if ch.isdigit():
|
||
if buf:
|
||
_emit_plain_run(paragraph, buf, size_pt=size_pt, cn_font=cn_font)
|
||
buf = ""
|
||
sub = paragraph.add_run(ch)
|
||
sub.font.size = Pt(size_pt)
|
||
_set_run_fonts(sub, cn_font=cn_font, en_font="Times New Roman")
|
||
_set_subscript(sub)
|
||
else:
|
||
buf += ch
|
||
if buf:
|
||
_emit_plain_run(paragraph, buf, size_pt=size_pt, cn_font=cn_font)
|
||
pos = m.end()
|
||
if pos < len(text):
|
||
_emit_plain_run(paragraph, text[pos:], size_pt=size_pt, cn_font=cn_font)
|
||
|
||
|
||
def _emit_plain_run(paragraph, text: str, *, size_pt: float, cn_font: str) -> None:
|
||
if not text:
|
||
return
|
||
run = paragraph.add_run(text)
|
||
run.font.size = Pt(size_pt)
|
||
_set_run_fonts(run, cn_font=cn_font, en_font="Times New Roman")
|
||
|
||
|
||
def _emit_plain_with_cites(paragraph, text: str, *, size_pt: float, cn_font: str,
|
||
make_citations: bool) -> None:
|
||
"""plain 段:处理引文上标超链接 + 化学式下标。"""
|
||
if not make_citations:
|
||
_emit_chem(paragraph, text, size_pt=size_pt, cn_font=cn_font)
|
||
return
|
||
pos = 0
|
||
prev_end = None
|
||
for m in _CITE_RE.finditer(text):
|
||
if m.start() > pos:
|
||
_emit_chem(paragraph, text[pos:m.start()], size_pt=size_pt, cn_font=cn_font)
|
||
# 连续 [1][3] 之间补一个上标逗号
|
||
if prev_end is not None and m.start() == prev_end:
|
||
comma = paragraph.add_run(",")
|
||
comma.font.size = Pt(size_pt * 0.85)
|
||
comma.font.color.rgb = PRIMARY_RGB
|
||
_set_subscript_super(comma)
|
||
cid = m.group(1)
|
||
add_internal_link(paragraph, f"ref_{cid}", cid, size_pt=size_pt * 0.85,
|
||
color=PRIMARY, superscript=True)
|
||
prev_end = m.end()
|
||
pos = m.end()
|
||
if pos < len(text):
|
||
_emit_chem(paragraph, text[pos:], size_pt=size_pt, cn_font=cn_font)
|
||
|
||
|
||
def _set_subscript_super(run) -> None:
|
||
rPr = run._element.get_or_add_rPr()
|
||
va = OxmlElement("w:vertAlign")
|
||
va.set(qn("w:val"), "superscript")
|
||
rPr.append(va)
|
||
|
||
|
||
def add_inline_rich(paragraph, text: str, *, size_pt=12.0, cn_font="宋体",
|
||
make_citations=True) -> None:
|
||
pos = 0
|
||
for m in _INLINE_RE.finditer(text):
|
||
if m.start() > pos:
|
||
_emit_plain_with_cites(paragraph, text[pos:m.start()], size_pt=size_pt,
|
||
cn_font=cn_font, make_citations=make_citations)
|
||
if m.group("bold"):
|
||
run = paragraph.add_run(m.group("bold_t"))
|
||
run.bold = True
|
||
run.font.size = Pt(size_pt)
|
||
_set_run_fonts(run, cn_font=cn_font)
|
||
elif m.group("italic"):
|
||
run = paragraph.add_run(m.group("italic_t"))
|
||
run.italic = True
|
||
run.font.size = Pt(size_pt)
|
||
_set_run_fonts(run, cn_font=cn_font)
|
||
elif m.group("code"):
|
||
run = paragraph.add_run(m.group("code_t"))
|
||
run.font.size = Pt(size_pt)
|
||
_set_run_fonts(run, cn_font=cn_font, en_font="Consolas")
|
||
pos = m.end()
|
||
if pos < len(text):
|
||
_emit_plain_with_cites(paragraph, text[pos:], size_pt=size_pt,
|
||
cn_font=cn_font, make_citations=make_citations)
|
||
|
||
|
||
# ───────────────────────── 标题 / 段落 ─────────────────────────
|
||
|
||
def add_heading(doc: Document, text: str, level: int, color: bool) -> None:
|
||
p = doc.add_paragraph(style=f"Heading {level}")
|
||
p.paragraph_format.first_line_indent = None
|
||
sizes = {1: 18.0, 2: 14.0, 3: 12.0}
|
||
if level == 1:
|
||
p.alignment = WD_ALIGN_PARAGRAPH.CENTER
|
||
add_inline_rich(p, text, size_pt=sizes[level], cn_font="黑体", make_citations=False)
|
||
for run in p.runs:
|
||
run.bold = True
|
||
if color and level <= 2:
|
||
_para_border(p, sides=("bottom",), color=PRIMARY, size=(12 if level == 1 else 6))
|
||
elif color and level == 3:
|
||
p.paragraph_format.left_indent = Pt(8)
|
||
_para_border(p, sides=("left",), color=PRIMARY, size=20, space=6)
|
||
|
||
|
||
def add_body_paragraph(doc: Document, text: str, *, indent=True) -> None:
|
||
p = doc.add_paragraph()
|
||
pf = p.paragraph_format
|
||
pf.line_spacing = 1.5
|
||
pf.first_line_indent = Pt(24) if indent else None
|
||
p.alignment = WD_ALIGN_PARAGRAPH.JUSTIFY
|
||
add_inline_rich(p, text)
|
||
|
||
|
||
def add_callout(doc: Document, text: str, fill: str, color: bool) -> None:
|
||
"""判断 / 引用块类强调框:底纹 + 左红条。"""
|
||
p = doc.add_paragraph()
|
||
pf = p.paragraph_format
|
||
pf.line_spacing = 1.4
|
||
pf.first_line_indent = None
|
||
pf.left_indent = Pt(8)
|
||
pf.space_before = Pt(3)
|
||
pf.space_after = Pt(3)
|
||
if color:
|
||
_para_shading(p, fill)
|
||
_para_border(p, sides=("left",), color=PRIMARY, size=22, space=5)
|
||
add_inline_rich(p, text)
|
||
|
||
|
||
def add_meta_band(doc: Document, text: str, color: bool) -> None:
|
||
"""标题下方的信息带(方向/时间窗/深度/数据源/受众):居中浅红底纹 + 上下细线。"""
|
||
p = doc.add_paragraph()
|
||
pf = p.paragraph_format
|
||
pf.first_line_indent = None
|
||
pf.space_before = Pt(2)
|
||
pf.space_after = Pt(12)
|
||
pf.line_spacing = 1.35
|
||
p.alignment = WD_ALIGN_PARAGRAPH.CENTER
|
||
if color:
|
||
_para_shading(p, "F3DADA")
|
||
_para_border(p, sides=("top", "bottom"), color=PRIMARY, size=6, space=3)
|
||
add_inline_rich(p, text, size_pt=10.5, make_citations=False)
|
||
|
||
|
||
def add_tldr_card(doc: Document, text: str, color: bool) -> None:
|
||
"""TL;DR 要点:每条做成浅红左条卡片,堆叠成卡片列。"""
|
||
p = doc.add_paragraph()
|
||
pf = p.paragraph_format
|
||
pf.first_line_indent = None
|
||
pf.left_indent = Pt(10)
|
||
pf.space_before = Pt(1)
|
||
pf.space_after = Pt(3)
|
||
pf.line_spacing = 1.3
|
||
if color:
|
||
_para_shading(p, TLDR_FILL)
|
||
_para_border(p, sides=("left",), color=PRIMARY, size=26, space=6)
|
||
add_inline_rich(p, text, size_pt=11.0)
|
||
|
||
|
||
def _add_field(paragraph, instr: str) -> None:
|
||
run = paragraph.add_run()
|
||
for typ, payload in (("begin", None), ("instr", instr), ("separate", None), ("end", None)):
|
||
if typ == "instr":
|
||
el = OxmlElement("w:instrText")
|
||
el.set(qn("xml:space"), "preserve")
|
||
el.text = payload
|
||
else:
|
||
el = OxmlElement("w:fldChar")
|
||
el.set(qn("w:fldCharType"), typ)
|
||
run._r.append(el)
|
||
|
||
|
||
def add_page_footer(doc: Document, color: bool) -> None:
|
||
p = doc.sections[0].footer.paragraphs[0]
|
||
p.alignment = WD_ALIGN_PARAGRAPH.CENTER
|
||
pre = p.add_run("第 ")
|
||
_add_field(p, " PAGE ")
|
||
post = p.add_run(" 页")
|
||
for r in p.runs:
|
||
r.font.size = Pt(9)
|
||
if color:
|
||
r.font.color.rgb = PRIMARY_RGB
|
||
_set_run_fonts(r, cn_font="宋体")
|
||
|
||
|
||
# ───────────────────────── 参考文献条目(可点击)─────────────────────────
|
||
|
||
_REF_RE = re.compile(r"^\[(W?\d+)\]\s+(.+)$")
|
||
_DOI_RE = re.compile(r"^10\.\d{4,9}/\S+$")
|
||
_URL_TOKEN_RE = re.compile(r"([a-z0-9][\w.\-]*\.[a-z]{2,}(?:/[^\s]+)?)", re.IGNORECASE)
|
||
|
||
|
||
def add_reference_item(doc: Document, cid: str, value: str, bm_id: int, color: bool) -> None:
|
||
p = doc.add_paragraph()
|
||
pf = p.paragraph_format
|
||
pf.first_line_indent = None
|
||
pf.left_indent = Pt(18)
|
||
pf.line_spacing = 1.3
|
||
_add_bookmark(p, f"ref_{cid}", bm_id)
|
||
# 编号标签 [n]
|
||
lab = p.add_run(f"[{cid}] ")
|
||
lab.bold = True
|
||
lab.font.size = Pt(10.5)
|
||
if color:
|
||
lab.font.color.rgb = PRIMARY_RGB
|
||
_set_run_fonts(lab, cn_font="宋体")
|
||
value = value.strip()
|
||
if _DOI_RE.match(value):
|
||
add_external_link(p, f"https://doi.org/{value}", value, size_pt=10.5)
|
||
return
|
||
# web 条目:把第一个像 URL 的 token 变成超链接
|
||
m = _URL_TOKEN_RE.search(value)
|
||
if m and ("/" in m.group(1) or m.group(1).count(".") >= 1) and " " not in m.group(1):
|
||
pre, mid, post = value[:m.start()], m.group(1), value[m.end():]
|
||
_emit_plain_run(p, pre, size_pt=10.5, cn_font="宋体")
|
||
url = mid if mid.startswith("http") else f"https://{mid}"
|
||
add_external_link(p, url, mid, size_pt=10.5)
|
||
if post:
|
||
_emit_plain_run(p, post, size_pt=10.5, cn_font="宋体")
|
||
else:
|
||
_emit_plain_run(p, value, size_pt=10.5, cn_font="宋体")
|
||
|
||
|
||
# ───────────────────────── 行类型识别 ─────────────────────────
|
||
|
||
_HEADING_RE = re.compile(r"^(#{1,6})\s+(.+)$")
|
||
_TABLE_LINE_RE = re.compile(r"^\s*\|.*\|\s*$")
|
||
_BLOCKQUOTE_RE = re.compile(r"^\s*>\s?")
|
||
_HR_RE = re.compile(r"^\s*-{3,}\s*$|^\s*={3,}\s*$|^\s*_{3,}\s*$")
|
||
_FENCE_RE = re.compile(r"^\s*(`{3,}|~{3,})\s*(\S*)\s*$")
|
||
_IMAGE_LINE_RE = re.compile(r"^\s*!\[(?P<cap>[^\]]*)\]\((?P<src>[^)\s]+)\)\s*$")
|
||
_MAX_IMG_WIDTH = Cm(15)
|
||
|
||
_LIST_PATTERNS = [
|
||
re.compile(r"^[-*+]\s"),
|
||
re.compile(r"^\d+[\.、．]\s*"),
|
||
re.compile(r"^\(\d+\)\s*"),
|
||
re.compile(r"^（\d+）\s*"),
|
||
re.compile(r"^[①②③④⑤⑥⑦⑧⑨⑩⑪⑫⑬⑭⑮]"),
|
||
]
|
||
|
||
|
||
def is_list_item(line: str) -> bool:
|
||
return any(p.match(line) for p in _LIST_PATTERNS)
|
||
|
||
|
||
# ───────────────────────── 表格 ─────────────────────────
|
||
|
||
def _split_md_row(line: str) -> list[str]:
|
||
return [c.strip() for c in line.strip().strip("|").split("|")]
|
||
|
||
|
||
def _is_sep_row(cells: list[str]) -> bool:
|
||
return all(re.match(r"^[-:\s]+$", c) for c in cells if c != "")
|
||
|
||
|
||
def render_table(doc: Document, table_lines: list[str], color: bool) -> None:
|
||
rows = []
|
||
for ln in table_lines:
|
||
cells = _split_md_row(ln)
|
||
if not cells or _is_sep_row(cells):
|
||
continue
|
||
rows.append(cells)
|
||
if not rows:
|
||
return
|
||
n_cols = max(len(r) for r in rows)
|
||
for r in rows:
|
||
while len(r) < n_cols:
|
||
r.append("")
|
||
table = doc.add_table(rows=len(rows), cols=n_cols)
|
||
try:
|
||
table.style = "Table Grid"
|
||
except KeyError:
|
||
pass
|
||
for ri, row in enumerate(rows):
|
||
for ci, val in enumerate(row):
|
||
cell = table.rows[ri].cells[ci]
|
||
cell.text = ""
|
||
p = cell.paragraphs[0]
|
||
p.paragraph_format.first_line_indent = None
|
||
p.paragraph_format.line_spacing = 1.2
|
||
add_inline_rich(p, val, size_pt=10.5, cn_font="宋体", make_citations=False)
|
||
if ri == 0:
|
||
if color:
|
||
_para_shading(p, TABLE_HEAD_FILL)
|
||
for run in p.runs:
|
||
run.bold = True
|
||
if color:
|
||
run.font.color.rgb = RGBColor(0xFF, 0xFF, 0xFF)
|
||
|
||
|
||
# ───────────────────────── 图片 ─────────────────────────
|
||
|
||
def _resolve_image_path(src: str, base_dir: Path) -> Path | None:
|
||
p = Path(src)
|
||
if not p.is_absolute():
|
||
p = (base_dir / p).resolve()
|
||
return p if p.is_file() else None
|
||
|
||
|
||
def add_image(doc: Document, png_path: Path, caption: str | None, ctx: dict) -> None:
|
||
p = doc.add_paragraph()
|
||
p.alignment = WD_ALIGN_PARAGRAPH.CENTER
|
||
p.paragraph_format.first_line_indent = None
|
||
p.paragraph_format.space_before = Pt(6)
|
||
p.paragraph_format.space_after = Pt(3)
|
||
run = p.add_run()
|
||
try:
|
||
run.add_picture(str(png_path), width=_MAX_IMG_WIDTH)
|
||
except Exception as e:
|
||
run.add_text(f"[image failed: {png_path.name}: {e}]")
|
||
return
|
||
ctx["fig_no"] = ctx.get("fig_no", 0) + 1
|
||
cap_p = doc.add_paragraph()
|
||
cap_p.alignment = WD_ALIGN_PARAGRAPH.CENTER
|
||
cap_p.paragraph_format.first_line_indent = None
|
||
cap_p.paragraph_format.space_after = Pt(6)
|
||
cap_text = f"图 {ctx['fig_no']} {caption}" if caption else f"图 {ctx['fig_no']}"
|
||
cap_run = cap_p.add_run(cap_text)
|
||
cap_run.font.size = Pt(10.5)
|
||
cap_run.bold = True
|
||
_set_run_fonts(cap_run, cn_font="宋体")
|
||
|
||
|
||
# ───────────────────────── 主渲染 ─────────────────────────
|
||
|
||
def render_md_block(doc: Document, md_text: str, ctx: dict) -> None:
|
||
color = ctx["color"]
|
||
lines = md_text.splitlines()
|
||
i, n = 0, len(lines)
|
||
in_refs = False # 进入「参考文献」段后,[n] 行按引文条目渲染
|
||
expect_meta = False # 紧跟 H1 标题的信息带(方向/时间窗...)
|
||
in_tldr = False # 「一句话要点」段:列表项做卡片
|
||
while i < n:
|
||
line = lines[i].rstrip()
|
||
if not line.strip():
|
||
i += 1
|
||
continue
|
||
|
||
if _HR_RE.match(line):
|
||
i += 1
|
||
continue
|
||
|
||
m_img = _IMAGE_LINE_RE.match(line)
|
||
if m_img:
|
||
png = _resolve_image_path(m_img.group("src"), ctx["sections_dir"])
|
||
if png is not None:
|
||
add_image(doc, png, m_img.group("cap").strip() or None, ctx)
|
||
else:
|
||
add_body_paragraph(doc, f"[image missing: {m_img.group('src')}]", indent=False)
|
||
i += 1
|
||
continue
|
||
|
||
m_fence = _FENCE_RE.match(line)
|
||
if m_fence:
|
||
fence = m_fence.group(1)
|
||
code = []
|
||
i += 1
|
||
while i < n:
|
||
mc = _FENCE_RE.match(lines[i])
|
||
if mc and mc.group(1)[0] == fence[0] and len(mc.group(1)) >= len(fence):
|
||
i += 1
|
||
break
|
||
code.append(lines[i])
|
||
i += 1
|
||
for ln in code:
|
||
p = doc.add_paragraph()
|
||
p.paragraph_format.first_line_indent = None
|
||
p.paragraph_format.line_spacing = 1.0
|
||
run = p.add_run(ln if ln else " ")
|
||
run.font.size = Pt(10.5)
|
||
_set_run_fonts(run, cn_font="新宋体", en_font="Consolas")
|
||
continue
|
||
|
||
if _TABLE_LINE_RE.match(line):
|
||
block = []
|
||
while i < n and _TABLE_LINE_RE.match(lines[i]):
|
||
block.append(lines[i])
|
||
i += 1
|
||
render_table(doc, block, color)
|
||
continue
|
||
|
||
m = _HEADING_RE.match(line)
|
||
if m:
|
||
title = m.group(2).strip()
|
||
level = min(len(m.group(1)), 3)
|
||
in_refs = "参考文献" in title
|
||
expect_meta = (level == 1)
|
||
if level <= 2:
|
||
in_tldr = ("要点" in title) or ("TL;DR" in title.upper())
|
||
add_heading(doc, title, level, color)
|
||
i += 1
|
||
continue
|
||
|
||
if _BLOCKQUOTE_RE.match(line):
|
||
# 引用块:并合连续 > 行,做浅红 callout(说明 / 取舍纪律等)
|
||
buf = [_BLOCKQUOTE_RE.sub("", line).strip()]
|
||
i += 1
|
||
while i < n and _BLOCKQUOTE_RE.match(lines[i]):
|
||
buf.append(_BLOCKQUOTE_RE.sub("", lines[i]).strip())
|
||
i += 1
|
||
add_callout(doc, " ".join(buf), TLDR_FILL, color)
|
||
continue
|
||
|
||
# 参考文献条目
|
||
if in_refs:
|
||
m_ref = _REF_RE.match(line.strip())
|
||
if m_ref:
|
||
ctx["bm_id"] += 1
|
||
add_reference_item(doc, m_ref.group(1), m_ref.group(2), ctx["bm_id"], color)
|
||
i += 1
|
||
continue
|
||
|
||
# 「判断」强调行 → callout
|
||
if line.strip().startswith("**判断**"):
|
||
add_callout(doc, line.strip(), CALLOUT_FILL, color)
|
||
i += 1
|
||
continue
|
||
|
||
if is_list_item(line):
|
||
if in_tldr:
|
||
add_tldr_card(doc, line.strip(), color)
|
||
else:
|
||
add_body_paragraph(doc, line.strip(), indent=False)
|
||
i += 1
|
||
continue
|
||
|
||
# 紧跟标题的信息带
|
||
if expect_meta and ("时间窗" in line):
|
||
add_meta_band(doc, line.strip(), color)
|
||
expect_meta = False
|
||
i += 1
|
||
continue
|
||
|
||
# 普通段落:并合软换行
|
||
buf = [line.strip()]
|
||
j = i + 1
|
||
while j < n:
|
||
nxt = lines[j].rstrip()
|
||
if not nxt.strip() or _HEADING_RE.match(nxt) or _BLOCKQUOTE_RE.match(nxt) \
|
||
or _TABLE_LINE_RE.match(nxt) or is_list_item(nxt) or _HR_RE.match(nxt):
|
||
break
|
||
buf.append(nxt.strip())
|
||
j += 1
|
||
add_body_paragraph(doc, " ".join(buf), indent=True)
|
||
i = j
|
||
|
||
|
||
def render_sections(sections_dir: Path, out: Path, color: bool) -> None:
|
||
if not sections_dir.is_dir():
|
||
print(f"[ERR] sections dir not found: {sections_dir}", file=sys.stderr)
|
||
sys.exit(2)
|
||
md_files = sorted(sections_dir.glob("*.md"))
|
||
if not md_files:
|
||
print(f"[ERR] no .md found in {sections_dir}", file=sys.stderr)
|
||
sys.exit(2)
|
||
|
||
ctx = {
|
||
"sections_dir": sections_dir,
|
||
"figures_dir": sections_dir.parent / "figures",
|
||
"fig_no": 0,
|
||
"bm_id": 0,
|
||
"color": color,
|
||
}
|
||
doc = init_doc(color)
|
||
add_page_footer(doc, color)
|
||
for idx, f in enumerate(md_files):
|
||
render_md_block(doc, f.read_text(encoding="utf-8"), ctx)
|
||
if idx != len(md_files) - 1:
|
||
doc.add_page_break()
|
||
|
||
out.parent.mkdir(parents=True, exist_ok=True)
|
||
doc.save(str(out))
|
||
|
||
paras = sum(1 for _ in doc.paragraphs)
|
||
chars = sum(len(p.text) for p in doc.paragraphs)
|
||
print(f"[OK] rendered {len(md_files)} sections -> {out}")
|
||
print(f" paragraphs: {paras} | tables: {len(doc.tables)} | figures: {ctx['fig_no']} | chars: {chars}")
|
||
print(f" theme: {'商务红 #C00000' if color else '黑白'} | 引文上标+超链接 | 化学式下标白名单")
|
||
|
||
|
||
def main() -> None:
|
||
ap = argparse.ArgumentParser(description="渲染章节 md → 科研方向简报 docx")
|
||
ap.add_argument("sections_dir", type=Path, help="sections/*.md 目录")
|
||
ap.add_argument("--no-color", dest="color", action="store_false",
|
||
help="关配色,出纯黑白(默认商务红主题)")
|
||
ap.add_argument("-o", "--output", type=Path, required=True, help="输出 .docx 路径")
|
||
args = ap.parse_args()
|
||
render_sections(args.sections_dir, args.output, args.color)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|