learning/add_toc.py at main · BobWang21/learning · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
import os
import re
from pathlib import Path

# 获取所有 HTML 文件
html_files = list(Path('/Users/wangbao/ts-model-survey').glob('*.html'))

# 排除首页
html_files = [f for f in html_files if f.name != 'index.html']

def extract_headings(content):
    """提取页面中的 h2 和 h3 标题"""
    headings = []

    # 查找所有 h2 和 h3 标签
    h2_pattern = r'<h2[^>]*>([^<]+)</h2>'
    h3_pattern = r'<h3[^>]*>([^<]+)</h3>'

    h2_matches = re.finditer(h2_pattern, content)
    h3_matches = re.finditer(h3_pattern, content)

    # 收集所有标题及其位置
    all_headings = []

    for match in h2_matches:
        all_headings.append({
            'level': 2,
            'text': match.group(1).strip(),
            'pos': match.start()
        })

    for match in h3_matches:
        all_headings.append({
            'level': 3,
            'text': match.group(1).strip(),
            'pos': match.start()
        })

    # 按位置排序
    all_headings.sort(key=lambda x: x['pos'])

    return all_headings

def generate_toc_html(headings):
    """生成目录 HTML"""
    if not headings:
        return ""

    toc_html = '<div class="page-toc">\n'
    toc_html += '    <h4 style="margin-bottom: 15px; color: #1e40af;">📑 页面目录</h4>\n'
    toc_html += '    <ul style="list-style: none; padding: 0; margin: 0;">\n'

    for heading in headings:
        # 生成 ID
        heading_id = re.sub(r'[^\w\s-]', '', heading['text']).strip().replace(' ', '-').lower()
        heading_id = re.sub(r'-+', '-', heading_id)

        if heading['level'] == 2:
            toc_html += f'        <li style="margin-bottom: 8px;"><a href="#{heading_id}" style="color: #1e40af; text-decoration: none; font-weight: 600;">{heading["text"]}</a></li>\n'
        else:  # h3
            toc_html += f'        <li style="margin-left: 20px; margin-bottom: 6px;"><a href="#{heading_id}" style="color: #7c3aed; text-decoration: none; font-size: 0.95rem;">{heading["text"]}</a></li>\n'

    toc_html += '    </ul>\n'
    toc_html += '</div>\n'

    return toc_html

def add_ids_to_headings(content, headings):
    """为标题添加 ID"""
    for heading in headings:
        heading_id = re.sub(r'[^\w\s-]', '', heading['text']).strip().replace(' ', '-').lower()
        heading_id = re.sub(r'-+', '-', heading_id)

        # 替换标题标签，添加 ID
        if heading['level'] == 2:
            old_tag = f"<h2{heading['text']}</h2>"
            new_tag = f'<h2 id="{heading_id}">{heading["text"]}</h2>'
        else:
            old_tag = f"<h3{heading['text']}</h3>"
            new_tag = f'<h3 id="{heading_id}">{heading["text"]}</h3>'

        # 更精确的替换
        pattern = f"<h{heading['level']}[^>]*>{re.escape(heading['text'])}</h{heading['level']}>"
        content = re.sub(pattern, new_tag, content)

    return content

# 处理每个文件
for html_file in sorted(html_files):
    print(f"处理: {html_file.name}")

    with open(html_file, 'r', encoding='utf-8') as f:
        content = f.read()

    # 检查是否已经有 TOC
    if 'page-toc' in content:
        print(f"  跳过: 已存在 TOC")
        continue

    # 提取标题
    headings = extract_headings(content)

    if not headings:
        print(f"  跳过: 没有找到标题")
        continue

    # 生成目录 HTML
    toc_html = generate_toc_html(headings)

    # 添加 ID 到标题
    content = add_ids_to_headings(content, headings)

    # 在 breadcrumb 后面插入目录
    # 查找 </div> 后的位置（breadcrumb 的结束）
    breadcrumb_pattern = r'(<div class="breadcrumb">.*?</div>)'
    match = re.search(breadcrumb_pattern, content, re.DOTALL)

    if match:
        insert_pos = match.end()
        content = content[:insert_pos] + '\n\n            ' + toc_html + content[insert_pos:]
    else:
        # 如果没有 breadcrumb，在 section 开始后插入
        section_pattern = r'(<section class="ai-section">)'
        match = re.search(section_pattern, content)
        if match:
            insert_pos = match.end()
            content = content[:insert_pos] + '\n\n            ' + toc_html + content[insert_pos:]

    # 写回文件
    with open(html_file, 'w', encoding='utf-8') as f:
        f.write(content)

    print(f"  成功: 添加了 {len(headings)} 个标题的目录")

print("\n完成！")