X-Git-Url: https://plomlompom.com/repos/feed.xml?a=blobdiff_plain;f=plomtask%2Fhttp.py;h=316fd0805588a00b59dc28fbd8094fc8634dcd87;hb=f19823e9e77ae0017022dbfe63f66d2b065ba33a;hp=2b41db82e00e5d4cfac862161fb352607d09dfbb;hpb=a99b13325a21042825450d2497ddf61f8c5c3644;p=plomtask

diff --git a/plomtask/http.py b/plomtask/http.py
index 2b41db8..316fd08 100644
--- a/plomtask/http.py
+++ b/plomtask/http.py
@@ -127,10 +127,10 @@ class TaskHandler(BaseHTTPRequestHandler):
 
         date = self.params.get_str('date', todays_date())
         top_todos = [t for t in Todo.by_date(self.conn, date) if not t.parents]
-        seen_todos: set[int] = set()
-        seen_conditions: set[int] = set()
-        todo_trees = [t.get_step_tree(seen_todos, seen_conditions)
-                      for t in top_todos]
+        todo_trees = [t.get_undone_steps_tree() for t in top_todos]
+        done_trees = []
+        for t in top_todos:
+            done_trees += t.get_done_steps_tree()
         condition_listings: list[ConditionListing] = []
         for cond in Condition.all(self.conn):
             enablers = Todo.enablers_for_at(self.conn, cond, date)
@@ -138,6 +138,7 @@ class TaskHandler(BaseHTTPRequestHandler):
             condition_listings += [ConditionListing(cond, enablers, disablers)]
         return {'day': Day.by_id(self.conn, date, create=True),
                 'todo_trees': todo_trees,
+                'done_trees': done_trees,
                 'processes': Process.all(self.conn),
                 'condition_listings': condition_listings}
 
@@ -207,6 +208,10 @@ class TaskHandler(BaseHTTPRequestHandler):
             todo.adopt_from(existing_todos)
             todo.make_missing_children(self.conn)
             todo.save(self.conn)
+        for todo_id in self.form_data.get_all_int('done'):
+            todo = Todo.by_id(self.conn, todo_id)
+            todo.is_done = True
+            todo.save(self.conn)
         return f'/day?date={date}'
 
     def do_POST_todo(self) -> str: