tasks.py
5.19 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
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
153
154
155
156
157
158
159
160
161
162
163
from __future__ import absolute_import
from django.conf import settings
from common.models import *
from celery.task import Task
from devices.PLC import PLCController
from utils.JDManipulator import *
import utils.Logger as L
import majordome.tasks
import alert_manager.tasks
log = L.setupLogger("MonitoringTaskLogger", "Monitoring")
DEBUG_FILE = False
'''
Infinite task created at the program's start.
Checks the plc status, parse it, analyse it, store it in db
'''
class Monitoring(Task):
timers = {}
functions = {}
def run(self):
self.createTask()
self.setContext()
self.setTime()
self.setTasks()
self.loop()
def createTask(self):
try:
TaskId.objects.find(task="monitoring").delete()
except Exception as e:
log.info(str(e))
return 1
TaskId.objects.create(task_id=self.request.id, task="monitoring")
return 0
def setTasks(self):
try:
self.majordome_task = TaskId.objects.get(task="majordome")
self.alert_task = TaskId.objects.get(task="alert_manager")
except Exception as e:
self.majordome_task = None
self.alert_task = None
return 0
def setContext(self):
self.plc = PLCController()
self.state = "RUNNING"
return (0)
def setTime(self):
self.timer_status = 2
self.timer_tasks = 5
self.timers = {"timer_status": self.timer_status,
"tasks": self.timer_tasks}
self.functions = {"timer_status": self.handleTimerStatus,
"tasks": self.handleTasks}
return (0)
def logDB(self, message: str):
Log.objects.create(agent='Monitoring', message=message)
'''
Function called by the main loop to check if the majordome and alert_manager are running
'''
def handleTasks(self):
self.timers["tasks"] = self.timer_tasks
# TODO check majordome and alert_manager status
if self.majordome_task is None:
try:
self.monitoring_task = TaskId.objects.get(task="majordome")
except Exception as e:
majordome.tasks.Majordome.apply_async()
if settings.DEBUG and DEBUG_FILE:
log.info(str(e))
if self.alert_task is None:
try:
self.alert_task = TaskId.objects.get(task="alert_manager")
except Exception as e:
alert_manager.tasks.AlertListener.apply_async()
if settings.DEBUG and DEBUG_FILE:
log.info(str(e))
return 0
'''
Parse status returned by plc
'''
def parseStatus(self, status_plc):
try:
status = {}
string_list = status_plc.splitlines()
for item in string_list:
res = item.split('=')
status[res[0]] = res[1]
except Exception as e:
if DEBUG_FILE and settings.DEBUG:
log.info(str(e))
self.status_plc = {}
return 1
self.status_plc = status
return 0
'''
Save status returned by plc
'''
def saveStatus(self):
outside = WeatherWatch()
inside = SiteWatch()
for key, value in self.status_plc.items():
if key.find("Weather") != -1:
outside.setAttribute(key, value)
else:
inside.setAttribute(key, value)
outside.setGlobalStatus()
inside.setGlobalStatus()
outside.save()
inside.save()
return 0
'''
Function called by the main loop to handle the plc status
'''
def handleTimerStatus(self):
self.timers["timer_status"] = self.timer_status
status_plc = self.plc.getStatus()
if (self.plc.isError(status_plc)):
if (settings.DEBUG and DEBUG_FILE):
log.info("Invalid status returned : " + str(self.status_plc))
return (1)
if self.parseStatus(status_plc):
if (settings.DEBUG and DEBUG_FILE):
log.info("Invalid status returned : " + str(self.status_plc))
return 1
return self.saveStatus()
'''
Main loop
'''
def loop(self):
while (self.state != "SHUTDOWN"):
minimal_timer = min(self.timers, key=self.timers.get)
time.sleep(self.timers[minimal_timer])
self.timers = {key: value - self.timers[minimal_timer] for key, value in self.timers.items()}
for timer_name, timer_value in self.timers.items():
if (timer_value <= 0):
if (timer_name in self.functions):
self.functions[timer_name]()
else:
if (settings.DEBUG and DEBUG_FILE):
log.info("Timer : " + str(timer_name) + "is not known by the monitoring")
self.logDB("Timer " + str(timer_name) + " unknown")
if (settings.DEBUG and DEBUG_FILE):
log.info("Timer : " + str(timer_name) + " executed by monitoring")
return (0)
if (__name__ == "__main__"):
m = Monitoring()
m.run()