X-Git-Url: https://plomlompom.com/repos/foo.html?a=blobdiff_plain;ds=sidebyside;f=plomtask%2Fconditions.py;h=cd147cb79cafcfd81f4672fae0a0b6296201ff6a;hb=696aed8a590fa9c67a6b9c723e2134b2663bd769;hp=9a442000a99befa563582346826244739bf13ae7;hpb=2d0d3a138de69e5e09208936ac094b53b0785c0b;p=plomtask
diff --git a/plomtask/conditions.py b/plomtask/conditions.py
index 9a44200..cd147cb 100644
--- a/plomtask/conditions.py
+++ b/plomtask/conditions.py
@@ -66,3 +66,26 @@ class Condition(BaseModel):
self.description.save(db_conn)
assert isinstance(self.id_, int)
db_conn.cached_conditions[self.id_] = self
+
+
+class ConditionsRelations:
+ """Methods for handling relations to Conditions, for Todo and Process."""
+
+ def set_conditions(self, db_conn: DatabaseConnection, ids: list[int],
+ target: str = 'conditions') -> None:
+ """Set self.[target] to Conditions identified by ids."""
+ target_list = getattr(self, target)
+ while len(target_list) > 0:
+ target_list.pop()
+ for id_ in ids:
+ target_list += [Condition.by_id(db_conn, id_)]
+
+ def set_enables(self, db_conn: DatabaseConnection,
+ ids: list[int]) -> None:
+ """Set self.enables to Conditions identified by ids."""
+ self.set_conditions(db_conn, ids, 'enables')
+
+ def set_disables(self, db_conn: DatabaseConnection,
+ ids: list[int]) -> None:
+ """Set self.disables to Conditions identified by ids."""
+ self.set_conditions(db_conn, ids, 'disables')