4.4.4 模块的有条件执行

优质

小牛编辑

138浏览

2023-12-01

4.4.4 模块的有条件执行

有些 Python 模块是可以直接执行的，一般称为程序或脚本；而另一些 Python 模块中只包含一些函数定义，本身并没有主程序入口，因而不能执行。标准库就属于后一种模块。有时我们希望创建一种混合式的模块——既可以作为独立执行的程序，又可以作为被其他程序

导入的库。在 Python 中，混合式模块可以通过在程序入口前加上特定条件而实现。如所熟知，我们一般都在程序文件的最后加上启动程序的一行语句：

main()

这是对程序入口（主函数 main）的调用，没有这一行，程序文件就不是可执行的文件。这就是直接执行的模块文件，在窗口系统中用鼠标双击即可启动程序。

Python 在导入一个模块的时候会执行模块中的每一行语句，执行函数定义语句 def 时就创建相应的函数但并不执行，而最后遇到启动程序的 main 时就启动了整个程序。有时我们希望导入模块时不要执行整个程序，例如交互环境下测试程序时，通常的做法是先导入模块，需要执行代码时才去调用 main 或其他函数。要想只导入不执行，当然可以删掉程序入口 main()，但这又会失去双击执行程序的可能。两全其美的做法是在主程序入口 main 之前加个条件：

if <条件>:
    main()

意思是当条件满足时启动程序，否则不启动程序。问题是条件怎么写？

如果是用 import 导入模块，Python 会将该模块的一个特殊变量 name 的值设置为模块的名字。例如：

>>> import math
>>> print math.__name__
math

第一行导入模块 math，并将 math 的变量 name 设置为'math'。第二行显示了这个变量的值。

但如果是直接执行模块（如双击模块文件图标等），Python 则将模块的特殊变量 name 设置为字符串'main'。因此可以通过特殊变量 name 的值来判断模块是被导入的还是被直接执行的。根据这个底层细节，我们可以将程序文件的最后一行改成：

if __name__ == ’__main__’:
    main()

这样就能确保当程序是直接执行时，main 能启动；当程序是被导入时，忽略 main。