supervisor-pagerduty.py

import urllib2
import json
from supervisor import childutils
import sys
import socket

 
class PagerDutyNotifier(object):
    def __init__(self, pd_service_key):
        self.pd_service_key = pd_service_key
        def run(self):
            while True:
                headers, payload = childutils.listener.wait()
                sys.stderr.write(str(headers) + '\n')
                payload = dict(v.split(':') for v in payload.split(' '))
                sys.stderr.write(str(payload) + '\n')
                if headers['eventname'] == 'PROCESS_STATE_EXITED' and not int(payload['expected']):
                    data = {'service_key': self.pd_service_key,
                            'event_type': 'trigger',
                            'description': '{} service has crashed unexpectedly on {}'.format(payload['processname'], socket.gethostname())
                    }
                    try:
                        res = urllib2.urlopen('https://events.pagerduty.com/generic/2010-04-15/create_event.json', json.dumps(data))
                    except urllib2.HTTPError, ex:
                        sys.stderr.write('{} - {}\n{}'.format(ex.code, ex.reason, ex.read()))
                    else:
                        sys.stderr.write('{}, {}\n'.format(res.code, res.msg))
                    childutils.listener.ok()
                    sys.stderr.flush()
    
if __name__ == '__main__':
    pager_duty_service_key = sys.argv[1]
    pager_duty_notifer = PagerDutyNotifier(pager_duty_service_key)
    pager_duty_notifer.run()