Python: Check if Program is being closed

Question

all.

Is there a way, using Python, to check if the script that is currently running is requested to close? For example, If I press the X-Button (close program button) on the top-right to close it, or end the script in any other way, can the script do some code before it ends? Example:

# script goes here...

if Script_To_Be_Closed: # replace this with an actual line of code.
    do_stuff

Answer 1

There are multiple options you may use, like trapping keyboardinterrupts, but the simplest is atexit, which executes a function whenever a scripts is ended (except of a hard process kill indeed).

import atexit

def my_exit_function(some_argument):
    // Your exit code goes here
    print(some_argument)


if __name__ == '__main__':
    atexit.register(my_exit_function, 'some argument', )

    // Your script goes here