[#138] Added ME-specific prioritizations and adjusted ME cocoa proj to the pri dialog.

core/data.py

@@ -7,10 +7,10 @@
 # http://www.hardcoded.net/licenses/bsd_license
 
 from collections import namedtuple
+import time
 
 from hscommon.util import format_time_decimal, format_size
-
-import time
+from . import prioritize
 
 Column = namedtuple('Column', 'attr display')
 
@@ -40,3 +40,6 @@ def format_dupe_count(c):
 
 def cmp_value(value):
     return value.lower() if isinstance(value, str) else value
+
+def prioritization_categories():
+    return prioritize.all_categories()

core/gui/prioritize_dialog.py

@@ -8,8 +8,6 @@
 
 from hscommon.gui.selectable_list import GUISelectableList
 
-from ..prioritize import all_categories
-
 class CriterionCategoryList(GUISelectableList):
     def __init__(self, dialog):
         self.dialog = dialog
@@ -44,7 +42,7 @@ class PrioritizationList(GUISelectableList):
 class PrioritizeDialog:
     def __init__(self, view, app):
         self.app = app
-        self.categories = [cat(app.results) for cat in all_categories()]
+        self.categories = [cat(app.results) for cat in app.data.prioritization_categories()]
         self.category_list = CriterionCategoryList(self)
         self.criteria = []
         self.criteria_list = GUISelectableList()

core/tests/data.py

@@ -10,6 +10,7 @@
 
 from hscommon.util import format_size
 from ..data import cmp_value, Column
+from .. import prioritize
 
 COLUMNS = [
     Column('name', 'Filename'),
@@ -41,4 +42,7 @@ def GetDupeSortKey(dupe, get_group, key, delta):
     return r
 
 def GetGroupSortKey(group, key):
-    return cmp_value(getattr(group.ref, COLUMNS[key].attr))
+    return cmp_value(getattr(group.ref, COLUMNS[key].attr))
+
+def prioritization_categories():
+    return prioritize.all_categories()