aha-eye uses getopt

aha/aha-eye.py

@@ -5,12 +5,31 @@
 #process vectors on the fly
 
 from ahalib import *
-
+import getopt
 logfile='aha.log'
-aha = AHAActions('../in','../out')
+aha = AHAActions(None,None)
 ptress = ProcessTrees()
 
-def extract_object(obj):
+def usage(exitcode):
+    print """ 
+Analyze log files of aha-worker and recovers process trees
+
+OPTIONS
+
+    -h Shows this screen
+    -e Specifies an export file (i.e. accessible through apache)
+    -l Specifies the log file generated by aha-worker
+
+AUTHOR
+    Gerard Wagener
+
+LICENSE
+
+    GPL
+"""
+    sys.exit(exitcode)
+
+def extract_object(obj,exportdir):
     try:
         #FIXME Until now discard decisions from aha
         if obj.has_key('block') and obj.has_key('insult'):
@@ -47,29 +66,50 @@ def extract_object(obj):
                 #User disconnected generate a report, to avoid that other 
                 #information is droped
                 print "List export is triggered for root ",pid
-                ptress.exportUserListTxt('userlist.txt')
+                ptress.exportUserListTxt(exportdir)
                 ptress.silent_remove_pid(pid)
                 #Cleanup annotated list
                 print "Clean annotated list"
                 ptress.clean_aplist(pid)
     except ValueError,e:
-        print "Failed to parse ",obj
+        sys.stderr.write("Failed to parse "+str(obj) + '\n')
     except KeyError,e:
-        print "Incomplete message"
+        sys.stderr.write("Incomplete message\n")
  
-line = None
 try:
-    f = open('aha.log','r')
+    line        = None
+    logfile     = None
+    exportdir   = None
+    opts,args = getopt.getopt(sys.argv[1:],"hl:e:",["help","logfile=", "export="])
+    for o,a in opts:
+        if o  in ('--help','-h'):
+            usage(0)
+        if o in ('--logfile','-l'):
+            logfile = a
+        if o in ('--export','-e'):
+            exportdir = a 
+    if logfile == None:
+        sys.stderr.write('A log file from aha-worker needs to be specified\n')
+        sys.exit(1)
+    #Load config file and get opts
+    if exportdir == None:
+        sys.stderr.write('An export file needs to be specified\n')
+        sys.exit(1)
+    f = open(logfile,'r')
     for line in f:
         (timestamp,key,serobj) = line.split('|',2)
         obj = aha.unserializeMessage(serobj)
-        extract_object(obj)
+        extract_object(obj,exportdir)
     f.close()
+    #Dump process trees
+    ptress.exportUserListTxt(exportdir)
+    sys.exit(0)
 except ValueError,e:
     #File may be incomplete
-    print "Value error"
-    print e
-    print line
+    sys.stderr("Value error, file may be incomplete\n")
+    sys.stderr.write(str(e) + '\n')
+    sys.stderr.write(line)
 
-#Dump process trees
-ptress.exportUserListTxt('userlist.txt')
+except getopt.GetoptError,e:
+    sys.stderr.write(str(e)+'\n')
+    sys.exit(1)