mirror of
https://github.com/arsenetar/dupeguru.git
synced 2024-10-31 22:05:58 +00:00
372 lines
16 KiB
Python
372 lines
16 KiB
Python
# Created By: Virgil Dupras
|
|
# Created On: 2009-04-25
|
|
# Copyright 2010 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
|
|
|
|
import sys
|
|
|
|
from PyQt4.QtCore import Qt, SIGNAL, QUrl, QRect
|
|
from PyQt4.QtGui import (QMainWindow, QMenu, QPixmap, QIcon, QLabel, QHeaderView, QMessageBox,
|
|
QInputDialog, QLineEdit, QDesktopServices, QFileDialog, QMenuBar, QWidget, QVBoxLayout,
|
|
QAbstractItemView, QStatusBar)
|
|
|
|
from hscommon.util import nonone
|
|
|
|
from .results_model import ResultsModel, ResultsView
|
|
from .stats_label import StatsLabel
|
|
from .util import createActions
|
|
|
|
class ResultWindow(QMainWindow):
|
|
def __init__(self, app):
|
|
QMainWindow.__init__(self, None)
|
|
self.app = app
|
|
self._last_filter = None
|
|
self._setupUi()
|
|
self.resultsModel = ResultsModel(self.app, self.resultsView)
|
|
self.stats = StatsLabel(app, self.statusLabel)
|
|
self._load_columns()
|
|
self._update_column_actions_status()
|
|
|
|
self.connect(self.menuColumns, SIGNAL('triggered(QAction*)'), self.columnToggled)
|
|
self.connect(self.resultsView, SIGNAL('doubleClicked()'), self.resultsDoubleClicked)
|
|
self.connect(self.resultsView, SIGNAL('spacePressed()'), self.resultsSpacePressed)
|
|
self.app.willSavePrefs.connect(self.appWillSavePrefs)
|
|
|
|
def _setupActions(self):
|
|
# (name, shortcut, icon, desc, func)
|
|
ACTIONS = [
|
|
('actionDetails', 'Ctrl+I', '', "Details", self.detailsTriggered),
|
|
('actionActions', '', '', "Actions", self.actionsTriggered),
|
|
('actionPowerMarker', 'Ctrl+1', '', "Show Dupes Only", self.powerMarkerTriggered),
|
|
('actionDelta', 'Ctrl+2', '', "Show Delta Values", self.deltaTriggered),
|
|
('actionDeleteMarked', 'Ctrl+D', '', "Send Marked to Recycle Bin", self.deleteTriggered),
|
|
('actionHardlinkMarked', 'Ctrl+Shift+D', '', "Delete Marked and Replace with Hardlinks", self.hardlinkTriggered),
|
|
('actionMoveMarked', 'Ctrl+M', '', "Move Marked to...", self.moveTriggered),
|
|
('actionCopyMarked', 'Ctrl+Shift+M', '', "Copy Marked to...", self.copyTriggered),
|
|
('actionRemoveMarked', 'Ctrl+R', '', "Remove Marked from Results", self.removeMarkedTriggered),
|
|
('actionRemoveSelected', 'Ctrl+Del', '', "Remove Selected from Results", self.removeSelectedTriggered),
|
|
('actionIgnoreSelected', 'Ctrl+Shift+Del', '', "Add Selected to Ignore List", self.addToIgnoreListTriggered),
|
|
('actionMakeSelectedReference', 'Ctrl+Space', '', "Make Selected Reference", self.makeReferenceTriggered),
|
|
('actionOpenSelected', 'Ctrl+O', '', "Open Selected with Default Application", self.openTriggered),
|
|
('actionRevealSelected', 'Ctrl+Shift+O', '', "Open Containing Folder of Selected", self.revealTriggered),
|
|
('actionRenameSelected', 'F2', '', "Rename Selected", self.renameTriggered),
|
|
('actionMarkAll', 'Ctrl+A', '', "Mark All", self.markAllTriggered),
|
|
('actionMarkNone', 'Ctrl+Shift+A', '', "Mark None", self.markNoneTriggered),
|
|
('actionInvertMarking', 'Ctrl+Alt+A', '', "Invert Marking", self.markInvertTriggered),
|
|
('actionMarkSelected', '', '', "Mark Selected", self.markSelectedTriggered),
|
|
('actionClearIgnoreList', '', '', "Clear Ignore List", self.clearIgnoreListTriggered),
|
|
('actionApplyFilter', 'Ctrl+F', '', "Apply Filter", self.applyFilterTriggered),
|
|
('actionCancelFilter', 'Ctrl+Shift+F', '', "Cancel Filter", self.cancelFilterTriggered),
|
|
('actionExport', '', '', "Export To HTML", self.exportTriggered),
|
|
('actionSaveResults', 'Ctrl+S', '', "Save Results...", self.saveResultsTriggered),
|
|
('actionInvokeCustomCommand', 'Ctrl+Alt+I', '', "Invoke Custom Command", self.app.invokeCustomCommand),
|
|
]
|
|
createActions(ACTIONS, self)
|
|
self.actionDelta.setCheckable(True)
|
|
self.actionPowerMarker.setCheckable(True)
|
|
|
|
def _setupMenu(self):
|
|
self.menubar = QMenuBar(self)
|
|
self.menubar.setGeometry(QRect(0, 0, 630, 22))
|
|
self.menuFile = QMenu(self.menubar)
|
|
self.menuFile.setTitle("File")
|
|
self.menuMark = QMenu(self.menubar)
|
|
self.menuMark.setTitle("Mark")
|
|
self.menuActions = QMenu(self.menubar)
|
|
self.menuActions.setTitle("Actions")
|
|
self.menuColumns = QMenu(self.menubar)
|
|
self.menuColumns.setTitle("Columns")
|
|
self.menuView = QMenu(self.menubar)
|
|
self.menuView.setTitle("View")
|
|
self.menuHelp = QMenu(self.menubar)
|
|
self.menuHelp.setTitle("Help")
|
|
self.setMenuBar(self.menubar)
|
|
|
|
self.menuActions.addAction(self.actionDeleteMarked)
|
|
self.menuActions.addAction(self.actionHardlinkMarked)
|
|
self.menuActions.addAction(self.actionMoveMarked)
|
|
self.menuActions.addAction(self.actionCopyMarked)
|
|
self.menuActions.addAction(self.actionRemoveMarked)
|
|
self.menuActions.addSeparator()
|
|
self.menuActions.addAction(self.actionRemoveSelected)
|
|
self.menuActions.addAction(self.actionIgnoreSelected)
|
|
self.menuActions.addAction(self.actionMakeSelectedReference)
|
|
self.menuActions.addSeparator()
|
|
self.menuActions.addAction(self.actionOpenSelected)
|
|
self.menuActions.addAction(self.actionRevealSelected)
|
|
self.menuActions.addAction(self.actionInvokeCustomCommand)
|
|
self.menuActions.addAction(self.actionRenameSelected)
|
|
self.menuActions.addSeparator()
|
|
self.menuActions.addAction(self.actionApplyFilter)
|
|
self.menuActions.addAction(self.actionCancelFilter)
|
|
self.menuMark.addAction(self.actionMarkAll)
|
|
self.menuMark.addAction(self.actionMarkNone)
|
|
self.menuMark.addAction(self.actionInvertMarking)
|
|
self.menuMark.addAction(self.actionMarkSelected)
|
|
self.menuView.addAction(self.actionPowerMarker)
|
|
self.menuView.addAction(self.actionDelta)
|
|
self.menuView.addSeparator()
|
|
self.menuView.addAction(self.actionDetails)
|
|
self.menuView.addAction(self.app.actionPreferences)
|
|
self.menuHelp.addAction(self.app.actionShowHelp)
|
|
self.menuHelp.addAction(self.app.actionRegister)
|
|
self.menuHelp.addAction(self.app.actionCheckForUpdate)
|
|
self.menuHelp.addAction(self.app.actionOpenDebugLog)
|
|
self.menuHelp.addAction(self.app.actionAbout)
|
|
self.menuFile.addAction(self.actionSaveResults)
|
|
self.menuFile.addAction(self.actionExport)
|
|
self.menuFile.addAction(self.actionClearIgnoreList)
|
|
self.menuFile.addSeparator()
|
|
self.menuFile.addAction(self.app.actionQuit)
|
|
|
|
self.menubar.addAction(self.menuFile.menuAction())
|
|
self.menubar.addAction(self.menuMark.menuAction())
|
|
self.menubar.addAction(self.menuActions.menuAction())
|
|
self.menubar.addAction(self.menuColumns.menuAction())
|
|
self.menubar.addAction(self.menuView.menuAction())
|
|
self.menubar.addAction(self.menuHelp.menuAction())
|
|
|
|
# Columns menu
|
|
menu = self.menuColumns
|
|
self._column_actions = []
|
|
for index, column in enumerate(self.app.data.COLUMNS):
|
|
action = menu.addAction(column['display'])
|
|
action.setCheckable(True)
|
|
action.column_index = index
|
|
self._column_actions.append(action)
|
|
menu.addSeparator()
|
|
action = menu.addAction("Reset to Defaults")
|
|
action.column_index = -1
|
|
|
|
# Action menu
|
|
actionMenu = QMenu('Actions', self.menubar)
|
|
actionMenu.addAction(self.actionDeleteMarked)
|
|
actionMenu.addAction(self.actionHardlinkMarked)
|
|
actionMenu.addAction(self.actionMoveMarked)
|
|
actionMenu.addAction(self.actionCopyMarked)
|
|
actionMenu.addAction(self.actionRemoveMarked)
|
|
actionMenu.addSeparator()
|
|
actionMenu.addAction(self.actionRemoveSelected)
|
|
actionMenu.addAction(self.actionIgnoreSelected)
|
|
actionMenu.addAction(self.actionMakeSelectedReference)
|
|
actionMenu.addSeparator()
|
|
actionMenu.addAction(self.actionOpenSelected)
|
|
actionMenu.addAction(self.actionRevealSelected)
|
|
actionMenu.addAction(self.actionInvokeCustomCommand)
|
|
actionMenu.addAction(self.actionRenameSelected)
|
|
self.actionActions.setMenu(actionMenu)
|
|
|
|
def _setupUi(self):
|
|
self.setWindowTitle("dupeGuru Results")
|
|
self.resize(630, 514)
|
|
self.centralwidget = QWidget(self)
|
|
self.verticalLayout_2 = QVBoxLayout(self.centralwidget)
|
|
self.verticalLayout_2.setMargin(0)
|
|
self.resultsView = ResultsView(self.centralwidget)
|
|
self.resultsView.setSelectionMode(QAbstractItemView.ExtendedSelection)
|
|
self.resultsView.setSelectionBehavior(QAbstractItemView.SelectRows)
|
|
self.resultsView.setSortingEnabled(True)
|
|
self.resultsView.verticalHeader().setVisible(False)
|
|
self.resultsView.verticalHeader().setDefaultSectionSize(18)
|
|
h = self.resultsView.horizontalHeader()
|
|
h.setHighlightSections(False)
|
|
h.setMovable(True)
|
|
h.setStretchLastSection(False)
|
|
h.setDefaultAlignment(Qt.AlignLeft)
|
|
self.verticalLayout_2.addWidget(self.resultsView)
|
|
self.setCentralWidget(self.centralwidget)
|
|
self._setupActions()
|
|
self._setupMenu()
|
|
self.statusbar = QStatusBar(self)
|
|
self.statusbar.setSizeGripEnabled(True)
|
|
self.setStatusBar(self.statusbar)
|
|
self.statusLabel = QLabel(self)
|
|
self.statusbar.addPermanentWidget(self.statusLabel, 1)
|
|
|
|
if self.app.prefs.resultWindowIsMaximized:
|
|
self.setWindowState(self.windowState() | Qt.WindowMaximized)
|
|
if self.app.prefs.resultWindowRect is not None and not self.app.prefs.resultWindowIsMaximized:
|
|
self.setGeometry(self.app.prefs.resultWindowRect)
|
|
|
|
# Platform-specific setup
|
|
if sys.platform == 'linux2':
|
|
self.actionCheckForUpdate.setVisible(False) # This only works on Windows
|
|
if sys.platform not in {'darwin', 'linux2'}:
|
|
self.actionHardlinkMarked.setVisible(False)
|
|
|
|
#--- Private
|
|
def _load_columns(self):
|
|
h = self.resultsView.horizontalHeader()
|
|
h.setResizeMode(QHeaderView.Interactive)
|
|
prefs = self.app.prefs
|
|
attrs = list(zip(prefs.columns_width, prefs.columns_visible))
|
|
for index, (width, visible) in enumerate(attrs):
|
|
h.resizeSection(index, width)
|
|
h.setSectionHidden(index, not visible)
|
|
h.setResizeMode(0, QHeaderView.Stretch)
|
|
|
|
def _update_column_actions_status(self):
|
|
h = self.resultsView.horizontalHeader()
|
|
for action in self._column_actions:
|
|
colid = action.column_index
|
|
action.setChecked(not h.isSectionHidden(colid))
|
|
|
|
#--- Actions
|
|
def actionsTriggered(self):
|
|
self.actionsButton.showMenu()
|
|
|
|
def addToIgnoreListTriggered(self):
|
|
self.app.add_selected_to_ignore_list()
|
|
|
|
def applyFilterTriggered(self):
|
|
title = "Apply Filter"
|
|
msg = "Type the filter you want to apply on your results. See help for details."
|
|
text = nonone(self._last_filter, '[*]')
|
|
answer, ok = QInputDialog.getText(self, title, msg, QLineEdit.Normal, text)
|
|
if not ok:
|
|
return
|
|
answer = str(answer)
|
|
self.app.apply_filter(answer)
|
|
self._last_filter = answer
|
|
|
|
def cancelFilterTriggered(self):
|
|
self.app.apply_filter('')
|
|
|
|
def clearIgnoreListTriggered(self):
|
|
title = "Clear Ignore List"
|
|
count = len(self.app.scanner.ignore_list)
|
|
if not count:
|
|
QMessageBox.information(self, title, "Nothing to clear.")
|
|
return
|
|
msg = "Do you really want to remove all {0} items from the ignore list?".format(count)
|
|
if self.app.confirm(title, msg, QMessageBox.No):
|
|
self.app.scanner.ignore_list.Clear()
|
|
QMessageBox.information(self, title, "Ignore list cleared.")
|
|
|
|
def copyTriggered(self):
|
|
self.app.copy_or_move_marked(True)
|
|
|
|
def deleteTriggered(self):
|
|
count = self.app.results.mark_count
|
|
if not count:
|
|
return
|
|
title = "Delete duplicates"
|
|
msg = "You are about to send {0} files to the recycle bin. Continue?".format(count)
|
|
if self.app.confirm(title, msg):
|
|
self.app.delete_marked()
|
|
|
|
def deltaTriggered(self):
|
|
self.resultsModel.delta_values = self.actionDelta.isChecked()
|
|
|
|
def detailsTriggered(self):
|
|
self.app.show_details()
|
|
|
|
def exportTriggered(self):
|
|
h = self.resultsView.horizontalHeader()
|
|
column_ids = []
|
|
for i in range(len(self.app.data.COLUMNS)):
|
|
if not h.isSectionHidden(i):
|
|
column_ids.append(str(i))
|
|
exported_path = self.app.export_to_xhtml(column_ids)
|
|
url = QUrl.fromLocalFile(exported_path)
|
|
QDesktopServices.openUrl(url)
|
|
|
|
def hardlinkTriggered(self):
|
|
count = self.app.results.mark_count
|
|
if not count:
|
|
return
|
|
title = "Delete and hardlink duplicates"
|
|
msg = "You are about to send {0} files to the trash and hardlink them afterwards. Continue?".format(count)
|
|
if self.app.confirm(title, msg):
|
|
self.app.delete_marked(replace_with_hardlinks=True)
|
|
|
|
def makeReferenceTriggered(self):
|
|
self.app.make_selected_reference()
|
|
|
|
def markAllTriggered(self):
|
|
self.app.mark_all()
|
|
|
|
def markInvertTriggered(self):
|
|
self.app.mark_invert()
|
|
|
|
def markNoneTriggered(self):
|
|
self.app.mark_none()
|
|
|
|
def markSelectedTriggered(self):
|
|
self.app.toggle_selected_mark_state()
|
|
|
|
def moveTriggered(self):
|
|
self.app.copy_or_move_marked(False)
|
|
|
|
def openTriggered(self):
|
|
self.app.open_selected()
|
|
|
|
def powerMarkerTriggered(self):
|
|
self.resultsModel.power_marker = self.actionPowerMarker.isChecked()
|
|
|
|
def preferencesTriggered(self):
|
|
self.app.show_preferences()
|
|
|
|
def removeMarkedTriggered(self):
|
|
count = self.app.results.mark_count
|
|
if not count:
|
|
return
|
|
title = "Remove duplicates"
|
|
msg = "You are about to remove {0} files from results. Continue?".format(count)
|
|
if self.app.confirm(title, msg):
|
|
self.app.remove_marked()
|
|
|
|
def removeSelectedTriggered(self):
|
|
self.app.remove_selected()
|
|
|
|
def renameTriggered(self):
|
|
self.resultsView.edit(self.resultsView.selectionModel().currentIndex())
|
|
|
|
def revealTriggered(self):
|
|
self.app.reveal_selected()
|
|
|
|
def saveResultsTriggered(self):
|
|
title = "Select a file to save your results to"
|
|
files = "dupeGuru Results (*.dupeguru)"
|
|
destination = QFileDialog.getSaveFileName(self, title, '', files)
|
|
if destination:
|
|
self.app.save_as(destination)
|
|
self.app.recentResults.insertItem(destination)
|
|
|
|
#--- Events
|
|
def appWillSavePrefs(self):
|
|
prefs = self.app.prefs
|
|
h = self.resultsView.horizontalHeader()
|
|
widths = []
|
|
visible = []
|
|
for i in range(len(self.app.data.COLUMNS)):
|
|
widths.append(h.sectionSize(i))
|
|
visible.append(not h.isSectionHidden(i))
|
|
prefs.columns_width = widths
|
|
prefs.columns_visible = visible
|
|
prefs.resultWindowIsMaximized = self.isMaximized()
|
|
prefs.resultWindowRect = self.geometry()
|
|
|
|
def columnToggled(self, action):
|
|
colid = action.column_index
|
|
if colid == -1:
|
|
self.app.prefs.reset_columns()
|
|
self._load_columns()
|
|
else:
|
|
h = self.resultsView.horizontalHeader()
|
|
h.setSectionHidden(colid, not h.isSectionHidden(colid))
|
|
self._update_column_actions_status()
|
|
|
|
def contextMenuEvent(self, event):
|
|
self.actionActions.menu().exec_(event.globalPos())
|
|
|
|
def resultsDoubleClicked(self):
|
|
self.app.open_selected()
|
|
|
|
def resultsSpacePressed(self):
|
|
self.app.toggle_selected_mark_state()
|
|
|