contextlib.ExitStack

import contextlib


def is_ok_job():
    try:
        print('do something')
        raise Exception('error')
        return True
    except Exception:
        return False


def cleanup():
    print('clean up')


def cleanup2():
    print('clean up2')


# try:
#     is_ok = is_ok_job()
#     print('more task')
# finally:
#     if not is_ok:
#         cleanup()


with contextlib.ExitStack() as stack:
    stack.callback(cleanup)
    stack.callback(cleanup2)

    @stack.callback
    def cleanup3():
        print('clean up3')


    is_ok = is_ok_job()
    print('more task')

    if is_ok:
        stack.pop_all()

実行結果：

do something
more task
clean up3
clean up2
clean up