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

KF8: Fix handling of multi-level ToCs

Browse Source
This commit is contained in:
Kovid Goyal 2012-04-23 10:48:32 +05:30
parent 5f00faaf5c
commit a187febc84
1 changed files with 21 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/calibre/ebooks/mobi/writer8/main.py
View File

@ -297,7 +297,6 @@ class KF8Writer(object):
self.chunk_records = ChunkIndex(self.chunk_table)() self.chunk_records = ChunkIndex(self.chunk_table)()
self.ncx_records = [] self.ncx_records = []
toc = self.oeb.toc toc = self.oeb.toc
max_depth = toc.depth()
entries = [] entries = []
is_periodical = self.opts.mobi_periodical is_periodical = self.opts.mobi_periodical
if toc.count() < 2: if toc.count() < 2:
@ -307,26 +306,37 @@ class KF8Writer(object):
# Flatten the ToC into a depth first list # Flatten the ToC into a depth first list
fl = toc.iter() if is_periodical else toc.iterdescendants() fl = toc.iter() if is_periodical else toc.iterdescendants()
for i, item in enumerate(fl): for i, item in enumerate(fl):
entry = {'index':i, 'depth': max_depth - item.depth() - (0 if entry = {'id': id(item), 'index': i, 'href':item.href,
is_periodical else 1), 'href':item.href, 'label':(item.title or 'label':(item.title or _('Unknown')),
_('Unknown'))} 'children':[]}
entries.append(entry) entry['depth'] = getattr(item, 'ncx_hlvl', 0)
for child in item:
child.ncx_parent = entry
p = getattr(item, 'ncx_parent', None) p = getattr(item, 'ncx_parent', None)
if p is not None: if p is not None:
entry['parent'] = p['index'] entry['parent_id'] = p
for child in item:
child.ncx_parent = entry['id']
child.ncx_hlvl = entry['depth'] + 1
entry['children'].append(id(child))
if is_periodical: if is_periodical:
if item.author: if item.author:
entry['author'] = item.author entry['author'] = item.author
if item.description: if item.description:
entry['description'] = item.description entry['description'] = item.description
entries.append(entry)
# The Kindle requires entries to be sorted by (depth, playorder)
entries.sort(key=lambda entry: (entry['depth'], entry['index']))
for i, entry in enumerate(entries):
entry['index'] = i
id_to_index = {entry['id']:entry['index'] for entry in entries}
for entry in entries: for entry in entries:
children = [e for e in entries if e.get('parent', -1) == entry['index']] children = entry.pop('children')
if children: if children:
entry['first_child'] = children[0]['index'] entry['first_child'] = id_to_index[children[0]]
entry['last_child'] = children[-1]['index'] entry['last_child'] = id_to_index[children[-1]]
if 'parent_id' in entry:
entry['parent'] = id_to_index[entry.pop('parent_id')]
href = entry.pop('href') href = entry.pop('href')
href, frag = href.partition('#')[0::2] href, frag = href.partition('#')[0::2]
aid = self.id_map.get((href, frag), None) aid = self.id_map.get((href, frag), None)