X-Git-Url: https://plomlompom.com/repos/feed.xml?a=blobdiff_plain;f=plomtask%2Ftodos.py;h=2b9fd1d8edb2fa159f3955f469a6b3d394aa1efa;hb=12195a1fe2c15334c866036eaa171eb8e92f0408;hp=7faea73b74df9c4b5a8f68f6dceced09f27c9b67;hpb=63249f5d7cefb97574610848ca3473cb1f9687e2;p=plomtask diff --git a/plomtask/todos.py b/plomtask/todos.py index 7faea73..2b9fd1d 100644 --- a/plomtask/todos.py +++ b/plomtask/todos.py @@ -18,13 +18,14 @@ 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: """Make Todo from database row, write to DB cache.""" todo = cls(id_=row[0], process=Process.by_id(db_conn, row[1]), - is_done=row[2], + is_done=bool(row[2]), day=Day.by_date(db_conn, row[3])) assert todo.id_ is not None db_conn.cached_todos[todo.id_] = 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."""