A decorator to close all given arguments

close_args.py

def close_args(*to_close):
    def decorator(func: Callable):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            res = func(*args, **kwargs)
            arg_names = inspect.getfullargspec(func)[0]
            all_args = {**kwargs, **dict(zip(arg_names, args))}
            for arg in to_close:
                all_args[arg].close()

            return res

        return wrapper
    return decorator