Refactor row_decoration management

This commit is contained in:
Kovid Goyal 2013-10-04 18:10:17 +05:30
parent 49686b7093
commit 4f87c0d39e
2 changed files with 4 additions and 1 deletions

View File

@ -212,6 +212,9 @@ class BooksModel(QAbstractTableModel): # {{{
def set_row_height(self, height): def set_row_height(self, height):
self.row_height = height self.row_height = height
def set_row_decoration(self, current_marked):
self.row_decoration = self.bool_blank_icon if current_marked else None
def change_alignment(self, colname, alignment): def change_alignment(self, colname, alignment):
if colname in self.column_map and alignment in ('left', 'right', 'center'): if colname in self.column_map and alignment in ('left', 'right', 'center'):
old = self.alignment_map.get(colname, 'left') old = self.alignment_map.get(colname, 'left')

View File

@ -696,7 +696,7 @@ class BooksView(QTableView): # {{{
self.row_header.headerDataChanged(Qt.Vertical, min(sections), max(sections)) self.row_header.headerDataChanged(Qt.Vertical, min(sections), max(sections))
else: else:
# Marked items have either appeared or all been removed # Marked items have either appeared or all been removed
self.model().row_decoration = self.model().bool_blank_icon if current_marked else None self.model().set_row_decoration(current_marked)
self.row_header.headerDataChanged(Qt.Vertical, 0, self.row_header.count()-1) self.row_header.headerDataChanged(Qt.Vertical, 0, self.row_header.count()-1)
self.row_header.geometriesChanged.emit() self.row_header.geometriesChanged.emit()