99 lines
4.3 KiB
Python
99 lines
4.3 KiB
Python
# Created By: Virgil Dupras
|
|
# Created On: 2009-11-25
|
|
# Copyright 2015 Hardcoded Software (http://www.hardcoded.net)
|
|
#
|
|
# This software is licensed under the "GPLv3" License as described in the "LICENSE" file,
|
|
# which should be included with this package. The terms are also available at
|
|
# http://www.gnu.org/licenses/gpl-3.0.html
|
|
|
|
from PyQt5.QtCore import Qt
|
|
from PyQt5.QtWidgets import QHeaderView
|
|
|
|
class Column:
|
|
def __init__(self, attrname, defaultWidth, editor=None, alignment=Qt.AlignLeft, cantTruncate=False, painter=None, resizeToFit=False):
|
|
self.attrname = attrname
|
|
self.defaultWidth = defaultWidth
|
|
self.editor = editor
|
|
# See moneyguru #15. Painter attribute was added to allow custom painting of amount value and
|
|
# currency information. Can be used as a pattern for custom painting of any column.
|
|
self.painter = painter
|
|
self.alignment = alignment
|
|
# This is to indicate, during printing, that a column can't have its data truncated.
|
|
self.cantTruncate = cantTruncate
|
|
self.resizeToFit = resizeToFit
|
|
|
|
|
|
class Columns:
|
|
def __init__(self, model, columns, headerView):
|
|
self.model = model
|
|
self._headerView = headerView
|
|
self._headerView.setDefaultAlignment(Qt.AlignLeft)
|
|
def setspecs(col, modelcol):
|
|
modelcol.default_width = col.defaultWidth
|
|
modelcol.editor = col.editor
|
|
modelcol.painter = col.painter
|
|
modelcol.resizeToFit = col.resizeToFit
|
|
modelcol.alignment = col.alignment
|
|
modelcol.cantTruncate = col.cantTruncate
|
|
if columns:
|
|
for col in columns:
|
|
modelcol = self.model.column_by_name(col.attrname)
|
|
setspecs(col, modelcol)
|
|
else:
|
|
col = Column('', 100)
|
|
for modelcol in self.model.column_list:
|
|
setspecs(col, modelcol)
|
|
self.model.view = self
|
|
self._headerView.sectionMoved.connect(self.headerSectionMoved)
|
|
self._headerView.sectionResized.connect(self.headerSectionResized)
|
|
|
|
# See moneyguru #14 and #15. This was added in order to allow automatic resizing of columns.
|
|
for column in self.model.column_list:
|
|
if column.resizeToFit:
|
|
self._headerView.setSectionResizeMode(column.logical_index, QHeaderView.ResizeToContents)
|
|
|
|
#--- Public
|
|
def setColumnsWidth(self, widths):
|
|
#`widths` can be None. If it is, then default widths are set.
|
|
columns = self.model.column_list
|
|
if not widths:
|
|
widths = [column.default_width for column in columns]
|
|
for column, width in zip(columns, widths):
|
|
if width == 0: # column was hidden before.
|
|
width = column.default_width
|
|
self._headerView.resizeSection(column.logical_index, width)
|
|
|
|
def setColumnsOrder(self, columnIndexes):
|
|
if not columnIndexes:
|
|
return
|
|
for destIndex, columnIndex in enumerate(columnIndexes):
|
|
# moveSection takes 2 visual index arguments, so we have to get our visual index first
|
|
visualIndex = self._headerView.visualIndex(columnIndex)
|
|
self._headerView.moveSection(visualIndex, destIndex)
|
|
|
|
#--- Events
|
|
def headerSectionMoved(self, logicalIndex, oldVisualIndex, newVisualIndex):
|
|
attrname = self.model.column_by_index(logicalIndex).name
|
|
self.model.move_column(attrname, newVisualIndex)
|
|
|
|
def headerSectionResized(self, logicalIndex, oldSize, newSize):
|
|
attrname = self.model.column_by_index(logicalIndex).name
|
|
self.model.resize_column(attrname, newSize)
|
|
|
|
#--- model --> view
|
|
def restore_columns(self):
|
|
columns = self.model.ordered_columns
|
|
indexes = [col.logical_index for col in columns]
|
|
self.setColumnsOrder(indexes)
|
|
widths = [col.width for col in self.model.column_list]
|
|
if not any(widths):
|
|
widths = None
|
|
self.setColumnsWidth(widths)
|
|
for column in self.model.column_list:
|
|
visible = self.model.column_is_visible(column.name)
|
|
self._headerView.setSectionHidden(column.logical_index, not visible)
|
|
|
|
def set_column_visible(self, colname, visible):
|
|
column = self.model.column_by_name(colname)
|
|
self._headerView.setSectionHidden(column.logical_index, not visible)
|