terminate a program gracefully from a thread

[Jabba Laci]

Hi,

I have a script (see below) that I want to terminate after X seconds.
The main loop of the program is waiting for user input.
The program enters the main loop and I try to shut down the program
after X seconds from a thread but I can't figure out how to do it. The
program should also do some cleanup before termination, so the shut
down must be graceful.

The code below is a simplified version. The whole idea is the
following: I have a script that has grown quite big over time. It
needs to read several data files, so when I start it for the first
time, it takes about 3-4 seconds to launch. The next start is much
faster since, I guess, the OS has done some caching. I use this script
a lot and the first slow launch bothers me. My idea: after booting, I
want to start the script in the background in suicide mode. OS does
the caching, so when I need it, it starts quickly.

See the code below with comments.

Thanks,

Laszlo

===============

import atexit
import sys
import time
from threading import Thread
import os

def suicide(wait):
    time.sleep(wait)
    print("should exit now...")
    sys.exit()      # exits this thread but not the main thread
#    os._exit(0)    # no cleanup with this :(

def cleanup():
    # I want it to run before termination.
    print("cleaning up...")

def main():
    Thread(target=suicide, kwargs={'wait': 3}).start()
    #
    while True:
        try:
            inp = raw_input("in> ")
            print(inp)
        except (KeyboardInterrupt, EOFError):
            print()
            sys.exit()

#####

if __name__ == "__main__":
    atexit.register(cleanup)
    main()