server.py

#-*- coding:utf-8 -*-
import os, random, time, atexit
from multiprocessing import Process, Queue
from github_event_handlers import get_handler
from web import get_web_app, set_data_queue
from signal import signal, SIGINT, SIG_IGN, siginterrupt

handler = get_handler()
data_queue = Queue()

set_data_queue(data_queue)

def run_web():
    get_web_app().run(host='0.0.0.0', port=1088, threaded=True)

web_process = Process(target=run_web)
web_process.start()

proc_pool = {}

def clean_up():
    print "stopping web process"
    web_process.terminate()
    web_process.join(5)
    print "stopping work process"
    for (pid, p) in proc_pool:
        if (p.is_alive()):
            print "stopping process " + str(pid)
            p.terminate()
            p.join()
        else:
            p.join()

atexit.register(clean_up)

def main():
    while True:
        item = data_queue.get()
        print "data get " + item['type']
        if (item['type'] == 'webhook' and handler.can_handle(item['data'])):
            proc = Process(target=handler.handle, args=(item['data'],data_queue))
            proc.start()
            proc_pool[proc.pid] = proc
            print 'process {} started'.format(proc.pid)
        elif (item['type'] == 'exit'):
            end_proc = proc_pool.pop(item['pid'])
            if (end_proc):
                print 'process {} ended'.format(end_proc.pid)
                end_proc.join()
        time.sleep(0.3)
        if (not web_process.is_alive()):
            print "webserver stop, restarting"
            web_process.terminate()
            web_process.join()
            web_process.start()

if __name__ == "__main__":
    main()