Blame view

src/core/pyros_django/agent/AgentBasic.py 17.1 KB
b29985c6   Etienne Pallier   Implementation pr...
1
2
3
#!/usr/bin/env python3


b7c6e890   Etienne Pallier   Test de tous les ...
4
import time
21843a6b   Alexis Koralewski   Add argparse & ne...
5
import sys, argparse
ef335590   Etienne Pallier   Agent v2 asynchro...
6

b29985c6   Etienne Pallier   Implementation pr...
7
8
9
##import utils.Logger as L

##from .Agent import Agent
6fbe2e20   Etienne Pallier   last minute.com m...
10
11
12
##sys.path.append("..")
###from agent.Agent import Agent, build_agent
sys.path.append("../../../..")
ef335590   Etienne Pallier   Agent v2 asynchro...
13
#from src.core.pyros_django.common.models import AgentCmd
dbc484ed   Etienne Pallier   CmdException* mov...
14
15
from src.core.pyros_django.agent.Agent import Agent, build_agent
#from src.core.pyros_django.common.models import AgentCmd
b29985c6   Etienne Pallier   Implementation pr...
16

f526191c   Etienne Pallier   Command (non runn...
17
18
from typing import List, Tuple, Union, Any

b29985c6   Etienne Pallier   Implementation pr...
19
20
21
22
23
24
25
26
27
28
29
30
##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
d59dd236   Etienne Pallier   bugfix TEST_MAX_D...
31
    TEST_MAX_DURATION_SEC = 400
b29985c6   Etienne Pallier   Implementation pr...
32
33
34
35
36
    # Who should I send commands to ?
    TEST_COMMANDS_DEST = "myself"
    #TEST_COMMANDS_DEST = "AgentB"
    # Scenario to be executed

0bcd6dd5   Etienne Pallier   renamed Agent met...
37
    
f526191c   Etienne Pallier   Command (non runn...
38
    # @override
baada9d5   Etienne Pallier   New PRIO CMD mana...
39
    _AGENT_SPECIFIC_COMMANDS = {
5bbf88b6   Etienne Pallier   ajout Specific Co...
40
        # Format : “cmd_name” : (timeout, exec_mode, tooltip)
baada9d5   Etienne Pallier   New PRIO CMD mana...
41
42

        # Error raising commands
5bbf88b6   Etienne Pallier   ajout Specific Co...
43
44
45
        "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, ''),
baada9d5   Etienne Pallier   New PRIO CMD mana...
46
47
48

        # Normal Commands 
        #("set_specific2", 5, 0),
5bbf88b6   Etienne Pallier   ajout Specific Co...
49
50
51
        "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'),
baada9d5   Etienne Pallier   New PRIO CMD mana...
52
53

    }
b29985c6   Etienne Pallier   Implementation pr...
54

d02249e6   Etienne Pallier   Added CMD_EXEC_ER...
55
56
    # Deactivate some tests, so that test scenario runs faster during DEV
    # on DEV
64b466e7   Etienne Pallier   Agent specific co...
57
    COMMIT_ONLY = False
d02249e6   Etienne Pallier   Added CMD_EXEC_ER...
58
    # on commit only
64b466e7   Etienne Pallier   Agent specific co...
59
    #COMMIT_ONLY = True
a0a295be   Etienne Pallier   Optimization of n...
60
61

    VALIDITY=120
d02249e6   Etienne Pallier   Added CMD_EXEC_ER...
62

f526191c   Etienne Pallier   Command (non runn...
63
    # @override
0bcd6dd5   Etienne Pallier   renamed Agent met...
64
    _TEST_COMMANDS_LIST = [
7b4ee4fe   Etienne Pallier   cleanup, mark som...
65

975bf9f6   Etienne Pallier   bugfix priority c...
66
        # Format : (DO_IT, "self cmd_name cmd_args", validity, "expected_result", expected_status),
ef335590   Etienne Pallier   Agent v2 asynchro...
67

6fbe2e20   Etienne Pallier   last minute.com m...
68
        #("self do_stop now", 200, '', Agent.CMD_STATUS.CMD_EXECUTED),
340ca843   Etienne Pallier   fixed get_specifi...
69

340ca843   Etienne Pallier   fixed get_specifi...
70
        # do_stop
340ca843   Etienne Pallier   fixed get_specifi...
71
72
73
        #("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),
795c826a   Etienne Pallier   Ajout de la gesti...
74

888d6d8e   Etienne Pallier   AgentBasic cleanu...
75
        # get_specific_cmds
baada9d5   Etienne Pallier   New PRIO CMD mana...
76
77
78
79
        # prio
        #(True, "    self      get_all_cmds    ", 100, 
        # noprio
        (True, "    self      get_all_cmds noprio    ", 100, 
888d6d8e   Etienne Pallier   AgentBasic cleanu...
80
            None,
a7b5f02c   Etienne Pallier   simplified agent ...
81
            #'do_specific10(arg1:int,arg2:int,arg3:float,arg4:str,arg5:typing.Tuple[int, str, int],arg6:typing.List[int]);do_specific30();do_cmd_raising_some_exception();do_cmd_unimplemented(U)', 
888d6d8e   Etienne Pallier   AgentBasic cleanu...
82
83
            Agent.CMD_STATUS.CMD_EXECUTED
        ),
795c826a   Etienne Pallier   Ajout de la gesti...
84

a0a295be   Etienne Pallier   Optimization of n...
85
        (COMMIT_ONLY, "self do_restart asap", VALIDITY, 'RESTARTING asap', Agent.CMD_STATUS.CMD_EXECUTED),
975bf9f6   Etienne Pallier   bugfix priority c...
86
87

        # do_restart
a0a295be   Etienne Pallier   Optimization of n...
88
        (True, "self do_cmd_with_long_exec_time", VALIDITY, None, Agent.CMD_STATUS.CMD_EXECUTED),
b7c6e890   Etienne Pallier   Test de tous les ...
89
        #("self do_stop", 200, 'STOPPING asap', Agent.CMD_STATUS.CMD_EXECUTED),
340ca843   Etienne Pallier   fixed get_specifi...
90

d02249e6   Etienne Pallier   Added CMD_EXEC_ER...
91
92
93
        # ----------------------------------------------------
        # ------ A - ERROR CASES - from PENDING status -------
        # ----------------------------------------------------
340ca843   Etienne Pallier   fixed get_specifi...
94

b7c6e890   Etienne Pallier   Test de tous les ...
95

888d6d8e   Etienne Pallier   AgentBasic cleanu...
96
        # - Error case 1 - CMD_INVALID
6d1f4089   Etienne Pallier   cleanup
97
        # a) Misnamed commands
a0a295be   Etienne Pallier   Optimization of n...
98
99
        (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),
888d6d8e   Etienne Pallier   AgentBasic cleanu...
100
        # b) Unknwon command
f526191c   Etienne Pallier   Command (non runn...
101
        ##FIXME: ("self unexisting_cmd", 200, '', Agent.CMD_STATUS.CMD_UNKNOWN),
a0a295be   Etienne Pallier   Optimization of n...
102
        (COMMIT_ONLY, "self do_cmd_unexisting", VALIDITY, 'EXCEPTION on command do_cmd_unexisting: Unknown command', Agent.CMD_STATUS.CMD_INVALID),
888d6d8e   Etienne Pallier   AgentBasic cleanu...
103
104
        # c) Bad parameters (missing or too many, or bad type)
        # - missing args
a0a295be   Etienne Pallier   Optimization of n...
105
        (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),
888d6d8e   Etienne Pallier   AgentBasic cleanu...
106
        # - missing args
a0a295be   Etienne Pallier   Optimization of n...
107
        (COMMIT_ONLY, "self do_specific10 2       ", VALIDITY, None, Agent.CMD_STATUS.CMD_INVALID),
b7c6e890   Etienne Pallier   Test de tous les ...
108
        # - too many args
a0a295be   Etienne Pallier   Optimization of n...
109
        (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),
b7c6e890   Etienne Pallier   Test de tous les ...
110
        # - bad type
a0a295be   Etienne Pallier   Optimization of n...
111
112
113
        (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),
b7c6e890   Etienne Pallier   Test de tous les ...
114
115
116
        # - 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),

da405f5b   Etienne Pallier   get_all_cmds remp...
117
        # - Error case 2 - Unimplemented command (in agent specific commands list)
b7c6e890   Etienne Pallier   Test de tous les ...
118
        # - Error case 3 - CMD_UNIMPLEMENTED
a0a295be   Etienne Pallier   Optimization of n...
119
        (True, "self do_cmd_unimplemented_and_declared", VALIDITY, None, Agent.CMD_STATUS.CMD_UNIMPLEMENTED),
f526191c   Etienne Pallier   Command (non runn...
120
        
b7c6e890   Etienne Pallier   Test de tous les ...
121
122
        # - Error case 4 - CMD_EXPIRED
        # This command has a validity of 0s and thus should be tagged as "expired"
ef335590   Etienne Pallier   Agent v2 asynchro...
123
        (COMMIT_ONLY, "self set_mode ATTENTIVE", 0, None, Agent.CMD_STATUS.CMD_EXPIRED),
b7c6e890   Etienne Pallier   Test de tous les ...
124
125
126

        # - Error case 5 - CMD_SKIPPED
        # a) In mode ROUTINE, SPECIFIC commands should be skipped as the agent is not ATTENTIVE
a0a295be   Etienne Pallier   Optimization of n...
127
128
        (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),
b7c6e890   Etienne Pallier   Test de tous les ...
129
        # b) Back to mode ATTENTIVE, SPECIFIC commands should be executed
a0a295be   Etienne Pallier   Optimization of n...
130
131
        (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),
d02249e6   Etienne Pallier   Added CMD_EXEC_ER...
132

da405f5b   Etienne Pallier   get_all_cmds remp...
133

d02249e6   Etienne Pallier   Added CMD_EXEC_ER...
134
135
136
        # ----------------------------------------------------
        # ------ B - ERROR CASES - from RUNNING status -------
        # ----------------------------------------------------
a7b5f02c   Etienne Pallier   simplified agent ...
137
        
a0a295be   Etienne Pallier   Optimization of n...
138
        (True, "self do_cmd_raising_some_exception", VALIDITY, 
a7b5f02c   Etienne Pallier   simplified agent ...
139
140
141
142
            #"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),
da405f5b   Etienne Pallier   get_all_cmds remp...
143
144


b7c6e890   Etienne Pallier   Test de tous les ...
145
        # ------------------------------
d02249e6   Etienne Pallier   Added CMD_EXEC_ER...
146
        # ------ C - NORMAL CASES -------
b7c6e890   Etienne Pallier   Test de tous les ...
147
148
149
150
151
152
153
        # ------------------------------
        
        # 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),

7b4ee4fe   Etienne Pallier   cleanup, mark som...
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
        # 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
        # -------------------------------

7b4ee4fe   Etienne Pallier   cleanup, mark som...
175
        # Agent general command
a0a295be   Etienne Pallier   Optimization of n...
176
177
178
179
        (COMMIT_ONLY, "self set_mode ROUTINE", VALIDITY, "MODE is ROUTINE", Agent.CMD_STATUS.CMD_EXECUTED),
        (COMMIT_ONLY, "self get_mode", VALIDITY, "MODE is 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, "MODE is ATTENTIVE", Agent.CMD_STATUS.CMD_EXECUTED),
6fbe2e20   Etienne Pallier   last minute.com m...
180
181
        
        # End test
7bb1c9b9   Etienne Pallier   Added 'noprio' ma...
182
        # prio
a0a295be   Etienne Pallier   Optimization of n...
183
        (True, "self do_stop", VALIDITY, 'STOPPING asap', Agent.CMD_STATUS.CMD_EXECUTED),
7bb1c9b9   Etienne Pallier   Added 'noprio' ma...
184
        # noprio
a0a295be   Etienne Pallier   Optimization of n...
185
        (True, "self do_stop noprio", VALIDITY, 'STOPPING asap', Agent.CMD_STATUS.CMD_EXECUTED),
b7c6e890   Etienne Pallier   Test de tous les ...
186

7b4ee4fe   Etienne Pallier   cleanup, mark som...
187
    ]
b29985c6   Etienne Pallier   Implementation pr...
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205


    """
    =================================================================
        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")
    '''
5bbf88b6   Etienne Pallier   ajout Specific Co...
206
    def __init__(self, name:str=None, simulated_computer=None):
b29985c6   Etienne Pallier   Implementation pr...
207
208
209
210
211
212
        if name is None:
            name = self.__class__.__name__
        super().__init__()
        #super().__init__(RUN_IN_THREAD)        

    # @override
0bcd6dd5   Etienne Pallier   renamed Agent met...
213
    def _init(self):
b29985c6   Etienne Pallier   Implementation pr...
214
215
216
        # --- Set the mode according the startmode value
        ##agent_alias = self.__class__.__name__
        ##self.set_mode_from_config(agent_alias)
6d5013fe   Etienne Pallier   Log agent state a...
217
        self.set_delay(3)
b29985c6   Etienne Pallier   Implementation pr...
218

6d5013fe   Etienne Pallier   Log agent state a...
219
    # @override
0bcd6dd5   Etienne Pallier   renamed Agent met...
220
    def _main_loop_start(self):
340ca843   Etienne Pallier   fixed get_specifi...
221
        print("AgentBasic LOOP START")
b29985c6   Etienne Pallier   Implementation pr...
222

6d5013fe   Etienne Pallier   Log agent state a...
223
    # @override
0bcd6dd5   Etienne Pallier   renamed Agent met...
224
    def _main_loop_end(self):
ef335590   Etienne Pallier   Agent v2 asynchro...
225
        if self.CC and self.CC.name != "get_specific_cmds":
927b033b   Etienne Pallier   update AgentBasic...
226
            self.sleep(3)
340ca843   Etienne Pallier   fixed get_specifi...
227
        print("AgentBasic LOOP END");
b29985c6   Etienne Pallier   Implementation pr...
228

6d5013fe   Etienne Pallier   Log agent state a...
229
    # @override
547df0ef   Etienne Pallier   routine_process_b...
230
    def _routine_process_iter_start_body(self):
ef335590   Etienne Pallier   Agent v2 asynchro...
231
        if self.CC and self.CC.name != "get_specific_cmds":
927b033b   Etienne Pallier   update AgentBasic...
232
            self.sleep(2)
6d5013fe   Etienne Pallier   Log agent state a...
233
234

    # @override
547df0ef   Etienne Pallier   routine_process_b...
235
    def _routine_process_iter_end_body(self):
ef335590   Etienne Pallier   Agent v2 asynchro...
236
        if self.CC and self.CC.name != "get_specific_cmds":
927b033b   Etienne Pallier   update AgentBasic...
237
            self.sleep(2)
6d5013fe   Etienne Pallier   Log agent state a...
238

b29985c6   Etienne Pallier   Implementation pr...
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
    '''
    # @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()

0cc81973   Etienne Pallier   added sleep after...
261
262
263
264
265

    # @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")
886e94d1   Etienne Pallier   bugfix CC_prev, c...
266
        if self.CC and self.CC.name not in ("get_all_cmds", "get_specific_cmds"): 
927b033b   Etienne Pallier   update AgentBasic...
267
            self.sleep(5)
0cc81973   Etienne Pallier   added sleep after...
268

5fddfc70   Etienne Pallier   Implementation pr...
269
    # @override
0bcd6dd5   Etienne Pallier   renamed Agent met...
270
    def _do_things_before_exit(self, stopper_agent_name=None):
5fddfc70   Etienne Pallier   Implementation pr...
271
        print("AgentBasic fait quelques trucs à lui avant de stopper...")
b29985c6   Etienne Pallier   Implementation pr...
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361


    """
    =================================================================
        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


f526191c   Etienne Pallier   Command (non runn...
362
363
364
365
366
367
368
369
370
371
    ###
    # ================================================================================================
    #   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:

a7b5f02c   Etienne Pallier   simplified agent ...
372
373
374
375
376
377
    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")
d02249e6   Etienne Pallier   Added CMD_EXEC_ER...
378

6d1f4089   Etienne Pallier   cleanup
379
380
    # Long time execution command
    def do_cmd_with_long_exec_time(self):
64b466e7   Etienne Pallier   Agent specific co...
381
        nbsec=5
975bf9f6   Etienne Pallier   bugfix priority c...
382
383
384
385
386
        
        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)
64b466e7   Etienne Pallier   Agent specific co...
387
        print(res)
975bf9f6   Etienne Pallier   bugfix priority c...
388
389
390
391
392
393
        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)
64b466e7   Etienne Pallier   Agent specific co...
394
        print(res)
975bf9f6   Etienne Pallier   bugfix priority c...
395
396
397
398
399
400
        self.sleep(nbsec)
        
        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)
64b466e7   Etienne Pallier   Agent specific co...
401
        print(res)
975bf9f6   Etienne Pallier   bugfix priority c...
402
403
404
405
406
407
        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)
64b466e7   Etienne Pallier   Agent specific co...
408
        print(res)
975bf9f6   Etienne Pallier   bugfix priority c...
409
410
411
412
413
414
        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)
64b466e7   Etienne Pallier   Agent specific co...
415
        print(res)
975bf9f6   Etienne Pallier   bugfix priority c...
416
417
        self.sleep(nbsec)

886e94d1   Etienne Pallier   bugfix CC_prev, c...
418
        return f"should have taken >= {5*nbsec}s to execute"
6d1f4089   Etienne Pallier   cleanup
419

f526191c   Etienne Pallier   Command (non runn...
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
    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

f526191c   Etienne Pallier   Command (non runn...
448

b29985c6   Etienne Pallier   Implementation pr...
449
450
451
452
453
454
455
"""
=================================================================
    MAIN FUNCTION
=================================================================
"""
if __name__ == "__main__":

21843a6b   Alexis Koralewski   Add argparse & ne...
456
457
458
459
    parser = argparse.ArgumentParser(description='Start the agent.')
    parser.add_argument("--computer",dest="computer",help='Launch agent with simulated computer hostname',action="store")
    args = vars(parser.parse_args())
    agent = build_agent(AgentBasic,param_constr=args)
b29985c6   Etienne Pallier   Implementation pr...
460
461
462
463
464
465
466
467

    '''
    TEST_MODE, configfile = extract_parameters()
    #agent = AgentX()
    agent = AgentA("AgentA", configfile, RUN_IN_THREAD)
    agent.setSimulatorMode(TEST_MODE)
    print(agent)
    '''
795c826a   Etienne Pallier   Ajout de la gesti...
468
469
470
471
472

    # Run only 4 iterations
    #agent.run(nb_iter=4)

    # Run indefinitely
b29985c6   Etienne Pallier   Implementation pr...
473
    agent.run()