diff --git a/src/core/pyros_django/majordome/agent/A_Basic.py b/src/core/pyros_django/majordome/agent/A_Basic.py new file mode 100755 index 0000000..cba3779 --- /dev/null +++ b/src/core/pyros_django/majordome/agent/A_Basic.py @@ -0,0 +1,492 @@ +#!/usr/bin/env python3 + + +import time +import sys, argparse +import os + +##import utils.Logger as L + +##from .Agent import Agent +##sys.path.append("..") +###from agent.Agent import Agent, build_agent + +###sys.path.append("../../../..") + +pwd = os.environ['PROJECT_ROOT_PATH'] +if pwd not in sys.path: + sys.path.append(pwd) + +short_paths = ['src', 'src/core/pyros_django'] +for short_path in short_paths: + path = os.path.join(pwd, short_path) + if path not in sys.path: + sys.path.insert(0, path) + + +#from src.core.pyros_django.common.models import AgentCmd +#from src.core.pyros_django.majordome.agent.Agent import Agent, build_agent +from majordome.agent.Agent import Agent, build_agent + +#from src.core.pyros_django.common.models import AgentCmd + +from typing import List, Tuple, Union, Any + +##log = L.setupLogger("AgentXTaskLogger", "AgentX") + + + +class A_Basic(Agent): + + # FOR TEST ONLY + # Run this agent in simulator mode + TEST_MODE = False + # Run the assertion tests at the end + TEST_WITH_FINAL_TEST = False + #TEST_MAX_DURATION_SEC = None + TEST_MAX_DURATION_SEC = 400 + # Who should I send commands to ? + TEST_COMMANDS_DEST = "myself" + #TEST_COMMANDS_DEST = "AgentB" + # Scenario to be executed + + # This agent SPECIFIC commands + # @override + _AGENT_SPECIFIC_COMMANDS = { + # Format : “cmd_name” : (timeout, exec_mode, tooltip) + + # Error raising commands + "do_cmd_unimplemented_and_declared": (3, Agent.EXEC_MODE.SEQUENTIAL, ''), + "cmd_misnamed_and_declared": (3, Agent.EXEC_MODE.SEQUENTIAL, ''), + "do_cmd_raising_some_exception": (3, Agent.EXEC_MODE.SEQUENTIAL, ''), + + # Normal Commands + #("set_specific2", 5, 0), + "do_specific10": (1, Agent.EXEC_MODE.SEQUENTIAL, ''), + "do_specific30": (3, Agent.EXEC_MODE.SEQUENTIAL, ''), + "do_cmd_with_long_exec_time": (50, Agent.EXEC_MODE.THREAD, 'A command that takes a long time to exec'), + + } + + # Deactivate some tests, so that test scenario runs faster during DEV + # on DEV + #COMMIT_ONLY = False + # on commit only + COMMIT_ONLY = True + + VALIDITY=120 + + # This agent TEST scenario (automatically executed in TEST mode) + # @override + _TEST_COMMANDS_LIST = [ + # Format : (DO_IT, "self cmd_name cmd_args", validity, "expected_result", expected_status), + + #("self do_stop now", 200, '', Agent.CMD_STATUS.CMD_EXECUTED), + + # do_stop + #("self do_stop now", 200, 'STOPPING now', Agent.CMD_STATUS.CMD_EXECUTED), + #("self do_stop", 200, 'STOPPING asap', Agent.CMD_STATUS.CMD_EXECUTED), + #("self do_stop asap", 200, 'STOPPING asap', Agent.CMD_STATUS.CMD_EXECUTED), + + # get_specific_cmds + # prio + #(True, " self get_all_cmds ", 100, + # noprio + (True, " self get_all_cmds noprio ", 100, + None, + #"do_eval(eval_str:str)/;get_status()/;get_state()/;get_mode()/;set_mode(mode:typing.Literal['IDLE', 'ROUTINE', 'ATTENTIVE'])/;do_stop(when:typing.Literal['asap', 'now', 'noprio'])/;do_exec_commands(what:typing.Literal['stop', 'resume', 'noprio'])/;do_stop_current(what:typing.Literal['cmd', 'routine', 'both'])/;do_flush_pending_commands()/;do_restart(when:typing.Literal['asap', 'now', 'noprio'])/;get_all_cmds(noprio:str)/;get_specific_cmds()/;do_cmd_unimplemented_and_declared(U)/;cmd_misnamed_and_declared(I)/;do_cmd_raising_some_exception()/;do_specific10(arg1:int|arg2:int|arg3:float|arg4:str|arg5:typing.Tuple[int, str, int]|arg6:typing.List[int])/;do_specific30()/;do_cmd_with_long_exec_time()/A command that takes a long time to exec", + Agent.CMD_STATUS.CMD_EXECUTED + ), + + (COMMIT_ONLY, "self do_restart asap", VALIDITY, 'RESTARTING asap', Agent.CMD_STATUS.CMD_EXECUTED), + + # do_restart + (True, "self do_cmd_with_long_exec_time", VALIDITY, None, Agent.CMD_STATUS.CMD_EXECUTED), + #("self do_stop", 200, 'STOPPING asap', Agent.CMD_STATUS.CMD_EXECUTED), + + # ---------------------------------------------------- + # ------ A - ERROR CASES - from PENDING status ------- + # ---------------------------------------------------- + + # - Error case 1 - CMD_INVALID + # a) Misnamed commands + (COMMIT_ONLY, "self cmd_misnamed_and_declared", VALIDITY, "Command misnamed, must start with do_, get_, or set_", Agent.CMD_STATUS.CMD_INVALID), + (COMMIT_ONLY, "self cmd_misnamed_and_undeclared", VALIDITY, None, Agent.CMD_STATUS.CMD_INVALID), + # b) Unknwon command + ##FIXME: ("self unexisting_cmd", 200, '', Agent.CMD_STATUS.CMD_UNKNOWN), + (COMMIT_ONLY, "self do_cmd_unexisting", VALIDITY, 'EXCEPTION on command do_cmd_unexisting: Unknown command', Agent.CMD_STATUS.CMD_INVALID), + # c) Bad parameters (missing or too many, or bad type) + # - missing args + (COMMIT_ONLY, "self do_specific10", VALIDITY, 'EXCEPTION on command do_specific10: Command has bad, missing, or too many argument(s)', Agent.CMD_STATUS.CMD_INVALID), + # - missing args + (COMMIT_ONLY, "self do_specific10 2 ", VALIDITY, None, Agent.CMD_STATUS.CMD_INVALID), + # - too many args + (COMMIT_ONLY, "self do_specific10 2 2 3.5 titi (3,'titi',5) [1,3,5,7,9] 4", VALIDITY, None, Agent.CMD_STATUS.CMD_INVALID), + # - bad type + (True, "self do_specific10 'toto' 2 3.5 titi (3,'titi',5) [1,3,5,7,9]", VALIDITY, None, Agent.CMD_STATUS.CMD_INVALID), + (True, "self set_mode unknown_mode", VALIDITY, None, Agent.CMD_STATUS.CMD_INVALID), + (True, "self do_eval bad_arg", VALIDITY, None, Agent.CMD_STATUS.CMD_INVALID), + # - OK + #("self do_specific10 2 2 3.5 titi (3,'titi',5) [1,3,5,7,9]", 200, '16.5', Agent.CMD_STATUS.CMD_EXECUTED), + + # - Error case 2 - Unimplemented command (in agent specific commands list) + # - Error case 3 - CMD_UNIMPLEMENTED + (True, "self do_cmd_unimplemented_and_declared", VALIDITY, None, Agent.CMD_STATUS.CMD_UNIMPLEMENTED), + + # - Error case 4 - CMD_EXPIRED + # This command has a validity of 0s and thus should be tagged as "expired" + (COMMIT_ONLY, "self set_mode ATTENTIVE", 0, None, Agent.CMD_STATUS.CMD_EXPIRED), + + # - Error case 5 - CMD_SKIPPED + # a) In mode ROUTINE, SPECIFIC commands should be skipped as the agent is not ATTENTIVE + (COMMIT_ONLY, "self set_mode ROUTINE", VALIDITY, "MODE is ROUTINE", Agent.CMD_STATUS.CMD_EXECUTED), + (COMMIT_ONLY, "self do_specific10 2 2 3.5 titi (3,'titi',5) [1,3,5,7,9]", VALIDITY, None, Agent.CMD_STATUS.CMD_SKIPPED), + # b) Back to mode ATTENTIVE, SPECIFIC commands should be executed + (True, "self set_mode ATTENTIVE", VALIDITY, "MODE is ATTENTIVE", Agent.CMD_STATUS.CMD_EXECUTED), + (COMMIT_ONLY, "self do_specific10 2 2 3.5 titi (3,'titi',5) [1,3,5,7,9]", VALIDITY, '16.5', Agent.CMD_STATUS.CMD_EXECUTED), + + + # ---------------------------------------------------- + # ------ B - ERROR CASES - from RUNNING status ------- + # ---------------------------------------------------- + + (True, "self do_cmd_raising_some_exception", VALIDITY, + #"EXCEPTION on command do_cmd_raising_some_exception: StopIteration", + "EXCEPTION on command do_cmd_raising_some_exception: StopIteration: Some exception was raised", + Agent.CMD_STATUS.CMD_EXEC_ERROR), + #(True, "self do_cmd_raising_some_exception", 200, "EXCEPTION on command do_cmd_raising_some_exception: Error during Execution", Agent.CMD_STATUS.CMD_EXEC_ERROR), + + + # ------------------------------ + # ------ C - NORMAL CASES ------- + # ------------------------------ + + # do_restart => #iteration should restart at 1 (so at least it should be <= 4) + #("self do_restart now", 200, 'RESTARTING now', Agent.CMD_STATUS.CMD_EXECUTED), + # FIXME: ajouter la possibilité d'utiliser des expressions pour tester le résultat : + #("self get_iteration_number", 200, 'EXPR:<=4', Agent.CMD_STATUS.CMD_EXECUTED), + + # 1) First, 3 EXCEPTION CASES (uncomment to activate exception) + # Each of these lines will stop execution with an exception + # ------------------------------------------------------------ + + # - Agent command, unknown => ko, UnknownCmdException + #("self do_unknown", 10, None,None), + + # - Agent general command malformed (missing arg) => ko, AgentCmdBadArgsException + #("Agent set_mode", 10, None,None), + + # - Agent specific command, known but not implemented => ko, AgentCmdUnimplementedException + #("self set_specific2", 10, None,None), + + # - Agent specific command, implemented but missing args => ko, AgentCmdBadArgsException + #(" self do_specific1 1 ", 10, None,None), + + + # 2) NORMAL CASES (test scenario) + # All these commands should be executed without error, from the 1st to the last one + # ------------------------------- + + # Agent general command + (COMMIT_ONLY, "self get_mode", VALIDITY, "ATTENTIVE", Agent.CMD_STATUS.CMD_EXECUTED), + (COMMIT_ONLY, "self set_mode ROUTINE", VALIDITY, "MODE is ROUTINE", Agent.CMD_STATUS.CMD_EXECUTED), + (COMMIT_ONLY, "self get_mode", VALIDITY, "ROUTINE", Agent.CMD_STATUS.CMD_EXECUTED), + (COMMIT_ONLY, "self set_mode ATTENTIVE", VALIDITY, "MODE is ATTENTIVE", Agent.CMD_STATUS.CMD_EXECUTED), + (COMMIT_ONLY, "self get_mode", VALIDITY, "ATTENTIVE", Agent.CMD_STATUS.CMD_EXECUTED), + + # End test + # prio + (True, "self do_stop", VALIDITY, 'STOPPING asap', Agent.CMD_STATUS.CMD_EXECUTED), + # noprio + (True, "self do_stop noprio", VALIDITY, 'STOPPING asap', Agent.CMD_STATUS.CMD_EXECUTED), + + ] + + + """ + ================================================================= + FUNCTIONS RUN INSIDE MAIN THREAD + ================================================================= + """ + + # @override + ''' + #def __init__(self, name:str=None, config_filename=None, RUN_IN_THREAD=True): + def __init__(self, config_filename=None): + ##if name is None: name = self.__class__.__name__ + #super().__init__(name, config_filename, RUN_IN_THREAD) + super().__init__(config_filename) + #self._log.print(f"init done for {name}") + self._log.print("init done") + ''' + def __init__(self, name:str=None, simulated_computer=None): + if name is None: + name = self.__class__.__name__ + super().__init__(simulated_computer=simulated_computer) + #super().__init__(RUN_IN_THREAD) + + # @override + def _init(self): + # --- Set the mode according the startmode value + ##agent_alias = self.__class__.__name__ + ##self.set_mode_from_config(agent_alias) + self.set_delay(3) + + # @override + def _main_loop_start(self): + print("AgentBasic LOOP START") + + # @override + def _main_loop_end(self): + if self.CC and self.CC.name != "get_specific_cmds": + self.sleep(3) + print("AgentBasic LOOP END"); + + # @override + def _routine_process_iter_start_body(self): + if self.CC and self.CC.name != "get_specific_cmds": + self.sleep(2) + + # @override + def _routine_process_iter_end_body(self): + if self.CC and self.CC.name != "get_specific_cmds": + self.sleep(2) + + ''' + # @override + def load_config(self): + super().load_config() + ''' + + ''' + # @override + def update_survey(self): + super().update_survey() + ''' + + ''' + # @override + def get_next_command(self): + return super().get_next_command() + ''' + + # @override + def do_log(self): + super().do_log() + + + # @override + def _sleep_as_soon_as_running(self): + print("Make agent sleep as soon as a command has started running") + print("so that we can see this command in the 'current command' column of agent web pages") + if self.CC and self.CC.name not in ("get_all_cmds", "get_specific_cmds"): + self.sleep(5) + + # @override + def _do_things_before_exit(self, stopper_agent_name=None): + print("AgentBasic fait quelques trucs à lui avant de stopper...") + + + """ + ================================================================= + FUNCTIONS RUN INSIDE A SUB-THREAD (OR A PROCESS) (thread_*()) + ================================================================= + """ + + # Define your own command step(s) here + def cmd_step1(self, step:int): + cmd = self._current_specific_cmd + cmd.result = f"in step #{step}/{self._thread_total_steps_number}" + cmd.save() + """ + if self.RUN_IN_THREAD: + print("(save from thread)") + cmd.save() + else: + #@transaction.atomic + print("(save from process)") + with transaction.atomic(): + cmd.save() + #Command.objects.select_for_update() + """ + + def cmd_step2(self, step:int): + self.cmd_step1(step) + def cmd_step3(self, step:int): + self.cmd_step1(step) + def cmd_step4(self, step:int): + self.cmd_step1(step) + + """ + # @override + def thread_exec_specific_cmd_step(self, step:int, sleep_time:float=1.0): + self.thread_stop_if_asked() + cmd = self._current_specific_cmd + print(f">>>>> Thread (cmd {cmd.name}): step #{step}/5") + self.sleep(sleep_time) + """ + + ''' + # @override + def exec_specific_cmd_start(self, cmd:Command, from_thread=True): + super().exec_specific_cmd_start(cmd, from_thread) + ''' + + + + # @override + def simulator_test_results_main(self, commands): + nb_asserted = 0 + for cmd in commands: + if cmd.name == "flush_commands": + assert cmd.is_executed() + nb_asserted+=1 + # 2 times + if cmd.name == "go_active": + assert cmd.is_executed() + nb_asserted+=1 + # 2 times + if cmd.name == "go_idle": + assert cmd.is_executed() + nb_asserted+=1 + """ + if cmd.name == "specific0": + assert cmd.is_skipped() + assert cmd.result == "in step #5/5" + nb_asserted+=1 + """ + if cmd.name == "specific1": + assert cmd.is_killed() + nb_asserted+=1 + if cmd.name == "specific2": + assert cmd.is_executed() + assert cmd.result == "in step #5/5" + nb_asserted+=1 + if cmd.name == "eval 4+3": + assert cmd.is_executed() + assert cmd.get_result() == "7" + nb_asserted+=1 + if cmd.name in ("abort"): + assert cmd.is_executed() + nb_asserted+=1 + if cmd.name in ("exit"): + assert cmd.is_executed() + nb_asserted+=1 + return nb_asserted + + + ### + # ================================================================================================ + # AGENT SPECIFIC COMMANDS (functions) + # (here just to serve as examples) + # ================================================================================================ + ### + + #def do_specific1(self, arg1:int, arg2:int=2, arg3:int=3) -> int: + #def do_specific1(self, arg1:int, arg2:int=2, arg3:float=3.1, arg4:list=[1,2,3]) -> float: + + def do_cmd_raising_some_exception(self): + ##raise CmdExceptionExecError(self.CC) + #raise Exception() + #raise Exception("Some exception was raised") + #raise StopIteration() + raise StopIteration("Some exception was raised") + + # Long time execution command + def do_cmd_with_long_exec_time(self): + nbsec=5 + + res = f"1 - now sleeping {nbsec} sec" + self.CC.set_result(res, True) + #self.CC.get_updated_result() + while self.CC.get_updated_result() != res: self.sleep(1) + print(res) + self.sleep(nbsec) + + res = f"2 - now sleeping {nbsec} sec" + self.CC.set_result(res, True) + #self.CC.get_updated_result() + while self.CC.get_updated_result() != res: self.sleep(1) + print(res) + self.sleep(nbsec) + + #raise Exception("exception bidon") + + res = f"3 - now sleeping {nbsec} sec" + self.CC.set_result(res, True) + #self.CC.get_updated_result() + while self.CC.get_updated_result() != res: self.sleep(1) + print(res) + self.sleep(nbsec) + + res = f"4 - now sleeping {nbsec} sec" + self.CC.set_result(res, True) + #self.CC.get_updated_result() + while self.CC.get_updated_result() != res: self.sleep(1) + print(res) + self.sleep(nbsec) + + res = f"5 - now sleeping {nbsec} sec" + self.CC.set_result(res, True) + #self.CC.get_updated_result() + while self.CC.get_updated_result() != res: self.sleep(1) + print(res) + self.sleep(nbsec) + + return f"should have taken >= {5*nbsec}s to execute" + + def do_specific10(self, + arg1:int, + arg2:int, + arg3:float=3.1, + arg4:str='toto', + arg5:Tuple[int,str,int]=(1,'toto',3), + #arg5:Tuple[int,int,int]=(1,2,3), + arg6:List[int]=[] + ) -> float: + ''' + arg1 = int(arg1) + arg2 = int(arg2) + arg3 = float(arg3) + arg5 = ast.literal_eval(arg5) + print(arg5[1]) + arg6 = ast.literal_eval(arg6) + ''' + #print(arg4) + res = arg1 + arg2 + arg3 + arg5[0] + arg5[2] + if arg6: res += arg6[0] + return res + + # Undefined specific cmd + #def set_specific2(self, arg1:str, arg2:int): pass + + def do_specific30(self): + pass + + +""" +================================================================= + MAIN FUNCTION +================================================================= +""" +if __name__ == "__main__": + + parser = argparse.ArgumentParser(description='Start the agent.') + parser.add_argument("--computer",dest="computer",help='Launch agent with simulated computer hostname',action="store") + parser.add_argument("-t", action="store_true") + args = vars(parser.parse_args()) + agent = build_agent(A_Basic, param_constr=args) + + ''' + TEST_MODE, configfile = extract_parameters() + #agent = AgentX() + agent = AgentA("AgentA", configfile, RUN_IN_THREAD) + agent.setSimulatorMode(TEST_MODE) + print(agent) + ''' + + # Run only 4 iterations + #agent.run(nb_iter=4) + + # Run indefinitely + agent.run() diff --git a/src/core/pyros_django/majordome/agent/AgentBasic.py b/src/core/pyros_django/majordome/agent/AgentBasic.py deleted file mode 100755 index 1cc3cb7..0000000 --- a/src/core/pyros_django/majordome/agent/AgentBasic.py +++ /dev/null @@ -1,476 +0,0 @@ -#!/usr/bin/env python3 - - -import time -import sys, argparse - -##import utils.Logger as L - -##from .Agent import Agent -##sys.path.append("..") -###from agent.Agent import Agent, build_agent -sys.path.append("../../../..") -#from src.core.pyros_django.common.models import AgentCmd -from src.core.pyros_django.majordome.agent.Agent import Agent, build_agent -#from src.core.pyros_django.common.models import AgentCmd - -from typing import List, Tuple, Union, Any - -##log = L.setupLogger("AgentXTaskLogger", "AgentX") - - - -class AgentBasic(Agent): - - # FOR TEST ONLY - # Run this agent in simulator mode - TEST_MODE = False - # Run the assertion tests at the end - TEST_WITH_FINAL_TEST = False - #TEST_MAX_DURATION_SEC = None - TEST_MAX_DURATION_SEC = 400 - # Who should I send commands to ? - TEST_COMMANDS_DEST = "myself" - #TEST_COMMANDS_DEST = "AgentB" - # Scenario to be executed - - # This agent SPECIFIC commands - # @override - _AGENT_SPECIFIC_COMMANDS = { - # Format : “cmd_name” : (timeout, exec_mode, tooltip) - - # Error raising commands - "do_cmd_unimplemented_and_declared": (3, Agent.EXEC_MODE.SEQUENTIAL, ''), - "cmd_misnamed_and_declared": (3, Agent.EXEC_MODE.SEQUENTIAL, ''), - "do_cmd_raising_some_exception": (3, Agent.EXEC_MODE.SEQUENTIAL, ''), - - # Normal Commands - #("set_specific2", 5, 0), - "do_specific10": (1, Agent.EXEC_MODE.SEQUENTIAL, ''), - "do_specific30": (3, Agent.EXEC_MODE.SEQUENTIAL, ''), - "do_cmd_with_long_exec_time": (50, Agent.EXEC_MODE.THREAD, 'A command that takes a long time to exec'), - - } - - # Deactivate some tests, so that test scenario runs faster during DEV - # on DEV - #COMMIT_ONLY = False - # on commit only - COMMIT_ONLY = True - - VALIDITY=120 - - # This agent TEST scenario (automatically executed in TEST mode) - # @override - _TEST_COMMANDS_LIST = [ - # Format : (DO_IT, "self cmd_name cmd_args", validity, "expected_result", expected_status), - - #("self do_stop now", 200, '', Agent.CMD_STATUS.CMD_EXECUTED), - - # do_stop - #("self do_stop now", 200, 'STOPPING now', Agent.CMD_STATUS.CMD_EXECUTED), - #("self do_stop", 200, 'STOPPING asap', Agent.CMD_STATUS.CMD_EXECUTED), - #("self do_stop asap", 200, 'STOPPING asap', Agent.CMD_STATUS.CMD_EXECUTED), - - # get_specific_cmds - # prio - #(True, " self get_all_cmds ", 100, - # noprio - (True, " self get_all_cmds noprio ", 100, - None, - #"do_eval(eval_str:str)/;get_status()/;get_state()/;get_mode()/;set_mode(mode:typing.Literal['IDLE', 'ROUTINE', 'ATTENTIVE'])/;do_stop(when:typing.Literal['asap', 'now', 'noprio'])/;do_exec_commands(what:typing.Literal['stop', 'resume', 'noprio'])/;do_stop_current(what:typing.Literal['cmd', 'routine', 'both'])/;do_flush_pending_commands()/;do_restart(when:typing.Literal['asap', 'now', 'noprio'])/;get_all_cmds(noprio:str)/;get_specific_cmds()/;do_cmd_unimplemented_and_declared(U)/;cmd_misnamed_and_declared(I)/;do_cmd_raising_some_exception()/;do_specific10(arg1:int|arg2:int|arg3:float|arg4:str|arg5:typing.Tuple[int, str, int]|arg6:typing.List[int])/;do_specific30()/;do_cmd_with_long_exec_time()/A command that takes a long time to exec", - Agent.CMD_STATUS.CMD_EXECUTED - ), - - (COMMIT_ONLY, "self do_restart asap", VALIDITY, 'RESTARTING asap', Agent.CMD_STATUS.CMD_EXECUTED), - - # do_restart - (True, "self do_cmd_with_long_exec_time", VALIDITY, None, Agent.CMD_STATUS.CMD_EXECUTED), - #("self do_stop", 200, 'STOPPING asap', Agent.CMD_STATUS.CMD_EXECUTED), - - # ---------------------------------------------------- - # ------ A - ERROR CASES - from PENDING status ------- - # ---------------------------------------------------- - - # - Error case 1 - CMD_INVALID - # a) Misnamed commands - (COMMIT_ONLY, "self cmd_misnamed_and_declared", VALIDITY, "Command misnamed, must start with do_, get_, or set_", Agent.CMD_STATUS.CMD_INVALID), - (COMMIT_ONLY, "self cmd_misnamed_and_undeclared", VALIDITY, None, Agent.CMD_STATUS.CMD_INVALID), - # b) Unknwon command - ##FIXME: ("self unexisting_cmd", 200, '', Agent.CMD_STATUS.CMD_UNKNOWN), - (COMMIT_ONLY, "self do_cmd_unexisting", VALIDITY, 'EXCEPTION on command do_cmd_unexisting: Unknown command', Agent.CMD_STATUS.CMD_INVALID), - # c) Bad parameters (missing or too many, or bad type) - # - missing args - (COMMIT_ONLY, "self do_specific10", VALIDITY, 'EXCEPTION on command do_specific10: Command has bad, missing, or too many argument(s)', Agent.CMD_STATUS.CMD_INVALID), - # - missing args - (COMMIT_ONLY, "self do_specific10 2 ", VALIDITY, None, Agent.CMD_STATUS.CMD_INVALID), - # - too many args - (COMMIT_ONLY, "self do_specific10 2 2 3.5 titi (3,'titi',5) [1,3,5,7,9] 4", VALIDITY, None, Agent.CMD_STATUS.CMD_INVALID), - # - bad type - (True, "self do_specific10 'toto' 2 3.5 titi (3,'titi',5) [1,3,5,7,9]", VALIDITY, None, Agent.CMD_STATUS.CMD_INVALID), - (True, "self set_mode unknown_mode", VALIDITY, None, Agent.CMD_STATUS.CMD_INVALID), - (True, "self do_eval bad_arg", VALIDITY, None, Agent.CMD_STATUS.CMD_INVALID), - # - OK - #("self do_specific10 2 2 3.5 titi (3,'titi',5) [1,3,5,7,9]", 200, '16.5', Agent.CMD_STATUS.CMD_EXECUTED), - - # - Error case 2 - Unimplemented command (in agent specific commands list) - # - Error case 3 - CMD_UNIMPLEMENTED - (True, "self do_cmd_unimplemented_and_declared", VALIDITY, None, Agent.CMD_STATUS.CMD_UNIMPLEMENTED), - - # - Error case 4 - CMD_EXPIRED - # This command has a validity of 0s and thus should be tagged as "expired" - (COMMIT_ONLY, "self set_mode ATTENTIVE", 0, None, Agent.CMD_STATUS.CMD_EXPIRED), - - # - Error case 5 - CMD_SKIPPED - # a) In mode ROUTINE, SPECIFIC commands should be skipped as the agent is not ATTENTIVE - (COMMIT_ONLY, "self set_mode ROUTINE", VALIDITY, "MODE is ROUTINE", Agent.CMD_STATUS.CMD_EXECUTED), - (COMMIT_ONLY, "self do_specific10 2 2 3.5 titi (3,'titi',5) [1,3,5,7,9]", VALIDITY, None, Agent.CMD_STATUS.CMD_SKIPPED), - # b) Back to mode ATTENTIVE, SPECIFIC commands should be executed - (True, "self set_mode ATTENTIVE", VALIDITY, "MODE is ATTENTIVE", Agent.CMD_STATUS.CMD_EXECUTED), - (COMMIT_ONLY, "self do_specific10 2 2 3.5 titi (3,'titi',5) [1,3,5,7,9]", VALIDITY, '16.5', Agent.CMD_STATUS.CMD_EXECUTED), - - - # ---------------------------------------------------- - # ------ B - ERROR CASES - from RUNNING status ------- - # ---------------------------------------------------- - - (True, "self do_cmd_raising_some_exception", VALIDITY, - #"EXCEPTION on command do_cmd_raising_some_exception: StopIteration", - "EXCEPTION on command do_cmd_raising_some_exception: StopIteration: Some exception was raised", - Agent.CMD_STATUS.CMD_EXEC_ERROR), - #(True, "self do_cmd_raising_some_exception", 200, "EXCEPTION on command do_cmd_raising_some_exception: Error during Execution", Agent.CMD_STATUS.CMD_EXEC_ERROR), - - - # ------------------------------ - # ------ C - NORMAL CASES ------- - # ------------------------------ - - # do_restart => #iteration should restart at 1 (so at least it should be <= 4) - #("self do_restart now", 200, 'RESTARTING now', Agent.CMD_STATUS.CMD_EXECUTED), - # FIXME: ajouter la possibilité d'utiliser des expressions pour tester le résultat : - #("self get_iteration_number", 200, 'EXPR:<=4', Agent.CMD_STATUS.CMD_EXECUTED), - - # 1) First, 3 EXCEPTION CASES (uncomment to activate exception) - # Each of these lines will stop execution with an exception - # ------------------------------------------------------------ - - # - Agent command, unknown => ko, UnknownCmdException - #("self do_unknown", 10, None,None), - - # - Agent general command malformed (missing arg) => ko, AgentCmdBadArgsException - #("Agent set_mode", 10, None,None), - - # - Agent specific command, known but not implemented => ko, AgentCmdUnimplementedException - #("self set_specific2", 10, None,None), - - # - Agent specific command, implemented but missing args => ko, AgentCmdBadArgsException - #(" self do_specific1 1 ", 10, None,None), - - - # 2) NORMAL CASES (test scenario) - # All these commands should be executed without error, from the 1st to the last one - # ------------------------------- - - # Agent general command - (COMMIT_ONLY, "self get_mode", VALIDITY, "ATTENTIVE", Agent.CMD_STATUS.CMD_EXECUTED), - (COMMIT_ONLY, "self set_mode ROUTINE", VALIDITY, "MODE is ROUTINE", Agent.CMD_STATUS.CMD_EXECUTED), - (COMMIT_ONLY, "self get_mode", VALIDITY, "ROUTINE", Agent.CMD_STATUS.CMD_EXECUTED), - (COMMIT_ONLY, "self set_mode ATTENTIVE", VALIDITY, "MODE is ATTENTIVE", Agent.CMD_STATUS.CMD_EXECUTED), - (COMMIT_ONLY, "self get_mode", VALIDITY, "ATTENTIVE", Agent.CMD_STATUS.CMD_EXECUTED), - - # End test - # prio - (True, "self do_stop", VALIDITY, 'STOPPING asap', Agent.CMD_STATUS.CMD_EXECUTED), - # noprio - (True, "self do_stop noprio", VALIDITY, 'STOPPING asap', Agent.CMD_STATUS.CMD_EXECUTED), - - ] - - - """ - ================================================================= - FUNCTIONS RUN INSIDE MAIN THREAD - ================================================================= - """ - - # @override - ''' - #def __init__(self, name:str=None, config_filename=None, RUN_IN_THREAD=True): - def __init__(self, config_filename=None): - ##if name is None: name = self.__class__.__name__ - #super().__init__(name, config_filename, RUN_IN_THREAD) - super().__init__(config_filename) - #self._log.print(f"init done for {name}") - self._log.print("init done") - ''' - def __init__(self, name:str=None, simulated_computer=None): - if name is None: - name = self.__class__.__name__ - super().__init__(simulated_computer=simulated_computer) - #super().__init__(RUN_IN_THREAD) - - # @override - def _init(self): - # --- Set the mode according the startmode value - ##agent_alias = self.__class__.__name__ - ##self.set_mode_from_config(agent_alias) - self.set_delay(3) - - # @override - def _main_loop_start(self): - print("AgentBasic LOOP START") - - # @override - def _main_loop_end(self): - if self.CC and self.CC.name != "get_specific_cmds": - self.sleep(3) - print("AgentBasic LOOP END"); - - # @override - def _routine_process_iter_start_body(self): - if self.CC and self.CC.name != "get_specific_cmds": - self.sleep(2) - - # @override - def _routine_process_iter_end_body(self): - if self.CC and self.CC.name != "get_specific_cmds": - self.sleep(2) - - ''' - # @override - def load_config(self): - super().load_config() - ''' - - ''' - # @override - def update_survey(self): - super().update_survey() - ''' - - ''' - # @override - def get_next_command(self): - return super().get_next_command() - ''' - - # @override - def do_log(self): - super().do_log() - - - # @override - def _sleep_as_soon_as_running(self): - print("Make agent sleep as soon as a command has started running") - print("so that we can see this command in the 'current command' column of agent web pages") - if self.CC and self.CC.name not in ("get_all_cmds", "get_specific_cmds"): - self.sleep(5) - - # @override - def _do_things_before_exit(self, stopper_agent_name=None): - print("AgentBasic fait quelques trucs à lui avant de stopper...") - - - """ - ================================================================= - FUNCTIONS RUN INSIDE A SUB-THREAD (OR A PROCESS) (thread_*()) - ================================================================= - """ - - # Define your own command step(s) here - def cmd_step1(self, step:int): - cmd = self._current_specific_cmd - cmd.result = f"in step #{step}/{self._thread_total_steps_number}" - cmd.save() - """ - if self.RUN_IN_THREAD: - print("(save from thread)") - cmd.save() - else: - #@transaction.atomic - print("(save from process)") - with transaction.atomic(): - cmd.save() - #Command.objects.select_for_update() - """ - - def cmd_step2(self, step:int): - self.cmd_step1(step) - def cmd_step3(self, step:int): - self.cmd_step1(step) - def cmd_step4(self, step:int): - self.cmd_step1(step) - - """ - # @override - def thread_exec_specific_cmd_step(self, step:int, sleep_time:float=1.0): - self.thread_stop_if_asked() - cmd = self._current_specific_cmd - print(f">>>>> Thread (cmd {cmd.name}): step #{step}/5") - self.sleep(sleep_time) - """ - - ''' - # @override - def exec_specific_cmd_start(self, cmd:Command, from_thread=True): - super().exec_specific_cmd_start(cmd, from_thread) - ''' - - - - # @override - def simulator_test_results_main(self, commands): - nb_asserted = 0 - for cmd in commands: - if cmd.name == "flush_commands": - assert cmd.is_executed() - nb_asserted+=1 - # 2 times - if cmd.name == "go_active": - assert cmd.is_executed() - nb_asserted+=1 - # 2 times - if cmd.name == "go_idle": - assert cmd.is_executed() - nb_asserted+=1 - """ - if cmd.name == "specific0": - assert cmd.is_skipped() - assert cmd.result == "in step #5/5" - nb_asserted+=1 - """ - if cmd.name == "specific1": - assert cmd.is_killed() - nb_asserted+=1 - if cmd.name == "specific2": - assert cmd.is_executed() - assert cmd.result == "in step #5/5" - nb_asserted+=1 - if cmd.name == "eval 4+3": - assert cmd.is_executed() - assert cmd.get_result() == "7" - nb_asserted+=1 - if cmd.name in ("abort"): - assert cmd.is_executed() - nb_asserted+=1 - if cmd.name in ("exit"): - assert cmd.is_executed() - nb_asserted+=1 - return nb_asserted - - - ### - # ================================================================================================ - # AGENT SPECIFIC COMMANDS (functions) - # (here just to serve as examples) - # ================================================================================================ - ### - - #def do_specific1(self, arg1:int, arg2:int=2, arg3:int=3) -> int: - #def do_specific1(self, arg1:int, arg2:int=2, arg3:float=3.1, arg4:list=[1,2,3]) -> float: - - def do_cmd_raising_some_exception(self): - ##raise CmdExceptionExecError(self.CC) - #raise Exception() - #raise Exception("Some exception was raised") - #raise StopIteration() - raise StopIteration("Some exception was raised") - - # Long time execution command - def do_cmd_with_long_exec_time(self): - nbsec=5 - - res = f"1 - now sleeping {nbsec} sec" - self.CC.set_result(res, True) - #self.CC.get_updated_result() - while self.CC.get_updated_result() != res: self.sleep(1) - print(res) - self.sleep(nbsec) - - res = f"2 - now sleeping {nbsec} sec" - self.CC.set_result(res, True) - #self.CC.get_updated_result() - while self.CC.get_updated_result() != res: self.sleep(1) - print(res) - self.sleep(nbsec) - - #raise Exception("exception bidon") - - res = f"3 - now sleeping {nbsec} sec" - self.CC.set_result(res, True) - #self.CC.get_updated_result() - while self.CC.get_updated_result() != res: self.sleep(1) - print(res) - self.sleep(nbsec) - - res = f"4 - now sleeping {nbsec} sec" - self.CC.set_result(res, True) - #self.CC.get_updated_result() - while self.CC.get_updated_result() != res: self.sleep(1) - print(res) - self.sleep(nbsec) - - res = f"5 - now sleeping {nbsec} sec" - self.CC.set_result(res, True) - #self.CC.get_updated_result() - while self.CC.get_updated_result() != res: self.sleep(1) - print(res) - self.sleep(nbsec) - - return f"should have taken >= {5*nbsec}s to execute" - - def do_specific10(self, - arg1:int, - arg2:int, - arg3:float=3.1, - arg4:str='toto', - arg5:Tuple[int,str,int]=(1,'toto',3), - #arg5:Tuple[int,int,int]=(1,2,3), - arg6:List[int]=[] - ) -> float: - ''' - arg1 = int(arg1) - arg2 = int(arg2) - arg3 = float(arg3) - arg5 = ast.literal_eval(arg5) - print(arg5[1]) - arg6 = ast.literal_eval(arg6) - ''' - #print(arg4) - res = arg1 + arg2 + arg3 + arg5[0] + arg5[2] - if arg6: res += arg6[0] - return res - - # Undefined specific cmd - #def set_specific2(self, arg1:str, arg2:int): pass - - def do_specific30(self): - pass - - -""" -================================================================= - MAIN FUNCTION -================================================================= -""" -if __name__ == "__main__": - - parser = argparse.ArgumentParser(description='Start the agent.') - parser.add_argument("--computer",dest="computer",help='Launch agent with simulated computer hostname',action="store") - parser.add_argument("-t", action="store_true") - args = vars(parser.parse_args()) - agent = build_agent(AgentBasic,param_constr=args) - - ''' - TEST_MODE, configfile = extract_parameters() - #agent = AgentX() - agent = AgentA("AgentA", configfile, RUN_IN_THREAD) - agent.setSimulatorMode(TEST_MODE) - print(agent) - ''' - - # Run only 4 iterations - #agent.run(nb_iter=4) - - # Run indefinitely - agent.run() diff --git a/src/core/pyros_django/scp_mgmt/A_SCP_Manager.py b/src/core/pyros_django/scp_mgmt/A_SCP_Manager.py index e796cac..22c8d3a 100644 --- a/src/core/pyros_django/scp_mgmt/A_SCP_Manager.py +++ b/src/core/pyros_django/scp_mgmt/A_SCP_Manager.py @@ -313,7 +313,7 @@ if __name__ == "__main__": print("ARGV OF AGENT SP :",sys.argv) if len(sys.argv) > 1 and sys.argv[1] == "test": print("i'm in test") - agentSP = AgentSP(use_db_test=True) + agentSP = A_SCP_Manager(use_db_test=True) agentSP.run() #agent = build_agent(agentSP, RUN_IN_THREAD=True) else: -- libgit2 0.21.2