可选参数如何成为必需的？

Question

基于Disable global variable lookup in Python（以及我自己的答案），我在使用带有可选参数的函数时遇到了问题，例如在这个最小示例中：

import types

def noglobal(f):
    return types.FunctionType(f.__code__, {})

@noglobal
def myFunction(x=0):
    pass

myFunction()

基本上，它会像这样失败：

Traceback (most recent call last):
  File "SetTagValue.py", line 10, in <module>
    myFunction()
TypeError: myFunction() missing 1 required positional argument: 'x'

为什么x突然被当作必填参数了？

Answer 1

这是因为你没有正确复制函数。如果您查看 types.FunctionType 的签名，您会发现它接受 5 个参数：

class function(object)
 |  function(code, globals, name=None, argdefs=None, closure=None)
 |
 |  Create a function object.
 |
 |  code
 |    a code object
 |  globals
 |    the globals dictionary
 |  name
 |    a string that overrides the name from the code object
 |  argdefs
 |    a tuple that specifies the default argument values
 |  closure
 |    a tuple that supplies the bindings for free variables

您没有传递任何argdefs，因此该函数不再有可选参数。复制函数的正确方法是

types.FunctionType(f.__code__,
                   {},
                   f.__name__,
                   f.__defaults__,
                   f.__closure__
                   )

---

然而，这会导致另一个问题：切断对全局变量的访问也会切断对内置函数的访问。如果您尝试使用print 或open 或dict 或myFunction 中的类似内容，您将得到NameError。所以真正正确的方式来写你的装饰器是这样的：

import builtins
import types

def noglobal(f):
    return types.FunctionType(f.__code__,
                              {'__builtins__': builtins},
                              f.__name__,
                              f.__defaults__,
                              f.__closure__
                              )

Answer 2

如果您想保留默认参数值，您还需要传递它们：

import types

def noglobal(f):
    return types.FunctionType(f.__code__, {}, f.__name__, f.__defaults__)

@noglobal
def myFunction(x=0):
    pass

myFunction()

您可以将最后一个closure 参数传递给types.FunctionType，如果您想保持带闭包的函数正常工作，您可能还想从f.__closure__ 继承该参数。