dupeguru/base/qt/app.py

#!/usr/bin/env python
# Unit Name: app
# Created By: Virgil Dupras
# Created On: 2009-04-25
# $Id$
# Copyright 2009 Hardcoded Software (http://www.hardcoded.net)

import logging
import os.path as op
import traceback

from PyQt4.QtCore import Qt, QTimer, QObject, QCoreApplication, QUrl, SIGNAL
from PyQt4.QtGui import QProgressDialog, QDesktopServices, QFileDialog, QDialog, QMessageBox

from hsutil import job
from hsutil.reg import RegistrationRequired

from dupeguru import data_pe
from dupeguru.app import (DupeGuru as DupeGuruBase, JOB_SCAN, JOB_LOAD, JOB_MOVE, JOB_COPY, 
    JOB_DELETE)

from main_window import MainWindow
from directories_dialog import DirectoriesDialog
from about_box import AboutBox
from reg import Registration
from error_report_dialog import ErrorReportDialog

JOBID2TITLE = {
    JOB_SCAN: "Scanning for duplicates",
    JOB_LOAD: "Loading",
    JOB_MOVE: "Moving",
    JOB_COPY: "Copying",
    JOB_DELETE: "Sending files to the recycle bin",
}

class Progress(QProgressDialog, job.ThreadedJobPerformer):
    def __init__(self, parent):
        flags = Qt.CustomizeWindowHint | Qt.WindowTitleHint | Qt.WindowSystemMenuHint
        QProgressDialog.__init__(self, '', u"Cancel", 0, 100, parent, flags)
        self.setModal(True)
        self.setAutoReset(False)
        self.setAutoClose(False)
        self._timer = QTimer()
        self._jobid = ''
        self.connect(self._timer, SIGNAL('timeout()'), self.updateProgress)
    
    def updateProgress(self):
        # the values might change before setValue happens
        last_progress = self.last_progress
        last_desc = self.last_desc
        if not self._job_running or last_progress is None:
            self._timer.stop()
            self.close()
            self.emit(SIGNAL('finished(QString)'), self._jobid)
            if self._last_error is not None:
                s = ''.join(traceback.format_exception(*self._last_error))
                dialog = ErrorReportDialog(self.parent(), s)
                dialog.exec_()
            return
        if self.wasCanceled():
            self.job_cancelled = True
            return
        if last_desc:
            self.setLabelText(last_desc)
        self.setValue(last_progress)
    
    def run(self, jobid, title, target, args=()):
        self._jobid = jobid
        self.reset()
        self.setLabelText('')
        self.run_threaded(target, args)
        self.setWindowTitle(title)
        self.show()
        self._timer.start(500)
    

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 = '<replace this>'
    NAME = '<replace this>'
    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
    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.SaveIgnoreList()
    
    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)