diff --git a/ums/management/process.py b/ums/management/process.py index 7921869..6f9bbf0 100644 --- a/ums/management/process.py +++ b/ums/management/process.py @@ -80,11 +80,17 @@ class MessageProcessor(): # check which step/ state the message requires the management to do if db_message.message.status.extract.required and not db_message.message.status.extract.finished: - self._do_extract(db_message.message) + # send to extract agents + self._send_messages(self.AGENTS_PROCESS, db_message.message) + elif db_message.message.status.solve.required and not db_message.message.status.solve.finished: - self._do_solve(db_message.message) + # send to solve agents + self._send_messages(self.AGENTS_SOLVE, db_message.message) + elif db_message.message.status.validate.required and not db_message.message.status.validate.finished: - self._do_validate(db_message.message) + # send to solve agents + self._send_messages(self.AGENTS_GATEKEEPER, db_message.message) + else: # all steps "done" if db_message.message.status.solved: # yay, message is solved @@ -99,18 +105,6 @@ class MessageProcessor(): # now message processed! self.db.set_processed(count=count, processed=True) - def _do_extract(self, message:AgentMessage): - # TODO - pass - - def _do_solve(self, message:AgentMessage): - # TODO - pass - - def _do_validate(self, message:AgentMessage): - # TODO - pass - def _do_again(self, message:AgentMessage): if message.status.trial < self.SOLUTION_MAX_TRIALS: # try again, recycle message