Blame view

src/core/pyros_django/monitoring/AgentM.py 6.38 KB
1cffbf1c   Etienne Pallier   moved pyros_djang...
1
2
3
4
5
6
7
8
9
10
11
#!/usr/bin/env python3

import sys
##import utils.Logger as L
#import threading #, multiprocessing, os
#import time

#from django.db import transaction
#from common.models import Command

sys.path.append("..")
57a9ffac   Etienne Pallier   bugfixing agentM
12
sys.path.append("../../../..")
12451fde   Etienne Pallier   Amélioration logging
13
from src.core.pyros_django.agent.Agent import Agent, build_agent, log
1cffbf1c   Etienne Pallier   moved pyros_djang...
14
15

# PM 20190416 recycle code
ad6d274a   Etienne Pallier   Restructuration d...
16
17
from src.core.pyros_django.devices.PLC import PLCController
from src.core.pyros_django.monitoring.plc_checker import PlcChecker
1cffbf1c   Etienne Pallier   moved pyros_djang...
18
19
20
21
22
23
24
25
26
27
28
from common.models import *

##log = L.setupLogger("AgentXTaskLogger", "AgentX")



class AgentM(Agent):


    # FOR TEST ONLY
    # Run this agent in simulator mode
71b8533d   Etienne Pallier   update agentA, ag...
29
    TEST_MODE = False
1cffbf1c   Etienne Pallier   moved pyros_djang...
30
    # Run the assertion tests at the end
71b8533d   Etienne Pallier   update agentA, ag...
31
32
33
    TEST_WITH_FINAL_TEST = True
    TEST_MAX_DURATION_SEC = None
    #TEST_MAX_DURATION_SEC = 120
1cffbf1c   Etienne Pallier   moved pyros_djang...
34
35

    # PM 20190416 fucking config path starting: /home/patrick/Dev/PYROS/start_agent.py agentM
a9d06e6f   Etienne Pallier   new PYROS_DOCKER_...
36
37
    ##_path_data = 'config'
    _path_data = 'config/old_config'
1cffbf1c   Etienne Pallier   moved pyros_djang...
38
39
40
41
42
    # PM 20190416 recycle code
    plcController = PLCController()
    print ("AGENT ENV: config PLC is (ip={}, port={})".format(plcController.ip, plcController.port))
    plc_checker = PlcChecker()

12451fde   Etienne Pallier   Amélioration logging
43
44
45
    log.debug("PLC instanciated")


1cffbf1c   Etienne Pallier   moved pyros_djang...
46
47
48

    '''
    # Who should I send commands to ?
71b8533d   Etienne Pallier   update agentA, ag...
49
50
    #TEST_COMMANDS_DEST = "myself"
    TEST_COMMANDS_DEST = "AgentA"
1cffbf1c   Etienne Pallier   moved pyros_djang...
51
    # Scenario to be executed
71b8533d   Etienne Pallier   update agentA, ag...
52
    TEST_COMMANDS_LIST = [
1cffbf1c   Etienne Pallier   moved pyros_djang...
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
        "go_active",
        "go_idle",
        "go_active",
        "go_idle",
        "go_active",
        "go_idle",
        "exit",
    ]
    '''

    """
    =================================================================
        FUNCTIONS RUN INSIDE MAIN THREAD
    =================================================================
    """

    # @override
71b8533d   Etienne Pallier   update agentA, ag...
70
71
72
73
    #def __init__(self, name:str=None, config_filename=None, RUN_IN_THREAD=True):
    def __init__(self, config_filename=None, RUN_IN_THREAD=True):
        ##if name is None: name = self.__class__.__name__
        super().__init__(config_filename, RUN_IN_THREAD)
1cffbf1c   Etienne Pallier   moved pyros_djang...
74
75
76
77

    # @override
    def init(self):
        super().init()
12451fde   Etienne Pallier   Amélioration logging
78
        log.debug("end init()")
1cffbf1c   Etienne Pallier   moved pyros_djang...
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
        # --- Set the mode according the startmode value
        ##agent_alias = self.__class__.__name__
        ##self.set_mode_from_config(agent_alias)

    '''
    # @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()



    """
    =================================================================
        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
74a196a1   Alexis Koralewski   fixing weather pa...
153
154
155
156
    # previous name of function : routine_process
    # Note : in Agent.py, routine_process_body seems to be the main function of routine of the agent
    # We need to override routine_process_body and not routine_process
    def routine_process_body(self):
12451fde   Etienne Pallier   Amélioration logging
157
        log.debug("in routine_process_body()")
1cffbf1c   Etienne Pallier   moved pyros_djang...
158
159
160
161
162
        print("TODO: we recycle code")
        status_plc = self.plcController.getStatus()
        if self.parseNewStatus(status_plc):
            self.saveWeather()

61df368d   Etienne Pallier   Mis à jour Enviro...
163
    def parseNewStatus(self,status_plc):
1cffbf1c   Etienne Pallier   moved pyros_djang...
164
165
166
167
        # """ PM 20181009 parse new status for config
        if status_plc.find('PLC_STATUS') >= 0:
            self.plc_checker.chk_config(status_plc)
            return True
61df368d   Etienne Pallier   Mis à jour Enviro...
168
        return False
1cffbf1c   Etienne Pallier   moved pyros_djang...
169
170
171

    def saveWeather(self):
        outside = WeatherWatch()
a78619ca   Patrick Maeght   weather config up...
172
        for sensor in self.plc_checker.monitoring_names.keys():
1cffbf1c   Etienne Pallier   moved pyros_djang...
173
174
175
176
177
            value = self.plc_checker.get_sensor(sensor)
            print(sensor, value)
            outside.setAttribute(sensor, value)
        outside.setGlobalStatus()
        outside.save()
12451fde   Etienne Pallier   Amélioration logging
178
179
        log.debug("saved weather")

1cffbf1c   Etienne Pallier   moved pyros_djang...
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217

    # @override
    def thread_exec_specific_cmd_main(self):
        # This is optional
        self.thread_set_total_steps_number(5)

        # HERE, write your own scenario

        # scenario OK
        self.thread_exec_specific_cmd_step(1, self.cmd_step1, 1)
        self.thread_exec_specific_cmd_step(2, self.cmd_step2, 3)
        self.thread_exec_specific_cmd_step(3, self.cmd_step1, 5)
        self.thread_exec_specific_cmd_step(4, self.cmd_step3, 10)
        self.thread_exec_specific_cmd_step(5, self.cmd_step1, 4)
        # ... as many as you need

        """ autre scenario
        self.thread_exec_specific_cmd_step(1, self.cmd_step1, 1)
        self.thread_exec_specific_cmd_step(2, self.cmd_step2, 2)
        self.thread_exec_specific_cmd_step(3, self.cmd_step1, 2)
        self.thread_exec_specific_cmd_step(4, self.cmd_step3, 2)
        self.thread_exec_specific_cmd_step(5, self.cmd_step1, 3)
        """

    '''
    # @override
    def exec_specific_cmd_end(self, cmd:Command, from_thread=True):
        super().exec_specific_cmd_end(cmd, from_thread)
    '''


if __name__ == "__main__":

    # with thread
    RUN_IN_THREAD=True
    # with process
    #RUN_IN_THREAD=False

71b8533d   Etienne Pallier   update agentA, ag...
218
219
    agent = build_agent(AgentM, RUN_IN_THREAD=RUN_IN_THREAD)
    '''
1cffbf1c   Etienne Pallier   moved pyros_djang...
220
221
222
    TEST_MODE, configfile = extract_parameters()
    agent = AgentM("AgentM", configfile, RUN_IN_THREAD)
    agent.setSimulatorMode(TEST_MODE)
71b8533d   Etienne Pallier   update agentA, ag...
223
    '''
1cffbf1c   Etienne Pallier   moved pyros_djang...
224
225
    print(agent)
    agent.run()