dupeguru/core/gui/prioritize_dialog.py

# Created By: Virgil Dupras
# Created On: 2011-09-06
# Copyright 2011 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

from hscommon.gui.selectable_list import SelectableList

from ..prioritize import KindCategory

class CriterionCategoryList(SelectableList):
    def __init__(self, dialog):
        self.dialog = dialog
        SelectableList.__init__(self, [c.NAME for c in dialog.categories])
    
    def _update_selection(self):
        self.dialog.select_category(self.dialog.categories[self.selected_index])


class PrioritizeDialog:
    def __init__(self, view, app):
        self.app = app
        self.categories = [KindCategory(app.results)]
        self.category_list = CriterionCategoryList(self)
        self.criteria = []
        self.criteria_list = SelectableList()
        self.prioritizations = []
    
    #--- Private
    def _sort_key(self, dupe):
        # Our sort key consists of a tuple of inverted bool values represented as ints. When a dupe
        # fits a criteria, we want it at the top of the listm and thus we'll give it the value 0.
        # When the dupe doesn't fit a criteria, we ant it at the bottom, and we give the value 1.
        result = (crit.test_dupe(dupe) for crit in self.prioritizations)
        return tuple((0 if value else 1) for value in result)
    
    #--- Public
    def select_category(self, category):
        self.criteria = category.criteria_list()
        self.criteria_list[:] = [c.value for c in self.criteria]

    def add_selected(self):
        # Add selected criteria in criteria_list to prioritization_list.
        crit = self.criteria[self.criteria_list.selected_index]
        self.prioritizations.append(crit)
    
    def perform_reprioritization(self):
        self.app.reprioritize_groups(self._sort_key)