dupeguru/qtlib/column.py

# 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)