function before exit python or ctr+c

exit.py

#!/usr/bin/env python
# encoding: utf-8

import atexit
import signal
import time


def shutdown():
    """function when exit, exception
    """
    print "in shutdown"


def breakp(signal, frame):
    """function when ctrl-c
    """
    print "in break"
    exit(0)


def main():
    atexit.register(shutdown)
    signal.signal(signal.SIGINT, breakp)
    while True:
        print "do something"
        time.sleep(5)
        raise Exception("exception happened")


if __name__ == "__main__":
    main()

# 1,
# do something
# ^c (control+c)
# in break
# in shutdown

# 2,
# do something (5s later)
# Traceback ....
# Exception: ....
# in shutdown