原创 古明地觉 2024-11-15 16:20 北京
前面我们介绍了函数的基本结构,它在底层由 PyFunctionObject 结构体表示,那么本篇文章来看看函数的创建过程。
正好明天周六日,可以慢慢看[doge]。
介绍函数结构时,我们看到内部有一个 func_code 字段,指向一个 PyCodeObject 对象,而函数就是根据 PyCodeObject 对象创建的。
因为一个 PyCodeObject 是对一段代码的静态表示,Python 编译器将源代码编译之后,针对里面的每一个代码块(code block)都会生成相应的 PyCodeObject 对象,该对象包含了这个代码块的一些静态信息,也就是可以从源代码中看到的信息。
比如某个函数对应的代码块里面有一个 a = 1 这样的表达式,那么符号 a 和整数 1、以及它们之间的联系就是静态信息。这些信息会被静态存储起来,符号 a 被存在符号表 co_varnames 中,整数 1 被存在常量池 co_consts 中。然后 a = 1 是一条赋值语句,因此会有两条指令 LOAD_CONST 和 STORE_FAST 存在字节码指令序列 co_code 中。
这些信息是在编译的时候就可以得到的,因此 PyCodeObject 对象是编译之后的结果。但 PyFunctionObject 对象是何时产生的呢?显然它是 Python 代码在运行时动态产生的,更准确的说,是在执行一个 def 语句的时候创建的。当虚拟机发现了 def 语句,那么就代表发现了新的 PyCodeObject 对象,因为它们是可以层层嵌套的。然后虚拟机会根据这个 PyCodeObject 对象创建对应的 PyFunctionObject 对象,并将变量名和 PyFunctionObject 对象(函数体)组成键值对放在当前的 local 空间中。
而在 PyFunctionObject 对象中,也需要拿到相关的静态信息,因此会有一个 func_code 字段指向 PyCodeObject。
除此之外,PyFunctionObject 对象还包含了一些函数在执行时所必需的动态信息,即上下文信息。比如 func_globals,就是函数在执行时关联的 global 名字空间,如果没有这个空间的话,函数就无法访问全局变量了。
由于 global 作用域中的符号和值必须在运行时才能确定,所以这部分必须在运行时动态创建,无法静态存储在 PyCodeObject 中。因此要基于 PyCodeObject 对象和 global 名字空间来创建 PyFunctionObject 对象,相当于一个封装。总之一切的目的,都是为了更好地执行字节码。
我们举个例子:
# 首先虚拟机从上到下执行字节码name = "古明地觉"age = 17# 啪,很快啊,出现了一个 defdef foo(): pass# 出现了 def,虚拟机就知道源代码进入了一个新的作用域了# 也就是遇到一个新的 PyCodeObject 对象了# 而通过 def 关键字知道这是一个函数,于是会进行封装# 将 PyCodeObject 封装成 PyFunctionObject,同时包含了全局名字空间# 所以当执行完 def 语句之后,一个函数就被创建了# 然后将变量名 foo 和函数体(PyFunctionObject)组成键值对存放在当前的 local 空间中# 当然对于模块而言,local 空间也是 global 空间print({k: v for k, v in locals().items() if k == "foo"})"""{'foo': <function foo at 0x102d5bf40>}"""# 函数内部也保存了 global 空间print(foo.__globals__ is globals() is locals())"""True"""print(foo.__globals__["foo"] is foo is locals()["foo"])"""True"""
调用的时候,会从 local 空间中取出符号 foo 对应的 PyFunctionObject 对象(函数对象)。然后根据函数对象创建栈帧对象,也就是为函数创建一个栈帧,随后将执行权交给新创建的栈帧,并在新创建的栈帧中执行字节码。
那么问题来了,函数(对象)是怎么创建的呢?或者说虚拟机是如何完成 PyCodeObject 对象到 PyFunctionObject 对象之间的转变呢?显然想了解这其中的奥秘,就必须从字节码入手。
import discode_string = """name = "satori"def foo(a, b): print(a, b)foo(1, 2)"""dis.dis(compile(code_string, "<func>", "exec"))
源代码很简单,定义一个变量 name 和一个函数 foo,然后调用函数。显然这里面会产生两个 PyCodeObject,我们来看一下。
0 RESUME 0 # name = "satori" 2 LOAD_CONST 0 ('satori') 4 STORE_NAME 0 (name) # 我们看到 PyCodeObject 也会作为常量被静态收集 # 这里是将常量池中索引为 1 的 PyCodeObject 压入运行时栈 6 LOAD_CONST 1 (<code object foo at 0x7f5d...>) # 从栈中弹出 PyCodeObject,然后构建函数对象 # 将函数对象(的指针)再压入运行时栈 8 MAKE_FUNCTION 0 # 从栈中弹出函数对象,并用符号 foo 绑定起来 # 到此函数就创建完毕了10 STORE_NAME 1 (foo) # 以下是 foo(1, 2) 对应的字节码12 PUSH_NULL # 加载全局变量 foo 并压栈14 LOAD_NAME 1 (foo) # 加载常量 1 和 2 并压栈16 LOAD_CONST 2 (1)18 LOAD_CONST 3 (2) # 从栈中弹出函数和参数,然后调用 # 将调用结果、即函数的返回值压入栈中20 CALL 2 # 从栈顶弹出返回值,因为我们没有使用变量保存,所以会直接丢弃 # 如果使用变量保存了,比如 res = foo(1, 2),那么这里的字节码就是 STORE_NAME28 POP_TOP30 RETURN_CONST 4 (None) # 以上是模块对应的字节码指令,下面是函数对应的字节码指令Disassembly of <code object foo at 0x7f5d..., file "<func>", line 3>: 0 RESUME 0 # 比较简单,就是 print(a, b) 对应的字节码 2 LOAD_GLOBAL 1 (NULL + print)12 LOAD_FAST 0 (a)14 LOAD_FAST 1 (b)16 CALL 224 POP_TOP26 RETURN_CONST 0 (None)
通过字节码我们看到,def 关键字实际上还是在定义变量,正所谓函数即变量,我们可以把函数当成普通的变量来处理。函数名就是变量名,它位于模块对应的 PyCodeObject 的符号表中。函数体就是变量指向的值,它是基于一个独立的 PyCodeObject 构建的。
至此,函数的结构就已经非常清晰了。
分析完结构之后,重点就要落在 MAKE_FUNCTION 指令上了,我们说当遇到 def 关键字的时候,就知道要创建函数了。在语法上这是函数的声明语句,但从虚拟机的角度来看,这其实是函数对象的创建语句。
所以函数是怎么创建的,就是执行 MAKE_FUNCTION 指令创建的,该指令执行完毕后,一个函数对象就被压入了运行时栈。等到 STORE_NAME 执行时,再将它从栈中弹出,然后和变量(函数名)绑定起来。
TARGET(MAKE_FUNCTION) { // 获取 PyCodeObject 对象 PyObject *codeobj = stack_pointer[-1]; // 编译时,解释器能够静态检测出函数有没有闭包变量、类型注解等属性,并体现在 oparg 中 // 构建函数时,通过 oparg 和一系列标志位做按位与,来判断函数是否包含指定属性 // 由于 oparg 是指令参数,所以这些属性是否存在、以及如何访问,在编译阶段就已经确定了 PyObject *closure = (oparg & 0x08) ? stack_pointer[...] : NULL; PyObject *annotations = (oparg & 0x04) ? stack_pointer[...] : NULL; PyObject *kwdefaults = (oparg & 0x02) ? stack_pointer[...] : NULL; PyObject *defaults = (oparg & 0x01) ? stack_pointer[...] : NULL; PyObject *func; #line 3267 "Python/bytecodes.c" // 基于 PyCodeObject 和全局名字空间,来构建 PyFunctionObject PyFunctionObject *func_obj = (PyFunctionObject *) PyFunction_New(codeobj, GLOBALS()); Py_DECREF(codeobj); if (func_obj == NULL) { goto error; } // 设置闭包变量、类型注解、默认值等属性 if (oparg & 0x08) { assert(PyTuple_CheckExact(closure)); func_obj->func_closure = closure; } if (oparg & 0x04) { assert(PyTuple_CheckExact(annotations)); func_obj->func_annotations = annotations; } if (oparg & 0x02) { assert(PyDict_CheckExact(kwdefaults)); func_obj->func_kwdefaults = kwdefaults; } if (oparg & 0x01) { assert(PyTuple_CheckExact(defaults)); func_obj->func_defaults = defaults; } func_obj->func_version = ((PyCodeObject *)codeobj)->co_version; // 函数创建之后,将栈里的元素弹出,然后将函数对象压入栈中 func = (PyObject *)func_obj; #line 4534 "Python/generated_cases.c.h" STACK_SHRINK(((oparg & 0x01) ? 1 : 0) + ((oparg & 0x02) ? 1 : 0) + ((oparg & 0x04) ? 1 : 0) + ((oparg & 0x08) ? 1 : 0)); stack_pointer[-1] = func; DISPATCH();}
整个步骤很好理解,然后创建函数体用的是 PyFunction_New,看一下它的逻辑。
// Objects/funcobject.cPyObject *PyFunction_New(PyObject *code, PyObject *globals){ return PyFunction_NewWithQualName(code, globals, NULL);}PyObject *PyFunction_NewWithQualName(PyObject *code, PyObject *globals, PyObject *qualname){ assert(globals != NULL); assert(PyDict_Check(globals)); // 给全局名字空间增加引用计数 Py_INCREF(globals); // 获取线程状态对象 PyThreadState *tstate = _PyThreadState_GET(); // 给 PyCodeObject 对象增加引用计数 PyCodeObject *code_obj = (PyCodeObject *)Py_NewRef(code); assert(code_obj->co_name != NULL); // 获取 co_name 并增加引用计数 PyObject *name = Py_NewRef(code_obj->co_name); // 获取 co_qualname,如果存在,增加引用计数 if (!qualname) { qualname = code_obj->co_qualname; } assert(qualname != NULL); Py_INCREF(qualname); // 获取常量池 PyObject *consts = code_obj->co_consts; assert(PyTuple_Check(consts)); // 函数的 docstring 也会被收集到常量池中,并且是常量池的第一个元素 // 如果函数没有 docstring,那么常量池的第一个元素会是 None PyObject *doc; if (PyTuple_Size(consts) >= 1) { doc = PyTuple_GetItem(consts, 0); // 如果第一个元素不是字符串,则说明函数没有 docstring if (!PyUnicode_Check(doc)) { doc = Py_None; } } else { doc = Py_None; } Py_INCREF(doc); // 获取 __module__,并增加引用计数 PyObject *module = PyDict_GetItemWithError( globals, &_Py_ID(__name__)); PyObject *builtins = NULL; if (module == NULL && _PyErr_Occurred(tstate)) { goto error; } Py_XINCREF(module); // 获取 __builtins__,并增加引用计数 builtins = _PyEval_BuiltinsFromGlobals(tstate, globals); if (builtins == NULL) { goto error; } Py_INCREF(builtins); // 为函数对象申请内存空间 PyFunctionObject *op = PyObject_GC_New( PyFunctionObject, &PyFunction_Type); if (op == NULL) { goto error; } // 初始化函数对象的内部属性 op->func_globals = globals; op->func_builtins = builtins; op->func_name = name; op->func_qualname = qualname; op->func_code = (PyObject*)code_obj; op->func_defaults = NULL; op->func_kwdefaults = NULL; op->func_closure = NULL; op->func_doc = doc; op->func_dict = NULL; op->func_weakreflist = NULL; op->func_module = module; op->func_annotations = NULL; op->func_typeparams = NULL; op->vectorcall = _PyFunction_Vectorcall; op->func_version = 0; _PyObject_GC_TRACK(op); handle_func_event(PyFunction_EVENT_CREATE, op, NULL); return (PyObject *)op;error: Py_DECREF(globals); Py_DECREF(code_obj); Py_DECREF(name); Py_DECREF(qualname); Py_DECREF(doc); Py_XDECREF(module); Py_XDECREF(builtins); return NULL;}
以上就是函数对象的创建过程,说白了就是对 PyCodeObject 进行了一个封装。等函数对象创建完毕后会回到 MAKE_FUNCTION,然后设置闭包、注解等属性,并将函数对象压入栈中。接着执行 STORE_NAME 从符号表中加载符号(函数名),并从栈顶弹出函数对象,然后将两者组成键值对存储在当前栈帧的 local 名字空间中,整体还是比较简单的。
但如果再加上类型注解、以及默认值,会有什么效果呢?
import discode_string = """name = "satori"def foo(a: int = 1, b: int = 2): print(a, b)"""dis.dis(compile(code_string, "<func>", "exec"))
我们看看加上了类型注解和默认值之后,它的字节码指令会有什么变化?
0 RESUME 0 2 LOAD_CONST 0 ('satori') 4 STORE_NAME 0 (name) 6 LOAD_CONST 5 ((1, 2)) 8 LOAD_CONST 1 ('a')10 LOAD_NAME 1 (int)12 LOAD_CONST 2 ('b')14 LOAD_NAME 1 (int)16 BUILD_TUPLE 418 LOAD_CONST 3 (<code object foo at 0x7f...>)20 MAKE_FUNCTION 5 (defaults, annotations)22 STORE_NAME 2 (foo)......
不难发现,在构建函数时会先将默认值以元组的形式压入运行时栈;然后再将使用了类型注解的参数和类型也构建一个元组,并压入运行时栈。
后续创建函数的时候,会将默认值保存在 func_defaults 字段中,类型注解对应的字典会保存在 func_annotations 字段中。
def foo(a: int = 1, b: int = 2): print(a, b)print(foo.__defaults__)"""(1, 2)"""print(foo.__annotations__)"""{'a': <class 'int'>, 'b': <class 'int'>}"""
基于类型注解,我们便可以额外施加一些手段,让 Python 像静态语言一样,实现函数参数的类型约束。
这里再说一下函数名,举个例子。
def foo(): passprint(foo.__name__) # foobar = fooprint(bar.__name__) # foo
我们定义了一个函数 foo,那么函数名就是 foo,这是没问题的,但怎么理解 bar 呢?
所以严格意义上讲,代码中的 foo 应该是一个变量。之前说过,定义函数、类、导入模块,其实都是创建了一个变量。所以代码中的 foo 也是一个变量,它指向了函数对象,而函数的名字是保存在函数对象里面的。
code_string = """def foo(): pass"""code_obj = compile(code_string, "<func>", "exec")# 我们是以模块的形式编译的,它里面只有一个变量 foo# 所以符号表就是 ('foo',)print(code_obj.co_names) # ('foo',)# 然后常量池里面存在一个 PyCodeObject# 这个 PyCodeObject 便是函数对应的 PyCodeObjectprint(code_obj.co_consts[0]) # <code object foo ...>print(code_obj.co_consts[0].co_name) # foo# 构建函数时,PyCodeObject 的 co_name 会被赋值给函数的 func_name# 所以严格意义上讲,def foo() 中的 foo 只能算做是变量名# 而真正的函数名是函数对象的 func_name,它来自于 co_name# 只不过在编译成 PyCodeObject 对象时,会进行词法分析# 因为 def 后面是 foo,所以编译之后的 PyCodeObject 的 co_name 也是 foo# 当然其它对象也是如此class A: pass# 这里的 A 指向了类型对象,但类型对象的名称是保存在类型对象里面的print(A.__name__) # A# A.__name__ 才是类名,class 后面的 A 只是一个变量名# 这里同样创建了一个类B = type("B1", (object,), {})print(B.__name__) # B1# 但是我们看到类名不是 B,而是 B1# 所以我们需要明白,不管是变量赋值、还是定义函数、类、方法,导入模块# 我们得到的只是一个变量,这个变量指向了具体的对象(它们是字典中的一个键值对)# 而对象的名称、类型等信息,都保存在对象里面,和变量无关# 因为变量只是一个符号,或者理解为代号,每个对象都可以有不同的代号def foo(): pass# 名称也可以自由更改foo.__name__ = "foo1"# 在更改过后,函数的名字就变成了 foo1print(foo.__name__) # foo1# bar = foo 之后,这个函数对象就有了两个代号,你通过 foo 和 bar 都可以找到它# 但函数对象的名字是不变的,还是 foo1,因为它的 __name__ 属性的值是 foo1bar = fooprint(bar.__name__) # foo1
我们之前说变量只是一个和对象绑定的符号,或者说代号,运行时会和某个对象(的地址)组成键值对保存在字典中。虚拟机通过变量可以找到它代表的对象,本质上就是将变量名作为 key,去字典中检索 value。至于获取到的对象叫什么名字,是保存在对象里面的。
如果变量指向的是整数、字符串等,那么该对象就没有名字。如果指向的是函数、类、模块,那么对象的 __name__ 就是对象的名字。只不过在默认情况下,定义函数(以及类)时,变量名默认和函数名是一样的,所以我们会把指向函数对象的变量的名称也叫做函数名。
关于这一点,大家一定要清晰。
name = "古明地觉"def foo(): passclass A: passimport osprint("name" in locals()) # Trueprint("foo" in locals()) # Trueprint("A" in locals()) # Trueprint("os" in locals()) # True
这里的 name、foo、A、os 都是变量,站在虚拟机的角度,它们没有任何的不同,只不过指向的对象不同罢了。而站在 Python 的角度,它们也是一样的,其名称都是字典里的一个 key,只不过关联的 value 不同罢了。
比如 name 指向的是字符串对象,foo 指向的是函数对象,A 指向的是类对象,os 指向的是模块对象。但我们也可以改变指向,比如让 foo 指向类对象,A 指向字符串对像等等,都是可以的。
总结:变量只是一个指针,可以保存任意对象的地址,也就是可以指向任意的对象。而对象的名字、类型等一切信息,都保存在对象中,和变量无关。
当然这些都是之前说过的内容,再来回顾一下,总之一定要了解 Python 变量的本质。
之前说 <class 'function'> 是函数的类型对象,而这个类底层没有暴露给我们,但我们依旧可以通过曲线救国的方式进行获取。
def foo(): passprint(type(foo)) # <class 'function'># lambda 匿名函数的类型也是 functionprint(type(lambda: None)) # <class 'function'>
那么下面就来创建函数:
gender = "female"def foo(name, age): return f"name: {name}, age: {age}, gender: {gender}"# 得到 PyCodeObject 对象code = foo.__code__# 根据 class function 创建函数对象# 接收三个参数: PyCodeObject 对象、名字空间、函数名new_foo = type(foo)(code, globals(), "根据 foo 创建的 new_foo")# 打印函数名print(new_foo.__name__)"""根据 foo 创建的 new_foo"""# 调用函数print(new_foo("古明地觉", 17))"""name: 古明地觉, age: 17, gender: female"""
是不是很神奇呢?另外函数之所以能访问全局变量,是因为在创建函数的时候将 global 名字空间传进去了,如果我们不传递呢?
gender = "female"def foo(name, age): return f"name: {name}, age: {age}, gender: {gender}"code = foo.__code__# 第二个参数必须是一个字典,不能传 Nonenew_foo = type(foo)(code, {}, "根据 foo 创建的 new_foo")try: print(new_foo("古明地觉", 17))except NameError as e: print(e) # name 'gender' is not defined
因此现在我们又从 Python 的角度理解了一遍,为什么在函数内部能够访问全局变量。原因就在于构建函数的时候,将 global 名字空间交给了函数,使得函数可以在 global 空间中进行变量查找,所以它才能够找到全局变量。而我们这里给了一个空字典,那么显然就找不到 gender 这个变量了。
gender = "female"def foo(name, age): return f"name: {name}, age: {age}, gender: {gender}"code = foo.__code__new_foo = type(foo)(code, {"gender": "萌妹子"}, "根据 foo 创建的 new_foo")# 我们可以手动传递一个字典进去# 此时传递的字典对于函数来说就是 global 名字空间print(new_foo("古明地觉", 17))"""name: 古明地觉, age: 17, gender: 萌妹子"""# 所以此时的 gender 不再是外部的 "female", 而是我们指定的 "萌妹子"
此外也可以为函数指定默认值:
def foo(name, age, gender): return f"name: {name}, age: {age}, gender: {gender}"# 必须接收一个 PyTupleObject 对象foo.__defaults__ = ("古明地觉", 17, "female")print(foo())"""name: 古明地觉, age: 17, gender: female"""
我们看到函数 foo 明明接收三个参数,但是调用时不传递居然也不会报错,原因就在于我们指定了默认值。而默认值可以在定义函数的时候指定,也可以通过 __defaults__ 指定,但很明显我们应该通过前者来指定。
如果你使用的是 PyCharm,那么会在 foo() 这个位置给你加波浪线,提示你参数没有传递。但我们知道,由于通过 __defaults__ 设置了默认值,所以这里是不会报错的。只不过 PyCharm 没有检测到,当然基本上所有的 IDE 都无法做到这一点,毕竟动态语言。
另外如果 __defaults__ 接收的元组里面的元素个数和参数个数不匹配怎么办?def foo(name, age, gender): return f"name: {name}, age: {age}, gender: {gender}"foo.__defaults__ = (15, "female")print(foo("古明地恋"))"""name: 古明地恋, age: 15, gender: female"""
由于元组里面只有两个元素,意味着我们在调用时需要至少传递一个参数,而这个参数会赋值给 name。原因就是在设置默认值的时候是从后往前设置的,也就是 "female" 会赋值给 gender,15 会赋值给 age。而 name 没有得到默认值,那么它就需要调用者显式传递了。
如果返回值从前往后设置的话,会出现什么后果?显然 15 会赋值给 name,"female" 会赋值给 age,此时函数就等价于如下:
def foo(name=15, age="female", gender): return f"name: {name}, age: {age}, gender: {gender}"
这样的函数显然无法通过编译,因为默认参数必须在非默认参数的后面。所以 Python 的这个做法是完全正确的,必须要从后往前进行设置。另外我们知道默认值的个数是小于等于参数个数的,如果大于会怎么样呢?
def foo(name, age, gender): return f"name: {name}, age: {age}, gender: {gender}"foo.__defaults__ = ("古明地觉", "古明地恋", 15, "female")print(foo())"""name: 古明地恋, age: 15, gender: female"""
依旧是从后往前进行设置,当所有参数都有默认值时,就结束了,多余的默认值会丢弃。当然,如果不使用 __defaults__,是不可能出现默认值个数大于参数个数的。可要是 __defaults__ 指向的元组先结束,那么没有得到默认值的参数就必须由调用者显式传递了。
最后,再来说一下如何深拷贝一个函数。首先如果是你的话,你会怎么拷贝一个函数呢?不出意外的话,你应该会使用 copy 模块。
import copydef foo(a, b): return [a, b]# 但是问题来了,这样能否实现深度拷贝呢?new_foo = copy.deepcopy(foo)# 修改 foo 的默认值foo.__defaults__ = (2, 3)# 但是 new_foo 也会受到影响print(new_foo()) # [2, 3]
打印结果提示我们并没有实现函数的深度拷贝,事实上 copy 模块无法对函数、方法、回溯栈、栈帧、模块、文件、套接字等类型实现深度拷贝。那我们应该怎么做呢?
from types import FunctionTypedef foo(a, b): return "result"# FunctionType 就是函数的类型对象# 它也是通过 type 得到的new_foo = FunctionType(foo.__code__, foo.__globals__, foo.__name__, foo.__defaults__, foo.__closure__)# 显然 function 还可以接收第四个参数和第五个参数# 分别是函数的默认值和闭包# 然后别忘记将属性字典也拷贝一份# 由于函数的属性字典几乎用不上,这里就浅拷贝了new_foo.__dict__.update(foo.__dict__)foo.__defaults__ = (2, 3)print(foo.__defaults__) # (2, 3)print(new_foo.__defaults__) # None
此时修改 foo 不会影响 new_foo,当然在拷贝的时候也可以自定义属性。其实上面实现的深拷贝,本质上就是定义了一个新的函数。由于是两个不同的函数,那么自然就没有联系了。
最后再来看看如何检测一个函数有哪些参数,首先函数的局部变量(包括参数)在编译时就已经确定,会存在符号表 co_varnames 中。
def foo(a, b, /, c, d, *args, e, f, **kwargs): g = 1 h = 2print(foo.__code__.co_varnames)"""('a', 'b', 'c', 'd', 'e', 'f', 'args', 'kwargs', 'g', 'h')"""
在定义函数的时候,* 和 ** 最多只能出现一次。然后这里的 a 和 b 必须通过位置参数传递,c 和 d 可以通过位置参数或者关键字参数传递,e 和 f 必须通过关键字参数传递。而从打印的符号表来看,里面的符号是有顺序的。参数永远在函数内部定义的局部变量的前面,比如 g 和 h 就是函数内部定义的局部变量,所以它在所有参数的后面。而对于参数,* 和 ** 会位于最后面,其它参数位置不变。所以除了 g 和 h,最后面的就是 args 和 kwargs。
有了这些信息,我们就可以进行检测了。
def foo(a, b, /, c, d, *args, e, f, **kwargs): g = 1 h = 2varnames = foo.__code__.co_varnames# 1. 寻找必须通过位置参数传递的参数posonlyargcount = foo.__code__.co_posonlyargcountprint(posonlyargcount) # 2print(varnames[: posonlyargcount]) # ('a', 'b')# 2. 寻找可以通过位置参数或者关键字参数传递的参数argcount = foo.__code__.co_argcountprint(argcount) # 4print(varnames[: argcount]) # ('a', 'b', 'c', 'd')print(varnames[posonlyargcount: argcount]) # ('c', 'd')# 3. 寻找必须通过关键字参数传递的参数kwonlyargcount = foo.__code__.co_kwonlyargcountprint(kwonlyargcount) # 2print(varnames[argcount: argcount + kwonlyargcount]) # ('e', 'f')# 4. 寻找 *args 和 **kwargsflags = foo.__code__.co_flags# 在介绍 PyCodeObject 的时候,我们说里面有一个 co_flags 成员# 它是函数的标识,可以对函数类型和参数进行检测# 如果 co_flags 和 4 按位与之后为真,那么就代表有 *args,否则没有# 如果 co_flags 和 8 按位与之后为真,那么就代表有 **kwargs,否则没有step = argcount + kwonlyargcountif flags & 0x04: print(varnames[step]) # args step += 1if flags & 0x08: print(varnames[step]) # kwargs
以上我们就检测出了函数都有哪些参数,你也可以将其封装成一个函数,实现代码的复用。然后还要注意一点,如果我们定义的时候不是 *args,而只是一个 *,那么它就不是参数了。def f(a, b, *, c): pass# 符号表里面只有 a、b、cprint(f.__code__.co_varnames) # ('a', 'b', 'c')# 显然此时也都为假print(f.__code__.co_flags & 0x04) # 0print(f.__code__.co_flags & 0x08) # 0
单独的一个 * 只是为了强制要求后面的参数必须通过关键字参数的方式传递。现在你是不是对函数有了一个更深刻的认识了呢?当然目前介绍的只是函数的一部分内容,还有更多内容等待我们挖掘,比如:
- 函数如何调用。位置参数和关键字参数如何解析。对于有默认值的参数,如何在不传参的时候使用默认值、在传参的时候使用我们传递的值。*args 和 **kwargs 如何解析。闭包怎么实现。装饰器怎么实现......
这些内容我们接下来慢慢说。