Cleaning up when killed (Python recipe)

cleanup.py

#!/usr/bin/env python

# From http://code.activestate.com/recipes/533117-cleaning-up-when-killed/

from signal import signal, SIGTERM
from sys import exit
import atexit

def cleanup():
    print "Cleanup"

if __name__ == "__main__":
    from time import sleep
    atexit.register(cleanup)

    # Normal exit when killed
    signal(SIGTERM, lambda signum, stack_frame: exit(1))

    sleep(10)