def init_frozen_view(self):
self._frozen_view = QTableView(self)
self._frozen_view.setFocusPolicy(Qt.NoFocus)
self._frozen_view.verticalHeader().hide()
self._frozen_view.setStyleSheet(FROZEN_STYLE)
self._frozen_view.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self._frozen_view.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
# Synchronize vertical scroll bars between the two table views
self._frozen_view.verticalScrollBar().valueChanged.connect(
self.verticalScrollBar().setValue)
self.verticalScrollBar().valueChanged.connect(
self._frozen_view.verticalScrollBar().setValue)
self.verticalHeader().sectionResized.connect(self.update_row_height)
hheader = self._frozen_view.horizontalHeader()
hheader.setContextMenuPolicy(Qt.CustomContextMenu)
hheader.customContextMenuRequested.connect(self._on_context_menu)
hheader.sectionResized.connect(self.frozen_section_resized)
self.viewport().stackUnder(self._frozen_view)
self._frozen_view.show()
class FreezeTableView(DataFrameTable):
""" A QTableView that supports freezing columns. Adapted from:
http://doc.qt.io/qt-5/qtwidgets-itemviews-frozencolumn-example.html
"""
def __init__(self, *args, **kwargs):
super(FreezeTableView, self).__init__(*args, **kwargs)
self._frozen_columns = []
self._frozen_positions = {}
self.init_frozen_view()
self.update_frozen_geometry()
def setModel(self, model):
super(FreezeTableView, self).setModel(model)
if model is None:
return
self._frozen_columns = []
self._frozen_view.setModel(self.model())
if self.selectionModel():
self._frozen_view.setSelectionModel(self.selectionModel())
for i in range(0, self.model().columnCount()):
self._frozen_view.setColumnHidden(i, True)
for i in range(0, self.model().rowCount()):
self._frozen_view.setRowHeight(i, self.rowHeight(i))
def init_frozen_view(self):
self._frozen_view = QTableView(self)
self._frozen_view.setFocusPolicy(Qt.NoFocus)
self._frozen_view.verticalHeader().hide()
self._frozen_view.setStyleSheet(FROZEN_STYLE)
self._frozen_view.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self._frozen_view.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
# Synchronize vertical scroll bars between the two table views
self._frozen_view.verticalScrollBar().valueChanged.connect(
self.verticalScrollBar().setValue)
self.verticalScrollBar().valueChanged.connect(
self._frozen_view.verticalScrollBar().setValue)
self.verticalHeader().sectionResized.connect(self.update_row_height)
hheader = self._frozen_view.horizontalHeader()
hheader.setContextMenuPolicy(Qt.CustomContextMenu)
hheader.customContextMenuRequested.connect(self._on_context_menu)
hheader.sectionResized.connect(self.frozen_section_resized)
self.viewport().stackUnder(self._frozen_view)
self._frozen_view.show()
def freeze_column(self, column):
if column not in self._frozen_columns:
self._frozen_columns.append(column)
self._frozen_view.setColumnHidden(column, False)
self._frozen_view.setColumnWidth(column, self.columnWidth(column))
self.horizontalHeader().moveSection(column,
len(self._frozen_columns) - 1)
self.update_frozen_geometry()
self._frozen_positions[len(self._frozen_columns) - 1] = column
def unfreeze_column(self, column):
if column in self._frozen_columns:
self._frozen_columns.remove(column)
self._frozen_view.setColumnHidden(column, True)
self.setColumnHidden(column, False)
self.update_frozen_geometry()
# Move the column back to its position before it was frozen
hheader = self.horizontalHeader()
v = hheader.visualIndex(column)
hheader.moveSection(v, self._frozen_positions[v])
del self._frozen_positions[v]
# Force a redraw, since some artifacts were being left behind
self.update()
def frozen_width(self):
return sum([self.columnWidth(i) for i in self._frozen_columns])
def update_frozen_geometry(self):
self._frozen_view.setGeometry(
self.verticalHeader().width() + self.frameWidth(),
self.frameWidth(), self.frozen_width(),
self.viewport().height() + self.horizontalHeader().height())
def update_row_height(self, index, old_size, size):
self._frozen_view.setRowHeight(index, size)
def frozen_section_resized(self, index, old_size, size):
if index in self._frozen_columns:
self.setColumnWidth(index, size)
self.update_frozen_geometry()
def resizeEvent(self, event):
super(FreezeTableView, self).resizeEvent(event)
self.update_frozen_geometry()
def moveCursor(self, action, modifiers):
current = super(FreezeTableView, self).moveCursor(action, modifiers)
corner_x = self.visualRect(current).topLeft().x()
if (action == QTableView.MoveLeft and current.column() > 0
and corner_x < self.frozen_width()):
value = self.horizontalScrollBar().value(
) + corner_x - self.frozen_width()
self.horizontalScrollBar().setValue(value)
return current
def context_menu(self, col):
menu = super(FreezeTableView, self).context_menu(col)
if col in self._frozen_columns:
action = QAction('Unfreeze Column', menu)
action.setData(col)
action.triggered.connect(self._on_unfreeze_column)
else:
action = QAction('Freeze Column', menu)
action.setData(col)
action.triggered.connect(self._on_freeze_column)
menu.addAction(action)
return menu
def _on_freeze_column(self):
col = self.sender().data()
self.freeze_column(col)
def _on_unfreeze_column(self):
col = self.sender().data()
self.unfreeze_column(col)
def create_widget(self):
self.widget = QTableView(self.parent_widget())
class QtTableView(QtAbstractItemView, ProxyTableView):
#: Proxy widget
widget = Typed(QTableView)
def create_widget(self):
self.widget = QTableView(self.parent_widget())
def init_widget(self):
super(QtTableView, self).init_widget()
d = self.declaration
self.set_show_grid(d.show_grid)
def init_model(self):
self.set_model(QAtomTableModel(parent=self.widget))
#--------------------------------------------------------------------------
# Widget settters
#--------------------------------------------------------------------------
def set_show_grid(self, show):
self.widget.setShowGrid(show)
def set_cell_padding(self, padding):
self.widget.setStyleSheet("QTableView::item { padding: %ipx }" %
padding)
def set_vertical_minimum_section_size(self, size):
self.widget.verticalHeader().setMinimumSectionSize(size)
def set_horizontal_minimum_section_size(self, size):
self.widget.horizontalHeader().setMinimumSectionSize(size)
def set_horizontal_stretch(self, stretch):
self.widget.horizontalHeader().setStretchLastSection(stretch)
def set_vertical_stretch(self, stretch):
self.widget.verticalHeader().setStretchLastSection(stretch)
def set_resize_mode(self, mode):
self.widget.horizontalHeader().setResizeMode(RESIZE_MODES[mode])
def set_show_horizontal_header(self, show):
header = self.widget.horizontalHeader()
header.show() if show else header.hide()
def set_show_vertical_header(self, show):
header = self.widget.verticalHeader()
header.show() if show else header.hide()
#--------------------------------------------------------------------------
# View refresh handlers
#--------------------------------------------------------------------------
def _refresh_visible_column(self, value):
self._pending_column_refreshes -= 1
if self._pending_column_refreshes == 0:
d = self.declaration
d.visible_column = max(
0, min(value,
self.model.columnCount() - d.visible_columns))
def _refresh_visible_row(self, value):
self._pending_row_refreshes -= 1
if self._pending_row_refreshes == 0:
d = self.declaration
d.visible_row = max(
0, min(value,
self.model.rowCount() - d.visible_rows))
def _refresh_visible_rows(self):
return
top = self.widget.rowAt(self.widget.rect().top())
bottom = self.widget.rowAt(self.widget.rect().bottom())
self.declaration.visible_rows = max(
1, (bottom - top)) * 2 # 2x for safety
def _refresh_visible_columns(self):
return
left = self.widget.rowAt(self.widget.rect().left())
right = self.widget.rowAt(self.widget.rect().right())
self.declaration.visible_columns = max(1, (right - left)) * 2
def create_widget(self):
self.widget = QTableView(self.parent_widget())
class QtTableView(QtAbstractItemView, ProxyTableView):
#: Proxy widget
widget = Typed(QTableView)
def create_widget(self):
self.widget = QTableView(self.parent_widget())
def init_widget(self):
super(QtTableView, self).init_widget()
d = self.declaration
self.set_show_grid(d.show_grid)
def init_model(self):
self.set_model(QAtomTableModel(parent=self.widget))
#--------------------------------------------------------------------------
# Widget settters
#--------------------------------------------------------------------------
def set_show_grid(self,show):
self.widget.setShowGrid(show)
def set_cell_padding(self, padding):
self.widget.setStyleSheet("QTableView::item { padding: %ipx }"%padding);
def set_vertical_minimum_section_size(self,size):
self.widget.verticalHeader().setMinimumSectionSize(size)
def set_horizontal_minimum_section_size(self,size):
self.widget.horizontalHeader().setMinimumSectionSize(size)
def set_horizontal_stretch(self,stretch):
self.widget.horizontalHeader().setStretchLastSection(stretch)
def set_vertical_stretch(self,stretch):
self.widget.verticalHeader().setStretchLastSection(stretch)
def set_resize_mode(self,mode):
self.widget.horizontalHeader().setResizeMode(RESIZE_MODES[mode])
def set_show_horizontal_header(self,show):
header = self.widget.horizontalHeader()
header.show() if show else header.hide()
def set_show_vertical_header(self,show):
header = self.widget.verticalHeader()
header.show() if show else header.hide()
#--------------------------------------------------------------------------
# View refresh handlers
#--------------------------------------------------------------------------
def _refresh_visible_column(self, value):
self._pending_column_refreshes -=1
if self._pending_column_refreshes==0:
d = self.declaration
d.visible_column = max(0,min(value,self.model.columnCount()-d.visible_columns))
def _refresh_visible_row(self, value):
self._pending_row_refreshes -=1
if self._pending_row_refreshes==0:
d = self.declaration
d.visible_row = max(0,min(value,self.model.rowCount()-d.visible_rows))
def _refresh_visible_rows(self):
return
top = self.widget.rowAt(self.widget.rect().top())
bottom = self.widget.rowAt(self.widget.rect().bottom())
self.declaration.visible_rows = max(1,(bottom-top))*2 # 2x for safety
def _refresh_visible_columns(self):
return
left = self.widget.rowAt(self.widget.rect().left())
right = self.widget.rowAt(self.widget.rect().right())
self.declaration.visible_columns = max(1,(right-left))*2
