蟒蛇，在装饰和关闭中混淆

Question

我有一些测试代码：

def num(num):
    def deco(func):
        def wrap(*args, **kwargs):
            inputed_num = num
            return func(*args, **kwargs)
        return wrap
    return deco


@num(5)
def test(a):
    return a + inputed_num

print test(1)

运行此代码时，我收到一个错误，显示未定义“inputed_num”

我的问题是：在wrap函数中，func是否可以得到“inputed_num”的闭包？

无论如何，如果没有，我应该如何实现我的目标：初始化一些值，并在主函数中直接使用这个值。

思考。

Answer 1

正如@Raymond所说的那样，定义函数后应用装饰器。这意味着，在编译函数体本身时，Pythn会看到inputed_num变量，当它不是本地定义的变量时，它会生成代码以尝试将其作为全局变量访问。

这意味着您可以在decorator中对其进行变通：decorator可以在函数globals（）空间中使用所需的名称设置一个全局变量，然后调用该函数。它应在单线程代码中可靠工作：

def num(num):
    def deco(func):
        def wrap(*args, **kwargs):
            glob = func.func_globals
            marker = object()
            original_value = glob.get("inputed_num", marker)
            glob["inputed_num"] = num
            result = func(*args, **kwargs)
            if original_value is marker:
                del glob["inputed_num"]
            else:
                glob["inputed_num"] = original_value
            return result
        return wrap
    return deco


@num(5)
def test(a):
    return a + inputed_num

和：

>>> print test(1)
6

Answer 2

我的问题是：在wrap函数中，func是否可以得到“inputed_num”的闭包？

抱歉，这不是装修工的工作方式。它们在函数最初定义后得到应用。到那时，已经太晚了。

当你写：

@num(5)
def test(a):
    return a + inputed_num

这相当于：

def test(a):
    return a + inputed_num

test = num(5)(test)       # note that num(5) is called after test() is defined.

为了实现你的目标，让inputed_num成为第一个要测试的论点。然后，让你的装修师通过这个论点：

def num(num):
    def deco(func):
        def wrap(*args, **kwargs):
            inputed_num = num
            return func(inputed_num, *args, **kwargs)  # this line changed
        return wrap
    return deco

@num(5)
def test(inputed_num, a):                              # this line changed
    return a + inputed_num

@num(6)
def test2(inputed_num, a):
    return a + inputed_num

print test(10)   # outputs 15
print test2(10)  # outputs 16

希望能为你扫清一切：-）

Answer 3

不，没有那样的结局。函数可以关闭周围词法上下文中的变量，而不是调用上下文中的变量。换句话说，如果您实际在另一个函数中写入一个函数，那么内部函数可以访问外部函数中的变量：

def f():
    g = 2
    def f2():
        print g
    f2()

但是函数永远不能访问调用它们的函数内部的变量。

一般来说，没有办法做你想做的事情，即在函数外部设置任意变量。最接近的是，您可以在装饰器中使用全局inputed_num将inputed_num分配为全局变量。然后test将访问全局值。

def num(num):
    def deco(func):
        def wrap(*args, **kwargs):
            global outsider
            outsider = num
            return func(*args, **kwargs)
        return wrap
    return deco
@num(5)
def test(a):
    print a+outsider

>>> test(2)
7

当然，变量设置是全局的，因此多个并发使用（例如递归）将不起作用。（只是为了好玩，你也可以在这里看到一个非常神秘的方法来做这件事，但它太疯狂了，在现实世界中没有用处。）