X-Git-Url: https://plomlompom.com/repos/berlin_corona.txt?a=blobdiff_plain;f=plomtask%2Fprocesses.py;h=9705f17a5672336371ba186ff5d52cdd5fe002ef;hb=4815fe5c7be508e67ceec144968a81bdd6a923d4;hp=6249d48445a3488a8f8c4bdd5748b63a5a727ae5;hpb=0ed3bc539d21d5536d0fc635760d88a2231587b9;p=plomtask diff --git a/plomtask/processes.py b/plomtask/processes.py index 6249d48..9705f17 100644 --- a/plomtask/processes.py +++ b/plomtask/processes.py @@ -18,7 +18,7 @@ class ProcessStepsNode: seen: bool -class Process(BaseModel, ConditionsRelations): +class Process(BaseModel[int], ConditionsRelations): """Template for, and metadata for, Todos, and their arrangements.""" table_name = 'processes' @@ -34,19 +34,6 @@ class Process(BaseModel, ConditionsRelations): self.enables: list[Condition] = [] self.disables: list[Condition] = [] - @classmethod - def all(cls, db_conn: DatabaseConnection) -> list[Process]: - """Collect all Processes and their connected VersionedAttributes.""" - processes = {} - for id_, process in db_conn.cached_processes.items(): - processes[id_] = process - already_recorded = processes.keys() - for id_ in db_conn.column_all('processes', 'id'): - if id_ not in already_recorded: - process = cls.by_id(db_conn, id_) - processes[process.id_] = process - return list(processes.values()) - @classmethod def by_id(cls, db_conn: DatabaseConnection, id_: int | None, create: bool = False) -> Process: @@ -165,8 +152,7 @@ class Process(BaseModel, ConditionsRelations): """Set self.explicit_steps in bulk.""" assert isinstance(self.id_, int) for step in self.explicit_steps: - assert isinstance(step.id_, int) - del db_conn.cached_process_steps[step.id_] + step.uncache() self.explicit_steps = [] db_conn.delete_where('process_steps', 'owner', self.id_) for step_tuple in steps: @@ -189,10 +175,9 @@ class Process(BaseModel, ConditionsRelations): db_conn.delete_where('process_steps', 'owner', self.id_) for step in self.explicit_steps: step.save(db_conn) - db_conn.cached_processes[self.id_] = self -class ProcessStep(BaseModel): +class ProcessStep(BaseModel[int]): """Sub-unit of Processes.""" table_name = 'process_steps' to_save = ['owner_id', 'step_process_id', 'parent_step_id'] @@ -209,7 +194,6 @@ class ProcessStep(BaseModel): """Retrieve ProcessStep by id_, or throw NotFoundException.""" step, _ = super()._by_id(db_conn, id_) if step: - assert isinstance(step, ProcessStep) return step raise NotFoundException(f'found no ProcessStep of ID {id_}')