now using python-genericore

1 files changed, 18 insertions, 36 deletions

diff --git a/src/main.py b/src/main.py
index 2d9bb206..473729f4 100755
--- a/src/main.py
+++ b/src/main.py
@@ -1,50 +1,32 @@
 #!/usr/bin/env python2
-
-import logging
-logging.basicConfig(level=logging.INFO)
-log = logging.getLogger('snmp_exchange')
-
-import pika
-import json,argparse,hashlib,sys,time
+import sys,json,time
 from snmp_users import snmp_users
+import logging
+import genericore as gen
+log = logging.getLogger('mail_proc_main')
+PROTO_VERSION = 1
+DESCRIPTION = 'performes statistical analysis against mails from stream'
 
-SNMP_EXCHANGE='snmp_src'
-PROTO_VERSION='1'
-
-parser = argparse.ArgumentParser(description='Generates a list of mac-addresses currently in the network via snmp ')
-parser.add_argument('--host',default='141.31.8.11',      help='AMQP host ip address')
-parser.add_argument('--port',type=int,default=5672,      help='AMQP host port')
-parser.add_argument('-u','--username',default='guest',   help='AMQP username') 
-parser.add_argument('-p','--password',default='guest',   help='AMQP password') 
-parser.add_argument('-r','--repeat',type=int,default=20, help='SNMP_Polling Delay') 
-parser.add_argument('--unique-key',action='store_true',   help='Unique Key') 
-args = parser.parse_args()
-if args.unique_key:
-    print hashlib.sha1(PROTO_VERSION+args.host+str(args.port)).hexdigest()
-    sys.exit(0)
 
-args = parser.parse_args()
+# set up instances of needed modules
+conf = gen.Configurator(PROTO_VERSION,DESCRIPTION)  
+amqp = gen.auto_amqp()   
+s = snmp_users()       # the magic mail parsing class
 
-connection = pika.AsyncoreConnection(pika.ConnectionParameters(
-          credentials = pika.PlainCredentials(args.username,args.password), 
-          host=args.host,port=args.port))
-channel = connection.channel()
+conf.configure([amqp,s]) #set up parser and eval parsed stuff
 
-channel.exchange_declare(exchange=SNMP_EXCHANGE,
-                             type='fanout')
+# start network connections
+amqp.create_connection()
 
 log.info('Starting up snmp_users')
-s = snmp_users()
 print ' Sending Messages in Intervals. To exit press CTRL+C'
 try:
   while True:
     log.info("collecting data from network")
     ret = s.collect()
     data = { 'type' : 'snmp', 'subtype' : 0, 'data' : ret}
-    log.info("writing data to queue : %s" % data)
-    channel.basic_publish(exchange=SNMP_EXCHANGE,
-        routing_key='',
-        body=json.dumps(data))
-    time.sleep(args.repeat)
-except :
-  print "something happened :("
+    log.debug("writing data to queue : %s" % data)
+    amqp.publish(json.dumps(data))
+    time.sleep(s.repeat)
+except Exception as e:
+  print "something happened :( " + str(e)