Add Conditions for Todos/Processes to be met or undone by other Todos.

[plomtask] / plomtask / http.py

diff --git a/plomtask/http.py b/plomtask/http.py
index 5d165ecf90f1ca6ebaed77e55b73a5d55ff428b8..b00ebeba3aa06b38332be265599e43d7e034a857 100644 (file)
--- a/plomtask/http.py
+++ b/plomtask/http.py
@@ -10,6 +10,7 @@ from plomtask.exceptions import HandledException, BadFormatException, \
         NotFoundException
 from plomtask.db import DatabaseConnection, DatabaseFile
 from plomtask.processes import Process
+from plomtask.conditions import Condition
 from plomtask.todos import Todo
 
 TEMPLATES_DIR = 'templates'
@@ -111,7 +112,8 @@ class TaskHandler(BaseHTTPRequestHandler):
         """Handle any GET request."""
         try:
             conn, site, params = self._init_handling()
-            if site in {'calendar', 'day', 'process', 'processes', 'todo'}:
+            if site in {'calendar', 'day', 'process', 'processes', 'todo',
+                        'condition', 'conditions'}:
                 html = getattr(self, f'do_GET_{site}')(conn, params)
             elif '' == site:
                 self._redirect('/day')
@@ -139,17 +141,41 @@ class TaskHandler(BaseHTTPRequestHandler):
         date = params.get_str('date', todays_date())
         day = Day.by_date(conn, date, create=True)
         todos = Todo.by_date(conn, date)
+        conditions_listing = []
+        for condition in Condition.all(conn):
+            enablers = Todo.enablers_for_at(conn, condition, date)
+            disablers = Todo.disablers_for_at(conn, condition, date)
+            conditions_listing += [{
+                    'condition': condition,
+                    'enablers': enablers,
+                    'disablers': disablers}]
         return self.server.jinja.get_template('day.html').render(
-                day=day, processes=Process.all(conn), todos=todos)
+                day=day, processes=Process.all(conn), todos=todos,
+                conditions_listing=conditions_listing)
 
     def do_GET_todo(self, conn: DatabaseConnection, params:
                     ParamsParser) -> str:
         """Show single Todo of ?id=."""
         id_ = params.get_int_or_none('id')
         todo = Todo.by_id(conn, id_)
-        candidates = Todo.by_date(conn, todo.day.date)
+        todo_candidates = Todo.by_date(conn, todo.day.date)
         return self.server.jinja.get_template('todo.html').render(
-                todo=todo, candidates=candidates)
+                todo=todo, todo_candidates=todo_candidates,
+                condition_candidates=Condition.all(conn))
+
+    def do_GET_conditions(self, conn: DatabaseConnection,
+                          _: ParamsParser) -> str:
+        """Show all Conditions."""
+        return self.server.jinja.get_template('conditions.html').render(
+                conditions=Condition.all(conn))
+
+    def do_GET_condition(self, conn: DatabaseConnection,
+                         params: ParamsParser) -> str:
+        """Show Condition of ?id=."""
+        id_ = params.get_int_or_none('id')
+        condition = Condition.by_id(conn, id_, create=True)
+        return self.server.jinja.get_template('condition.html').render(
+                condition=condition)
 
     def do_GET_process(self, conn: DatabaseConnection,
                        params: ParamsParser) -> str:
@@ -159,7 +185,8 @@ class TaskHandler(BaseHTTPRequestHandler):
         owners = process.used_as_step_by(conn)
         return self.server.jinja.get_template('process.html').render(
                 process=process, steps=process.get_steps(conn),
-                owners=owners, candidates=Process.all(conn))
+                owners=owners, process_candidates=Process.all(conn),
+                condition_candidates=Condition.all(conn))
 
     def do_GET_processes(self, conn: DatabaseConnection,
                          _: ParamsParser) -> str:
@@ -175,7 +202,7 @@ class TaskHandler(BaseHTTPRequestHandler):
             postvars = parse_qs(self.rfile.read(length).decode(),
                                 keep_blank_values=True, strict_parsing=True)
             form_data = PostvarsParser(postvars)
-            if site in ('day', 'process', 'todo'):
+            if site in ('day', 'process', 'todo', 'condition'):
                 getattr(self, f'do_POST_{site}')(conn, params, form_data)
                 conn.commit()
             else:
@@ -209,8 +236,15 @@ class TaskHandler(BaseHTTPRequestHandler):
         if child_id is not None:
             child = Todo.by_id(conn, child_id)
             todo.add_child(child)
+        todo.set_conditions(conn, form_data.get_all_int('condition'))
+        todo.set_fulfills(conn, form_data.get_all_int('fulfills'))
+        todo.set_undoes(conn, form_data.get_all_int('undoes'))
         todo.is_done = len(form_data.get_all_str('done')) > 0
         todo.save(conn)
+        for condition in todo.fulfills:
+            condition.save(conn)
+        for condition in todo.undoes:
+            condition.save(conn)
 
     def do_POST_process(self, conn: DatabaseConnection, params: ParamsParser,
                         form_data: PostvarsParser) -> None:
@@ -220,6 +254,9 @@ class TaskHandler(BaseHTTPRequestHandler):
         process.title.set(form_data.get_str('title'))
         process.description.set(form_data.get_str('description'))
         process.effort.set(form_data.get_float('effort'))
+        process.set_conditions(conn, form_data.get_all_int('condition'))
+        process.set_fulfills(conn, form_data.get_all_int('fulfills'))
+        process.set_undoes(conn, form_data.get_all_int('undoes'))
         process.save_without_steps(conn)
         assert process.id_ is not None  # for mypy
         process.explicit_steps = []
@@ -236,6 +273,15 @@ class TaskHandler(BaseHTTPRequestHandler):
             process.add_step(conn, None, step_process_id, None)
         process.fix_steps(conn)
 
+    def do_POST_condition(self, conn: DatabaseConnection, params: ParamsParser,
+                          form_data: PostvarsParser) -> None:
+        """Update/insert Condition of ?id= and fields defined in postvars."""
+        id_ = params.get_int_or_none('id')
+        condition = Condition.by_id(conn, id_, create=True)
+        condition.title.set(form_data.get_str('title'))
+        condition.description.set(form_data.get_str('description'))
+        condition.save(conn)
+
     def _init_handling(self) -> tuple[DatabaseConnection, str, ParamsParser]:
         conn = DatabaseConnection(self.server.db)
         parsed_url = urlparse(self.path)