Skip to content

Commit

Permalink
import 机制是怎么实现的?
Browse files Browse the repository at this point in the history
  • Loading branch information
satori1995 committed Jan 6, 2025
1 parent 8692fe6 commit 4b44a6c
Show file tree
Hide file tree
Showing 2 changed files with 331 additions and 0 deletions.
331 changes: 331 additions & 0 deletions src/84.import 机制是怎么实现的?.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,331 @@
## 楔子

通过上一篇的 import 黑盒探测,我们已经对 import 机制有了一个非常清晰的认识,Python 的 import 机制基本上可以切分为三个不同的功能。

- Python运行时的全局模块池的维护和搜索;
- 解析与搜索模块路径的树状结构;
- 对不同文件格式的模块执行动态加载机制;

尽管 import 的表现形式千变万化,但是都可以归结为:**import x.y.z** 的形式,当然 **import sys** 也可以看成是 x.y.z 的一种特殊形式。而诸如 from、as 与 import 的结合,实际上同样会进行 **import x.y.z** 的动作,只是最后在当前名字空间中引入的符号各有不同。

## \_\_import\_\_

然后导入模块,虚拟机会调用 \_\_import\_\_,那么我们就来看看这个函数长什么样子。

~~~C
// Python/bitinmodule.c
static PyObject *
builtin___import__(PyObject *self, PyObject *args, PyObject *kwds)
{
static char *kwlist[] = {"name", "globals", "locals", "fromlist",
"level", 0};
PyObject *name, *globals = NULL, *locals = NULL, *fromlist = NULL;
int level = 0; // 表示默认绝对导入
// 从 PyTupleObject 和 PyDictObject 中解析出需要的信息
if (!PyArg_ParseTupleAndKeywords(args, kwds, "U|OOOi:__import__",
kwlist, &name, &globals, &locals, &fromlist, &level))
return NULL;
// 导入模块
return PyImport_ImportModuleLevelObject(name, globals, locals,
fromlist, level);
}
~~~
里面有一个 PyArg_ParseTupleAndKeywords 函数,我们需要提一下,它在虚拟机中是一个被广泛使用的函数,原型如下:
~~~C
// Python/getargs.c
int PyArg_ParseTupleAndKeywords(PyObject *, PyObject *,
const char *, char **, ...);
~~~

这个函数的作用是参数解析,负责将 args 和 kwds 中所包含的所有对象(指针)按指定的 format 解析成各种目标对象,可以是 Python 的对象,例如 PyListObject、PyLongObject,也可以是 C 的原生对象。我们知道这个 builtin\_\_import\_\_ 里面的参数 args 指向一个 PyTupleObject ,包含了 \_\_import\_\_ 函数运行所需要的参数和信息,它是虚拟机在执行 IMPORT_NAME 指令的时候打包产生的。

然而在这里,虚拟机进行了一个逆动作,将打包后的这个 PyTupleObject 拆开,重新获得当初的参数。Python 在自身的实现中大量使用了这样的打包、拆包策略,使得可变数量的对象能够很容易地在函数之间传递。

而在完成了对参数的拆包动作之后,会进入 PyImport_ImportModuleLevelObject ,这个我们在 import_name 中已经看到了,当然它内部也是调用了 \_\_import\_\_

另外每个包和模块都有一个 \_\_name\_\_\_\_path\_\_ 属性。

~~~Python
import six
import numpy as np
import numpy.core

print(np.__name__, np.__path__)
"""
numpy ['C:\\python38\\lib\\site-packages\\numpy']
"""

print(np.core.__name__, np.core.__path__)
"""
numpy.core ['C:\\python38\\lib\\site-packages\\numpy\\core']
"""

print(six.__name__, six.__path__)
"""
six []
"""
~~~

\_\_name\_\_ 就是模块名或者包名,如果是包下面的包或者模块,那么就是**包名.包名**或者**包名.模块名**。至于 \_\_path\_\_ 则是导入包内模块时会搜索的目录列表,对于模块而言, \_\_path\_\_ 为空列表。

此外还有一个 \_\_file\_\_ 属性,对于模块而言就是其自身的完整路径;对于包而言则分两种情况,如果包内部存在 \_\_init\_\_.py 文件,那么得到的就是 \_\_init\_\_.py 文件的完整路径,没有则为 None。

下面来看一下不同的导入方式对应的字节码,然后在虚拟机的层面来理解这些导入方式。

## 单模块导入

以一个简单的模块导入为例:

~~~python
import sys
"""
0 LOAD_CONST 0 (0)
2 LOAD_CONST 1 (None)
4 IMPORT_NAME 0 (sys)
6 STORE_NAME 0 (sys)
8 LOAD_CONST 1 (None)
10 RETURN_VALUE
"""
~~~

这是一开始考察的例子,现在我们已经很清楚地了解了 IMPORT_NAME 的行为。在 IMPORT_NAME 指令的最后,虚拟机会将 PyModuleObject 对象(指针)压入运行时栈,随后会将 <font color="blue">"sys": <PyModuleObject \*></font> 存放到当前的 local 名字空间中。

## 级联导入

级联导入就是类似于 import a.b.c 这种导入方式。

~~~Python
import sklearn.linear_model.ridge
"""
0 LOAD_CONST 0 (0)
2 LOAD_CONST 1 (None)
4 IMPORT_NAME 0 (sklearn.linear_model.ridge)
6 STORE_NAME 1 (sklearn)
8 LOAD_CONST 1 (None)
10 RETURN_VALUE
"""
~~~

如果是级联导入,那么 IMPORT_NAME 的指令参数则是完整的路径信息,该指令的内部将解析这个路径,并为 sklearn、sklearn.linear_model、sklearn.linear_model.ridge 都创建一个 PyModuleObject 对象,这三者都存在于 sys.modules 里面。

但我们看到 STORE_NAME 是 sklearn,表示只有 sklearn 这个符号暴露在了当前模块的 local 空间里面。可为什么是 sklearn 呢?难道不应该是 sklearn.linear_model.ridge 吗?

其实经过我们之前的分析这一点已经不再是问题了,因为 <font color="blue">import sklearn.linear_model.ridge</font> 并不是说导入一个模块或包叫做 sklearn.linear_model.ridge,而是先导入 sklearn,然后把 linear_model 放在 sklearn 的属性字典里面,再把 ridge 放在 linear_model 的属性字典里面。

同理 sklearn.linear_model.ridge 代表的是先从 local 空间里面找到 sklearn,再从 sklearn 的属性字典中找到 linear_model,然后在 linear_model 的属性字典里面找到 ridge。因为 linear_model 和 ridge 已经在相应的属性字典里面,我们通过 sklearn 一级一级往下找是可以找到的,因此只需要将符号 skearn 暴露给 local 空间即可。

或者说暴露 sklearn.linear_model.ridge 本身就是不合理的,因为这表示导入一个名字就叫做 sklearn.linear_model.ridge 的模块或者包,但显然不存在。而即便我们创建了这样的一个模块或包,由于 Python 的语法解析规范依旧不会得到想要的结果。不然的话,假设 <font color="blue">import test_import.a</font>,那是导入名为 test_import.a 的模块或包呢?还是导入 test_import 下的 a 呢?

也正如我们之前分析的 test_import.a,我们在导入 test_import.a 的时候,会把 test_import 加载进来,然后把 a 加到 test_import 的属性字典里面,最后只需要把 test_import 返回即可。因为通过 test_import 可以找到 a,或者说 test_import.a 代表的含义就是从 test_import 的属性字典里面获取 a,所以 <font color="blue">import test_import.a</font> 必须要返回 test_import,而且只需返回 test_import。

至于 sys.modules 里面虽然存在字符串名为 "test_import.a" 的 key,但这是为了避免重复加载所采取的策略,它依旧表示从 test_import 的属性字典里面获取 a。

~~~Python
import pandas.core

print(pandas.DataFrame({"a": [1, 2, 3]}))
"""
a
0 1
1 2
2 3
"""
# 所以通过 pandas.DataFrame 是可以调用的
~~~

导入 pandas.core 会先导入 pandas,也就是执行 pandas 内部的 \_\_init\_\_ 文件。虽然 sys.modules 里面同时有 "pandas" 和 "pandas.core",但是暴露在 local 空间的只有 pandas,所以调用 pandas.DataFrame 是完全合理的。至于 pandas.core 显然它无法暴露,因为这不符合 Python 的变量命名规范,变量的名称里面不能出现小数点,它只是单纯地表示从 pandas 的属性字典中加载 core。

## from & import

基于 from & import 可以实现精确导入,看一下字节码。

~~~python
from sklearn.linear_model import ridge
"""
0 LOAD_CONST 0 (0)
2 LOAD_CONST 1 (('ridge',))
4 IMPORT_NAME 0 (sklearn.linear_model)
6 IMPORT_FROM 1 (ridge)
8 STORE_NAME 1 (ridge)
10 POP_TOP
12 LOAD_CONST 2 (None)
14 RETURN_VALUE
"""
~~~

注意此时的 2 LOAD_CONST 不再是 None 了,而是一个元组,虚拟机将 ridge 放到了当前模块的 local 空间中。并且 sklearn.linear_model 和 sklearn 都被导入了,存在 sys.modules 里面。

但是 sklearn 却并不在当前 local 空间中,尽管它被创建了,但是又被隐藏了。IMPORT_NAME 是 sklearn.linear_model,也表示导入 sklearn,然后把 sklearn 下面的 linear_model 加入到 sklearn 的属性字典里面。

而之所以 sklearn 没在 local 空间里面,可以这样理解。当只出现 import 的时候,那么我们必须从头开始一级一级向下调用,所以顶层的包必须加入到 local 空间里面。但这里通过 <font color="blue">from ... import ...</font> 把 ridge 导出了,此时 ridge 已经指向了 sklearn 下面的 linear_model 下面的 ridge,那么就不需要 sklearn 了,或者说 sklearn 就没必要暴露在 local 空间里面了,但它确实被导入进来了。

并且 sys.modules 里面也不存在 "ridge" 这个key,存在的是 "sklearn.linear_model.ridge",暴露给 local 空间的符号是 ridge。所以正如上面所说,不管什么导入,都可以归结为 <font color="blue">import x.y.z</font> 的形式,只是暴露出来的符号不同罢了。

## import & as

通过 import & as 可以在导入的时候给模块起一个别名。

~~~Python
import sklearn.linear_model.ridge as xxx
"""
0 LOAD_CONST 0 (0)
2 LOAD_CONST 1 (None)
4 IMPORT_NAME 0 (sklearn.linear_model.ridge)
6 IMPORT_FROM 1 (linear_model)
8 ROT_TWO
10 POP_TOP
12 IMPORT_FROM 2 (ridge)
14 STORE_NAME 3 (xxx)
16 POP_TOP
18 LOAD_CONST 1 (None)
20 RETURN_VALUE
"""
~~~

这个和上面的 from & import 类似,"sklearn"、"sklearn.linear_model"、"sklearn.linear_model.ridge" 都在 sys.modules 里面。但是我们加上了 as xxx,那么这个 xxx 就直接指向了 sklearn 下面的 linear_model 下面的 ridge,此时就不需要 sklearn 了。

因此只有 xxx 暴露在了当前模块的 local 空间里面,而 sklearn 虽然也被导入了,但它只在 sys.modules 里面,没有暴露给当前模块的 local 空间。

## from & import & as

通过 from & import & as 可以在精确导入的同时起别名。

~~~Python
from sklearn.linear_model import ridge as xxx
"""
0 LOAD_CONST 0 (0)
2 LOAD_CONST 1 (('ridge',))
4 IMPORT_NAME 0 (sklearn.linear_model)
6 IMPORT_FROM 1 (ridge)
8 STORE_NAME 2 (xxx)
10 POP_TOP
12 LOAD_CONST 2 (None)
14 RETURN_VALUE
"""
~~~

这个和之前的 from & import 一样,只是最后暴露给 local 空间的 ridge 变成了我们自己指定的 xxx。

## 与 module 对象有关的名字空间问题

同函数、类一样,每个 PyModuleObject 也有自己的名字空间。一个模块不能直接访问另一个模块的内容,尽管模块内部的作用域比较复杂,比如:遵循 LEGB 规则,但是模块与模块之间的划分则是很明显的。

~~~python
# test.py
name = "古明地觉"

def print_name():
return name

# main.py
from test1 import name, print_name
name = "古明地恋"
print(print_name()) # 古明地觉
~~~

执行 main.py 之后,发现打印的依旧是"古明地觉"。我们说 Python 是根据 LEGB 规则进行查找,而 print_name 函数里面没有 name,那么去外层找。可 main.py 里面的 name 是"古明地恋",但打印的依旧是 test.py 里面的 "古明地觉"。为什么?

还是那句话,模块与模块之间的作用域划分的非常明显,print_name 是 test.py 里面的函数,所以在返回 name 的时候,只会从 test.py 中搜索,无论如何都是不会跳过 test.py、跑到 main.py 里面的。

再来看个例子:

~~~Python
# test.py
name = "古明地觉"
nicknames = ["小五""少女觉"]

# main.py
import test
test.name = "❤古明地觉❤"
test.nicknames = ["觉大人"]

from test import name, nicknames
print(name) # ❤古明地觉❤
print(nicknames) # ['觉大人']
~~~

此时打印的结果变了,很简单,这里是直接把 test.py 里面的变量修改了。因为这种方式,相当于直接修改 test.py 的属性字典。那么后续再导入的时候,打印的就是修改之后的值。

~~~python
# test.py
name = "古明地觉"
nicknames = ["小五""少女觉"]

# main.py
from test import name, nicknames
name = "古明地恋"
nicknames.remove("小五")

from test import name, nicknames
print(name) # 古明地觉
print(nicknames) # ["少女觉"]
~~~

如果是 from test1 import name, nicknames,那么相当于在当前的 local 空间中新创建变量 name 和 nicknames,它们和 test.py 中的 name 和 nicknames 指向相同的对象。

而 name = "古明地觉" 相当于重新赋值了,所以不会影响 test.py 里的 name。而 nicknames.remove 则是在本地进行修改,所以会产生影响。

## 模块的 reload 问题

在上一篇文章中我们说到,如果模块在导入之后,源文件被修改了,那么可以通过 importlib.reload 更新模块。但 reload 的行为和我们想的不太一样,reload 会重新执行模块的代码,但是它会<font color="blue">重用同一个模块对象</font>,而不是创建新的。

~~~Python
# test.py
print("模块被加载")
x = 1
~~~

我们来导入 test.py 测试一下。

~~~Python
>>> import test
模块被加载
>>> id(test) # 记录模块对象的 id
140712834927872
>>> test.x
1
>>> test.y = 2 # 手动添加一个属性(但源文件里面是没有 y = 2 的)
>>>
>>> import importlib
>>> test = importlib.reload(test)
模块被加载
>>> id(test) # reload 后是同一个对象
140712834927872
>>> test.x # 模块中定义的变量被更新了
1
>>> test.y # 手动添加的属性仍然存在
2
~~~

所以 reload 不会创建新的模块对象,它会重新执行模块代码,更新模块的 \_\_dict\_\_,但不会清除模块对象中已存在的属性。如果我们真的想完全重新加载一个模块(包括删除旧属性),需要先将模块从 sys.modules 中删掉,然后重新 import。

~~~Python
>>> import sys
>>> del sys.modules['test']
>>> import test # 完全重新加载
模块被加载
>>> test.y # 之前手动添加的属性不存在了
AttributeError: module 'test' has no attribute 'y'
~~~

这种设计保持了模块级别的状态,可以避免破坏其它可能持有该模块引用的代码,但可能导致"幽灵属性"存在。所以实际开发中如果需要完全重新加载一个模块,最好使用删除 sys.modules 的方式,而不是依赖 reload。当然啦,重新加载模块这个需求本身就不常见,大家了解一下就好。

## 小结

以上就是模块(包)相关的内容,虽然一个项目可以有很多个文件,但是每个文件的执行原理是一致的。无论一个文件是作为模块被导入,还是直接作为启动文件被执行,虚拟机的执行流程都没有变化。

通过模块和包,我们便可以对项目进行功能上的划分,从而更好地组织项目。

-----

&nbsp;

**欢迎大家关注我的公众号:古明地觉的编程教室。**

![](./images/qrcode_for_gh.jpg)

**如果觉得文章对你有所帮助,也可以请作者吃个馒头,Thanks♪(・ω・)ノ。**

![](./images/supports.png)
Binary file modified src/images/275.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.

0 comments on commit 4b44a6c

Please sign in to comment.