Python中import机制实用分析

import

对于语句 import ,实际将module_name.py文件的代码作为一个对象加载到内存,放在当前文件上下文中。
这是,在当前上下文中,可以用点操作符调用module中的成员(import的目标对象也是成员)。比如:

// lib.py
a = 1

// main.py
import lib
print(lib.a)

需要注意的是,这个module文件必须在一些目录下(一层),才能够被这样import。
这些目录保存于数组sys.path
也就是说,解释器对于import ,会去sys.path包含的目录下,逐个去找名为module_name.py的文件。

一般情况下,sys.path 数组中首个路径是所执行文件所在路径,即入口文件中的os.path.dirname(__file__)。其余是python自身提供的一些路径,包括第三方模块所在路径。
所以,导入时,首个路径下的同名文件,可以覆盖第三方模块。

from

import

对于语句 from x import y,第一个参数x,是引入路径,可以有多层。最后一层可以是module_name
第二个参数y是引入对象,可以是module_name,也可以是item_name,即模块中的成员。

from x import y相对于import y,增加了从多级路径中导入,和只导入部分成员的能力。

比如:

// lib/lib.py
a = 1

// main.py
from lib imort lib
from lib.lib import a
print(lib.a)
print(a)

上面展示的引入路径,是绝对路径(相对于sys.path的)。path的首层,必须是sys.path目录的成员(文件或目录)。

还有一种相对引入路径,使用.开头,用于子目录中相互引用。比如:

// lib/lib1.py
from . import lib2
from .lib2 import a

// lib/lib2.py
a = 1

// main.py
from lib import lib1
from lib.lib1 import lib2, a

这里.就表示导入文件中的os.path.dirname(__file__)。其余规则同上。

修改sys.path

sys.path是可写的。这使得我们可以用它来做一些特别处理。

一个案例是:

python使用grpc开发时,需要将x.proto原型定义文件,编译为x_pb2.py和x_pb2_grpc.py文件。

使用命令行工具进行编译,编译结果会输出到一个指定的目录。如果这个目录和引入x_pb2.py和x_pb2_grpc.py的文件所在是同个目录,那就没有问题。

如果希望将之存放到一个子目录,就会产生异常了:

- main.py
- proto/
- - x_pb2.py
- - x_pb2_grpc.py

因为x_pb2_grpc.py会引入x_pb2.py,语句为import x_pb2 as x__pb2。而proto/不在sys.path中,这个引用是错误的。

因此,可以通过增加目录到sys.path来实现修正:

// main.py
import os
import sys

sys.path.append(os.path.join(os.path.dirname(__file__), 'proto'))

from proto import x_pb2_grpc
from proto import x_pb2

你可能感兴趣的:(Python中import机制实用分析)