def count_tasks_by_project(self, project_name: str,
**kwargs) -> List[Snapshot]:
assert type(project_name) is str
snapshots = Snapshots(self.tasks)
snapshots.count_tasks_by_project(project_name)
return self.display_snapshots(snapshots, **kwargs)
def count_tasks_by_due_date(self, due_date: str,
**kwargs) -> List[Snapshot]:
assert type(due_date) is str
snapshots = Snapshots(self.tasks)
snapshots.count_tasks_by_due_date(due_date)
return self.display_snapshots(snapshots, **kwargs)
def count_tasks_by_due_date_range(self, min_date: str, max_date: str,
**kwargs) -> List[Snapshot]:
assert type(min_date) and type(max_date) is str
snapshots = Snapshots(self.tasks)
snapshots.count_tasks_by_due_date_range(min_date, max_date)
return self.display_snapshots(snapshots, **kwargs)
def test_invalid_operation(self):
snapshots = Snapshots(self.tasks)
summary, snapshot_list = snapshots.get_snapshot()
self.assertIsNotNone(summary)
self.assertEqual(summary.count, 0)
self.assertEqual(summary.completed, 0)
self.assertEqual(summary.incomplete, 0)
self.assertEqual(summary.deleted, 0)
self.assertTrue(len(snapshot_list) == 0)
def test_count_project(self):
self.tasks.add("task1", "label1", "project1", "today")
self.tasks.add("task2", "label1", "project2", "today")
self.tasks.add("task3", "label1", "project2", "today")
self.tasks.add("task4", "label1", "project2", "today")
snapshots = Snapshots(self.tasks)
snapshots.count_tasks_by_project("project2")
summary, snapshot_list = snapshots.get_snapshot()
self.assertIsNotNone(summary)
self.assertEqual(summary.count, 3)
self.assertEqual(summary.completed, 0)
self.assertEqual(summary.incomplete, 3)
self.assertEqual(summary.deleted, 0)
self.assertTrue(len(snapshot_list) == 1)
def test_count_by_due_date_range(self):
self.tasks.add("task1", "label1", "project1", "2021-07-14")
self.tasks.add("task2", "label1", "project1", "2021-09-01")
self.tasks.add("task3", "label1", "project1", "2021-09-01")
self.tasks.add("task4", "label1", "project1", "2021-11-01")
snapshots = Snapshots(self.tasks)
snapshots.count_tasks_by_due_date_range("2021-07-13", "2021-11-02")
summary, snapshot_list = snapshots.get_snapshot()
self.assertIsNotNone(summary)
self.assertEqual(summary.count, 4)
self.assertEqual(summary.completed, 0)
self.assertEqual(summary.incomplete, 4)
self.assertEqual(summary.deleted, 0)
self.assertTrue(len(snapshot_list) == 3)
def display_snapshots(self, snapshots: Snapshots, **kwargs):
summary, snapshot_list = snapshots.get_snapshot()
return {
"snapshot": {
"summary": summary.compose_summary(),
"list": [dict(snapshot) for snapshot in snapshot_list]
}
}
def test_count_all(self):
self.tasks.add("task1", "label1", "project1", "today")
t1 = self.tasks.get_task_by_name("task1")
self.tasks.complete(t1.unique_id)
self.tasks.add("task2", "label1", "project1", "today")
t2 = self.tasks.get_task_by_name("task2")
self.tasks.delete(t2.unique_id)
self.tasks.add("task3", "label1", "project1", "today")
self.tasks.add("task4", "label1", "project1", "today")
snapshots = Snapshots(self.tasks)
snapshots.count_all_tasks()
summary, snapshot_list = snapshots.get_snapshot()
self.assertIsNotNone(summary)
self.assertTrue(summary.count == 4)
self.assertTrue(summary.deleted == 1)
self.assertTrue(summary.incomplete == 2)
self.assertTrue(summary.completed == 1)
def count_all_tasks(self, **kwargs) -> List[Snapshot]:
snapshots = Snapshots(self.tasks)
snapshots.count_all_tasks()
return self.display_snapshots(snapshots, **kwargs)
def display_snapshots(self, snapshots: Snapshots, **kwargs):
summary, snapshot_list = snapshots.get_snapshot()
self.__print_snapshot_summary_table(summary)
return self.__print_snapshot_list_table(snapshot_list,
kwargs.get("page"))