Initial commit to git.yoin
This commit is contained in:
hmo
18
deep-research/README.md
Normal file
18
deep-research/README.md
Normal file
@@ -0,0 +1,18 @@
|
||||
# Deep Research
|
||||
|
||||
深度调研技能，对技术主题进行调研，输出 Markdown + Word 报告。
|
||||
|
||||
## 依赖
|
||||
|
||||
```bash
|
||||
# 必需
|
||||
brew install pandoc
|
||||
pip install python-docx
|
||||
|
||||
# 可选（生成配图）
|
||||
# 需要 image-service skill 并配置 API Key
|
||||
```
|
||||
|
||||
## 使用
|
||||
|
||||
加载 skill 后，直接告诉 Agent 要调研的主题即可。
|
||||
346
deep-research/SKILL.md
Normal file
346
deep-research/SKILL.md
Normal file
@@ -0,0 +1,346 @@
|
||||
---
|
||||
name: deep-research
|
||||
description: 当用户要求"调研"、"深度调研"、"帮我研究"、"调研下这个"，或提到需要搜索、整理、汇总指定主题的技术内容时，应使用此技能。
|
||||
metadata:
|
||||
version: "1.0.0"
|
||||
---
|
||||
|
||||
# 深度调研技能（Deep Research Skill）
|
||||
|
||||
## 技能概述
|
||||
|
||||
此技能用于对技术主题进行深度调研，输出专业的调研报告文档。
|
||||
|
||||
| 能力 | 说明 |
|
||||
|-----|------|
|
||||
| 内容提取 | 从 URL、文档中提取核心信息 |
|
||||
| 深度调研 | 联网搜索补充背景、对比、最新进展 |
|
||||
| 报告生成 | **默认生成 Markdown 和 Word 两个版本** |
|
||||
| 图解生成 | 为核心概念生成技术信息图 |
|
||||
| Word 格式化 | 自动处理目录、标题加粗、表格实线等样式 |
|
||||
|
||||
## 触发规则
|
||||
|
||||
当用户消息包含以下关键词时使用此技能：
|
||||
- 调研、深度调研、调研报告
|
||||
- 帮我研究、帮我分析
|
||||
- 调研下这个、看看这个
|
||||
|
||||
## 输出规范
|
||||
|
||||
每次调研任务必须同时提供：
|
||||
1. **Markdown 版本**：用于 Obsidian 知识库沉淀和双链关联
|
||||
2. **Word 版本**：用于正式汇报和外部分享，需经过脚本格式化处理
|
||||
|
||||
## 目录结构
|
||||
|
||||
每个调研主题创建独立文件夹，保持整洁：
|
||||
|
||||
```
|
||||
{output_dir}/
|
||||
├── Ralph-Loop/ # 主题文件夹（英文短横线命名）
|
||||
│ ├── images/ # 该主题的信息图
|
||||
│ │ ├── architecture.png
|
||||
│ │ └── comparison.png
|
||||
│ ├── Ralph-Loop调研报告.md # Markdown 报告
|
||||
│ └── Ralph-Loop调研报告.docx # Word 报告
|
||||
├── MCP-Protocol/
|
||||
│ ├── images/
|
||||
│ ├── MCP-Protocol调研报告.md
|
||||
│ └── MCP-Protocol调研报告.docx
|
||||
└── ...
|
||||
```
|
||||
|
||||
命名规范：
|
||||
- 文件夹名：英文，单词间用短横线连接，如 `Ralph-Loop`、`MCP-Protocol`
|
||||
- 报告文件：`{主题名}调研报告.md` 和 `{主题名}调研报告.docx`
|
||||
- 图片目录：每个主题文件夹下单独的 `images/` 目录
|
||||
|
||||
## 调研流程
|
||||
|
||||
### 第一步：创建主题目录
|
||||
|
||||
根据调研主题创建独立文件夹：
|
||||
|
||||
```bash
|
||||
mkdir -p "{output_dir}/{主题名}/images"
|
||||
```
|
||||
|
||||
### 第二步：内容获取
|
||||
|
||||
1. 如果用户提供 URL，使用 webfetch 获取内容
|
||||
2. 提炼核心概念、技术原理、关键信息
|
||||
3. 识别需要深入调研的点
|
||||
|
||||
### 第三步：深度调研
|
||||
|
||||
使用 Task 工具进行联网搜索，补充：
|
||||
- 技术背景和发展历程
|
||||
- 竞品对比和差异化
|
||||
- 社区讨论和实际案例
|
||||
- GitHub 仓库和开源实现
|
||||
- 最新进展和趋势
|
||||
|
||||
### 第四步：图解生成
|
||||
|
||||
使用预设风格脚本生成统一手绘风格的信息图。
|
||||
|
||||
#### 生图触发规则
|
||||
|
||||
| 内容类型 | 是否生图 | 图解类型 | 说明 |
|
||||
|---------|---------|---------|------|
|
||||
| 核心架构/原理 | 必须 | arch | 系统结构、技术栈、模块组成 |
|
||||
| 流程/步骤 | 必须 | flow | 工作流、执行顺序、操作步骤 |
|
||||
| A vs B 对比 | 必须 | compare | 两种方案/技术的对比 |
|
||||
| 3个以上要素 | 建议 | concept | 核心概念、多个方面组成 |
|
||||
| 纯文字表格 | 不需要 | - | 用 Markdown 表格即可 |
|
||||
| 代码示例 | 不需要 | - | 用代码块即可 |
|
||||
|
||||
#### 预设风格模板
|
||||
|
||||
所有配图统一使用手绘体可视化风格，保持系列一致性：
|
||||
|
||||
| 类型 | 命令参数 | 配色 | 布局 |
|
||||
|------|---------|------|------|
|
||||
| 架构图 | `-t arch` | 科技蓝 #4A90D9 | 分层/模块化 |
|
||||
| 流程图 | `-t flow` | 蓝+绿+橙 | 从上到下 |
|
||||
| 对比图 | `-t compare` | 蓝 vs 橙 | 左右分栏 |
|
||||
| 概念图 | `-t concept` | 蓝紫渐变 | 中心发散 |
|
||||
|
||||
#### 生成命令
|
||||
|
||||
使用 `research_image.py` 脚本生成：
|
||||
|
||||
```bash
|
||||
# 架构图
|
||||
python .opencode/skills/image-service/scripts/research_image.py \
|
||||
-t arch \
|
||||
-n "Ralph Loop 核心架构" \
|
||||
-c "展示 Prompt、Agent、Stop Hook、Files 四个模块的循环关系" \
|
||||
-o "{output_dir}/{主题名}/images/architecture.png"
|
||||
|
||||
# 流程图
|
||||
python .opencode/skills/image-service/scripts/research_image.py \
|
||||
-t flow \
|
||||
-n "Stop Hook 工作流程" \
|
||||
-c "Agent尝试退出、Hook触发、检查条件、允许或阻止退出的完整流程" \
|
||||
-o "{output_dir}/{主题名}/images/flow.png"
|
||||
|
||||
# 对比图
|
||||
python .opencode/skills/image-service/scripts/research_image.py \
|
||||
-t compare \
|
||||
-n "ReAct vs Ralph Loop" \
|
||||
-c "左侧ReAct依赖自我评估停止，右侧Ralph使用外部Hook控制" \
|
||||
-o "{output_dir}/{主题名}/images/comparison.png"
|
||||
|
||||
# 概念图
|
||||
python .opencode/skills/image-service/scripts/research_image.py \
|
||||
-t concept \
|
||||
-n "状态持久化要素" \
|
||||
-c "中心是Agent，周围是progress.txt、prd.json、Git历史、代码文件" \
|
||||
-o "{output_dir}/{主题名}/images/concept.png"
|
||||
```
|
||||
|
||||
#### 图片命名规范
|
||||
|
||||
| 图解类型 | 文件名 |
|
||||
|---------|--------|
|
||||
| 架构图 | `architecture.png` 或 `{具体名称}_arch.png` |
|
||||
| 流程图 | `flow.png` 或 `{具体名称}_flow.png` |
|
||||
| 对比图 | `comparison.png` 或 `{A}_vs_{B}.png` |
|
||||
| 概念图 | `concept.png` 或 `{具体名称}_concept.png` |
|
||||
|
||||
### 第五步：报告撰写
|
||||
|
||||
按标准模板撰写 Markdown 报告，存放到主题文件夹：
|
||||
|
||||
```
|
||||
{output_dir}/{主题名}/{主题名}调研报告.md
|
||||
```
|
||||
|
||||
报告中引用图片使用相对路径：
|
||||
```markdown
|
||||
|
||||
```
|
||||
|
||||
### 第六步：Word 导出
|
||||
|
||||
```bash
|
||||
# 进入主题目录
|
||||
cd "{output_dir}/{主题名}"
|
||||
|
||||
# 生成 Word（--resource-path=. 确保图片正确引用）
|
||||
# 注意：不要使用 --toc 参数，因为 Markdown 中已有手写目录
|
||||
pandoc "{主题名}调研报告.md" -o "{主题名}调研报告.docx" --resource-path=.
|
||||
|
||||
# 格式化 Word
|
||||
python ../../../.opencode/skills/deep-research/scripts/format_docx.py "{主题名}调研报告.docx"
|
||||
```
|
||||
|
||||
## 写作原则
|
||||
|
||||
调研报告的核心价值：深入研究、降低团队吸收成本、提供专家级建议。
|
||||
|
||||
1. 理解透彻：不能一知半解或大段拷贝，必须消化吸收后用自己的话表达
|
||||
2. 体现思考：有判断、有建议，而非仅仅陈述现状
|
||||
3. 细节佐证：有过程和细节支撑结论，不空谈
|
||||
4. 逻辑清晰：有分段、有结构、有编号
|
||||
5. 配图说明：核心概念必须配信息图
|
||||
6. 去除 AI 味：
|
||||
- 不使用「」、" " 等特殊符号
|
||||
- 不用过多强调符号和 emoji
|
||||
- 行文自然流畅，像人写的专业文档
|
||||
- 避免"首先、其次、总之"等套话
|
||||
|
||||
## 报告模板
|
||||
|
||||
```markdown
|
||||
---
|
||||
date: YYYY-MM-DD
|
||||
type: 调研报告
|
||||
领域: {技术领域}
|
||||
tags: [调研, {主题关键词}]
|
||||
---
|
||||
|
||||
# XX调研报告
|
||||
|
||||
> 调研日期：YYYY年M月D日
|
||||
|
||||
---
|
||||
|
||||
## 目录
|
||||
|
||||
- 一、简介
|
||||
- 二、启示
|
||||
- 三、核心介绍
|
||||
- 3.1 XXX
|
||||
- 3.2 XXX
|
||||
- 四、附录
|
||||
- 4.1 详细文档
|
||||
- 4.2 参考资料
|
||||
|
||||
---
|
||||
|
||||
## 一、简介
|
||||
|
||||
（快速说明调研内容，简短重点）
|
||||
|
||||
是什么，主要用来做什么，属于什么类别。有哪些能力，有什么特点。和竞品相比，有哪些区别，主打什么。
|
||||
|
||||
1. 要点一
|
||||
2. 要点二
|
||||
3. 要点三
|
||||
|
||||
---
|
||||
|
||||
## 二、启示
|
||||
|
||||
（调研内容带来的启示、值得学习借鉴之处、与现有产品如何结合、是否值得推荐）
|
||||
|
||||
1. 启示一
|
||||
2. 启示二
|
||||
3. 启示三
|
||||
|
||||
---
|
||||
|
||||
## 三、核心介绍
|
||||
|
||||
（正文部分，详细说明调研内容的原理/搭建/操作/使用过程，含信息图及流程说明）
|
||||
|
||||
### 3.1 XXX
|
||||
|
||||
|
||||
|
||||
上图展示了...（图解说明，让读者看图就能理解）
|
||||
|
||||
详细内容...
|
||||
|
||||
### 3.2 XXX
|
||||
|
||||
详细内容...
|
||||
|
||||
---
|
||||
|
||||
## 四、附录
|
||||
|
||||
### 4.1 详细文档
|
||||
|
||||
（更详细的配置/操作过程）
|
||||
|
||||
### 4.2 参考资料
|
||||
|
||||
**官方文档**
|
||||
|
||||
- 文档名称: https://xxx
|
||||
|
||||
**开源实现**
|
||||
|
||||
- 项目名称: https://github.com/xxx
|
||||
|
||||
**社区讨论**
|
||||
|
||||
- 讨论来源: https://xxx
|
||||
```
|
||||
|
||||
## 脚本说明
|
||||
|
||||
### format_docx.py
|
||||
|
||||
Word 文档格式化脚本，功能包括：
|
||||
|
||||
1. 标题居中，黑色字体（去除 pandoc 默认蓝色）
|
||||
2. "Table of Contents" 替换为中文"目录"
|
||||
3. 目录页单独一页
|
||||
4. 一级标题（简介、启示等）前自动分页
|
||||
5. 表格保持完整不跨页断开
|
||||
6. 代码块保持完整不断开
|
||||
7. 日期行居中
|
||||
|
||||
用法：
|
||||
```bash
|
||||
python .opencode/skills/deep-research/scripts/format_docx.py "输入.docx" ["输出.docx"]
|
||||
```
|
||||
|
||||
## 完整调研示例
|
||||
|
||||
用户输入：
|
||||
> 调研下 Ralph Loop
|
||||
|
||||
执行流程：
|
||||
|
||||
```bash
|
||||
# 1. 创建主题目录
|
||||
mkdir -p "{output_dir}/Ralph-Loop/images"
|
||||
|
||||
# 2. 获取内容（如有 URL）
|
||||
webfetch https://example.com/article
|
||||
|
||||
# 3. 深度调研（使用 Task 工具联网搜索）
|
||||
|
||||
# 4. 生成信息图
|
||||
python .opencode/skills/image-service/scripts/text_to_image.py "技术架构图..." --output "{output_dir}/Ralph-Loop/images/architecture.png"
|
||||
|
||||
# 5. 撰写报告
|
||||
# 写入 {output_dir}/Ralph-Loop/Ralph-Loop调研报告.md
|
||||
|
||||
# 6. 导出 Word（不使用 --toc，Markdown 已有手写目录）
|
||||
cd "{output_dir}/Ralph-Loop"
|
||||
pandoc "Ralph-Loop调研报告.md" -o "Ralph-Loop调研报告.docx" --resource-path=.
|
||||
python ../../../.opencode/skills/deep-research/scripts/format_docx.py "Ralph-Loop调研报告.docx"
|
||||
```
|
||||
|
||||
输出文件：
|
||||
```
|
||||
{output_dir}/Ralph-Loop/
|
||||
├── images/
|
||||
│ ├── architecture.png
|
||||
│ └── comparison.png
|
||||
├── Ralph-Loop调研报告.md
|
||||
└── Ralph-Loop调研报告.docx
|
||||
```
|
||||
|
||||
## 依赖
|
||||
|
||||
- pandoc：Markdown 转 Word
|
||||
- python-docx：Word 格式化
|
||||
- image-service 技能：生成信息图
|
||||
332
deep-research/scripts/format_docx.py
Normal file
332
deep-research/scripts/format_docx.py
Normal file
@@ -0,0 +1,332 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Word 文档格式化脚本
|
||||
|
||||
Author: 翟星人
|
||||
|
||||
功能：
|
||||
1. 标题居中，黑色字体，加粗
|
||||
2. 在日期后插入目录
|
||||
3. 一级标题前分页
|
||||
4. 表格实线边框，不跨页断开
|
||||
5. 日期居中
|
||||
6. 图片说明小字居中
|
||||
7. 1/2/3级标题加粗
|
||||
8. 附录参考文献左对齐
|
||||
"""
|
||||
|
||||
import sys
|
||||
import re
|
||||
from docx import Document
|
||||
from docx.shared import Pt, RGBColor
|
||||
from docx.enum.text import WD_ALIGN_PARAGRAPH
|
||||
from docx.enum.table import WD_TABLE_ALIGNMENT
|
||||
from docx.oxml.ns import qn
|
||||
from docx.oxml import OxmlElement
|
||||
|
||||
|
||||
def add_page_break_before(paragraph):
|
||||
"""在段落前添加分页符"""
|
||||
p = paragraph._p
|
||||
pPr = p.get_or_add_pPr()
|
||||
pageBreakBefore = OxmlElement('w:pageBreakBefore')
|
||||
pPr.insert(0, pageBreakBefore)
|
||||
|
||||
|
||||
def set_table_border(table):
|
||||
"""设置表格实线边框"""
|
||||
tbl = table._tbl
|
||||
tblPr = tbl.tblPr if tbl.tblPr is not None else OxmlElement('w:tblPr')
|
||||
|
||||
tblBorders = OxmlElement('w:tblBorders')
|
||||
|
||||
for border_name in ['top', 'left', 'bottom', 'right', 'insideH', 'insideV']:
|
||||
border = OxmlElement(f'w:{border_name}')
|
||||
border.set(qn('w:val'), 'single')
|
||||
border.set(qn('w:sz'), '4')
|
||||
border.set(qn('w:space'), '0')
|
||||
border.set(qn('w:color'), '000000')
|
||||
tblBorders.append(border)
|
||||
|
||||
tblPr.append(tblBorders)
|
||||
if tbl.tblPr is None:
|
||||
tbl.insert(0, tblPr)
|
||||
|
||||
|
||||
def keep_table_together(table):
|
||||
"""保持表格不跨页断开"""
|
||||
for row in table.rows:
|
||||
for cell in row.cells:
|
||||
for paragraph in cell.paragraphs:
|
||||
pPr = paragraph._p.get_or_add_pPr()
|
||||
keepNext = OxmlElement('w:keepNext')
|
||||
keepLines = OxmlElement('w:keepLines')
|
||||
pPr.append(keepNext)
|
||||
pPr.append(keepLines)
|
||||
|
||||
|
||||
def keep_paragraph_together(paragraph):
|
||||
"""保持段落不断开"""
|
||||
pPr = paragraph._p.get_or_add_pPr()
|
||||
keepNext = OxmlElement('w:keepNext')
|
||||
keepLines = OxmlElement('w:keepLines')
|
||||
pPr.append(keepNext)
|
||||
pPr.append(keepLines)
|
||||
|
||||
|
||||
def set_heading_style(paragraph, level=1):
|
||||
"""设置标题样式：黑色加粗"""
|
||||
for run in paragraph.runs:
|
||||
run.font.color.rgb = RGBColor(0, 0, 0)
|
||||
run.font.bold = True
|
||||
if level == 1:
|
||||
run.font.size = Pt(16)
|
||||
elif level == 2:
|
||||
run.font.size = Pt(14)
|
||||
elif level == 3:
|
||||
run.font.size = Pt(12)
|
||||
|
||||
|
||||
def set_caption_style(paragraph):
|
||||
"""设置图片说明样式：小字居中"""
|
||||
paragraph.alignment = WD_ALIGN_PARAGRAPH.CENTER
|
||||
for run in paragraph.runs:
|
||||
run.font.size = Pt(9)
|
||||
run.font.color.rgb = RGBColor(80, 80, 80)
|
||||
|
||||
|
||||
def is_image_caption(text, prev_has_image):
|
||||
"""判断是否为图片说明"""
|
||||
if prev_has_image and text and len(text) < 100:
|
||||
# 必须以特定词开头才算图片说明
|
||||
if text.startswith("上图") or text.startswith("图：") or text.startswith("图:"):
|
||||
return True
|
||||
return False
|
||||
|
||||
|
||||
def paragraph_has_image(paragraph):
|
||||
"""检查段落是否包含图片"""
|
||||
for run in paragraph.runs:
|
||||
if run._element.xpath('.//w:drawing') or run._element.xpath('.//w:pict'):
|
||||
return True
|
||||
return False
|
||||
|
||||
|
||||
def is_horizontal_rule(paragraph):
|
||||
"""检查是否为分割线（文本或绘图元素）"""
|
||||
text = paragraph.text.strip()
|
||||
# 检查文本形式的分割线
|
||||
if text == "---" or text == "***" or text == "___" or (len(text) > 0 and all(c == '-' for c in text)):
|
||||
return True
|
||||
# 检查 pandoc 生成的绘图形式水平线（包含 line 或 rect 且文本为空，但不包含图片）
|
||||
if text == "":
|
||||
xml_str = paragraph._p.xml
|
||||
has_drawing = 'w:pict' in xml_str or 'w:drawing' in xml_str
|
||||
has_line = 'v:line' in xml_str or 'v:rect' in xml_str or '<a:ln' in xml_str
|
||||
has_image = 'a:blip' in xml_str or 'v:imagedata' in xml_str or 'r:embed' in xml_str
|
||||
# 只有有绘图、有线条、但没有图片时才是水平线
|
||||
if has_drawing and has_line and not has_image:
|
||||
return True
|
||||
return False
|
||||
|
||||
|
||||
def is_reference_item(text):
|
||||
"""判断是否为参考文献条目"""
|
||||
if re.match(r'^\d+\.\s', text):
|
||||
if 'http' in text or '地址' in text or 'github' in text.lower():
|
||||
return True
|
||||
return False
|
||||
|
||||
|
||||
def add_toc_field(paragraph):
|
||||
"""向段落添加目录域"""
|
||||
run = paragraph.add_run()
|
||||
fldChar1 = OxmlElement('w:fldChar')
|
||||
fldChar1.set(qn('w:fldCharType'), 'begin')
|
||||
|
||||
instrText = OxmlElement('w:instrText')
|
||||
instrText.set(qn('xml:space'), 'preserve')
|
||||
instrText.text = ' TOC \\o "1-3" \\h \\z \\u '
|
||||
|
||||
fldChar2 = OxmlElement('w:fldChar')
|
||||
fldChar2.set(qn('w:fldCharType'), 'separate')
|
||||
|
||||
fldChar3 = OxmlElement('w:fldChar')
|
||||
fldChar3.set(qn('w:fldCharType'), 'end')
|
||||
|
||||
run._r.append(fldChar1)
|
||||
run._r.append(instrText)
|
||||
run._r.append(fldChar2)
|
||||
run.add_text('请右键点击此处，选择"更新域"以生成目录')
|
||||
run._r.append(fldChar3)
|
||||
|
||||
|
||||
def format_docx(input_path, output_path=None):
|
||||
"""格式化 Word 文档"""
|
||||
if output_path is None:
|
||||
output_path = input_path
|
||||
|
||||
doc = Document(input_path)
|
||||
|
||||
is_first_heading1 = True
|
||||
prev_was_code = False
|
||||
prev_has_image = False
|
||||
in_appendix = False
|
||||
date_para_index = -1
|
||||
paragraphs_to_remove = []
|
||||
|
||||
# 第一遍遍历：找到并删除 Markdown 转换来的手写目录章节
|
||||
# 查找 Heading 2 样式且文本为"目录"的段落，以及紧随其后的 Compact 列表
|
||||
in_md_toc_section = False
|
||||
for i, paragraph in enumerate(doc.paragraphs):
|
||||
text = paragraph.text.strip()
|
||||
style_name = paragraph.style.name if paragraph.style else ""
|
||||
if style_name is None:
|
||||
style_name = ""
|
||||
|
||||
# 检测 Markdown 转换来的目录章节开始（Heading 2 且文本为"目录"）
|
||||
if "Heading 2" in style_name and text == "目录":
|
||||
in_md_toc_section = True
|
||||
paragraphs_to_remove.append(paragraph)
|
||||
continue
|
||||
|
||||
# 在目录章节内，删除 Compact 样式的列表项（手写目录内容）
|
||||
if in_md_toc_section:
|
||||
# 遇到 Heading 样式，说明目录章节结束
|
||||
if "Heading" in style_name:
|
||||
in_md_toc_section = False
|
||||
continue
|
||||
# 删除 Compact 样式的列表项
|
||||
if style_name == "Compact":
|
||||
paragraphs_to_remove.append(paragraph)
|
||||
continue
|
||||
elif text == "":
|
||||
paragraphs_to_remove.append(paragraph)
|
||||
continue
|
||||
else:
|
||||
in_md_toc_section = False
|
||||
|
||||
# 第二遍遍历：处理样式、标记删除分割线
|
||||
for i, paragraph in enumerate(doc.paragraphs):
|
||||
text = paragraph.text.strip()
|
||||
style_name = paragraph.style.name if paragraph.style else ""
|
||||
if style_name is None:
|
||||
style_name = ""
|
||||
|
||||
# 跳过已标记删除的段落
|
||||
if paragraph in paragraphs_to_remove:
|
||||
continue
|
||||
|
||||
# 删除分割线段落
|
||||
if is_horizontal_rule(paragraph):
|
||||
paragraphs_to_remove.append(paragraph)
|
||||
continue
|
||||
|
||||
# 检查是否进入附录部分
|
||||
if text.startswith("四、") or "附录" in text:
|
||||
in_appendix = True
|
||||
|
||||
# 附录中的参考文献左对齐
|
||||
if in_appendix and is_reference_item(text):
|
||||
paragraph.alignment = WD_ALIGN_PARAGRAPH.LEFT
|
||||
continue
|
||||
|
||||
# 检查当前段落是否有图片
|
||||
current_has_image = paragraph_has_image(paragraph)
|
||||
|
||||
# 处理图片说明
|
||||
if is_image_caption(text, prev_has_image):
|
||||
set_caption_style(paragraph)
|
||||
prev_has_image = False
|
||||
continue
|
||||
|
||||
prev_has_image = current_has_image
|
||||
|
||||
# First Paragraph 样式应该左对齐（pandoc 可能设置为居中）
|
||||
if style_name == "First Paragraph":
|
||||
paragraph.alignment = WD_ALIGN_PARAGRAPH.LEFT
|
||||
|
||||
# 处理文档标题
|
||||
if "Heading 1" in style_name or style_name == "Title":
|
||||
set_heading_style(paragraph, 1)
|
||||
|
||||
if is_first_heading1 and ("调研报告" in text or i < 3):
|
||||
paragraph.alignment = WD_ALIGN_PARAGRAPH.CENTER
|
||||
is_first_heading1 = False
|
||||
continue
|
||||
|
||||
# 处理日期行，记录索引
|
||||
if "调研日期" in text or re.match(r'.*\d{4}年\d{1,2}月\d{1,2}日.*', text):
|
||||
paragraph.alignment = WD_ALIGN_PARAGRAPH.CENTER
|
||||
date_para_index = i
|
||||
|
||||
# 一级标题前分页
|
||||
if "Heading" in style_name:
|
||||
if text.startswith("一、") or text.startswith("二、") or text.startswith("三、") or text.startswith("四、"):
|
||||
add_page_break_before(paragraph)
|
||||
set_heading_style(paragraph, 1)
|
||||
elif "Heading 2" in style_name:
|
||||
set_heading_style(paragraph, 2)
|
||||
elif "Heading 3" in style_name:
|
||||
set_heading_style(paragraph, 3)
|
||||
else:
|
||||
set_heading_style(paragraph, 1)
|
||||
|
||||
# 代码块保持不断开
|
||||
if paragraph.style and "Code" in str(paragraph.style.name):
|
||||
keep_paragraph_together(paragraph)
|
||||
prev_was_code = True
|
||||
elif prev_was_code and text.startswith("```"):
|
||||
prev_was_code = False
|
||||
|
||||
# 删除待删除的段落
|
||||
for paragraph in paragraphs_to_remove:
|
||||
p = paragraph._element
|
||||
parent = p.getparent()
|
||||
if parent is not None:
|
||||
parent.remove(p)
|
||||
|
||||
# 插入 Word 目录域：在日期行后面插入
|
||||
target_para = None
|
||||
for i, p in enumerate(doc.paragraphs):
|
||||
if "调研日期" in p.text or re.match(r'.*\d{4}年\d{1,2}月\d{1,2}日.*', p.text):
|
||||
if i + 1 < len(doc.paragraphs):
|
||||
target_para = doc.paragraphs[i+1]
|
||||
break
|
||||
|
||||
if target_para:
|
||||
toc_title = target_para.insert_paragraph_before("目录")
|
||||
toc_title.alignment = WD_ALIGN_PARAGRAPH.CENTER
|
||||
for run in toc_title.runs:
|
||||
run.font.bold = True
|
||||
run.font.size = Pt(16)
|
||||
run.font.color.rgb = RGBColor(0, 0, 0)
|
||||
|
||||
toc_field = target_para.insert_paragraph_before()
|
||||
add_toc_field(toc_field)
|
||||
|
||||
# 最后一遍：修复 First Paragraph 样式的对齐（pandoc 默认居中）
|
||||
for paragraph in doc.paragraphs:
|
||||
style_name = paragraph.style.name if paragraph.style else ""
|
||||
if style_name == "First Paragraph":
|
||||
paragraph.alignment = WD_ALIGN_PARAGRAPH.LEFT
|
||||
|
||||
# 处理表格
|
||||
for table in doc.tables:
|
||||
set_table_border(table)
|
||||
keep_table_together(table)
|
||||
table.alignment = WD_TABLE_ALIGNMENT.CENTER
|
||||
|
||||
doc.save(output_path)
|
||||
print(f"格式化完成: {output_path}")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
if len(sys.argv) < 2:
|
||||
print("用法: python format_docx.py <input.docx> [output.docx]")
|
||||
sys.exit(1)
|
||||
|
||||
input_file = sys.argv[1]
|
||||
output_file = sys.argv[2] if len(sys.argv) > 2 else None
|
||||
|
||||
format_docx(input_file, output_file)
|
||||
Reference in New Issue
Block a user