Don't add toc if there are no headers besides title.
This commit is contained in:
parent
546b2f0424
commit
f0ca7f7382
1 changed files with 3 additions and 1 deletions
|
@ -195,6 +195,9 @@ def add_toc(soup, max_level=None):
|
||||||
raise ValueError('max_level must be >= 1.')
|
raise ValueError('max_level must be >= 1.')
|
||||||
|
|
||||||
header_pattern = re.compile(rf'^h[1-{max_level}]$')
|
header_pattern = re.compile(rf'^h[1-{max_level}]$')
|
||||||
|
headers = soup.find_all(header_pattern)
|
||||||
|
if headers == [first_h1]:
|
||||||
|
return
|
||||||
|
|
||||||
toc = new_list(root=True)
|
toc = new_list(root=True)
|
||||||
toc.ol['id'] = 'table_of_contents'
|
toc.ol['id'] = 'table_of_contents'
|
||||||
|
@ -202,7 +205,6 @@ def add_toc(soup, max_level=None):
|
||||||
current_list = toc.ol
|
current_list = toc.ol
|
||||||
current_list['level'] = None
|
current_list['level'] = None
|
||||||
|
|
||||||
headers = soup.find_all(header_pattern)
|
|
||||||
for header in headers:
|
for header in headers:
|
||||||
if header == first_h1:
|
if header == first_h1:
|
||||||
continue
|
continue
|
||||||
|
|
Loading…
Reference in a new issue