pvl/syslog/args.py
changeset 31 3e6d0feb115c
child 43 9d13b101beab
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/pvl/syslog/args.py	Wed Oct 24 21:02:33 2012 +0300
@@ -0,0 +1,66 @@
+import optparse, sys
+
+from pvl.syslog.parser import SyslogParser
+from pvl.syslog.syslog import SyslogSource
+from pvl.syslog import fifo, tail
+
+# XXX: use optparse parser.error()?
+import logging; log = logging.getLogger('pvl.syslog.args')
+
+def parser (parser, prog=None) :
+    """
+        Optparse option group
+
+            prog        - filter to only process lines from given process
+    """
+    
+    syslog = optparse.OptionGroup(parser, 'Syslog collector')
+
+    syslog.add_option('--syslog-fifo',          metavar='PATH',
+            help="Read syslog messages from given fifo")
+
+    syslog.add_option('--syslog-file',          metavar='FILE',
+            help="Read syslog messages from given file")
+
+    syslog.add_option('--syslog-tail',          type='float', metavar='POLL',
+            help="Continuously poll file")
+
+    syslog.add_option('--syslog-raw',           action='store_true',
+            help="Parse raw syslog lines without timestamp/etc")
+
+    syslog.add_option('--syslog-prog',          metavar='PROG',     default=prog,
+            help="Filter by given prog: %default")
+
+    return syslog
+
+def apply (options, optional=False) :
+    """
+        Handle options, returning a SyslogSource, if any.
+
+        May log.error/sys.exit
+    """
+
+    if options.syslog_fifo :
+        # read fifo
+        source = fifo.Fifo(options.syslog_fifo)
+
+    elif options.syslog_file :
+        # tail file
+        source = tail.TailFile(options.syslog_file)
+
+    elif optional :
+        log.debug("No --syslog source given")
+        return None
+
+    else :
+        log.error("No --syslog source given")
+        sys.exit(2)
+    
+    parser = SyslogParser(
+        raw     = options.syslog_raw,
+        prog    = options.syslog_prog,
+    )
+
+    # build
+    return SyslogSource(source, parser)
+