terom@31: import optparse, sys
terom@31: 
terom@31: from pvl.syslog.parser import SyslogParser
terom@31: from pvl.syslog.syslog import SyslogSource
terom@31: from pvl.syslog import fifo, tail
terom@31: 
terom@31: # XXX: use optparse parser.error()?
terom@31: import logging; log = logging.getLogger('pvl.syslog.args')
terom@31: 
terom@31: def parser (parser, prog=None) :
terom@31:     """
terom@31:         Optparse option group
terom@31: 
terom@31:             prog        - filter to only process lines from given process
terom@31:     """
terom@31:     
terom@31:     syslog = optparse.OptionGroup(parser, 'Syslog collector')
terom@31: 
terom@31:     syslog.add_option('--syslog-fifo',          metavar='PATH',
terom@31:             help="Read syslog messages from given fifo")
terom@31: 
terom@31:     syslog.add_option('--syslog-file',          metavar='FILE',
terom@31:             help="Read syslog messages from given file")
terom@31: 
terom@31:     syslog.add_option('--syslog-tail',          type='float', metavar='POLL',
terom@31:             help="Continuously poll file")
terom@31: 
terom@31:     syslog.add_option('--syslog-raw',           action='store_true',
terom@31:             help="Parse raw syslog lines without timestamp/etc")
terom@31: 
terom@31:     syslog.add_option('--syslog-prog',          metavar='PROG',     default=prog,
terom@31:             help="Filter by given prog: %default")
terom@31: 
terom@31:     return syslog
terom@31: 
terom@31: def apply (options, optional=False) :
terom@31:     """
terom@31:         Handle options, returning a SyslogSource, if any.
terom@31: 
terom@31:         May log.error/sys.exit
terom@31:     """
terom@31: 
terom@31:     if options.syslog_fifo :
terom@31:         # read fifo
terom@31:         source = fifo.Fifo(options.syslog_fifo)
terom@31: 
terom@31:     elif options.syslog_file :
terom@31:         # tail file
terom@31:         source = tail.TailFile(options.syslog_file)
terom@31: 
terom@31:     elif optional :
terom@31:         log.debug("No --syslog source given")
terom@31:         return None
terom@31: 
terom@31:     else :
terom@31:         log.error("No --syslog source given")
terom@31:         sys.exit(2)
terom@31:     
terom@31:     parser = SyslogParser(
terom@31:         raw     = options.syslog_raw,
terom@31:         prog    = options.syslog_prog,
terom@31:     )
terom@31: 
terom@31:     # build
terom@31:     return SyslogSource(source, parser)
terom@31: