voussoir/epubfile
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Use enumerate instead of manual counting, now that headers is exact.

Browse source
This commit is contained in:
Ethan Dalool 2020-02-03 23:11:22 -08:00
parent 3cfb8030a7
commit 81fe7cd47b
1 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
epubfile.py
View file

@ -1027,7 +1027,6 @@ class Epub:
toc = new_list(root=True) toc = new_list(root=True)
current_level = None current_level = None
current_list = toc.ol current_list = toc.ol
toc_line_index = 1
spine = self.get_spine_order(linear_only=linear_only) spine = self.get_spine_order(linear_only=linear_only)
spine = [s for s in spine if s != nav_id] spine = [s for s in spine if s != nav_id]
@ -1036,12 +1035,12 @@ class Epub:
file_path = self.get_filepath(file_id) file_path = self.get_filepath(file_id)
soup = self.read_file(file_id, soup=True) soup = self.read_file(file_id, soup=True)
for header in soup.find_all(header_pattern): headers = soup.find_all(header_pattern)
for (toc_line_index, header) in enumerate(headers, start=1):
# 'hX' -> X # 'hX' -> X
level = int(header.name[1]) level = int(header.name[1])
header['id'] = f'toc_{toc_line_index}' header['id'] = f'toc_{toc_line_index}'
toc_line_index += 1
toc_line = toc.new_tag('li') toc_line = toc.new_tag('li')
toc_line['text'] = header.text toc_line['text'] = header.text