最近在学习Python,在很多Python脚本中在最后的部分会执行一个判断语句if __name__ == "__main__:
,之后还可能会有一些执行语句。那添加这个判断的目的何在?为此搜集了些资料终于明白了,分享给大家。
在Python编译器读取源文件的时候会执行它找到的所有代码,而在执行之前会根据当运行的模块是否为主程序而定义变量__name__
的值为__main__
还是模块名。
因此,该判断语句为真的时候(也就是判断语句if __name__ == "__main__:
中的if
为真时),说明当前运行的脚本为主程序,而非主程序所引用(import
)的一个模块。
这在当你想要运行一些只有在将模块当做程序运行时而非当做模块引用时才执行的命令,只要将它们放到if __name__ == "__main__:"
判断语句之后就可以了。由此可见,此句常常在测试的时候使用。
下面具体举个栗子方便理解:
将one.py
和two.py
分别写入文件,内容为:
# file one.py
def func():
print("func() in one.py")
print("top-level in one.py")
if __name__ == "__main__":
print("one.py is being run directly") #主程序时执行此句
else:
print("one.py is being imported into another module") #被另外的文件import时执行此句
# file two.py
import one # start executing one.py
print("top-level in two.py")
one.func()
if __name__ == "__main__":
print("two.py is being run directly")
else:
print("two.py is being imported into another module")
当运行python one.py
,
输出:
top-level in one.py
one.py is being run directly
当运行python two.py
,
输出:
top-level in one.py
one.py is being imported into another module
top-level in two.py
func() in one.py
two.py is being run directly
具体的理解可以拷贝代码测试一下,对照代码和运行结果很快就明白了。