##// END OF EJS Templates
schain-jc
RSS
Merge branch 'schain_alarm' of http://jro-dev.igp.gob.pe/rhodecode/schain into v2.3
jespinoza -
r1131:e8ab492363ef merge
parent child
Show More
Side by Side Unified
Context file:
r1131:e8ab492363ef Loading diff...:
Show file before | Show file after | Hide comments
1 NO CONTENT: new file 100644, binary diff hidden
NO CONTENT: new file 100644, binary diff hidden
Show file before | Show file after | Hide comments
1 NO CONTENT: new file 100644, binary diff hidden
NO CONTENT: new file 100644, binary diff hidden
Show file before | Show file after | Hide comments
@@ -5,19 +5,114 notification class and a standard error handing class.
5
5
6 $Id: admin.py 3966 2015-12-01 14:32:29Z miguel.urco $
6 $Id: admin.py 3966 2015-12-01 14:32:29Z miguel.urco $
7 """
7 """
8 import os, sys
8 import os
9 import sys
10 import time
9 import traceback
11 import traceback
10 import smtplib
12 import smtplib
11 import ConfigParser
13 import ConfigParser
12 import StringIO
14 import StringIO
13
15 from threading import Thread
16 from multiprocessing import Process
14 from email.mime.text import MIMEText
17 from email.mime.text import MIMEText
15 from email.mime.application import MIMEApplication
18 from email.mime.application import MIMEApplication
16 from email.mime.multipart import MIMEMultipart
19 from email.mime.multipart import MIMEMultipart
17
20
21 from schainpy.utils import log
22 from schainpy.model.graphics.jroplot_data import popup
23
24 def get_path():
25 '''
26 Return schainpy path
27 '''
28
29 try:
30 root = __file__
31 if os.path.islink(root):
32 root = os.path.realpath(root)
33
34 return os.path.dirname(os.path.abspath(root))
35 except:
36 log.error('I am sorry, but something is wrong... __file__ not found')
37
38 class Alarm(Process):
39 '''
40 modes:
41 0 - All
42 1 - Sound alarm
43 2 - Send email
44 3 - Popup message
45 4 - Send to alarm system TODO
46 '''
47
48 def __init__(self, modes=[1], **kwargs):
49 Process.__init__(self)
50 self.modes = modes
51 self.kwargs = kwargs
52
53 @staticmethod
54 def play_sound():
55 sound = os.path.join(get_path(), 'alarm1.oga')
56 if os.path.exists(sound):
57 for __ in range(2):
58 os.system('paplay {}'.format(sound))
59 time.sleep(0.5)
60 else:
61 log.warning('Unable to play alarm, sound file not found', 'ADMIN')
62
63 @staticmethod
64 def send_email(**kwargs):
65 notifier = SchainNotify()
66 notifier.notify(**kwargs)
67
68 @staticmethod
69 def show_popup(message='Error'):
70 popup(message)
71
72 @staticmethod
73 def send_alarm():
74 pass
75
76 @staticmethod
77 def get_kwargs(kwargs, keys):
78 ret = {}
79 for key in keys:
80 ret[key] = kwargs[key]
81 return ret
82
83 def run(self):
84 tasks = {
85 1 : self.send_email,
86 2 : self.play_sound,
87 3 : self.show_popup,
88 4 : self.send_alarm,
89 }
90
91 tasks_args = {
92 1: ['email', 'message', 'subject', 'subtitle', 'filename'],
93 2: [],
94 3: ['message'],
95 4: [],
96 }
97 procs = []
98 for mode in self.modes:
99 if 0 in self.modes:
100 for x in tasks:
101 t = Thread(target=tasks[x], kwargs=self.get_kwargs(self.kwargs, tasks_args[x]))
102 t.start()
103 procs.append(t)
104 break
105 else:
106 t = Thread(target=tasks[mode], kwargs=self.get_kwargs(self.kwargs, tasks_args[mode]))
107 t.start()
108 procs.append(t)
109 for t in procs:
110 t.join()
111
112
18 class SchainConfigure():
113 class SchainConfigure():
19
114
20 __DEFAULT_ADMINISTRATOR_EMAIL = ""
115 __DEFAULT_ADMINISTRATOR_EMAIL = "juan.espinoza@jro.igp.gob.pe"
21 __DEFAULT_EMAIL_SERVER = "jro-zimbra.igp.gob.pe"
116 __DEFAULT_EMAIL_SERVER = "jro-zimbra.igp.gob.pe"
22 __DEFAULT_SENDER_EMAIL = "notifier-schain@jro.igp.gob.pe"
117 __DEFAULT_SENDER_EMAIL = "notifier-schain@jro.igp.gob.pe"
23 __DEFAULT_SENDER_PASS = ""
118 __DEFAULT_SENDER_PASS = ""
@@ -183,6 +278,8 class SchainNotify:
183 if not self.__emailServer:
278 if not self.__emailServer:
184 return 0
279 return 0
185
280
281 log.success('Sending email to {}...'.format(email_to), 'System')
282
186 msg = MIMEMultipart()
283 msg = MIMEMultipart()
187 msg['Subject'] = subject
284 msg['Subject'] = subject
188 msg['From'] = "(Python SChain API): " + email_from
285 msg['From'] = "(Python SChain API): " + email_from
@@ -204,7 +301,7 class SchainNotify:
204
301
205 msg.attach(part)
302 msg.attach(part)
206
303
207 if os.path.isfile(filename):
304 if filename and os.path.isfile(filename):
208 # This is the binary part(The Attachment):
305 # This is the binary part(The Attachment):
209 part = MIMEApplication(open(filename,"rb").read())
306 part = MIMEApplication(open(filename,"rb").read())
210 part.add_header('Content-Disposition',
307 part.add_header('Content-Disposition',
@@ -216,7 +313,7 class SchainNotify:
216 try:
313 try:
217 smtp = smtplib.SMTP(self.__emailServer)
314 smtp = smtplib.SMTP(self.__emailServer)
218 except:
315 except:
219 print "***** Could not connect to server %s *****" %self.__emailServer
316 log.error('Could not connect to server {}'.format(self.__emailServer), 'System')
220 return 0
317 return 0
221
318
222 # Start the server:
319 # Start the server:
@@ -228,12 +325,14 class SchainNotify:
228 try:
325 try:
229 smtp.sendmail(msg['From'], msg['To'], msg.as_string())
326 smtp.sendmail(msg['From'], msg['To'], msg.as_string())
230 except:
327 except:
231 print "***** Could not send the email to %s *****" %msg['To']
328 log.error('Could not send the email to {}'.format(msg['To']), 'System')
232 smtp.quit()
329 smtp.quit()
233 return 0
330 return 0
234
331
235 smtp.quit()
332 smtp.quit()
236
333
334 log.success('Email sent ', 'System')
335
237 return 1
336 return 1
238
337
239 def sendAlert(self, message, subject = "", subtitle="", filename=""):
338 def sendAlert(self, message, subject = "", subtitle="", filename=""):
@@ -264,8 +363,6 class SchainNotify:
264 if not sent:
363 if not sent:
265 return 0
364 return 0
266
365
267 print "***** Your system administrator has been notified *****"
268
269 return 1
366 return 1
270
367
271 def notify(self, email, message, subject = "", subtitle="", filename=""):
368 def notify(self, email, message, subject = "", subtitle="", filename=""):
@@ -280,16 +377,18 class SchainNotify:
280 Exceptions: None.
377 Exceptions: None.
281 """
378 """
282
379
283 print "Notifying to %s ..." %email
380 if email is None:
381 email = self.__emailToAddress
284
382
285 self.sendEmail(email_from=self.__emailFromAddress,
383 self.sendEmail(
384 email_from=self.__emailFromAddress,
286 email_to=email,
385 email_to=email,
287 subject=subject,
386 subject=subject,
288 message=message,
387 message=message,
289 subtitle=subtitle,
388 subtitle=subtitle,
290 filename=filename)
389 filename=filename
390 )
291
391
292 print "***** Your system administrator has been notified *****"
293
392
294 class SchainError(Exception):
393 class SchainError(Exception):
295 """SchainError is an exception class that is thrown for all known errors using Schain Py lib.
394 """SchainError is an exception class that is thrown for all known errors using Schain Py lib.
@@ -386,6 +485,10 class SchainError(Exception):
386
485
387 return excStr
486 return excStr
388
487
488 class SchainWarning(Exception):
489 pass
490
491
389 if __name__ == '__main__':
492 if __name__ == '__main__':
390
493
391 test = SchainNotify()
494 test = SchainNotify()
Show file before | Show file after | Hide comments
@@ -15,7 +15,7 from xml.etree.ElementTree import ElementTree, Element, SubElement, tostring
15 from xml.dom import minidom
15 from xml.dom import minidom
16
16
17 import schainpy
17 import schainpy
18 import schainpy.admin
18 from schainpy.admin import Alarm, SchainWarning
19 from schainpy.model import *
19 from schainpy.model import *
20 from schainpy.utils import log
20 from schainpy.utils import log
21
21
@@ -906,11 +906,10 class Project(Process):
906
906
907 Process.__init__(self)
907 Process.__init__(self)
908 self.id = None
908 self.id = None
909 # self.name = None
910 self.description = None
909 self.description = None
911
910 self.email = None
911 self.alarm = [0]
912 self.plotterQueue = plotter_queue
912 self.plotterQueue = plotter_queue
913
914 self.procUnitConfObjDict = {}
913 self.procUnitConfObjDict = {}
915
914
916 def __getNewId(self):
915 def __getNewId(self):
@@ -952,13 +951,12 class Project(Process):
952 procUnitConfObj = self.procUnitConfObjDict[procKey]
951 procUnitConfObj = self.procUnitConfObjDict[procKey]
953 idProcUnit = str(int(self.id) * 10 + n)
952 idProcUnit = str(int(self.id) * 10 + n)
954 procUnitConfObj.updateId(idProcUnit, parentId=self.id)
953 procUnitConfObj.updateId(idProcUnit, parentId=self.id)
955
956 newProcUnitConfObjDict[idProcUnit] = procUnitConfObj
954 newProcUnitConfObjDict[idProcUnit] = procUnitConfObj
957 n += 1
955 n += 1
958
956
959 self.procUnitConfObjDict = newProcUnitConfObjDict
957 self.procUnitConfObjDict = newProcUnitConfObjDict
960
958
961 def setup(self, id, name='', description=''):
959 def setup(self, id, name='', description='', email=None, alarm=[0]):
962
960
963 print
961 print
964 print '*' * 60
962 print '*' * 60
@@ -967,10 +965,13 class Project(Process):
967 print
965 print
968 self.id = str(id)
966 self.id = str(id)
969 self.description = description
967 self.description = description
968 self.email = email
969 self.alarm = alarm
970
970
971 def update(self, name, description):
971 def update(self, **kwargs):
972
972
973 self.description = description
973 for key, value in kwargs:
974 setattr(self, key, value)
974
975
975 def clone(self):
976 def clone(self):
976
977
@@ -1179,24 +1180,23 class Project(Process):
1179
1180
1180 self.__connect(puObjIN, thisPUObj)
1181 self.__connect(puObjIN, thisPUObj)
1181
1182
1182 def __handleError(self, procUnitConfObj, send_email=False):
1183 def __handleError(self, procUnitConfObj, modes=None):
1183
1184
1184 import socket
1185 import socket
1185
1186
1187 if modes is None:
1188 modes = self.alarm
1189
1186 err = traceback.format_exception(sys.exc_info()[0],
1190 err = traceback.format_exception(sys.exc_info()[0],
1187 sys.exc_info()[1],
1191 sys.exc_info()[1],
1188 sys.exc_info()[2])
1192 sys.exc_info()[2])
1189
1193
1190 print '***** Error occurred in %s *****' % (procUnitConfObj.name)
1194 log.error('{}'.format(err[-1]), procUnitConfObj.name)
1191 print '***** %s' % err[-1]
1192
1195
1193 message = ''.join(err)
1196 message = ''.join(err)
1194
1197
1195 sys.stderr.write(message)
1198 sys.stderr.write(message)
1196
1199
1197 if not send_email:
1198 return
1199
1200 subject = 'SChain v%s: Error running %s\n' % (
1200 subject = 'SChain v%s: Error running %s\n' % (
1201 schainpy.__version__, procUnitConfObj.name)
1201 schainpy.__version__, procUnitConfObj.name)
1202
1202
@@ -1218,11 +1218,16 class Project(Process):
1218 subtitle += '[Start time = %s]\n' % readUnitConfObj.startTime
1218 subtitle += '[Start time = %s]\n' % readUnitConfObj.startTime
1219 subtitle += '[End time = %s]\n' % readUnitConfObj.endTime
1219 subtitle += '[End time = %s]\n' % readUnitConfObj.endTime
1220
1220
1221 adminObj = schainpy.admin.SchainNotify()
1221 a = Alarm(
1222 adminObj.sendAlert(message=message,
1222 modes=modes,
1223 email=self.email,
1224 message=message,
1223 subject=subject,
1225 subject=subject,
1224 subtitle=subtitle,
1226 subtitle=subtitle,
1225 filename=self.filename)
1227 filename=self.filename
1228 )
1229
1230 a.start()
1226
1231
1227 def isPaused(self):
1232 def isPaused(self):
1228 return 0
1233 return 0
@@ -1292,12 +1297,14 class Project(Process):
1292 try:
1297 try:
1293 sts = procUnitConfObj.run()
1298 sts = procUnitConfObj.run()
1294 is_ok = is_ok or sts
1299 is_ok = is_ok or sts
1300 except SchainWarning:
1301 self.__handleError(procUnitConfObj, modes=[2, 3])
1295 except KeyboardInterrupt:
1302 except KeyboardInterrupt:
1296 is_ok = False
1303 is_ok = False
1297 break
1304 break
1298 except ValueError, e:
1305 except ValueError, e:
1299 time.sleep(0.5)
1306 time.sleep(0.5)
1300 self.__handleError(procUnitConfObj, send_email=True)
1307 self.__handleError(procUnitConfObj)
1301 is_ok = False
1308 is_ok = False
1302 break
1309 break
1303 except:
1310 except:
Show file before | Show file after | Hide comments
@@ -36,6 +36,12 def figpause(interval):
36 canvas.start_event_loop(interval)
36 canvas.start_event_loop(interval)
37 return
37 return
38
38
39 def popup(message):
40 fig = plt.figure(figsize=(12, 8), facecolor='r')
41 fig.text(0.5, 0.5, message, ha='center', va='center', size='20', weight='heavy', color='w')
42 fig.show()
43 figpause(1000)
44
39
45
40 class PlotData(Operation, Process):
46 class PlotData(Operation, Process):
41 '''
47 '''
Show file before | Show file after | Hide comments
@@ -41,6 +41,7 setup(name = "schainpy",
41 ext_package = 'schainpy',
41 ext_package = 'schainpy',
42 package_data = {'': ['schain.conf.template'],
42 package_data = {'': ['schain.conf.template'],
43 'schainpy.gui.figures': ['*.png', '*.jpg'],
43 'schainpy.gui.figures': ['*.png', '*.jpg'],
44 'schainpy.files': ['*.oga']
44 },
45 },
45 include_package_data = False,
46 include_package_data = False,
46 scripts = ['schainpy/gui/schainGUI'],
47 scripts = ['schainpy/gui/schainGUI'],
General Comments 0
You need to be logged in to leave comments. Login now