Python带可选参数装饰器多次调用后nonlocal变量未重置问题

为什么s不会自动重置？ #

这是闭包的作用域特性导致的：

• 每次调用deco()装饰函数时，都会生成一个独立的deco执行上下文，参数s是这个上下文里的局部变量。
• 被装饰后的test1和test2各自绑定了一个wrapper函数，而wrapper通过nonlocal引用了对应deco上下文里的s。
• 第一次调用test1(1)时，s被改成了"some_value_depending_on_other_args_i_removed"，这个修改会直接存在于deco的上下文里，下次再调用test1时，s已经是修改后的值，不会自己变回None。

两种解决思路 #

思路1：每次调用都重新初始化s #

把s的初始化放到wrapper内部，不依赖闭包变量，这样每次调用都会从deco传入的默认值开始：

from functools import wraps

def deco(s=None):
    def _decorate(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            print(locals())
            
            # 每次调用都用deco的默认值重新初始化
            current_s = s
            if not current_s:
                current_s = "some_value_depending_on_other_args_i_removed"
            
            # do_smth_smart(current_s)           
            return None

        return wrapper
    return _decorate

思路2：调用后手动重置s #

就是你代码里注释的s=None，适合需要临时保存状态但每次调用后必须重置的场景：

from functools import wraps

def deco(s=None):
    def _decorate(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            print(locals())
            
            nonlocal s
            if not s:
                s = "some_value_depending_on_other_args_i_removed"
            
            # do_smth_smart(s)           
            
            # 调用完成后把s改回None，确保下次调用从头开始
            s = None
            return None

        return wrapper
    return _decorate

测试效果 #

用上面任意一种方式修改后，再运行你的测试代码，每次调用输出里的s都会是None，符合你的预期。

内容的提问来源于stack exchange，提问作者SOKS