testing atexit

gistfile1.txt

""" Test script to debug atexit SIGTERM handling """
import atexit
import os
import signal
import sys
import time

@atexit.register
def cleanup():
    print("atexit handler called")


def killme(pid):
    os.kill(pid, signal.SIGTERM)


if __name__ == "__main__":
    pid = os.getpid()
    if sys.argv[1] == 'wait':
        print(f"Sleeping for 60 seconds, go run kill {pid} from cli")
        time.sleep(60)
    if sys.argv[1] == 'kill':
        killme(pid)