1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
'''
Task Coach - Your friendly task manager
Copyright (C) 2004-2014 Task Coach developers <developers@taskcoach.org>
Task Coach is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
Task Coach is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
'''
from taskcoachlib.domain import base
from wx.lib.pubsub import pub
import task
class Sorter(base.TreeSorter):
DomainObjectClass = task.Task # What are we sorting
TaskStatusAttributes = ('prerequisites', 'dueDateTime',
'plannedStartDateTime', 'actualStartDateTime',
'completionDateTime')
def __init__(self, *args, **kwargs):
self.__treeMode = kwargs.pop('treeMode', False)
self.__sortByTaskStatusFirst = kwargs.pop('sortByTaskStatusFirst', True)
super(Sorter, self).__init__(*args, **kwargs)
for eventType in (task.Task.prerequisitesChangedEventType(),
task.Task.dueDateTimeChangedEventType(),
task.Task.plannedStartDateTimeChangedEventType(),
task.Task.actualStartDateTimeChangedEventType(),
task.Task.completionDateTimeChangedEventType()):
pub.subscribe(self.onAttributeChanged, eventType)
def setTreeMode(self, treeMode=True):
self.__treeMode = treeMode
try:
self.observable().setTreeMode(treeMode)
except AttributeError:
pass
self.reset(forceEvent=True)
def treeMode(self):
return self.__treeMode
def sortByTaskStatusFirst(self, sortByTaskStatusFirst):
self.__sortByTaskStatusFirst = sortByTaskStatusFirst
# We don't need to invoke self.reset() here since when this property is
# changed, the sort order also changes which in turn will cause
# self.reset() to be called.
def createSortKeyFunction(self, sortKey):
statusSortKey = self.__createStatusSortKey()
regularSortKey = super(Sorter, self).createSortKeyFunction(sortKey)
return lambda task: statusSortKey(task) + [regularSortKey(task)]
def __createStatusSortKey(self):
if self.__sortByTaskStatusFirst:
if self.isAscending():
return lambda task: [task.completed(), task.inactive()]
else:
return lambda task: [not task.completed(), not task.inactive()]
else:
return lambda task: []
def _registerObserverForAttribute(self, attribute):
# Sorter is always observing task dates and prerequisites because
# sorting by status depends on those attributes. Hence we don't need
# to subscribe to these attributes when they become the sort key.
if attribute not in self.TaskStatusAttributes:
super(Sorter, self)._registerObserverForAttribute(attribute)
def _removeObserverForAttribute(self, attribute):
# See comment at _registerObserverForAttribute.
if attribute not in self.TaskStatusAttributes:
super(Sorter, self)._removeObserverForAttribute(attribute)
|
