6

我正在尝试根据其<h2>和<h3>标签从 HTML 块（不是完整的文件 - 只是内容）生成目录。

到目前为止，我的计划是：

  • 使用提取标题列表beautifulsoup

  • 在内容上使用正则表达式在标题标签之前/内部放置锚链接（以便用户可以单击目录） - 可能有一种替换 inside 的方法beautifulsoup？

  • 在预定义的位置输出指向标题的嵌套链接列表。

当我这样说的时候，这听起来很容易，但事实证明这让我有点痛苦。

有没有什么东西可以一次性为我完成所有这些，所以我不会浪费接下来的几个小时重新发明轮子？

一个例子：

<p>This is an introduction</p>

<h2>This is a sub-header</h2>
<p>...</p>

<h3>This is a sub-sub-header</h3>
<p>...</p>

<h2>This is a sub-header</h2>
<p>...</p>
4

4 回答 4

3

一些快速破解了丑陋的代码：

soup = BeautifulSoup(html)

toc = []
header_id = 1
current_list = toc
previous_tag = None

for header in soup.findAll(['h2', 'h3']):
    header['id'] = header_id

    if previous_tag == 'h2' and header.name == 'h3':
        current_list = []
    elif previous_tag == 'h3' and header.name == 'h2':
        toc.append(current_list)
        current_list = toc

    current_list.append((header_id, header.string))

    header_id += 1
    previous_tag = header.name

if current_list != toc:
    toc.append(current_list)


def list_to_html(lst):
    result = ["<ul>"]
    for item in lst:
        if isinstance(item, list):
            result.append(list_to_html(item))
        else:
            result.append('<li><a href="#%s">%s</a></li>' % item)
    result.append("</ul>")
    return "\n".join(result)

# Table of contents
print list_to_html(toc)

# Modified HTML
print soup
于 2010-03-25T12:43:38.387 回答
1

使用lxml.html.

于 2010-03-25T11:24:09.353 回答
0

如何在 Python 中为 HTML 文本生成目录？

但我认为你走在正确的轨道上，重新发明轮子会很有趣。

于 2010-03-25T12:08:52.007 回答
0

我附带了 Łukasz 提出的解决方案的扩展版本。

def list_to_html(lst):
    result = ["<ul>"]
    for item in lst:
        if isinstance(item, list):
            result.append(list_to_html(item))
        else:
            result.append('<li><a href="#{}">{}</a></li>'.format(item[0], item[1]))
    result.append("</ul>")
    return "\n".join(result)

soup = BeautifulSoup(article, 'html5lib')

toc = []
h2_prev = 0
h3_prev = 0
h4_prev = 0
h5_prev = 0

for header in soup.findAll(['h2', 'h3', 'h4', 'h5', 'h6']):
    data = [(slugify(header.string), header.string)]

    if header.name == "h2":
        toc.append(data)
        h3_prev = 0
        h4_prev = 0
        h5_prev = 0
        h2_prev = len(toc) - 1
    elif header.name == "h3":
        toc[int(h2_prev)].append(data)
        h3_prev = len(toc[int(h2_prev)]) - 1
    elif header.name == "h4":
        toc[int(h2_prev)][int(h3_prev)].append(data)
        h4_prev = len(toc[int(h2_prev)][int(h3_prev)]) - 1
    elif header.name == "h5":
        toc[int(h2_prev)][int(h3_prev)][int(h4_prev)].append(data)
        h5_prev = len(toc[int(h2_prev)][int(h3_prev)][int(h4_prev)]) - 1
    elif header.name == "h6":
        toc[int(h2_prev)][int(h3_prev)][int(h4_prev)][int(h5_prev)].append(data)

toc_html = list_to_html(toc)
于 2018-07-30T15:14:25.567 回答