From d77d0661697655a7c2d0e9e0b270cf21e286d7fb Mon Sep 17 00:00:00 2001 From: Christian Heller Date: Sat, 13 Apr 2024 03:20:33 +0200 Subject: [PATCH] Add Todo.parents. --- plomtask/todos.py | 5 +++++ tests/todos.py | 4 ++++ 2 files changed, 9 insertions(+) diff --git a/plomtask/todos.py b/plomtask/todos.py index 7faea73..e8e00ac 100644 --- a/plomtask/todos.py +++ b/plomtask/todos.py @@ -18,6 +18,7 @@ class Todo: self.is_done = is_done self.day = day self.children: list[Todo] = [] + self.parents: list[Todo] = [] @classmethod def from_table_row(cls, db_conn: DatabaseConnection, row: Row) -> Todo: @@ -46,6 +47,9 @@ class Todo: for row in db_conn.exec('SELECT child FROM todo_children ' 'WHERE parent = ?', (id_,)): todo.children += [cls.by_id(db_conn, row[0])] + for row in db_conn.exec('SELECT parent FROM todo_children ' + 'WHERE child = ?', (id_,)): + todo.parents += [cls.by_id(db_conn, row[0])] assert isinstance(todo, Todo) return todo @@ -72,6 +76,7 @@ class Todo: raise BadFormatException('cannot adopt same child twice') walk_steps(child) self.children += [child] + child.parents += [self] def save(self, db_conn: DatabaseConnection) -> None: """Write self and children to DB and its cache.""" diff --git a/tests/todos.py b/tests/todos.py index 7ae6b64..a377920 100644 --- a/tests/todos.py +++ b/tests/todos.py @@ -118,6 +118,7 @@ class TestsWithServer(TestCaseWithServer): self.check_post(form_data, '/todo?id=1', 302, '/') todo1 = Todo.by_date(self.db_conn, '2024-01-01')[0] self.assertEqual(todo1.children, []) + self.assertEqual(todo1.parents, []) form_data = {'adopt': 'foo'} self.check_post(form_data, '/todo?id=1', 400) form_data = {'adopt': 1} @@ -132,6 +133,9 @@ class TestsWithServer(TestCaseWithServer): todo1 = Todo.by_date(self.db_conn, '2024-01-01')[0] todo2 = Todo.by_date(self.db_conn, '2024-01-01')[1] self.assertEqual(todo1.children, [todo2]) + self.assertEqual(todo1.parents, []) + self.assertEqual(todo2.children, []) + self.assertEqual(todo2.parents, [todo1]) self.check_post(form_data, '/todo?id=1', 400, '/') def test_do_GET_todo(self) -> None: -- 2.30.2