X-Git-Url: https://plomlompom.com/repos/berlin_corona.txt?a=blobdiff_plain;f=plomtask%2Fprocesses.py;h=bebd39456a804de397b6357919a0ceb0f245be3d;hb=8bbb9ac156bdca7b9dd015b62db3f07f1e7a9e17;hp=0300e73ccbaec2f15a5478a892b9fcc9d231eaa3;hpb=13bce001bd86de2c20dff2f93d9f08cfae95a16b;p=plomtask

diff --git a/plomtask/processes.py b/plomtask/processes.py
index 0300e73..bebd394 100644
--- a/plomtask/processes.py
+++ b/plomtask/processes.py
@@ -16,6 +16,7 @@ class Process:
         self.title = VersionedAttribute(self, 'title', 'UNNAMED')
         self.description = VersionedAttribute(self, 'description', '')
         self.effort = VersionedAttribute(self, 'effort', 1.0)
+        self.child_ids: list[int] = []
 
     @classmethod
     def from_table_row(cls, row: Row) -> Process:
@@ -60,8 +61,15 @@ class Process:
             for row in db_conn.exec('SELECT * FROM process_efforts '
                                     'WHERE process_id = ?', (process.id_,)):
                 process.effort.history[row[1]] = row[2]
+            for row in db_conn.exec('SELECT * FROM process_children '
+                                    'WHERE parent_id = ?', (process.id_,)):
+                process.child_ids += [row[1]]
         return process
 
+    def children(self, db_conn: DatabaseConnection) -> list[Process]:
+        """Return child Processes as determined by self.child_ids."""
+        return [self.__class__.by_id(db_conn, id_) for id_ in self.child_ids]
+
     def save(self, db_conn: DatabaseConnection) -> None:
         """Add (or re-write) self and connected VersionedAttributes to DB."""
         cursor = db_conn.exec('REPLACE INTO processes VALUES (?)', (self.id_,))
@@ -69,6 +77,11 @@ class Process:
         self.title.save(db_conn)
         self.description.save(db_conn)
         self.effort.save(db_conn)
+        db_conn.exec('DELETE FROM process_children WHERE parent_id = ?',
+                     (self.id_,))
+        for child_id in self.child_ids:
+            db_conn.exec('INSERT INTO process_children VALUES (?, ?)',
+                         (self.id_, child_id))
 
 
 class VersionedAttribute: