tasks.py
6.07 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
from __future__ import absolute_import
from celery.task import Task
from common.RequestBuilder import RequestBuilder
from alert_manager.StrategyBuilder import StrategyBuilder
from django.conf import settings
from django.templatetags.static import static
import socket
from common.models import *
import time
import voeventparse
import os
from os.path import isfile, join
import observation_manager
import majordome.TaskManager
from decimal import Decimal
TIMESTAMP_JD = 2440587.500000
VOEVENTS_PATH = "alert_manager/events_received"
class alert_listener(Task):
'''
Launched at server start
Listens to VOEvents, and create appropriate request when an event is received
'''
def run(self):
'''
Is called at the beginning of the task
Calls the function to get the new events in the VOEVENTS_PATH directory
For each event, calls a fonction to analyze it
'''
print("alert listener started")
Log.objects.create(agent="Alert manager", message="Start alert manager")
self.old_files = [
f for f in os.listdir(VOEVENTS_PATH) if isfile(join(VOEVENTS_PATH, f))]
while True:
fresh_events = self.get_fresh_events()
for event in fresh_events:
self.analyze_event(event)
time.sleep(1)
def get_fresh_events(self):
'''
Reads in the VOEVENTS_PATH directory to see if there are new events to analyze
:returns : A list containing the new files to analyze
'''
fresh_events = []
files = [
f for f in os.listdir(VOEVENTS_PATH) if isfile(join(VOEVENTS_PATH, f))]
diff = list(set(files).difference(set(self.old_files)))
for event in diff:
print("New file found : %s" % (event,))
Log.objects.create(agent="Alert manager", message="New file found : %s" % (event,))
fresh_events.append(event)
self.old_files = files
return fresh_events
def analyze_event(self, event_file):
'''
Opens and parse the voevent file
Will create the request & the alert object related
'''
with open(os.path.join(VOEVENTS_PATH, event_file), 'rb') as f:
voevent = voeventparse.load(f)
# TODO: Faire un try/except pour gérer les mauvais fichiers
self.create_related_request(voevent, event_file)
def create_related_request(self, voevent, event_file):
'''
Creates a request object related to the voevent received.
For the moment, it doesn't take care of the voevent content, and hardcode the sequences etc
:param voevent: Object resulting of the VOEvent parsing with voeventparse library
:returns : The request
'''
pyros_user = PyrosUser.objects.get(user__username="pyros")
scientific_program = ScientificProgram.objects.get(name="GRB")
alert = self.get_alert_attributes(voevent, event_file)
if alert == None or StrategyObs.objects.filter(is_default=True).exists() == False:
# TODO: Supprimer le fichier reçu ? Ou le stocker dans un dossier à part
return
alert.strat = StrategyObs.objects.filter(is_default=True)[0]
name = "GRB - " + str(alert.trig_id)
Log.objects.create(agent="Alert manager", message="Creating alert from file : %s ..." % (alert.strat.xml_file,))
sb = StrategyBuilder()
sb.create_request_from_strategy(alert.strat.xml_file, pyros_user, scientific_program, name)
req = sb.validate()
for seq in req.sequences.all():
seq.ra = alert.burst_ra
seq.dec = alert.burst_dec
seq.save()
alert.request = req
alert.save()
Log.objects.create(agent="Alert manager", message="Alert created.")
def get_alert_attributes(self, voevent, event_file):
'''
Parses the VOEvent to get all the required attributes
Creates the alert object
Handles errors (missing params, wrong params, ...)
:param voevent: Object resulting of the VOEvent parsing with voeventparse library
:returns : The alert object
'''
'''
The parameters are a dictionnary with the "<What>" <Group>s as keys,
and a dictionnary of their <Param> as values.
The nested dictionnaries have the 'name' attribute as keys,
and another dictionnary as values, containing {attribute : value} pairs
The <Param>s without <Group> are in the dictionnary with 'None' as key
'''
v_params = voeventparse.pull_params(voevent)
alert = Alert(voevent_file=event_file)
try:
alert.trig_id = v_params[None]["TrigID"]["value"]
alert.editor = v_params[None]["Packet_Type"]["value"]
alert.soln_status = v_params[None]["Soln_Status"]["value"]
alert.pkt_ser_num = v_params[None]["Pkt_Ser_Num"]["value"]
tjd = Decimal(v_params[None]["Burst_TJD"]["value"])
sod = Decimal(v_params[None]["Burst_SOD"]["value"])
alert.burst_jd = tjd - 13370 - 1 + 2453371 + (sod / 86400)
alert.defly_not_grb = int(alert.soln_status[0]) & 0xF0
alert.author = voevent.Who.AuthorIVORN
alert.jd_send = time.mktime(time.strptime(str(voevent.Who.Date), "%Y-%m-%dT%H:%M:%S")) / 86400 + TIMESTAMP_JD
alert.jd_received = time.time() / 86400 + TIMESTAMP_JD
alert.astro_coord_system = voevent.WhereWhen.ObsDataLocation.ObservationLocation.AstroCoordSystem
alert.burst_ra = voevent.WhereWhen.ObsDataLocation.ObservationLocation.AstroCoords.Position2D.Value2.C1
alert.burst_dec = voevent.WhereWhen.ObsDataLocation.ObservationLocation.AstroCoords.Position2D.Value2.C2
alert.error_radius = voevent.WhereWhen.ObsDataLocation.ObservationLocation.AstroCoords.Position2D.Error2Radius
except KeyError or AttributeError as e:
print("Error while parsing VOEvent : ", e)
return None
return alert