78 lines
2.4 KiB
Python
78 lines
2.4 KiB
Python
from __future__ import annotations
|
||
|
||
import re
|
||
from typing import Any
|
||
|
||
from .classify import SECTION_ORDER
|
||
from .models import NewsItem
|
||
from .validate import validate_markdown
|
||
|
||
|
||
END_PUNCTUATION = "。！？；.!?;"
|
||
|
||
|
||
def _clean_text(text: str) -> str:
|
||
value = re.sub(r"^```(?:\w+)?\s*\n?", "", (text or "").strip())
|
||
value = re.sub(r"\n?```\s*$", "", value)
|
||
value = re.sub(r"^\s*>\s*", "", value)
|
||
value = re.sub(r"\[\d+\]|\[N\]", "", value)
|
||
value = re.sub(r"主线判断[：:]\s*", "", value)
|
||
value = re.sub(r"\s+", " ", value).strip()
|
||
return value
|
||
|
||
|
||
def _ensure_sentence(text: str) -> str:
|
||
value = _clean_text(text)
|
||
if value and value[-1] not in END_PUNCTUATION:
|
||
value += "。"
|
||
return value
|
||
|
||
|
||
def _source_link(item: NewsItem) -> str:
|
||
source = item.source_label or item.source_group or "来源"
|
||
if item.url:
|
||
return f"[{source} ↗]({item.url})"
|
||
return source
|
||
|
||
|
||
def assemble_markdown(items: list[NewsItem], guide: dict[str, Any] | None = None) -> tuple[str, dict[str, Any]]:
|
||
guide = guide or {"theme": "", "threads": []}
|
||
lines: list[str] = []
|
||
|
||
theme = _clean_text(str(guide.get("theme") or ""))
|
||
if theme:
|
||
lines.extend(["## 导览", "", f"> {theme}", ""])
|
||
|
||
item_number = 1
|
||
for section in SECTION_ORDER:
|
||
section_items = [item for item in items if item.section == section]
|
||
if not section_items:
|
||
continue
|
||
lines.extend([f"## {section}", ""])
|
||
for item in section_items:
|
||
title = _clean_text(item.title or item.title_raw)
|
||
summary = _ensure_sentence(item.summary or item.summary_raw or "该条目暂无摘要。")
|
||
lines.extend(
|
||
[
|
||
f"**{item_number}. {title}**",
|
||
"",
|
||
f"> {summary}{_source_link(item)}",
|
||
"",
|
||
]
|
||
)
|
||
item_number += 1
|
||
|
||
threads = guide.get("threads", []) or []
|
||
if threads:
|
||
lines.extend(["## 今日脉络", ""])
|
||
for thread in threads:
|
||
title = _clean_text(str(thread.get("title") or ""))
|
||
text = _ensure_sentence(str(thread.get("text") or ""))
|
||
if not title or not text:
|
||
continue
|
||
lines.extend([f"- **{title}**", f" {text}", ""])
|
||
|
||
markdown = "\n".join(lines).strip()
|
||
report = validate_markdown(markdown, items)
|
||
return markdown, report
|