Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Table of Contents Editor: Handle generating ToC from headings if the first heading level in the document is not h1 or the document skips some intermediate heading levels

Browse Source
This commit is contained in:
Kovid Goyal 2013-12-04 10:00:56 +05:30
parent 9fc66ab7bb
commit a2dff45219
1 changed files with 19 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/calibre/ebooks/oeb/polish/toc.py
View File

@ -9,8 +9,9 @@ __docformat__ = 'restructuredtext en'
import re
from urlparse import urlparse
from collections import deque, Counter
from collections import deque, Counter, OrderedDict
from functools import partial
from operator import itemgetter
from lxml import etree
@ -206,10 +207,26 @@ def from_xpaths(container, xpaths):
level_prev = {i+1:None for i in xrange(len(xpaths))}
level_prev[0] = tocroot
# Find those levels that have no elements in all spine items
maps = OrderedDict()
empty_levels = {i+1 for i, xp in enumerate(xpaths)}
for spinepath in container.spine_items:
name = container.abspath_to_name(spinepath)
root = container.parsed(name)
level_item_map = {i+1:frozenset(xp(root)) for i, xp in enumerate(xpaths)}
level_item_map = maps[name] = {i+1:frozenset(xp(root)) for i, xp in enumerate(xpaths)}
for lvl, elems in level_item_map.iteritems():
if elems:
empty_levels.discard(lvl)
# Remove empty levels from all level_maps
if empty_levels:
for name, lmap in tuple(maps.iteritems()):
lmap = {lvl:items for lvl, items in lmap.iteritems() if lvl not in empty_levels}
lmap = sorted(lmap.iteritems(), key=itemgetter(0))
lmap = {i+1:items for i, (l, items) in enumerate(lmap)}
maps[name] = lmap
for name, level_item_map in maps.iteritems():
root = container.parsed(name)
item_level_map = {e:i for i, elems in level_item_map.iteritems() for e in elems}
item_dirtied = False