Changeset 854 for trunk/Monitoring/Watch

Timestamp:

04/26/13 16:57:16 (11 years ago)

Author:

jripsl

Message:

add JSON deserialization, mail notification, logging.

File:

• trunk/Monitoring/Watch/watch (modified) (5 diffs)

trunk/Monitoring/Watch/watch

@@ -1 @@
-#!/usr/bin/python -u
+#!/usr/bin/env python
 # -*- coding: ISO-8859-1 -*-
 
@@ -13 +13 @@
 ##################################
 
-from smon import dao
+#from smon import dao
+import pika
+import base64
+import json
+import sys
+import traceback
+import smtplib
+from email.mime.text import MIMEText
 
 class Watcher():
+        message_code_action_mapping = {"0000":["log","mail"],"1000":["log"],"2000":["log"],"3000":["log"],"9000":["log"],"9999":["log","mail"]}
 
         @classmethod
@@ -23 +31 @@
         @classmethod
         def start(cls):
-                dao.insert_progress_messages(cls.get_fake_progress_messages())
+                #dao.insert_progress_messages(cls.get_fake_progress_messages())
+                pass
 
         @classmethod
         def stop(cls):
                 pass
- 
+
+        @classmethod
+        def add(cls,message):
+                pass
+
+        @classmethod
+        def mail_example(cls):
+                me="jripsl@ipsl.jussieu.fr"
+                you="jripsl@ipsl.jussieu.fr"
+                body="Alarm"
+                object="Supervisor"
+
+                cls.mail(me,you,object,body)
+
+        @classmethod
+        def mail(cls):
+                cls.mail_example()
+
+        @classmethod
+        def send_mail(cls,me,you,object,body):
+                msg = MIMEText(body)
+                msg['Subject'] = object
+                msg['From'] = me
+                msg['To'] = you
+
+                # Send the message via our own SMTP server, but don't include the # envelope header.
+                s = smtplib.SMTP('localhost')
+                s.sendmail(me,[you], msg.as_string())
+                s.quit()
+
+        @classmethod
+        def log(cls,message):
+
+                with open("/home/jripsl/supervisor/log/supervisor.log", "a") as log_file:
+                        log_file.write("%s %s\n"%(message["code"],message["jobid"]))
+
+        @classmethod
+        def execActions(cls,message):
+
+                message_code=message["code"]
+
+                for action in cls.message_code_action_mapping[message_code]:
+                        proc_name=action
+
+                        try:
+                                getattr(cls, proc_name)(message)
+                        except Exception,e:
+                                traceback.print_exc()
+
+                                raise Exception("WATCH-ERR002","procedure error (%s,%s)"%(proc_name,str(e)))
+
 def main():
 
+        """
         # parse args
         parser = argparse.ArgumentParser(prog='watcher')
@@ -41 +101 @@
 
         SMON.init_singleton()
+        """
+
+        connection = pika.BlockingConnection(pika.ConnectionParameters( host='localhost'))
+        channel = connection.channel()
+
+        #channel.queue_declare(queue='myqueue')
+
+        print ' [*] Waiting for messages. To exit press CTRL+C'
+
+        def callback(ch, method, properties, raw_msg):
+                #
+                #return
+
+                base64_decoded_msg=base64.b64decode(raw_msg)
+
+                # debug
+                #print " [x] Received %s" % raw_msg
+                #print " [x] Received %s (uudecoded)" % base64_decoded_msg 
+
+                try:
+                        message=json.loads(base64_decoded_msg)
+                except Exception,e:
+                        print base64_decoded_msg
+                
+                # message code based action
+                Watcher.execActions(message)
+
+        channel.basic_consume(callback, queue='myqueue', no_ack=True)
+
+        channel.start_consuming()
 
 
-
+        """
         SMON.free_singleton()
+        """
 
 if __name__ == '__main__':
@@ -54 +145 @@
         except Exception, e:
 
-                #traceback.print_exc()
+                traceback.print_exc()
 
                 sys.exit(1)