mirror of
https://github.com/arsenetar/dupeguru.git
synced 2024-11-14 03:29:02 +00:00
90 lines
3.4 KiB
Python
90 lines
3.4 KiB
Python
# Created By: Virgil Dupras
|
|
# Created On: 2006/11/11
|
|
# Copyright 2010 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 hsutil import cocoa, job
|
|
from hsutil.cocoa import install_exception_hook
|
|
from hsutil.cocoa.objcmin import (NSNotificationCenter, NSUserDefaults,
|
|
NSSearchPathForDirectoriesInDomains, NSApplicationSupportDirectory, NSUserDomainMask,
|
|
NSWorkspace, NSWorkspaceRecycleOperation)
|
|
from hsutil.reg import RegistrationRequired
|
|
|
|
from . import app, fs
|
|
|
|
JOBID2TITLE = {
|
|
app.JOB_SCAN: "Scanning for duplicates",
|
|
app.JOB_LOAD: "Loading",
|
|
app.JOB_MOVE: "Moving",
|
|
app.JOB_COPY: "Copying",
|
|
app.JOB_DELETE: "Sending to Trash",
|
|
}
|
|
|
|
def demo_method(method):
|
|
def wrapper(self, *args, **kwargs):
|
|
try:
|
|
return method(self, *args, **kwargs)
|
|
except RegistrationRequired:
|
|
NSNotificationCenter.defaultCenter().postNotificationName_object_('RegistrationRequired', self)
|
|
|
|
return wrapper
|
|
|
|
class DupeGuru(app.DupeGuru):
|
|
def __init__(self, data_module, appdata_subdir, appid):
|
|
LOGGING_LEVEL = logging.DEBUG if NSUserDefaults.standardUserDefaults().boolForKey_('debug') else logging.WARNING
|
|
logging.basicConfig(level=LOGGING_LEVEL, format='%(levelname)s %(message)s')
|
|
logging.debug('started in debug mode')
|
|
install_exception_hook()
|
|
appsupport = NSSearchPathForDirectoriesInDomains(NSApplicationSupportDirectory, NSUserDomainMask, True)[0]
|
|
appdata = op.join(appsupport, appdata_subdir)
|
|
app.DupeGuru.__init__(self, data_module, appdata, appid)
|
|
self.progress = cocoa.ThreadedJobPerformer()
|
|
|
|
#--- Override
|
|
@staticmethod
|
|
def _open_path(path):
|
|
NSWorkspace.sharedWorkspace().openFile_(unicode(path))
|
|
|
|
@staticmethod
|
|
def _recycle_dupe(dupe):
|
|
# local import because first appkit import takes a lot of memory. we want to avoid it.
|
|
directory = unicode(dupe.path[:-1])
|
|
filename = dupe.name
|
|
result, tag = NSWorkspace.sharedWorkspace().performFileOperation_source_destination_files_tag_(
|
|
NSWorkspaceRecycleOperation, directory, '', [filename], None)
|
|
|
|
@staticmethod
|
|
def _reveal_path(path):
|
|
NSWorkspace.sharedWorkspace().selectFile_inFileViewerRootedAtPath_(unicode(path), '')
|
|
|
|
def _start_job(self, jobid, func):
|
|
try:
|
|
j = self.progress.create_job()
|
|
self.progress.run_threaded(func, args=(j, ))
|
|
except job.JobInProgressError:
|
|
NSNotificationCenter.defaultCenter().postNotificationName_object_('JobInProgress', self)
|
|
else:
|
|
ud = {'desc': JOBID2TITLE[jobid], 'jobid':jobid}
|
|
NSNotificationCenter.defaultCenter().postNotificationName_object_userInfo_('JobStarted', self, ud)
|
|
|
|
#---Public
|
|
copy_or_move_marked = demo_method(app.DupeGuru.copy_or_move_marked)
|
|
delete_marked = demo_method(app.DupeGuru.delete_marked)
|
|
|
|
def start_scanning(self):
|
|
self._select_dupes([])
|
|
try:
|
|
app.DupeGuru.start_scanning(self)
|
|
return 0
|
|
except app.NoScannableFileError:
|
|
return 3
|
|
except app.AllFilesAreRefError:
|
|
return 1
|
|
|