Run atexit() when python process is killed

Question

I have a python process which runs in background, and I would like it to generate some output only when the script is terminated.

def handle_exit(): print('\nAll files saved in ' + directory) generate_output() atexit.register(handle_exit)

Calling raising a KeyboardInterupt exception and sys.exit() calls handle_exit() properly, but if I were to do kill {PID} from the terminal it terminates the script without calling handle_exit().

Is there a way to terminate the process that is running in the background, and still have it run handle_exit() before terminating?

Not possible with atexit alone. As the documentation states The functions registered via this module are not called when the program is killed by a signal not handled by Python, when a Python fatal internal error is detected, or when os._exit() is called. here — Alexander Ejbekov
– Alexander Ejbekov, Commented Nov 29, 2016 at 13:19

Gennady Kandaurov · Accepted Answer · 2016-11-29 13:19:12Z

37

Try signal.signal. It allows to catch any system signal:

import signal def handle_exit(): print('\nAll files saved in ' + directory) generate_output() atexit.register(handle_exit) signal.signal(signal.SIGTERM, handle_exit) signal.signal(signal.SIGINT, handle_exit)

Now you can kill {pid} and handle_exit will be executed.

answered Nov 29, 2016 at 13:19

Gennady Kandaurov

1,9741 gold badge15 silver badges21 bronze badges

Sign up to request clarification or add additional context in comments.

4 Comments

fish2000 Over a year ago

N.B., your handler function signatures should afford two positional arguments – a signal number and a stack frame, e.g. handle_exit(signum, frame) or handle_exit(*args) – as that’s how those functions invoked via signal.signal(…) will be called. If your functions take explicit arguments, you can set up values for atexit.register(…) to pass on its own invocation, e.g. atexit.register(handle_exit, None, None) or similar.

IlyaP Over a year ago

Doesn't work on Windows when closing via Task Manager or via Stop button in PyCharm

KamilCuk Over a year ago

This will execute handle_exit twice - once from handle_exit and then again from atexit.register.

Akay Over a year ago

it does not works on pycharm when STOP MAIN button is pressed, any solution for that please?

John Zhang · Accepted Answer · 2023-05-15 13:03:45Z

If you want to handle signals yourself, according to the Python documentation, the handler function needs two arguments.

If you handle the SIGINT signal, you must call sys.exit(0) within the handler function; otherwise, the program will not terminate upon pressing Ctrl + C.

As KamiCuk mentioned, if the function registered with atexit is the same as the signal handler function, the function will be called twice.

Therefore, I prefer using separate functions for the signal handler and for the atexit registration. Here is an example:

import atexit import signal import sys def on_exit(): print('real clean code here') def handle_exit(signum, frame): sys.exit(0) atexit.register(on_exit) signal.signal(signal.SIGTERM, handle_exit) signal.signal(signal.SIGINT, handle_exit)

Handling these signals correctly can be tricky, especially for cross-platform applications.

To address this, I have created a package called safe-exit.

The documentation can be found here: https://safe-exit.readthedocs.io/en/latest/

ChaimG · Accepted Answer · 2021-01-11 17:01:49Z

To enable signals when debugging PyCharm on Windows:

Within PyCharm hit Ctrl + Shift + A to bring up the "Find Actions..." menu
Search for "Registry" and hit enter
Find the key kill.windows.processes.softly and enable it (you can start typing "kill" and it will search for the key)
Restart PyCharm

does not works on pycharm when STOP MAIN button is pressed even after doing this procedure and restarting pycharm, am I missing something?.

ChaimG · Accepted Answer · 2021-01-11 17:46:25Z

To check your system and see which signal is being called:

import signal import time def handle_signal(sig_id, frame): sig = {x.value: x for x in signal.valid_signals()}.get(sig_id) print(f'{sig.name}, {sig_id=}, {frame=}') exit(-1) for sig in signal.valid_signals(): print(f'{sig.value}: signal.{sig.name},') signal.signal(sig, handle_signal) time.sleep(30)

Collectives™ on Stack Overflow

Run atexit() when python process is killed

4 Answers 4

4 Comments

Comments

1 Comment

Comments

Linked

Hot Network Questions

Collectives™ on Stack Overflow

4 Answers 4

4 Comments

Comments

1 Comment

Comments

Linked

Related