schain-jc Commit - r1129:986a05a7d203

Now alarm is a process, add SchainWarning exception for handling non-stop exceptions

jespinoza -

r1129:986a05a7d203

parent child

Context file:

r1129:986a05a7d203

Collapse all files

schainpy/admin.py +40 -19

             import ConfigParser
             import StringIO
             from threading import Thread
+            from multiprocessing import Process
             from email.mime.text import MIMEText
             from email.mime.application import MIMEApplication
             from email.mime.multipart import MIMEMultipart
                 except:
                     log.error('I am sorry, but something is wrong... __file__ not found')
-            def alarm(modes=[1], **kwargs):
+            class Alarm(Process):
                 '''
                 modes:
 - All
 - Send to alarm system TODO
                 '''
+                def __init__(self, modes=[1], **kwargs):
+                    Process.__init__(self)
+                    self.modes = modes
+                    self.kwargs = kwargs
+                @staticmethod
                 def play_sound():
                     sound = os.path.join(get_path(), 'alarm1.oga')
                     if os.path.exists(sound):
                     else:
                         log.warning('Unable to play alarm, sound file not found', 'ADMIN')
+                @staticmethod
                 def send_email(**kwargs):
                     notifier = SchainNotify()
                     notifier.notify(**kwargs)
+                @staticmethod
                 def show_popup(message='Error'):
                     popup(message)
+                @staticmethod
                 def send_alarm():
                     pass
+                @staticmethod
                 def get_kwargs(kwargs, keys):
                     ret = {}
                     for key in keys:
                         ret[key] = kwargs[key]
                     return ret
+                def run(self):
                     tasks = {
-: send_email,
+: self.send_email,
-: play_sound,
+: self.play_sound,
-: show_popup,
+: self.show_popup,
-: send_alarm,
+: self.send_alarm,
                     }
                     tasks_args = {
 : ['message'],
 : [],
                     }
+                    procs = []
-                for mode in modes:
+                    for mode in self.modes:
-                    if mode == 0:
+                        if 0 in self.modes:
                             for x in tasks:
-                            t = Thread(target=tasks[x], kwargs=get_kwargs(kwargs, tasks_args[x]))
+                                t = Thread(target=tasks[x], kwargs=self.get_kwargs(self.kwargs, tasks_args[x]))
                                 t.start()
+                                procs.append(t)
                             break
                         else:
-                        t = Thread(target=tasks[mode], kwargs=get_kwargs(kwargs, tasks_args[x]))
+                            t = Thread(target=tasks[mode], kwargs=self.get_kwargs(self.kwargs, tasks_args[mode]))
                             t.start()
+                            procs.append(t)
+                    for t in procs:
+                        t.join()
             class SchainConfigure():
                     if not self.__emailServer:
                         return 0
+                    log.success('Sending email to {}...'.format(email_to), 'System')
                     msg = MIMEMultipart()
                     msg['Subject'] = subject
                     msg['From'] = "(Python SChain API): " + email_from
                     try:
                         smtp = smtplib.SMTP(self.__emailServer)
                     except:
-                        print "***** Could not connect to server %s *****" %self.__emailServer
+                        log.error('Could not connect to server {}'.format(self.__emailServer), 'System')
                         return 0
                     # Start the server:
                     try:
                         smtp.sendmail(msg['From'], msg['To'], msg.as_string())
                     except:
-                        print "***** Could not send the email to %s *****" %msg['To']
+                        log.error('Could not send the email to {}'.format(msg['To']), 'System')
                         smtp.quit()
                         return 0
                     smtp.quit()
+                    log.success('Email sent ', 'System')
                     return 1
                 def sendAlert(self, message, subject = "", subtitle="", filename=""):
                     if not sent:
                         return 0
-                    print "***** Your system administrator has been notified *****"
                     return 1
                 def notify(self, email, message, subject = "", subtitle="", filename=""):
                     if email is None:
                         email = self.__emailToAddress
-                    log.success('Notifying to %s ...'.format(email), 'ADMIN')
+                    self.sendEmail(
+                        email_from=self.__emailFromAddress,
-                    self.sendEmail(email_from=self.__emailFromAddress,
                         email_to=email,
                         subject=subject,
                         message=message,
                         subtitle=subtitle,
-                                   filename=filename)
+                        filename=filename
+                        )
-                    log.success('Email sent', 'ADMIN')
             class SchainError(Exception):
                 """SchainError is an exception class that is thrown for all known errors using Schain Py lib.
                     return excStr
+            class SchainWarning(Exception):
+                pass
             if __name__ == '__main__':
                 test = SchainNotify()

schainpy/controller.py +11 -5

             from xml.dom import minidom
             import schainpy
-            import schainpy.admin
+            from schainpy.admin import Alarm, SchainWarning
             from schainpy.model import *
             from schainpy.utils import log
                         self.__connect(puObjIN, thisPUObj)
-                def __handleError(self, procUnitConfObj):
+                def __handleError(self, procUnitConfObj, modes=None):
                     import socket
+                    if modes is None:
+                        modes = self.alarm
                     err = traceback.format_exception(sys.exc_info()[0],
                                                      sys.exc_info()[1],
                                                      sys.exc_info()[2])
                         subtitle += '[Start time = %s]\n' % readUnitConfObj.startTime
                         subtitle += '[End time = %s]\n' % readUnitConfObj.endTime
-                    schainpy.admin.alarm(
+                    a = Alarm(
-                        modes=self.alarm,
+                        modes=modes,
                         email=self.email,
                         message=message,
                         subject=subject,
                         filename=self.filename
                     )
+                    a.start()
                 def isPaused(self):
                     return 0
                             try:
                                 sts = procUnitConfObj.run()
                                 is_ok = is_ok or sts
+                            except SchainWarning:
+                                self.__handleError(procUnitConfObj, modes=[2, 3])
                             except KeyboardInterrupt:
                                 is_ok = False
                                 break

schainpy/model/graphics/jroplot_data.py +1 -2

                         return
             def popup(message):
-                fig = plt.figure(figsize=(12, 9), facecolor='r')
+                fig = plt.figure(figsize=(12, 8), facecolor='r')
                 fig.text(0.5, 0.5, message, ha='center', va='center', size='20', weight='heavy', color='w')
                 fig.show()
                 figpause(1000)
             class PlotData(Operation, Process):
                 '''
                 Base class for Schain plotting operations

General Comments 0

Write
Preview

You need to be logged in to leave comments. Login now

No TODOs yet

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages