ThinkChat2.0新版上线,更智能更精彩,支持会话、画图、阅读、搜索等,送10W Token,即刻开启你的AI之旅 广告
### 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。