dupeguru/qt/base/results_model.py

# Created By: Virgil Dupras
# Created On: 2009-04-23
# Copyright 2011 Hardcoded Software (http://www.hardcoded.net)
# 
# This software is licensed under the "BSD" License as described in the "LICENSE" file, 
# which should be included with this package. The terms are also available at 
# http://www.hardcoded.net/licenses/bsd_license

from PyQt4.QtCore import Qt, pyqtSignal
from PyQt4.QtGui import QBrush, QFont, QFontMetrics, QTableView, QColor

from qtlib.table import Table

class ResultsModel(Table):
    def __init__(self, app, view):
        model = app.model.result_table
        Table.__init__(self, model, view)
        
        app.prefsChanged.connect(self.appPrefsChanged)
        app.willSavePrefs.connect(self.appWillSavePrefs)
    
    def columnCount(self, parent):
        return len(self.model.COLUMNS)
    
    def data(self, index, role):
        if not index.isValid():
            return None
        row = self.model[index.row()]
        column = self.model.COLUMNS[index.column()]
        if role == Qt.DisplayRole:
            data = row.data_delta if self.model.delta_values else row.data
            return data[column.name]
        elif role == Qt.CheckStateRole:
            if index.column() == 0 and row.markable:
                return Qt.Checked if row.marked else Qt.Unchecked
        elif role == Qt.ForegroundRole:
            if row.isref:
                return QBrush(Qt.blue)
            elif self.model.delta_values and column.name in self.model.DELTA_COLUMNS:
                return QBrush(QColor(255, 142, 40)) # orange
        elif role == Qt.FontRole:
            isBold = row.isref
            font = QFont(self.view.font())
            font.setBold(isBold)
            return font
        elif role == Qt.EditRole:
            if column.name == 'name':
                return row.data[column.name]
        return None
    
    def flags(self, index):
        if not index.isValid():
            return Qt.ItemIsEnabled
        flags = Qt.ItemIsEnabled | Qt.ItemIsSelectable
        if index.column() == 0:
            flags |= Qt.ItemIsEditable
            row = self.model[index.row()]
            if row.markable:
                flags |= Qt.ItemIsUserCheckable
        return flags
    
    def headerData(self, section, orientation, role):
        if role == Qt.DisplayRole:
            if orientation == Qt.Horizontal and section < len(self.model.COLUMNS):
                return self.model.COLUMNS[section].display
        return None
    
    def setData(self, index, value, role):
        if not index.isValid():
            return False
        row = self.model[index.row()]
        column = self.model.COLUMNS[index.column()]
        if role == Qt.CheckStateRole:
            if column.name == 'name':
                self.model.app.mark_dupe(row._dupe, value.toBool())
                return True
        elif role == Qt.EditRole:
            if column.name == 'name':
                value = str(value.toString())
                return self.model.rename_selected(value)
        return False
    
    def sort(self, column, order):
        column = self.model.COLUMNS[column]
        self.model.sort(column.name, order == Qt.AscendingOrder)
    
    #--- Properties
    @property
    def power_marker(self):
        return self.model.power_marker
    
    @power_marker.setter
    def power_marker(self, value):
        self.model.power_marker = value
    
    @property
    def delta_values(self):
        return self.model.delta_values
    
    @delta_values.setter
    def delta_values(self, value):
        self.model.delta_values = value
    
    #--- Events
    def appPrefsChanged(self, prefs):
        font = self.view.font()
        font.setPointSize(prefs.tableFontSize)
        self.view.setFont(font)
        fm = QFontMetrics(font)
        self.view.verticalHeader().setDefaultSectionSize(fm.height()+2)
    
    def appWillSavePrefs(self):
        self.model.columns.save_columns()
    
    #--- model --> view
    def invalidate_markings(self):
        # redraw view
        # HACK. this is the only way I found to update the widget without reseting everything
        self.view.scroll(0, 1)
        self.view.scroll(0, -1)
    

class ResultsView(QTableView):
    #--- Override
    def keyPressEvent(self, event):
        if event.text() == ' ':
            self.spacePressed.emit()
            return
        QTableView.keyPressEvent(self, event)
    
    def mouseDoubleClickEvent(self, event):
        self.doubleClicked.emit(None)
        # We don't call the superclass' method because the default behavior is to rename the cell.
    
    #--- Signals
    spacePressed = pyqtSignal()