mirror of
https://github.com/arsenetar/dupeguru.git
synced 2024-10-29 21:05:57 +00:00
122 lines
4.1 KiB
Python
122 lines
4.1 KiB
Python
# Created By: Virgil Dupras
|
|
# Created On: 2009-04-23
|
|
# 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, pyqtSignal, QModelIndex
|
|
from PyQt5.QtGui import QBrush, QFont, QFontMetrics
|
|
from PyQt5.QtWidgets import QTableView
|
|
|
|
from qt.table import Table
|
|
|
|
|
|
class ResultsModel(Table):
|
|
def __init__(self, app, view, **kwargs):
|
|
model = app.model.result_table
|
|
super().__init__(model, view, **kwargs)
|
|
view.horizontalHeader().setSortIndicator(1, Qt.AscendingOrder)
|
|
font = view.font()
|
|
font.setPointSize(app.prefs.tableFontSize)
|
|
view.setFont(font)
|
|
fm = QFontMetrics(font)
|
|
view.verticalHeader().setDefaultSectionSize(fm.height() + 2)
|
|
|
|
app.willSavePrefs.connect(self.appWillSavePrefs)
|
|
self.prefs = app.prefs
|
|
|
|
def _getData(self, row, column, role):
|
|
if column.name == "marked":
|
|
if role == Qt.BackgroundRole and row.isref:
|
|
return QBrush(self.prefs.result_table_ref_background_color)
|
|
if role == Qt.CheckStateRole and row.markable:
|
|
return Qt.Checked if row.marked else Qt.Unchecked
|
|
return None
|
|
if role == Qt.DisplayRole:
|
|
data = row.data_delta if self.model.delta_values else row.data
|
|
return data[column.name]
|
|
elif role == Qt.ForegroundRole:
|
|
if row.isref:
|
|
return QBrush(self.prefs.result_table_ref_foreground_color)
|
|
elif row.is_cell_delta(column.name):
|
|
return QBrush(self.prefs.result_table_delta_foreground_color)
|
|
elif role == Qt.BackgroundRole:
|
|
if row.isref:
|
|
return QBrush(self.prefs.result_table_ref_background_color)
|
|
elif role == Qt.FontRole:
|
|
font = QFont(self.view.font())
|
|
if self.prefs.reference_bold_font:
|
|
font.setBold(row.isref)
|
|
return font
|
|
elif role == Qt.EditRole and column.name == "name":
|
|
return row.data[column.name]
|
|
return None
|
|
|
|
def _getFlags(self, row, column):
|
|
flags = Qt.ItemIsEnabled | Qt.ItemIsSelectable
|
|
if column.name == "marked":
|
|
if row.markable:
|
|
flags |= Qt.ItemIsUserCheckable
|
|
elif column.name == "name":
|
|
flags |= Qt.ItemIsEditable
|
|
return flags
|
|
|
|
def _setData(self, row, column, value, role):
|
|
if role == Qt.CheckStateRole:
|
|
if column.name == "marked":
|
|
row.marked = bool(value)
|
|
return True
|
|
elif role == Qt.EditRole and column.name == "name":
|
|
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 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
|
|
super().keyPressEvent(event)
|
|
|
|
def mouseDoubleClickEvent(self, event):
|
|
self.doubleClicked.emit(QModelIndex())
|
|
# We don't call the superclass' method because the default behavior is to rename the cell.
|
|
|
|
# --- Signals
|
|
spacePressed = pyqtSignal()
|