vigiboard / vigiboard / controllers / root.py @ c1ce3d6a
History | View | Annotate | Download (15.1 KB)
1 |
# -*- coding: utf-8 -*-
|
---|---|
2 |
# vim:set expandtab tabstop=4 shiftwidth=4:
|
3 |
"""Vigiboard Controller"""
|
4 |
|
5 |
from tg import expose, validate, require, flash, \ |
6 |
tmpl_context, request, config, session, redirect, url |
7 |
from tw.forms import validators |
8 |
from pylons.i18n import ugettext as _ |
9 |
from pylons.controllers.util import abort |
10 |
from sqlalchemy import asc |
11 |
from vigiboard.model import DBSession |
12 |
from vigiboard.model import Event, EventHistory, EventsAggregate, \ |
13 |
Host, HostGroup, \ |
14 |
User
|
15 |
from repoze.what.predicates import Any, not_anonymous |
16 |
from vigiboard.widgets.edit_event import edit_event_status_options |
17 |
from vigiboard.controllers.vigiboardrequest import VigiboardRequest |
18 |
from vigiboard.controllers.vigiboard_controller import VigiboardRootController |
19 |
|
20 |
__all__ = ['RootController']
|
21 |
|
22 |
class RootController(VigiboardRootController): |
23 |
|
24 |
"""
|
25 |
Le controller général de vigiboard
|
26 |
"""
|
27 |
|
28 |
def process_form_errors(self, *argv, **kwargv): |
29 |
"""
|
30 |
Gestion des erreurs de validation : On affiche les erreurs
|
31 |
puis on redirige vers la dernière page accédée.
|
32 |
"""
|
33 |
for k in tmpl_context.form_errors: |
34 |
flash("'%s': %s" % (k, tmpl_context.form_errors[k]), 'error') |
35 |
if request.environ.get('HTTP_REFERER') : |
36 |
redirect(request.environ.get('HTTP_REFERER'
|
37 |
).split(request.environ.get('HTTP_HOST'))[1]) |
38 |
else :
|
39 |
redirect('/')
|
40 |
|
41 |
@expose('vigiboard.templates.vigiboard') |
42 |
@require(Any(not_anonymous(), msg=_("You need to be authenticated"))) |
43 |
def default(self, page = None, host = None, service = None, output = None, |
44 |
trouble_ticket=None, *argv, **krgv):
|
45 |
|
46 |
"""
|
47 |
Page d'accueil de Vigiboard. Elle affiche, suivant la page demandée
|
48 |
(page 1 par defaut), la liste des évènements, rangés par ordre de prise
|
49 |
en compte, puis de sévérité.
|
50 |
Pour accéder à cette page, l'utilisateur doit être authentifié.
|
51 |
|
52 |
@param page: Numéro de la page souhaitée, commence à 1
|
53 |
@param host: Si l'utilisateur souhaite sélectionner seulement certains
|
54 |
évènements suivant leur hôte, il peut placer une expression
|
55 |
ici en suivant la structure du LIKE en SQL
|
56 |
@param service: Idem que host mais sur les services
|
57 |
@param output: Idem que host mais sur le text explicatif
|
58 |
@param trouble_ticket: Idem que host mais sur les tickets attribués
|
59 |
"""
|
60 |
if page is None: |
61 |
page = 1
|
62 |
|
63 |
try:
|
64 |
page = int(page)
|
65 |
except ValueError: |
66 |
abort(404)
|
67 |
|
68 |
if page < 1: |
69 |
page = 1
|
70 |
|
71 |
username = request.environ['repoze.who.identity']['repoze.who.userid'] |
72 |
events = VigiboardRequest(User.by_user_name(username)) |
73 |
|
74 |
search = { |
75 |
'host': '', |
76 |
'service': '', |
77 |
'output': '', |
78 |
'tt': '' |
79 |
} |
80 |
# Application des filtres si nécessaire
|
81 |
if host :
|
82 |
search['host'] = host
|
83 |
host = host.replace('%', '\\%').replace('_', '\\_') \ |
84 |
.replace('*', '%').replace('?', '_') |
85 |
events.add_filter(Event.hostname.like('%%%s%%' % host))
|
86 |
|
87 |
if service :
|
88 |
search['service'] = service
|
89 |
service = service.replace('%', '\\%').replace('_', '\\_') \ |
90 |
.replace('*', '%').replace('?', '_') |
91 |
events.add_filter(Event.servicename.like('%%%s%%' % service))
|
92 |
|
93 |
if output :
|
94 |
search['output'] = output
|
95 |
output = output.replace('%', '\\%').replace('_', '\\_') \ |
96 |
.replace('*', '%').replace('?', '_') |
97 |
events.add_filter(Event.message.like('%%%s%%' % output))
|
98 |
|
99 |
if trouble_ticket :
|
100 |
search['tt'] = trouble_ticket
|
101 |
trouble_ticket = trouble_ticket.replace('%', '\\%') \ |
102 |
.replace('_', '\\_').replace('*', '%').replace('?', '_') |
103 |
events.add_filter(EventsAggregate.trouble_ticket.like( |
104 |
'%%%s%%' % trouble_ticket))
|
105 |
|
106 |
# Calcul des éléments à afficher et du nombre de pages possibles
|
107 |
total_rows = events.num_rows() |
108 |
|
109 |
item_per_page = int(config['vigiboard_item_per_page']) |
110 |
|
111 |
if total_rows <= item_per_page * (page-1) : |
112 |
page = 1
|
113 |
id_first_row = item_per_page * (page-1)
|
114 |
id_last_row = min(id_first_row + item_per_page, total_rows)
|
115 |
|
116 |
events.format_events(id_first_row, id_last_row) |
117 |
events.generate_tmpl_context() |
118 |
return dict( |
119 |
events = events.events, |
120 |
rows_info = { |
121 |
'id_first_row': id_first_row + 1, |
122 |
'id_last_row': id_last_row,
|
123 |
'total_rows': total_rows,
|
124 |
}, |
125 |
pages = range(1, (total_rows / item_per_page) + 2), |
126 |
page = page, |
127 |
event_edit_status_options = edit_event_status_options, |
128 |
history = [], |
129 |
hist_error = False,
|
130 |
plugin_context = events.context_fct, |
131 |
search = search, |
132 |
) |
133 |
|
134 |
@validate(validators={'idevent':validators.String(not_empty=True)}, |
135 |
error_handler=process_form_errors) |
136 |
@expose('json') |
137 |
@require(Any(not_anonymous(), msg=_("You need to be authenticated"))) |
138 |
def history_dialog(self, idevent): |
139 |
|
140 |
"""
|
141 |
JSon renvoyant les éléments pour l'affichage de la fenêtre de dialogue
|
142 |
contenant des liens internes et externes.
|
143 |
Pour accéder à cette page, l'utilisateur doit être authentifié.
|
144 |
|
145 |
@param id: identifiant de l'évènement
|
146 |
"""
|
147 |
|
148 |
# Obtention de données sur l'évènement et sur son historique
|
149 |
username = request.environ.get('repoze.who.identity'
|
150 |
).get('repoze.who.userid')
|
151 |
user = User.by_user_name(username) |
152 |
|
153 |
event = DBSession.query( |
154 |
EventsAggregate.severity, |
155 |
Event.hostname, |
156 |
Event.servicename, |
157 |
).join( |
158 |
(Event, EventsAggregate.idcause == Event.idevent), |
159 |
(HostGroup, Event.hostname == HostGroup.hostname), |
160 |
).filter(HostGroup.groupname.in_(user.groups) |
161 |
).filter(EventsAggregate.idcause == idevent).one() |
162 |
|
163 |
initial_state = DBSession.query( |
164 |
EventHistory, |
165 |
).filter(EventHistory.idevent == idevent |
166 |
).order_by(asc(EventHistory.timestamp) |
167 |
).order_by(asc(EventHistory.type_action)) |
168 |
|
169 |
if initial_state.count() > 0 : |
170 |
for i in initial_state: |
171 |
if i.value != '' and i.value is not None: |
172 |
initial_state = i.value |
173 |
break
|
174 |
else :
|
175 |
initial_state = 0
|
176 |
|
177 |
severity = ( |
178 |
_('None'), _('OK'), _('Suppressed'), _('Initial'), |
179 |
_('Maintenance'), _('Minor'), _('Major'), _('Critical') |
180 |
) |
181 |
eventdetails = {} |
182 |
for edname, edlink in \ |
183 |
config['vigiboard_links.eventdetails'].iteritems():
|
184 |
|
185 |
eventdetails[edname] = edlink[1] % {
|
186 |
'idevent': idevent,
|
187 |
'host': event.hostname,
|
188 |
'service': event.servicename
|
189 |
} |
190 |
|
191 |
return dict( |
192 |
initial_state = severity[int(initial_state)],
|
193 |
current_state = severity[event.severity], |
194 |
idevent = idevent, |
195 |
host = event.hostname, |
196 |
service = event.servicename, |
197 |
eventdetails = eventdetails |
198 |
) |
199 |
|
200 |
@validate(validators={'idevent':validators.String(not_empty=True)}, |
201 |
error_handler=process_form_errors) |
202 |
@expose('vigiboard.templates.vigiboard') |
203 |
@require(Any(not_anonymous(), msg=_("You need to be authenticated"))) |
204 |
def event(self, idevent): |
205 |
"""
|
206 |
Affichage de l'historique d'un évènement.
|
207 |
Pour accéder à cette page, l'utilisateur doit être authentifié.
|
208 |
|
209 |
@param idevent: identifiant de l'évènement souhaité
|
210 |
"""
|
211 |
|
212 |
username = request.environ['repoze.who.identity']['repoze.who.userid'] |
213 |
events = VigiboardRequest(User.by_user_name(username)) |
214 |
events.add_filter(EventsAggregate.idcause == idevent) |
215 |
|
216 |
# Vérification que l'évènement existe
|
217 |
if events.num_rows() != 1 : |
218 |
flash(_('Error in DB'), 'error') |
219 |
redirect('/')
|
220 |
|
221 |
events.format_events(0, 1) |
222 |
events.format_history() |
223 |
events.generate_tmpl_context() |
224 |
|
225 |
return dict( |
226 |
events = events.events, |
227 |
rows_info = { |
228 |
'id_first_row': 1, |
229 |
'id_last_row': 1, |
230 |
'total_rows': 1, |
231 |
}, |
232 |
pages = [1],
|
233 |
page = 1,
|
234 |
event_edit_status_options = edit_event_status_options, |
235 |
history = events.hist, |
236 |
hist_error = True,
|
237 |
plugin_context = events.context_fct, |
238 |
search = { |
239 |
'host': None, |
240 |
'service': None, |
241 |
'output': None, |
242 |
'tt': None |
243 |
} |
244 |
) |
245 |
|
246 |
@validate(validators={'host':validators.NotEmpty(), |
247 |
'service':validators.NotEmpty()}, error_handler=process_form_errors)
|
248 |
@expose('vigiboard.templates.vigiboard') |
249 |
@require(Any(not_anonymous(), msg=_("You need to be authenticated"))) |
250 |
def host_service(self, host, service): |
251 |
|
252 |
"""
|
253 |
Affichage de l'historique de l'ensemble des évènements correspondant
|
254 |
au host et service demandé.
|
255 |
Pour accéder à cette page, l'utilisateur doit être authentifié.
|
256 |
|
257 |
@param host: Nom de l'hôte souhaité.
|
258 |
@param service: Nom du service souhaité
|
259 |
"""
|
260 |
|
261 |
username = request.environ['repoze.who.identity']['repoze.who.userid'] |
262 |
events = VigiboardRequest(User.by_user_name(username)) |
263 |
events.add_filter(Event.hostname == host, |
264 |
Event.servicename == service) |
265 |
del events.filter[2] |
266 |
|
267 |
# Vérification qu'il y a au moins 1 évènement qui correspond
|
268 |
if events.num_rows() == 0 : |
269 |
redirect('/')
|
270 |
|
271 |
events.format_events(0, events.num_rows())
|
272 |
events.format_history() |
273 |
events.generate_tmpl_context() |
274 |
return dict( |
275 |
events = events.events, |
276 |
rows_info = { |
277 |
'id_first_row': 1, |
278 |
'id_last_row': 1, |
279 |
'total_rows': 1, |
280 |
}, |
281 |
pages = [1],
|
282 |
page = 1,
|
283 |
event_edit_status_options = edit_event_status_options, |
284 |
history = events.hist, |
285 |
hist_error = True,
|
286 |
plugin_context = events.context_fct, |
287 |
search = { |
288 |
'host': None, |
289 |
'service': None, |
290 |
'output': None, |
291 |
'tt': None |
292 |
} |
293 |
) |
294 |
|
295 |
@validate(validators={
|
296 |
"id":validators.Regex(r'^[^,]+(,[^,]*)*,?$'), |
297 |
"trouble_ticket":validators.Regex(r'^[0-9]*$'), |
298 |
"status":validators.OneOf(['NoChange', 'None', 'Acknowledged', |
299 |
'AAClosed'])
|
300 |
}, error_handler=process_form_errors) |
301 |
@require(Any(not_anonymous(), msg=_("You need to be authenticated"))) |
302 |
def update(self,**krgv): |
303 |
|
304 |
"""
|
305 |
Mise à jour d'un évènement suivant les arguments passés.
|
306 |
Cela peut être un changement de ticket ou un changement de statut.
|
307 |
|
308 |
@param krgv['id']: Le ou les identifiants des évènements à traiter
|
309 |
@param krgv['tt']: Nouveau numéro du ticket associé.
|
310 |
@param krgv['status']: Nouveau status de/des évènements.
|
311 |
"""
|
312 |
|
313 |
# Si l'utilisateur édite plusieurs évènements à la fois,
|
314 |
# il nous faut chacun des identifiants
|
315 |
|
316 |
ids = krgv['id'].split(',') |
317 |
|
318 |
if len(ids) > 1 : |
319 |
ids = ids[:-1]
|
320 |
|
321 |
username = request.environ['repoze.who.identity']['repoze.who.userid'] |
322 |
events = VigiboardRequest(User.by_user_name(username)) |
323 |
events.add_filter(EventsAggregate.idcause.in_(ids)) |
324 |
|
325 |
# Vérification que au moins un des identifiants existe et est éditable
|
326 |
if events.num_rows() <= 0 : |
327 |
flash(_('No access to this event'), 'error') |
328 |
redirect('/')
|
329 |
|
330 |
# Modification des évènements et création d'un historique
|
331 |
# pour chacun d'eux
|
332 |
|
333 |
username = request.environ['repoze.who.identity']['repoze.who.userid'] |
334 |
|
335 |
for req in events.req : |
336 |
if isinstance(req, EventsAggregate): |
337 |
event = req |
338 |
else:
|
339 |
event = req[0]
|
340 |
|
341 |
if krgv['trouble_ticket'] != '' : |
342 |
event.trouble_ticket = krgv['trouble_ticket']
|
343 |
history = EventHistory( |
344 |
type_action="Ticket change",
|
345 |
idevent=event.idcause, |
346 |
value='',
|
347 |
text='',
|
348 |
username=username, |
349 |
) |
350 |
DBSession.add(history) |
351 |
|
352 |
if krgv['status'] != 'NoChange' : |
353 |
event.status = krgv['status']
|
354 |
history = EventHistory( |
355 |
type_action="Acknowlegement change state",
|
356 |
idevent=event.idcause, |
357 |
value='',
|
358 |
text='',
|
359 |
username=username, |
360 |
) |
361 |
DBSession.add(history) |
362 |
|
363 |
flash(_('Updated successfully'))
|
364 |
redirect(request.environ.get('HTTP_REFERER', url('/'))) |
365 |
|
366 |
|
367 |
@validate(validators={"plugin_name":validators.OneOf( |
368 |
[i for [i,j] in config.get('vigiboard_plugins', [])])}, |
369 |
error_handler = process_form_errors) |
370 |
@expose('json') |
371 |
def get_plugin_value(self, plugin_name, *arg, **krgv): |
372 |
"""
|
373 |
Permet aux plugins de pouvoir récupérer des valeurs Json
|
374 |
"""
|
375 |
plugins = config['vigiboard_plugins']
|
376 |
if plugins is None: |
377 |
return
|
378 |
|
379 |
plugin = [i for i in plugins if i[0] == plugin_name][0] |
380 |
try:
|
381 |
mypac = __import__(
|
382 |
'vigiboard.controllers.vigiboard_plugin.' + plugin[0], |
383 |
globals(), locals(), [plugin[1]], -1) |
384 |
plug = getattr(mypac, plugin[1])() |
385 |
return plug.controller(*arg, **krgv)
|
386 |
except:
|
387 |
raise
|
388 |
|
389 |
# @validate(validators= {"fontsize": validators.Int()},
|
390 |
# error_handler = process_form_errors)
|
391 |
@expose('json') |
392 |
def set_fontsize(self, fontsize): |
393 |
"""
|
394 |
Save font size
|
395 |
"""
|
396 |
session['fontsize'] = fontsize
|
397 |
session.save() |
398 |
return dict(ret= 'ok') |
399 |
|
400 |
@validate(validators= {"refresh": validators.Int()}, |
401 |
error_handler = process_form_errors) |
402 |
@expose('json') |
403 |
def set_refresh(self, refresh): |
404 |
"""
|
405 |
Save refresh time
|
406 |
"""
|
407 |
session['refresh'] = refresh
|
408 |
session.save() |
409 |
return dict(ret= 'ok') |