dupeguru/cocoa/inter/app.py

import logging

from objp.util import pyref, dontwrap
from hscommon.path import Path, pathify
from cocoa import install_exception_hook, install_cocoa_logger, patch_threaded_job_performer
from cocoa.inter import PyBaseApp, BaseAppView

import core.pe.photo
from core.app import DupeGuru as DupeGuruBase, AppMode
from .directories import Directories, Bundle
from .photo import Photo

class DupeGuru(DupeGuruBase):
    def __init__(self, view):
        DupeGuruBase.__init__(self, view)
        self.directories = Directories()
    
    def selected_dupe_path(self):
        if not self.selected_dupes:
            return None
        return self.selected_dupes[0].path
    
    def selected_dupe_ref_path(self):
        if not self.selected_dupes:
            return None
        ref = self.results.get_group_of_duplicate(self.selected_dupes[0]).ref
        if ref is self.selected_dupes[0]: # we don't want the same pic to be displayed on both sides
            return None
        return ref.path
    
    def _get_fileclasses(self):
        result = DupeGuruBase._get_fileclasses(self)
        if self.app_mode == AppMode.Standard:
            result = [Bundle] + result
        return result

class DupeGuruView(BaseAppView):
    def askYesNoWithPrompt_(self, prompt: str) -> bool: pass
    def createResultsWindow(self): pass
    def showResultsWindow(self): pass
    def showProblemDialog(self): pass
    def selectDestFolderWithPrompt_(self, prompt: str) -> str: pass
    def selectDestFileWithPrompt_extension_(self, prompt: str, extension: str) -> str: pass

class PyDupeGuru(PyBaseApp):
    @dontwrap
    def __init__(self):
        core.pe.photo.PLAT_SPECIFIC_PHOTO_CLASS = Photo
        logging.basicConfig(level=logging.WARNING, format='%(levelname)s %(message)s')
        install_exception_hook('https://github.com/hsoft/dupeguru/issues')
        install_cocoa_logger()
        patch_threaded_job_performer()
        self.model = DupeGuru(self)
    
    #---Sub-proxies
    def detailsPanel(self) -> pyref:
        return self.model.details_panel
    
    def directoryTree(self) -> pyref:
        return self.model.directory_tree
    
    def problemDialog(self) -> pyref:
        return self.model.problem_dialog
    
    def statsLabel(self) -> pyref:
        return self.model.stats_label
    
    def resultTable(self) -> pyref:
        return self.model.result_table
    
    def ignoreListDialog(self) -> pyref:
        return self.model.ignore_list_dialog
    
    def progressWindow(self) -> pyref:
        return self.model.progress_window
    
    def deletionOptions(self) -> pyref:
        return self.model.deletion_options
    
    #---Directories
    def addDirectory_(self, directory: str):
        self.model.add_directory(directory)
    
    #---Results
    def doScan(self):
        self.model.start_scanning()
    
    def exportToXHTML(self):
        self.model.export_to_xhtml()
    
    def exportToCSV(self):
        self.model.export_to_csv()
    
    def loadSession(self):
        self.model.load()
    
    def loadResultsFrom_(self, filename: str):
        self.model.load_from(filename)
    
    def markAll(self):
        self.model.mark_all()
    
    def markNone(self):
        self.model.mark_none()
    
    def markInvert(self):
        self.model.mark_invert()
    
    def purgeIgnoreList(self):
        self.model.purge_ignore_list()
    
    def toggleSelectedMark(self):
        self.model.toggle_selected_mark_state()
    
    def saveSession(self):
        self.model.save()
    
    def saveResultsAs_(self, filename: str):
        self.model.save_as(filename)
    
    #---Actions
    def addSelectedToIgnoreList(self):
        self.model.add_selected_to_ignore_list()
    
    def deleteMarked(self):
        self.model.delete_marked()
    
    def applyFilter_(self, filter: str):
        self.model.apply_filter(filter)
    
    def makeSelectedReference(self):
        self.model.make_selected_reference()
    
    def copyMarked(self):
        self.model.copy_or_move_marked(copy=True)
    
    def moveMarked(self):
        self.model.copy_or_move_marked(copy=False)
    
    def openSelected(self):
        self.model.open_selected()
    
    def removeMarked(self):
        self.model.remove_marked()
    
    def removeSelected(self):
        self.model.remove_selected()
    
    def revealSelected(self):
        self.model.reveal_selected()

    def invokeCustomCommand(self):
        self.model.invoke_custom_command()
    
    def showIgnoreList(self):
        self.model.ignore_list_dialog.show()
    
    def clearPictureCache(self):
        self.model.clear_picture_cache()
    
    #---Information
    def getScanOptions(self) -> list:
        return [o.label for o in self.model.SCANNER_CLASS.get_scan_options()]
    
    def resultsAreModified(self) -> bool:
        return self.model.results.is_modified
    
    def getSelectedDupePath(self) -> str:
        return str(self.model.selected_dupe_path())
    
    def getSelectedDupeRefPath(self) -> str:
        return str(self.model.selected_dupe_ref_path())
    
    #---Properties
    def getAppMode(self) -> int:
        return self.model.app_mode

    def setAppMode_(self, app_mode: int):
        self.model.app_mode = app_mode
    
    def setScanType_(self, scan_type_index: int):
        scan_options = self.model.SCANNER_CLASS.get_scan_options()
        try:
            so = scan_options[scan_type_index]
            self.model.options['scan_type'] = so.scan_type
        except IndexError:
            pass

    def setMinMatchPercentage_(self, percentage: int):
        self.model.options['min_match_percentage'] = int(percentage)
    
    def setWordWeighting_(self, words_are_weighted: bool):
        self.model.options['word_weighting'] = words_are_weighted
    
    def setMatchSimilarWords_(self, match_similar_words: bool):
        self.model.options['match_similar_words'] = match_similar_words
    
    def setSizeThreshold_(self, size_threshold: int):
        self.model.options['size_threshold'] = size_threshold

    def enable_scanForTag_(self, enable: bool, scan_tag: str):
        if 'scanned_tags' not in self.model.options:
            self.model.options['scanned_tags'] = set()
        if enable:
            self.model.options['scanned_tags'].add(scan_tag)
        else:
            self.model.options['scanned_tags'].discard(scan_tag)

    def setMatchScaled_(self, match_scaled: bool):
        self.model.options['match_scaled'] = match_scaled
    
    def setMixFileKind_(self, mix_file_kind: bool):
        self.model.options['mix_file_kind'] = mix_file_kind
    
    def setEscapeFilterRegexp_(self, escape_filter_regexp: bool):
        self.model.options['escape_filter_regexp'] = escape_filter_regexp
    
    def setRemoveEmptyFolders_(self, remove_empty_folders: bool):
        self.model.options['clean_empty_dirs'] = remove_empty_folders
    
    def setIgnoreHardlinkMatches_(self, ignore_hardlink_matches: bool):
        self.model.options['ignore_hardlink_matches'] = ignore_hardlink_matches
    
    def setCopyMoveDestType_(self, copymove_dest_type: int):
        self.model.options['copymove_dest_type'] = copymove_dest_type
    
    #--- model --> view
    @dontwrap
    def ask_yes_no(self, prompt):
        return self.callback.askYesNoWithPrompt_(prompt)
    
    @dontwrap
    def create_results_window(self):
        self.callback.createResultsWindow()
    
    @dontwrap
    def show_results_window(self):
        self.callback.showResultsWindow()
    
    @dontwrap
    def show_problem_dialog(self):
        self.callback.showProblemDialog()
    
    @dontwrap
    def select_dest_folder(self, prompt):
        return self.callback.selectDestFolderWithPrompt_(prompt)
    
    @dontwrap
    def select_dest_file(self, prompt, extension):
        return self.callback.selectDestFileWithPrompt_extension_(prompt, extension)