SetBased / py-enarksh

                errors = self._test_node_run_status_of_successor(rnd_id, (enarksh.ENK_RST_ID_ERROR,), set())

                if errors:

                    message['actions'][enarksh.ENK_ACT_ID_RESTART_FAILED]['act_enabled'] = True

        elif rst_id == enarksh.ENK_RST_ID_COMPLETED:

            # Node has completed successfully.

            busy = self._test_node_run_status_of_successor(rnd_id, (enarksh.ENK_RST_ID_RUNNING,), set())

            if not busy:

                message['actions'][enarksh.ENK_ACT_ID_RESTART]['act_enabled'] = True

        elif rst_id == enarksh.ENK_RST_ID_ERROR:

            if node.is_complex_node():

                message['actions'][enarksh.ENK_ACT_ID_RESTART]['act_enabled'] = True

                message['actions'][enarksh.ENK_ACT_ID_RESTART_FAILED]['act_enabled'] = True

            elif node.is_simple_node():

                message['actions'][enarksh.ENK_ACT_ID_RESTART]['act_enabled'] = True

            else:

                raise Exception('Internal error.')

        else:

            raise Exception("Unexpected rst_id '%d'." % rst_id)

        return message

                errors = self._test_node_run_status_of_successor(rnd_id, (enarksh.ENK_RST_ID_ERROR,), set())

                if errors:

                    message['actions'][enarksh.ENK_ACT_ID_RESTART_FAILED]['act_enabled'] = True

            elif rst_id == enarksh.ENK_RST_ID_QUEUED:

                # No actions are possible.

                pass

            elif rst_id == enarksh.ENK_RST_ID_RUNNING:

                errors = self._test_node_run_status_of_successor(rnd_id, (enarksh.ENK_RST_ID_ERROR,), set())

                if errors:

                    message['actions'][enarksh.ENK_ACT_ID_RESTART_FAILED]['act_enabled'] = True

            elif rst_id == enarksh.ENK_RST_ID_COMPLETED:

                message['actions'][enarksh.ENK_ACT_ID_TRIGGER]['act_enabled'] = True

            elif rst_id == enarksh.ENK_RST_ID_ERROR:

                message['actions'][enarksh.ENK_ACT_ID_TRIGGER]['act_enabled'] = True

                message['actions'][enarksh.ENK_ACT_ID_RESTART_FAILED]['act_enabled'] = True

            else:

                raise Exception("Unexpected rst_id '%s'." % rst_id)

            return message