# Created By: Virgil Dupras # Created On: 2009-04-25 # $Id$ # Copyright 2009 Hardcoded Software (http://www.hardcoded.net) # # This software is licensed under the "HS" 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/hs_license import logging import os.path as op from PyQt4.QtCore import Qt, QTimer, QObject, QCoreApplication, QUrl, SIGNAL from PyQt4.QtGui import QProgressDialog, QDesktopServices, QFileDialog, QDialog, QMessageBox import hsfs as fs from hsutil import job from hsutil.reg import RegistrationRequired from dupeguru.app import (DupeGuru as DupeGuruBase, JOB_SCAN, JOB_LOAD, JOB_MOVE, JOB_COPY, JOB_DELETE) from qtlib.progress import Progress from . import platform from .main_window import MainWindow from .directories_dialog import DirectoriesDialog from .about_box import AboutBox from .reg import Registration JOBID2TITLE = { JOB_SCAN: "Scanning for duplicates", JOB_LOAD: "Loading", JOB_MOVE: "Moving", JOB_COPY: "Copying", JOB_DELETE: "Sending files to the recycle bin", } def demo_method(method): def wrapper(self, *args, **kwargs): try: return method(self, *args, **kwargs) except RegistrationRequired: msg = "The demo version of dupeGuru only allows 10 actions (delete/move/copy) per session." QMessageBox.information(self.main_window, 'Demo', msg) return wrapper class DupeGuru(DupeGuruBase, QObject): LOGO_NAME = '' NAME = '' DELTA_COLUMNS = frozenset() def __init__(self, data_module, appid): appdata = unicode(QDesktopServices.storageLocation(QDesktopServices.DataLocation)) DupeGuruBase.__init__(self, data_module, appdata, appid) QObject.__init__(self) self._setup() #--- Private def _setup(self): self.selected_dupe = None self.prefs = self._create_preferences() self.prefs.load() self._update_options() self.main_window = self._create_main_window() self._progress = Progress(self.main_window) self.directories_dialog = DirectoriesDialog(self.main_window, self) self.details_dialog = self._create_details_dialog(self.main_window) self.preferences_dialog = self._create_preferences_dialog(self.main_window) self.about_box = AboutBox(self.main_window, self) self.reg = Registration(self) self.set_registration(self.prefs.registration_code, self.prefs.registration_email) if not self.registered: self.reg.show_nag() self.main_window.show() self.load() self.connect(QCoreApplication.instance(), SIGNAL('aboutToQuit()'), self.application_will_terminate) self.connect(self._progress, SIGNAL('finished(QString)'), self.job_finished) def _setup_as_registered(self): self.prefs.registration_code = self.registration_code self.prefs.registration_email = self.registration_email self.main_window.actionRegister.setVisible(False) self.about_box.registerButton.hide() self.about_box.registeredEmailLabel.setText(self.prefs.registration_email) def _update_options(self): self.scanner.mix_file_kind = self.prefs.mix_file_kind self.options['escape_filter_regexp'] = self.prefs.use_regexp self.options['clean_empty_dirs'] = self.prefs.remove_empty_folders #--- Virtual def _create_details_dialog(self, parent): raise NotImplementedError() def _create_main_window(self): return MainWindow(app=self) def _create_preferences(self): raise NotImplementedError() def _create_preferences_dialog(self, parent): raise NotImplementedError() #--- Override @staticmethod def _recycle_dupe(dupe): platform.recycle_file(dupe.path) def _start_job(self, jobid, func): title = JOBID2TITLE[jobid] try: j = self._progress.create_job() self._progress.run(jobid, title, func, args=(j, )) except job.JobInProgressError: msg = "A previous action is still hanging in there. You can't start a new one yet. Wait a few seconds, then try again." QMessageBox.information(self.main_window, 'Action in progress', msg) #--- Public def add_dupes_to_ignore_list(self, duplicates): for dupe in duplicates: self.add_to_ignore_list(dupe) self.remove_duplicates(duplicates) def apply_filter(self, filter): DupeGuruBase.apply_filter(self, filter) self.emit(SIGNAL('resultsChanged()')) def ask_for_reg_code(self): if self.reg.ask_for_code(): self._setup_ui_as_registered() @demo_method def copy_or_move_marked(self, copy): opname = 'copy' if copy else 'move' title = "Select a directory to {0} marked files to".format(opname) flags = QFileDialog.ShowDirsOnly destination = unicode(QFileDialog.getExistingDirectory(self.main_window, title, '', flags)) if not destination: return recreate_path = self.prefs.destination_type DupeGuruBase.copy_or_move_marked(self, copy, destination, recreate_path) delete_marked = demo_method(DupeGuruBase.delete_marked) def make_reference(self, duplicates): DupeGuruBase.make_reference(self, duplicates) self.emit(SIGNAL('resultsChanged()')) def mark_all(self): self.results.mark_all() self.emit(SIGNAL('dupeMarkingChanged()')) def mark_invert(self): self.results.mark_invert() self.emit(SIGNAL('dupeMarkingChanged()')) def mark_none(self): self.results.mark_none() self.emit(SIGNAL('dupeMarkingChanged()')) def open_selected(self): if self.selected_dupe is None: return url = QUrl.fromLocalFile(unicode(self.selected_dupe.path)) QDesktopServices.openUrl(url) def remove_duplicates(self, duplicates): self.results.remove_duplicates(duplicates) self.emit(SIGNAL('resultsChanged()')) def remove_marked_duplicates(self): marked = [d for d in self.results.dupes if self.results.is_marked(d)] self.remove_duplicates(marked) def rename_dupe(self, dupe, newname): try: dupe.move(dupe.parent, newname) return True except (IndexError, fs.FSError) as e: logging.warning("dupeGuru Warning: %s" % unicode(e)) return False def reveal_selected(self): if self.selected_dupe is None: return url = QUrl.fromLocalFile(unicode(self.selected_dupe.path[:-1])) QDesktopServices.openUrl(url) def select_duplicate(self, dupe): self.selected_dupe = dupe self.emit(SIGNAL('duplicateSelected()')) def show_about_box(self): self.about_box.show() def show_details(self): self.details_dialog.show() def show_directories(self): self.directories_dialog.show() def show_help(self): url = QUrl.fromLocalFile(op.abspath('help/intro.htm')) QDesktopServices.openUrl(url) def show_preferences(self): self.preferences_dialog.load() result = self.preferences_dialog.exec_() if result == QDialog.Accepted: self.preferences_dialog.save() self.prefs.save() self._update_options() def toggle_marking_for_dupes(self, dupes): for dupe in dupes: self.results.mark_toggle(dupe) self.emit(SIGNAL('dupeMarkingChanged()')) #--- Events def application_will_terminate(self): self.save() self.save_ignore_list() def job_finished(self, jobid): self.emit(SIGNAL('resultsChanged()')) if jobid == JOB_LOAD: self.emit(SIGNAL('directoriesChanged()')) if jobid in (JOB_MOVE, JOB_COPY, JOB_DELETE) and self.last_op_error_count > 0: msg = "{0} files could not be processed.".format(self.results.mark_count) QMessageBox.warning(self.main_window, 'Warning', msg)