class Tasks(Visitor): def __init__(self, root_folder, root_context): Visitor.__init__(self) self.root_folder = root_folder self.root_context = root_context self.project = Project(name="Tasks") self.context = Context(name="Tasks") def end_project(self, item): for child in item.children: self.project.add_child(child) item.children = [] def end_folder(self, item): if item == self.root_folder: self.root_folder.children = [] self.root_folder.add_child(self.project) def end_context(self, item): if item != self.root_context: for child in item.children: if child.type == TASK: self.context.add_child(child) item.children = [] else: self.root_context.children = [] self.root_context.add_child(self.context) def __str__(self): return "Tasks"
class Tasks (Visitor): def __init__(self, root_folder, root_context): Visitor.__init__(self) self.root_folder = root_folder self.root_context = root_context self.project = Project (name='Tasks') self.context = Context (name='Tasks') def end_project (self, item): for child in item.children: self.project.add_child(child) item.children = [] def end_folder (self, item): if item == self.root_folder: self.root_folder.children = [] self.root_folder.add_child(self.project) def end_context (self, item): if item != self.root_context: for child in item.children: if child.type == TASK: self.context.add_child(child) item.children = [] else: self.root_context.children = [] self.root_context.add_child(self.context) def __str__ (self): return 'Tasks'
def test_ContextNameFilterVisitor_include_ignores_children(self): n1 = Context(name=u'n1 xxx') n2 = Context(name=u'n2') n1.add_child(n2) visitor = ContextNameFilterVisitor('xxx') traverse(visitor, n1) self.assertTrue(n1.marked) self.assertTrue(n2.marked)
def test_ContextNameFilterVisitor_include_ignores_children (self): n1 = Context (name=u'n1 xxx') n2 = Context (name=u'n2') n1.add_child(n2) visitor = ContextNameFilterVisitor ('xxx') traverse (visitor, n1) self.assertTrue(n1.marked) self.assertTrue(n2.marked)
def test_context_add_child (self): context = Context () child = Context () context.add_child (child) self.assertEquals(context, child.parent) context = Context () child = Task () context.add_child (child) self.assertEquals(context, child.context) self.assertEquals(None, child.parent)
def test_context_add_child(self): context = Context() child = Context() context.add_child(child) self.assertEquals(context, child.parent) context = Context() child = Task() context.add_child(child) self.assertEquals(context, child.context) self.assertEquals(None, child.parent)
def test_Scenario_4(self): ''' In context mode select a single deeply nested task for exclusion ''' f1 = Folder(name='1') p1 = Project(name='2', parent=f1) t1 = Task(name='3', parent=p1) t2_selected = Task(name='xxx', parent=t1) t3 = Task(name='5', parent=t2_selected) t4 = Task(name='3', parent=t1) f2 = Folder(name='1', parent=f1) p2 = Project(name='2', parent=f2) t5 = Task(name='3', parent=p2) t6 = Task(name='3', parent=t5) t7 = Task(name='5', parent=t6) c1_on_path = Context(name='6') c2_on_path = Context(name='6', parent=c1_on_path) c2_on_path.add_child(t2_selected) c3 = Context(name='7', parent=c2_on_path) traverse_list(Filter([PROJECT, CONTEXT, TASK, FOLDER], lambda x: match_name(x, 'xxx'), False, 'pretty'), [c1_on_path], project_mode=False) self.assertTrue(f1.marked) self.assertTrue(p1.marked) self.assertTrue(t1.marked) self.assertFalse(t2_selected.marked) self.assertTrue(t3.marked) self.assertTrue(t4.marked) self.assertTrue(t5.marked) self.assertTrue(t6.marked) self.assertTrue(t7.marked) self.assertTrue(f2.marked) self.assertTrue(p2.marked) self.assertTrue(t5.marked) self.assertTrue(t6.marked) self.assertTrue(t7.marked) self.assertTrue(c1_on_path.marked) self.assertTrue(c2_on_path.marked) self.assertTrue(c3.marked)
def test_Scenario_4 (self): ''' In context mode select a single deeply nested task for exclusion ''' f1 = Folder (name='1') p1 = Project (name='2', parent=f1) t1 = Task (name='3', parent=p1) t2_selected = Task (name='xxx', parent=t1) t3 = Task (name='5', parent=t2_selected) t4 = Task (name='3', parent=t1) f2 = Folder (name='1', parent=f1) p2 = Project (name='2', parent=f2) t5 = Task (name='3', parent=p2) t6 = Task (name='3', parent=t5) t7 = Task (name='5', parent=t6) c1_on_path = Context (name='6') c2_on_path = Context (name='6', parent=c1_on_path) c2_on_path.add_child(t2_selected) c3 = Context (name='7', parent=c2_on_path) traverse_list (TaskNameFilterVisitor ('xxx', include=False), [c1_on_path], project_mode=False) self.assertTrue(f1.marked) self.assertTrue(p1.marked) self.assertTrue(t1.marked) self.assertFalse(t2_selected.marked) self.assertTrue(t3.marked) self.assertTrue(t4.marked) self.assertTrue(t5.marked) self.assertTrue(t6.marked) self.assertTrue(t7.marked) self.assertTrue(f2.marked) self.assertTrue(p2.marked) self.assertTrue(t5.marked) self.assertTrue(t6.marked) self.assertTrue(t7.marked) self.assertTrue(c1_on_path.marked) self.assertTrue(c2_on_path.marked) self.assertTrue(c3.marked)
def test_Scenario_4 (self): ''' In context mode select a single deeply nested task for exclusion ''' f1 = Folder (name='1') p1 = Project (name='2', parent=f1) t1 = Task (name='3', parent=p1) t2_selected = Task (name='xxx', parent=t1) t3 = Task (name='5', parent=t2_selected) t4 = Task (name='3', parent=t1) f2 = Folder (name='1', parent=f1) p2 = Project (name='2', parent=f2) t5 = Task (name='3', parent=p2) t6 = Task (name='3', parent=t5) t7 = Task (name='5', parent=t6) c1_on_path = Context (name='6') c2_on_path = Context (name='6', parent=c1_on_path) c2_on_path.add_child(t2_selected) c3 = Context (name='7', parent=c2_on_path) traverse_list (Filter ([PROJECT, CONTEXT, TASK, FOLDER], lambda x: match_name(x, 'xxx'), False, 'pretty'), [c1_on_path], project_mode=False) self.assertTrue(f1.marked) self.assertTrue(p1.marked) self.assertTrue(t1.marked) self.assertFalse(t2_selected.marked) self.assertTrue(t3.marked) self.assertTrue(t4.marked) self.assertTrue(t5.marked) self.assertTrue(t6.marked) self.assertTrue(t7.marked) self.assertTrue(f2.marked) self.assertTrue(p2.marked) self.assertTrue(t5.marked) self.assertTrue(t6.marked) self.assertTrue(t7.marked) self.assertTrue(c1_on_path.marked) self.assertTrue(c2_on_path.marked) self.assertTrue(c3.marked)
def test_Scenario_2(self): ''' In project mode select a single deeply nested task for exclusion ''' f_on_path = Folder(name='1') p_on_path = Project(name='2', parent=f_on_path) t1_on_path = Task(name='3', parent=p_on_path) t2_selected = Task(name='xxx', parent=t1_on_path) t3_on_path = Task(name='5', parent=t2_selected) t4 = Task(name='3', parent=t1_on_path) f = Folder(name='1', parent=f_on_path) p = Project(name='2', parent=f) t1 = Task(name='3', parent=p) t2 = Task(name='3', parent=t1) t3 = Task(name='5', parent=t2) c1 = Context(name='6') c1.add_child(t2_selected) c2 = Context(name='7') c2.add_child(t3) traverse_list( Filter([PROJECT, CONTEXT, TASK, FOLDER], lambda x: match_name(x, 'xxx'), False, 'pretty'), [f_on_path]) self.assertTrue(f_on_path.marked) self.assertTrue(p_on_path.marked) self.assertTrue(t1_on_path.marked) self.assertFalse(t2_selected.marked) self.assertFalse(t3_on_path.marked) self.assertTrue(c1.marked) self.assertTrue(c2.marked) self.assertTrue(f.marked) self.assertTrue(p.marked) self.assertTrue(t1.marked) self.assertTrue(t2.marked) self.assertTrue(t3.marked) self.assertTrue(t4.marked)
def test_Scenario_2 (self): ''' In project mode select a single deeply nested task for exclusion ''' f_on_path = Folder (name='1') p_on_path = Project (name='2', parent=f_on_path) t1_on_path = Task (name='3', parent=p_on_path) t2_selected = Task (name='xxx', parent=t1_on_path) t3_on_path = Task (name='5', parent=t2_selected) t4 = Task (name='3', parent=t1_on_path) f = Folder (name='1', parent=f_on_path) p = Project (name='2', parent=f) t1 = Task (name='3', parent=p) t2 = Task (name='3', parent=t1) t3 = Task (name='5', parent=t2) c1 = Context (name='6') c1.add_child(t2_selected) c2 = Context (name='7') c2.add_child(t3) traverse_list (TaskNameFilterVisitor ('xxx', include=False), [f_on_path]) self.assertTrue(f_on_path.marked) self.assertTrue(p_on_path.marked) self.assertTrue(t1_on_path.marked) self.assertFalse(t2_selected.marked) self.assertFalse(t3_on_path.marked) self.assertTrue(c1.marked) self.assertTrue(c2.marked) self.assertTrue(f.marked) self.assertTrue(p.marked) self.assertTrue(t1.marked) self.assertTrue(t2.marked) self.assertTrue(t3.marked) self.assertTrue(t4.marked)
def test_Scenario_1 (self): ''' In project mode select a single deeply nested task for inclusion ''' f_on_path = Folder (name='1') p_on_path = Project (name='2', parent=f_on_path) t1_on_path = Task (name='3', parent=p_on_path) t2_selected = Task (name='xxx', parent=t1_on_path) t3_on_path = Task (name='5', parent=t2_selected) t4 = Task (name='3', parent=t1_on_path) f = Folder (name='1', parent=f_on_path) p = Project (name='2', parent=f) t1 = Task (name='3', parent=p) t2 = Task (name='3', parent=t1) t3 = Task (name='5', parent=t2) c1 = Context (name='6') c1.add_child(t2_selected) c2 = Context (name='7') c2.add_child(t3) traverse_list (Filter ([PROJECT, CONTEXT, TASK, FOLDER], lambda x: match_name(x, 'xxx'), True, 'pretty'), [f_on_path]) self.assertTrue(f_on_path.marked) self.assertTrue(p_on_path.marked) self.assertTrue(t1_on_path.marked) self.assertTrue(t2_selected.marked) self.assertTrue(t3_on_path.marked) self.assertTrue(c1.marked) self.assertTrue(c2.marked) self.assertFalse(f.marked) self.assertFalse(p.marked) self.assertFalse(t1.marked) self.assertFalse(t2.marked) self.assertFalse(t3.marked) self.assertFalse(t4.marked)
def build_model (db): conn = sqlite3.connect(db) contexts = query (conn, clazz=OFContext) no_context = Context(name = 'No Context') inbox = Project (name='Inbox') project_infos = query (conn, clazz=ProjectInfo) folders = query (conn, clazz=OFFolder) tasks = query (conn, clazz=OFTask) projects = transmute_projects (project_infos, tasks) wire_projects_and_folders(projects, folders, tasks) wire_task_hierarchy(tasks) wire_tasks_to_enclosing_projects (project_infos, tasks, inbox) wire_tasks_and_contexts(contexts, tasks, no_context) wire_folder_hierarchy (folders) wire_context_hierarchy (contexts) conn.close () # Find top level items project_roots = only_roots (projects.values()) folder_roots = only_roots (folders.values()) root_projects_and_folders = project_roots + folder_roots root_contexts = only_roots (contexts.values()) root_contexts.insert(0, no_context) root_projects_and_folders.insert(0, inbox) sort(root_projects_and_folders) sort(root_contexts) root_folder = Folder (name='') for child in root_projects_and_folders: root_folder.add_child(child) root_context = Context (name='', status='active') for child in root_contexts: root_context.add_child(child) return root_folder, root_context
def build_model(db): conn = sqlite3.connect(db) contexts = query(conn, clazz=OFContext) no_context = Context(name='No Context') inbox = Project(name='Inbox') project_infos = query(conn, clazz=ProjectInfo) folders = query(conn, clazz=OFFolder) tasks = query(conn, clazz=OFTask) projects = transmute_projects(project_infos, tasks) wire_projects_and_folders(projects, folders, tasks) wire_task_hierarchy(tasks) wire_tasks_to_enclosing_projects(project_infos, tasks, inbox) wire_tasks_and_contexts(contexts, tasks, no_context) wire_folder_hierarchy(folders) wire_context_hierarchy(contexts) conn.close() # Find top level items project_roots = only_roots(projects.values()) folder_roots = only_roots(folders.values()) root_projects_and_folders = project_roots + folder_roots root_contexts = only_roots(contexts.values()) root_contexts.insert(0, no_context) root_projects_and_folders.insert(0, inbox) sort(root_projects_and_folders) sort(root_contexts) root_folder = Folder(name='') for child in root_projects_and_folders: root_folder.add_child(child) root_context = Context(name='', status='active') for child in root_contexts: root_context.add_child(child) return root_folder, root_context