万朱浩 / Venue-Ops

Go to a project
Toggle navigation Toggle navigation pinning
马一丁
2c21832b 1 parent 08fddfbf

Add test file

Inline Side-by-side
Showing 1 changed file with 587 additions and 0 deletions
#!/usr/bin/env python3
"""
生成覆盖全部允许block类型的演示 IR，用于验证 HTML 与 PDF 渲染。
执行后会在 `final_reports/ir` 写入一份带时间戳的 IR，
并分别在 `final_reports/html` 与 `final_reports/pdf` 输出对应的渲染文件。
"""
from __future__ import annotations
import json
import sys
from datetime import datetime
from pathlib import Path
# 允许直接以脚本形式运行
ROOT = Path(__file__).resolve().parents[2]
if str(ROOT) not in sys.path:
sys.path.insert(0, str(ROOT))
from ReportEngine.core import DocumentComposer
from ReportEngine.ir import IRValidator
from ReportEngine.ir.schema import ENGINE_AGENT_TITLES
from ReportEngine.renderers import HTMLRenderer, PDFRenderer
from ReportEngine.utils.config import settings
def build_inline_marks_demo() -> dict:
"""生成覆盖全部内联标记的 paragraph block。"""
return {
"type": "paragraph",
"inlines": [
{"text": "这一段覆盖全部内联标记："},
{"text": "粗体", "marks": [{"type": "bold"}]},
{"text": " / 斜体", "marks": [{"type": "italic"}]},
{"text": " / 下划线", "marks": [{"type": "underline"}]},
{"text": " / 删除线", "marks": [{"type": "strike"}]},
{"text": " / 代码", "marks": [{"type": "code"}]},
{
"text": " / 链接",
"marks": [
{
"type": "link",
"href": "https://example.com/demo",
"title": "示例链接",
}
],
},
{"text": " / 颜色", "marks": [{"type": "color", "value": "#c0392b"}]},
{
"text": " / 字体",
"marks": [
{
"type": "font",
"family": "Georgia, serif",
"size": "15px",
"weight": "600",
}
],
},
{"text": " / 高亮", "marks": [{"type": "highlight"}]},
{"text": " / 下标", "marks": [{"type": "subscript"}]},
{"text": " / 上标", "marks": [{"type": "superscript"}]},
{"text": " / 行内公式", "marks": [{"type": "math", "value": "E=mc^2"}]},
{"text": "。"},
],
}
def build_widget_block() -> dict:
"""构造一个合法的 Chart.js widget block。"""
return {
"type": "widget",
"widgetId": "demo-volume-trend",
"widgetType": "chart.js/line",
"props": {
"type": "line",
"options": {
"responsive": True,
"plugins": {"legend": {"position": "bottom"}},
"scales": {"y": {"title": {"display": True, "text": "提及量"}}},
},
},
"data": {
"labels": ["T0", "T0+6h", "T0+12h", "T0+18h", "T0+24h"],
"datasets": [
{
"label": "主流媒体",
"data": [12, 18, 23, 30, 26],
"borderColor": "#2980b9",
"backgroundColor": "rgba(41,128,185,0.18)",
"tension": 0.25,
"fill": False,
},
{
"label": "社交平台",
"data": [8, 10, 15, 28, 40],
"borderColor": "#c0392b",
"backgroundColor": "rgba(192,57,43,0.2)",
"tension": 0.35,
"fill": False,
},
],
},
}
def build_chapters() -> list[dict]:
"""构造覆盖所有 block 类型的章节列表。"""
inline_demo = build_inline_marks_demo()
bullet_list = {
"type": "list",
"listType": "bullet",
"items": [
[
{
"type": "paragraph",
"inlines": [{"text": "社交媒体热度在 48 小时内翻倍"}],
}
],
[
{
"type": "paragraph",
"inlines": [{"text": "主流媒体报道集中在早间时段"}],
},
{
"type": "list",
"listType": "ordered",
"items": [
[
{
"type": "paragraph",
"inlines": [{"text": "07:00-09:00：首轮报道"}],
}
],
[
{
"type": "paragraph",
"inlines": [{"text": "10:00-12:00：评论扩散"}],
}
],
],
},
],
[
{
"type": "paragraph",
"inlines": [{"text": "地方政务号开始回应并同步线下通稿"}],
}
],
],
}
task_list = {
"type": "list",
"listType": "task",
"items": [
[
{
"type": "paragraph",
"inlines": [{"text": "跟踪权威辟谣素材是否上线"}],
}
],
[
{
"type": "paragraph",
"inlines": [{"text": "监测新增关联关键词与长尾问题"}],
}
],
[
{
"type": "paragraph",
"inlines": [{"text": "准备 FAQ 供客服统一答复"}],
}
],
],
}
table_block = {
"type": "table",
"caption": "核心信源与传播路径",
"zebra": True,
"colgroup": [{"width": "22%"}, {"width": "38%"}, {"width": "40%"}],
"rows": [
{
"cells": [
{
"align": "center",
"blocks": [
{
"type": "paragraph",
"inlines": [{"text": "时间节点", "marks": [{"type": "bold"}]}],
}
],
},
{
"align": "center",
"blocks": [
{
"type": "paragraph",
"inlines": [{"text": "事件内容", "marks": [{"type": "bold"}]}],
}
],
},
{
"align": "center",
"blocks": [
{
"type": "paragraph",
"inlines": [{"text": "主要渠道", "marks": [{"type": "bold"}]}],
}
],
},
]
},
{
"cells": [
{"blocks": [{"type": "paragraph", "inlines": [{"text": "T0"}]}]},
{
"blocks": [
{
"type": "paragraph",
"inlines": [{"text": "线下冲突视频首次上传"}],
}
]
},
{
"blocks": [
{
"type": "paragraph",
"inlines": [{"text": "短视频平台 / 私聊转发"}],
}
]
},
]
},
{
"cells": [
{"blocks": [{"type": "paragraph", "inlines": [{"text": "T0+6h"}]}]},
{
"blocks": [
{
"type": "paragraph",
"inlines": [{"text": "登上热搜，出现二次剪辑"}],
}
]
},
{
"blocks": [
{
"type": "paragraph",
"inlines": [{"text": "微博 / 朋友圈"}],
}
]
},
]
},
{
"cells": [
{"blocks": [{"type": "paragraph", "inlines": [{"text": "T0+18h"}]}]},
{
"blocks": [
{
"type": "paragraph",
"inlines": [{"text": "官方回应并发布事实澄清"}],
}
]
},
{
"blocks": [
{
"type": "paragraph",
"inlines": [{"text": "政务号 / 新闻客户端"}],
}
]
},
]
},
{
"cells": [
{"blocks": [{"type": "paragraph", "inlines": [{"text": "T0+24h"}]}]},
{
"blocks": [
{
"type": "paragraph",
"inlines": [{"text": "专家解读，舆论重心转向责任归属"}],
}
]
},
{
"blocks": [
{
"type": "paragraph",
"inlines": [{"text": "视频号直播 / 行业社群"}],
}
]
},
]
},
],
}
blockquote_block = {
"type": "blockquote",
"variant": "accent",
"blocks": [
{
"type": "paragraph",
"inlines": [{"text": "“公众最关心的信息是真相与责任边界。”"}],
},
{
"type": "paragraph",
"inlines": [{"text": "—— 模拟引用，验证引用块样式"}],
},
],
}
engine_quote_block = {
"type": "engineQuote",
"engine": "insight",
"title": ENGINE_AGENT_TITLES["insight"],
"blocks": [
{
"type": "paragraph",
"inlines": [
{
"text": "模型认为 24 小时内保持回应频次，可避免信息真空。",
"marks": [{"type": "bold"}],
}
],
},
{
"type": "paragraph",
"inlines": [
{"text": "建议同时准备简短 FAQ，便于多渠道统一口径。"}
],
},
],
}
swot_block = {
"type": "swotTable",
"title": "舆论场 SWOT 速览",
"summary": "覆盖当前情绪分布、潜在风险与机会。",
"strengths": [
{"title": "官方快速响应", "detail": "首条澄清视频 3 小时内上线"},
{"title": "同城媒体配合", "impact": "高", "score": 8},
],
"weaknesses": [
{"title": "早期谣言存量大", "detail": "相关转发仍占 30%"},
"外部专家尚未统一口径",
],
"opportunities": [
{
"title": "社区共建讨论",
"detail": "自发组织“辟谣志愿者”话题，情绪正向",
},
{"title": "公益合作窗口", "impact": "中"},
],
"threats": [
{"title": "跨平台剪辑继续发酵", "impact": "高", "score": 9},
{"title": "个别自媒体煽动情绪", "evidence": "存在地域标签化倾向"},
],
}
callout_block = {
"type": "callout",
"tone": "warning",
"title": "排版边界提示",
"blocks": [
{
"type": "paragraph",
"inlines": [
{"text": "callout 内部仅放轻量内容，超出部分会自动溢出到外层。"}
],
},
{
"type": "list",
"listType": "bullet",
"items": [
[
{
"type": "paragraph",
"inlines": [{"text": "支持嵌套列表 / 表格 / 数学公式"}],
}
],
[
{
"type": "paragraph",
"inlines": [{"text": "可在这里放置提醒或操作步骤"}],
}
],
],
},
],
}
code_block = {
"type": "code",
"lang": "json",
"caption": "演示代码块",
"content": '{\n "event": "热点示例",\n "topic": "公共事件",\n "status": "monitoring"\n}',
}
math_block = {
"type": "math",
"latex": r"E = mc^2",
"displayMode": True,
}
figure_block = {
"type": "figure",
"img": {
"src": "https://dummyimage.com/600x320/eeeeee/333333&text=Placeholder",
"alt": "占位示意图",
"width": 600,
"height": 320,
},
"caption": "图像外链被替换为友好提示，可验证 figure 占位效果。",
"responsive": True,
}
widget_block = build_widget_block()
chapter_1 = {
"chapterId": "S1",
"title": "封面与目录",
"anchor": "overview",
"order": 10,
"blocks": [
{"type": "heading", "level": 2, "text": "一、封面与目录", "anchor": "overview"},
{
"type": "paragraph",
"inlines": [
{
"text": "模拟社会公共热点事件的摘要，便于快速确认排版与字体效果。",
}
],
},
inline_demo,
{
"type": "kpiGrid",
"items": [
{"label": "24h提及量", "value": "98K", "delta": "+41%", "deltaTone": "up"},
{"label": "正向占比", "value": "32%", "delta": "+5pp", "deltaTone": "up"},
{"label": "负向占比", "value": "18%", "delta": "-3pp", "deltaTone": "down"},
{"label": "高频渠道", "value": "短视频 / 微博"},
],
"cols": 4,
},
{"type": "toc"},
{"type": "hr"},
],
}
chapter_2 = {
"chapterId": "S2",
"title": "块类型演示",
"anchor": "blocks-showcase",
"order": 20,
"blocks": [
{
"type": "heading",
"level": 2,
"text": "二、块类型演示",
"anchor": "blocks-showcase",
},
{
"type": "paragraph",
"inlines": [
{"text": "以下内容逐一覆盖 paragraph/list/table/swot/table/widget 等全部块类型。"}
],
},
{
"type": "heading",
"level": 3,
"text": "2.1 列表与表格",
"anchor": "lists-and-tables",
},
bullet_list,
task_list,
table_block,
{
"type": "heading",
"level": 3,
"text": "2.2 高阶块与富媒体",
"anchor": "advanced-blocks",
},
blockquote_block,
callout_block,
engine_quote_block,
swot_block,
widget_block,
code_block,
math_block,
figure_block,
{
"type": "hr",
"variant": "dashed",
},
{
"type": "paragraph",
"align": "justify",
"inlines": [
{
"text": "本章节的 inline math 兜底验证：",
},
{"text": "p(t)=p_0 e^{\\lambda t}", "marks": [{"type": "math"}]},
{"text": "；以上覆盖所有允许块及标记。"},
],
},
],
}
return [chapter_1, chapter_2]
def validate_chapters(chapters: list[dict]) -> None:
"""使用 IRValidator 校验章节结构，发现错误时抛出异常。"""
validator = IRValidator()
for chapter in chapters:
ok, errors = validator.validate_chapter(chapter)
if not ok:
raise ValueError(f"{chapter.get('chapterId', 'unknown')} 校验失败: {errors}")
def render_and_save(document_ir: dict, timestamp: str) -> tuple[Path, Path, Path]:
"""将 IR 保存为 JSON，并渲染 HTML / PDF，返回三个路径。"""
ir_dir = Path(settings.DOCUMENT_IR_OUTPUT_DIR)
html_dir = Path(settings.OUTPUT_DIR) / "html"
pdf_dir = Path(settings.OUTPUT_DIR) / "pdf"
ir_dir.mkdir(parents=True, exist_ok=True)
html_dir.mkdir(parents=True, exist_ok=True)
pdf_dir.mkdir(parents=True, exist_ok=True)
ir_path = ir_dir / f"report_ir_all_blocks_demo_{timestamp}.json"
ir_path.write_text(json.dumps(document_ir, ensure_ascii=False, indent=2), encoding="utf-8")
html_renderer = HTMLRenderer()
html_content = html_renderer.render(document_ir)
html_path = html_dir / f"report_html_all_blocks_demo_{timestamp}.html"
html_path.write_text(html_content, encoding="utf-8")
pdf_renderer = PDFRenderer()
pdf_path = pdf_dir / f"report_pdf_all_blocks_demo_{timestamp}.pdf"
pdf_renderer.render_to_pdf(document_ir, pdf_path)
return ir_path, html_path, pdf_path
def main() -> int:
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
report_id = f"all-blocks-demo-{timestamp}"
metadata = {
"title": "社会公共热点事件渲染测试",
"subtitle": "覆盖全部 IR 块类型的示例数据",
"query": "公共事件渲染能力自检",
"toc": {"title": "目录", "depth": 3},
"hero": {
"summary": "用于验证 Report Engine 在 HTML / PDF 渲染时对各类区块的兼容性。",
"kpis": [
{"label": "示例块数量", "value": "14", "delta": "+100%", "tone": "up"},
{"label": "图表数", "value": "1", "delta": "安全检查", "tone": "neutral"},
],
"highlights": ["覆盖全部 block", "含行内/块级公式", "Chart.js 数据有效"],
"actions": ["重新生成", "导出 PDF"],
},
}
chapters = build_chapters()
validate_chapters(chapters)
composer = DocumentComposer()
document_ir = composer.build_document(report_id, metadata, chapters)
ir_path, html_path, pdf_path = render_and_save(document_ir, timestamp)
print("✅ 演示 IR 生成完成")
print(f"IR: {ir_path}")
print(f"HTML: {html_path}")
print(f"PDF: {pdf_path}")
return 0
if __name__ == "__main__":
raise SystemExit(main())
... ...