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 +59 -38

             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
-                tasks = {
+                def run(self):
-: send_email,
+                    tasks = {
-: play_sound,
+: self.send_email,
-: show_popup,
+: self.play_sound,
-: send_alarm,
+: self.show_popup,
+: self.send_alarm,
+                    }
-                tasks_args = {
-: ['email', 'message', 'subject', 'subtitle', 'filename'],
+                    tasks_args = {
-: [],
+: ['email', 'message', 'subject', 'subtitle', 'filename'],
-: ['message'],
+: [],
-: [],
+: ['message'],
+: [],
+                    }
-                for mode in modes:
+                    procs = []
-                    if mode == 0:
+                    for mode in self.modes:
-                        for x in tasks:
+                        if 0 in self.modes:
-                            t = Thread(target=tasks[x], kwargs=get_kwargs(kwargs, tasks_args[x]))
+                            for x in tasks:
+                                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=self.get_kwargs(self.kwargs, tasks_args[mode]))
                             t.start()
-                        break
+                            procs.append(t)
-                    else:
+                    for t in procs:
-                        t = Thread(target=tasks[mode], kwargs=get_kwargs(kwargs, tasks_args[x]))
+                        t.join()
-                        t.start()
             class SchainConfigure():
                     self.__emailServer  = confObj.getEmailServer()
                 def sendEmail(self, email_from, email_to, subject='Error running ...', message="", subtitle="", filename="", html_format=True):
                     if not email_to:
                         return 0
                     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
                     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,
-                                   email_to=email,
-                                   subject=subject,
-                                   message=message,
-                                   subtitle=subtitle,
-                                   filename=filename)
-                    log.success('Email sent', 'ADMIN')
+                    self.sendEmail(
+                        email_from=self.__emailFromAddress,
+                        email_to=email,
+                        subject=subject,
+                        message=message,
+                        subtitle=subtitle,
+                        filename=filename
+                        )
             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