Python文件的两种使用方式:
- 作为脚本直接运行。
- 通过Import作为其他Python文件的模块执行。
在很多Python脚本中的最后的部分会执行一个判断语句if __name__ == "__main__:"
,在这个判断语句之后还会有些语句,那这个判断语句的作用是什么?
在python编译器读取源文件的时候会执行它找到的所有代码,而在执行之前会根据当前运行的模块是否为主程序而定义变量__name__的值为__main__还是模块名。因此,该判断语句为真的时候,说明当前运行的脚本为主程序,而非主程序所引用的一个模块。这在当你想要运行一些只有在将模块当做程序运行时而非当做模块引用时才执行的命令,只要将它们放到if __name__ == "__main__:"
判断语句之后就可以了。
示例:
#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")
输出:
top-level in one.py
one.py is being run directly
#file two.py
import one
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")
输出:
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
每个Python模块都包含内置的变量__name__
,当该模块被直接执行的时候,__name__
等于文件名(包含后缀);如果该模块import到其他模块中,则该模块__name__等于模块名称(不包含后缀)。
__main__
始终指当前模块的名称(包含后缀)。进而当模块被直接执行时,__name__ == "main"
结果为真。