Как вызвать функцию перед закрытием программы?

если требуется безусловно выполнить после завершения программы, обычном или при ошибке

def save_file():
    print('exit')

def main():
    print('работа программы')
    raise KeyboardInterrupt('выход')

if __name__ == '__main__':
    try:
        main()
    finally:
        save_file()

out:

работа программы
exit
KeyboardInterrupt: выход