zcommit.py

   1 #!/usr/bin/python
   2
   3 import cherrypy
   4 from flup.server.fcgi import WSGIServer
   5 import logging
   6 import json
   7 import subprocess
   8 import sys
   9 import traceback
  10
  11 LOG_FILENAME = 'logs/zcommit.log'
  12
  13 # Set up a specific logger with our desired output level
  14 logger = logging.getLogger(__name__)
  15 logger.setLevel(logging.DEBUG)
  16
  17 # Add the log message handler to the logger
  18 handler = logging.FileHandler(LOG_FILENAME)
  19 logger.addHandler(handler)
  20
  21 def zephyr(klass, instance, zsig, msg):
  22     # TODO: spoof the sender
  23     logger.info("""About to send zephyr:
  24 class: %(klass)s
  25 instance: %(instance)s
  26 zsig: %(zsig)s
  27 msg: %(msg)s""" % {'klass' : klass, 'instance' : instance,
  28                  'zsig' : zsig, 'msg' : msg})
  29     cmd = ['zwrite', '-c', klass, '-i', instance,
  30            '-s', zsig, '-d', '-m', msg]
  31     subprocess.check_call(cmd)
  32
  33 class Application(object):
  34     @cherrypy.expose
  35     def index(self):
  36         logger.debug('Hello world app reached')
  37         return 'Hello world!'
  38
  39     class Github(object):
  40         @cherrypy.expose
  41         def default(self, *args, **query):
  42             try:
  43                 return self._default(*args, **query)
  44             except Exception, e:
  45                 logger.error('Caught exception %s:\n%s' % (e, traceback.format_exc()))
  46                 raise
  47
  48         def _default(self, *args, **query):
  49             logger.info('A %s request with args: %r and query: %r' %
  50                         (cherrypy.request.method, args, query))
  51             opts = {}
  52             if len(args) % 2:
  53                 raise cherrypy.HTTPError(400, 'Invalid submission URL')
  54             logger.debug('Passed validation')
  55             for i in xrange(0, len(args), 2):
  56                 opts[args[i]] = args[i + 1]
  57             logger.debug('Set opts')
  58             if 'class' not in opts:
  59                 raise cherrypy.HTTPError(400, 'Must specify a zephyr class name')
  60             logger.debug('Specified a class')
  61             if cherrypy.request.method == 'POST':
  62                 logger.debug('About to load data')
  63                 payload = json.loads(query['payload'])
  64                 logger.debug('Loaded payload data')
  65                 zsig = payload['ref']
  66                 if 'zsig' in opts:
  67                     zsig = '%s: %s' % (opts['zsig'], zsig)
  68                 logger.debug('Set zsig')
  69                 for c in reversed(payload['commits']):
  70                     inst = opts.get('instance', c['id'][:8])
  71                     info = {'name' : c['author']['name'],
  72                             'email' : c['author']['email'],
  73                             'message' : c['message'],
  74                             'timestamp' : c['timestamp'],
  75                             'added' :  '\n'.join(c['added'])}
  76                     msg = """%(name)s <%(email)s>
  77 %(message)s
  78 %(timestamp)s
  79 --
  80 %(added)s""" % info
  81                     zephyr(opts['class'], inst, zsig, msg)
  82                 msg = 'Thanks for posting!'
  83             else:
  84                 msg = ('If you had sent a POST request to this URL, would have sent'
  85                        ' a zepyhr to -c %s' % opts['class'])
  86             return msg
  87
  88     github = Github()
  89
  90 def main():
  91     app = cherrypy.tree.mount(Application(), '/zcommit')
  92     cherrypy.server.unsubscribe()
  93     cherrypy.engine.start()
  94     try:
  95         WSGIServer(app, environ={'SCRIPT_NAME' : '/zcommit'}).run()
  96     finally:
  97         cherrypy.engine.stop()
  98
  99 if __name__ == '__main__':
 100     sys.exit(main())