terom@31: import optparse, sys
terom@31: 
terom@31: from pvl.syslog.parser import SyslogParser
terom@44: from pvl.syslog.filter import SyslogFilter
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@98:     syslog.add_option('--syslog-tail',          metavar='FILE',
terom@98:             help="Continuously poll syslog messages given 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@74:     syslog.add_option('--syslog-facility',      metavar='FACILITY',
terom@74:             help="Set/filter by given facility")
terom@74: 
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@50:         # fifo pipe
terom@31:         source = fifo.Fifo(options.syslog_fifo)
terom@50:         poll = True # select(source)
terom@43: 
terom@43:     elif options.syslog_tail :
terom@43:         # tail file
terom@98:         source = tail.TailFile(options.syslog_tail, skip=True)
terom@98:         poll = 2.0 # select(float)
terom@31: 
terom@31:     elif options.syslog_file :
terom@43:         # read file
terom@65:         source = tail.Tail(open(options.syslog_file))
terom@50:         poll = False # do not loop, just read up to EOF
terom@31: 
terom@31:     elif optional :
terom@31:         return None
terom@31: 
terom@31:     else :
terom@48:         # from stdin
terom@50:         if sys.stdin.isatty() :
terom@50:             log.warning("Reading syslog messages from TTY?")
terom@50:         
terom@50:         source = tail.Tail(sys.stdin)
terom@50:         poll = False # XXX: tty vs pipe vs file?
terom@50:     
terom@50:     # options
terom@31:     parser = SyslogParser(
terom@74:         raw         = options.syslog_raw,
terom@74:         facility    = options.syslog_facility,
terom@44:     )
terom@44: 
terom@44:     filter = SyslogFilter(
terom@52:         # glob pattern
terom@74:         prog        = options.syslog_prog,
terom@74:         facility    = options.syslog_facility,
terom@31:     )
terom@31: 
terom@50:     # polling
terom@60:     return SyslogSource(source, parser, filter, poll)