Exit hanlder for function

at_function_exit.py

# SPDX-FileCopyrightText: 2025 geisserml <[email protected]>
# SPDX-License-Identifier: Apache-2.0 OR BSD-3-Clause

import functools

class finally_decorator:
    
    def __init__(self, callback):
        self._callback = callback
    
    def __call__(self, func):
        @functools.wraps(func)
        def wrapped(*args, **kwargs):
            try:
                return func(*args, **kwargs)
            finally:
                self._callback()
        return wrapped

@finally_decorator(lambda: print("Exit test"))
def test(arg, kwarg=1):
    print(f"Enter test({arg}, {kwarg})")

test(0)

My initial motivation for this was to clean up global state, but refactoring to avoid globals in the first place turned out to be the better solution in that case.