X-Git-Url: https://plomlompom.com/repos/berlin_corona.txt?a=blobdiff_plain;f=plomtask%2Fhttp.py;h=0aa692545d3e5ae38f4be7190a697df40259dcb0;hb=1ac1d92bba8caeeadf2f853cf15b9158c6024908;hp=44c710b3b9bd46e04f67a210d6c3e526b0d43724;hpb=4691a05e64acb2239f9d449b36f4ff5c678a1d6c;p=plomtask diff --git a/plomtask/http.py b/plomtask/http.py index 44c710b..0aa6925 100644 --- a/plomtask/http.py +++ b/plomtask/http.py @@ -1,4 +1,6 @@ """Web server stuff.""" +from __future__ import annotations +from dataclasses import dataclass from typing import Any from base64 import b64encode, b64decode from http.server import BaseHTTPRequestHandler @@ -11,13 +13,23 @@ from plomtask.days import Day from plomtask.exceptions import HandledException, BadFormatException, \ NotFoundException from plomtask.db import DatabaseConnection, DatabaseFile -from plomtask.processes import Process, ProcessStep +from plomtask.processes import Process, ProcessStep, ProcessStepsNode from plomtask.conditions import Condition from plomtask.todos import Todo TEMPLATES_DIR = 'templates' +@dataclass +class TodoStepsNode: + """Collect what's useful for Todo steps tree display.""" + id_: int + todo: Todo | None + process: Process | None + children: list[TodoStepsNode] + fillable: bool = False + + class TaskServer(HTTPServer): """Variant of HTTPServer that knows .jinja as Jinja Environment.""" @@ -46,7 +58,7 @@ class InputsParser: return self.inputs[key][0] def get_first_strings_starting(self, prefix: str) -> dict[str, str]: - """Retrieve list of (first) strings at key starting with prefix.""" + """Retrieve dict of (first) strings at key starting with prefix.""" ret = {} for key in [k for k in self.inputs.keys() if k.startswith(prefix)]: ret[key] = self.inputs[key][0] @@ -173,11 +185,68 @@ class TaskHandler(BaseHTTPRequestHandler): def do_GET_todo(self) -> dict[str, object]: """Show single Todo of ?id=.""" + + def walk_process_steps(id_: int, + process_step_nodes: list[ProcessStepsNode], + steps_nodes: list[TodoStepsNode]) -> None: + for process_step_node in process_step_nodes: + id_ += 1 + node = TodoStepsNode(id_, None, process_step_node.process, []) + steps_nodes += [node] + walk_process_steps(id_, list(process_step_node.steps.values()), + node.children) + + def walk_todo_steps(id_: int, todos: list[Todo], + steps_nodes: list[TodoStepsNode]) -> None: + for todo in todos: + matched = False + for match in [item for item in steps_nodes + if item.process + and item.process == todo.process]: + match.todo = todo + matched = True + for child in match.children: + child.fillable = True + walk_todo_steps(id_, todo.children, match.children) + if not matched: + id_ += 1 + node = TodoStepsNode(id_, todo, None, []) + steps_nodes += [node] + walk_todo_steps(id_, todo.children, node.children) + + def collect_adoptables_keys(steps_nodes: list[TodoStepsNode] + ) -> set[int]: + ids = set() + for node in steps_nodes: + if not node.todo: + assert isinstance(node.process, Process) + assert isinstance(node.process.id_, int) + ids.add(node.process.id_) + ids = ids | collect_adoptables_keys(node.children) + return ids + id_ = self.params.get_int('id') todo = Todo.by_id(self.conn, id_) - return {'todo': todo, + todo_steps = [step.todo for step in todo.get_step_tree(set()).children] + process_tree = todo.process.get_steps(self.conn, None) + steps_todo_to_process: list[TodoStepsNode] = [] + walk_process_steps(0, list(process_tree.values()), + steps_todo_to_process) + for steps_node in steps_todo_to_process: + steps_node.fillable = True + walk_todo_steps(len(steps_todo_to_process), todo_steps, + steps_todo_to_process) + adoptables: dict[int, list[Todo]] = {} + any_adoptables = [Todo.by_id(self.conn, t.id_) + for t in Todo.by_date(self.conn, todo.date) + if t != todo] + for id_ in collect_adoptables_keys(steps_todo_to_process): + adoptables[id_] = [t for t in any_adoptables + if t.process.id_ == id_] + return {'todo': todo, 'steps_todo_to_process': steps_todo_to_process, + 'adoption_candidates_for': adoptables, 'process_candidates': Process.all(self.conn), - 'todo_candidates': Todo.by_date(self.conn, todo.date), + 'todo_candidates': any_adoptables, 'condition_candidates': Condition.all(self.conn)} def do_GET_todos(self) -> dict[str, object]: @@ -370,17 +439,27 @@ class TaskHandler(BaseHTTPRequestHandler): return '/' todo = Todo.by_id(self.conn, id_) adopted_child_ids = self.form_data.get_all_int('adopt') + processes_to_make = self.form_data.get_all_int('make') + fill_fors = self.form_data.get_first_strings_starting('fill_for_') + for v in fill_fors.values(): + if v.startswith('make_'): + processes_to_make += [int(v[5:])] + elif v != 'ignore': + adopted_child_ids += [int(v)] + to_remove = [] for child in todo.children: + assert isinstance(child.id_, int) if child.id_ not in adopted_child_ids: - assert isinstance(child.id_, int) - child = Todo.by_id(self.conn, child.id_) - todo.remove_child(child) + to_remove += [child.id_] + for id_ in to_remove: + child = Todo.by_id(self.conn, id_) + todo.remove_child(child) for child_id in adopted_child_ids: if child_id in [c.id_ for c in todo.children]: continue child = Todo.by_id(self.conn, child_id) todo.add_child(child) - for process_id in self.form_data.get_all_int('make'): + for process_id in processes_to_make: made = Todo.create_with_children(self.conn, process_id, todo.date) todo.add_child(made) effort = self.form_data.get_str('effort', ignore_strict=True) @@ -389,8 +468,7 @@ class TaskHandler(BaseHTTPRequestHandler): todo.set_blockers(self.conn, self.form_data.get_all_int('blocker')) todo.set_enables(self.conn, self.form_data.get_all_int('enables')) todo.set_disables(self.conn, self.form_data.get_all_int('disables')) - if len(self.form_data.get_all_str('done')) > 0 != todo.is_done: - todo.is_done = len(self.form_data.get_all_str('done')) > 0 + todo.is_done = len(self.form_data.get_all_str('done')) > 0 todo.calendarize = len(self.form_data.get_all_str('calendarize')) > 0 todo.comment = self.form_data.get_str('comment', ignore_strict=True) todo.save(self.conn)