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

E-book viewer: Allow expanding/collapsing all items in the Table of Contents view by right clicking

Browse Source
This commit is contained in:
Kovid Goyal 2015-04-12 10:23:33 +05:30
parent eec1f44dee
commit d49d227e4a
1 changed files with 27 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
src/calibre/gui2/viewer/toc.py
View File

@ -8,9 +8,11 @@ __copyright__ = '2012, Kovid Goyal <kovid@kovidgoyal.net>'
__docformat__ = 'restructuredtext en' __docformat__ = 'restructuredtext en'
import re import re
from functools import partial
from PyQt5.Qt import ( from PyQt5.Qt import (
QStandardItem, QStandardItemModel, Qt, QFont, QTreeView, QWidget, QStandardItem, QStandardItemModel, Qt, QFont, QTreeView, QWidget,
QHBoxLayout, QToolButton, QIcon, QModelIndex, pyqtSignal) QHBoxLayout, QToolButton, QIcon, QModelIndex, pyqtSignal, QMenu)
from calibre.ebooks.metadata.toc import TOC as MTOC from calibre.ebooks.metadata.toc import TOC as MTOC
from calibre.gui2 import error_dialog from calibre.gui2 import error_dialog
@ -45,6 +47,8 @@ class TOCView(QTreeView):
border-radius: 6px; border-radius: 6px;
} }
''') ''')
self.setContextMenuPolicy(Qt.CustomContextMenu)
self.customContextMenuRequested.connect(self.context_menu)
def mouseMoveEvent(self, ev): def mouseMoveEvent(self, ev):
if self.indexAt(ev.pos()).isValid(): if self.indexAt(ev.pos()).isValid():
@ -53,6 +57,26 @@ class TOCView(QTreeView):
self.unsetCursor() self.unsetCursor()
return QTreeView.mouseMoveEvent(self, ev) return QTreeView.mouseMoveEvent(self, ev)
def expand_tree(self, index):
self.expand(index)
i = -1
while True:
i += 1
child = index.child(i, 0)
if not child.isValid():
break
self.expand_tree(child)
def context_menu(self, pos):
index = self.indexAt(pos)
m = QMenu(self)
if index.isValid():
m.addAction(_('Expand all items under %s') % index.data(), partial(self.expand_tree, index))
m.addSeparator()
m.addAction(_('Expand all items'), self.expandAll)
m.addAction(_('Collapse all items'), self.collapseAll)
m.exec_(self.mapToGlobal(pos))
class TOCSearch(QWidget): class TOCSearch(QWidget):
def __init__(self, toc_view, parent=None): def __init__(self, toc_view, parent=None):
@ -263,8 +287,8 @@ class TOC(QStandardItemModel):
self.appendRow(TOCItem(spine, t, 0, depth_first)) self.appendRow(TOCItem(spine, t, 0, depth_first))
for x in depth_first: for x in depth_first:
possible_enders = [t for t in depth_first if t.depth <= x.depth possible_enders = [t for t in depth_first if t.depth <= x.depth and
and t.starts_at >= x.starts_at and t is not x and t not in t.starts_at >= x.starts_at and t is not x and t not in
x.ancestors] x.ancestors]
if possible_enders: if possible_enders:
min_spine = min(t.starts_at for t in possible_enders) min_spine = min(t.starts_at for t in possible_enders)