barn-growl/barn-growl.py

   1 #!/usr/bin/env python
   2
   3 """
   4 Subscribes to zephyr via tzc and sends messages to notification drivers (growl or libnotify).
   5 """
   6
   7 import sexpr
   8 import os
   9 import subprocess
  10 import fcntl
  11 import select
  12 import sys
  13 from abstfilter import AbstractConsumer
  14 import optparse
  15
  16 class Notifier(AbstractConsumer):
  17     def __init__(self, usegrowl, usenotify, useprint):
  18         self.usegrowl = usegrowl
  19         self.usenotify = usenotify
  20         if usenotify:
  21             import pynotify
  22             pynotify.init("Zephyr")
  23             self.pings = {}
  24             self.pynotify = pynotify
  25         self.useprint = useprint
  26         return
  27     def feed(self, s):
  28         if s is None or type(s) is type(''): return
  29         d = dict([(ss[0], len(ss) > 2 and ss[2] or None) for ss in s])
  30         if d['tzcspew'] == 'message':
  31             zclass = d['class'].lower()
  32             zinstance = d['instance'].lower()
  33             zop = d['opcode'].lower()
  34             zsender = d['sender'].lower()
  35             zauth = d['auth'].lower() == 'yes'
  36             ztime = ':'.join(d['time'].split(' ')[3].split(':')[0:2])
  37             zmessage = d['message']
  38             id = '%s/\n%s/\n%s\n %s' % (zclass, zinstance, zsender, ztime)
  39             if zop == 'ping':
  40                 header = '%s (%s)' % (id, zsender)
  41                 message = '...'
  42             elif zop == 'nil':
  43                 header = '%s (%s)' % (id, len(zmessage) > 0 and zmessage[0] or zsender)
  44                 message = '%s' % (len(zmessage) > 1 and zmessage[1] or '')
  45             else:
  46                 return
  47             if self.useprint:
  48                 print (id, header)
  49                 print message
  50             if self.usegrowl:
  51                 growlnotify = ['growlnotify', '-a', 'MacZephyr', '-n', 'zephyr', '-d', id, '-t', header]
  52                 g = subprocess.Popen(growlnotify, stdin=subprocess.PIPE)
  53                 g.stdin.write(message)
  54                 g.stdin.close()
  55             if self.usenotify:
  56                 if id in self.pings:
  57                     self.pings[id].close()
  58                 self.pings[id] = self.pynotify.Notification(header, message)
  59                 self.pings[id].show()
  60     def close(self):
  61         return
  62
  63 def main(argv):
  64     parser = optparse.OptionParser(usage = '%prog [-s "username@machine"] (--growl | --notify | --print)',
  65             description = __doc__.strip())
  66     parser.add_option('-s', '--ssh',
  67             type = 'string',
  68             default = None,
  69             dest = 'ssh',
  70             help = 'optional remote host to run tzc')
  71     parser.add_option('-g', '--growl',
  72             action = 'store_true',
  73             default = False,
  74             dest = 'growl',
  75             help = 'use growlnotify for output')
  76     parser.add_option('-n', '--notify',
  77             action = 'store_true',
  78             default = False,
  79             dest = 'notify',
  80             help = 'use notify-send for output')
  81     parser.add_option('-p', '--print',
  82             action = 'store_true',
  83             default = False,
  84             dest = 'useprint',
  85             help = 'use stdout for output')
  86     opts, args = parser.parse_args()
  87
  88     usegrowl = opts.growl
  89     usenotify = opts.notify
  90     useprint = opts.useprint
  91     if not usegrowl and not usenotify and not useprint:
  92         parser.print_help(sys.stderr)
  93         return 1
  94     ssh = opts.ssh
  95
  96     if ssh is None:
  97         retval = subprocess.call(['which', 'tzc'], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
  98         if retval:
  99             print 'tzc not in path.  Please add -s username@machine to specify remote host.'
 100             return 1
 101
 102     if ssh is not None:
 103         command = "ssh -K %s 'tzc -si'" % ssh
 104     else:
 105         command = "tzc -si"
 106     p = os.popen(command)
 107     r = sexpr.SExprReader(Notifier(usegrowl, usenotify, useprint))
 108
 109     flags = fcntl.fcntl(p, fcntl.F_GETFL)
 110     fcntl.fcntl(p, fcntl.F_SETFL, flags | os.O_NONBLOCK)
 111
 112     try:
 113         while 1:
 114             [i,o,e] = select.select([p], [], [], 5)
 115             if i: s = p.read(1024)
 116             else: s = ''
 117
 118             if s != '':
 119                 r.feed(s)
 120     except KeyboardInterrupt:
 121         pass
 122     return 0
 123
 124 if __name__ == "__main__":
 125     sys.exit(main(sys.argv))