前言
在大多数编排得好一点的脚本或者程序里面都有这段if __name__ == 'main':
,虽然一直知道他的作用,但是一直比较模糊,收集资料详细理解之后与大家分享。
代码功能
一个python的文件有两种使用的方法,第一是直接作为脚本执行,第二是import到其他的python脚本中被调用(模块重用)执行。因此if __name__ == 'main':
的作用就是控制这两种情况执行代码的过程,if __name__ == 'main':
下的代码只有在第一种情况下(即文件作为脚本直接执行)才会被执行,而import到其他脚本中是不会被执行的。
举个例子,下面在test.py中写入如下代码:
1
2
3print("A")
if __name__ == '__main__' :
print("B")
直接执行test.py,输出A
和B
。即,if __name__=="__main__":
语句之前和之后的代码都被执行。
然后在同一文件夹新建名称为import_test.py的脚本,只输入如代码·import test
,执行脚本,只输出A
。即,if __name__=="__main__":
之前的语句被执行,之后的没有被执行。
运行原理
每个python模块(python文件,也就是此处的test.py和import_test.py)都包含内置的变量__name__,当运行模块被执行的时候,name__等于文件名(包含了后缀.py);如果import到其他模块中,则__name__等于模块名称(不包含后缀.py)。而“__main__”等于当前执行文件的名称(包含了后缀.py)。进而当模块被直接执行时,__name == ‘main’结果为真。
举例说明,我们在test.py脚本的if __name__=="__main__":
之前加入print (__name__)
,即将__name__
打印出来。输出__main__
。再执行import_test.py,输出test
。此时,test.py中的__name__变量值为test,不满足__name__==”main“的条件,因此,无法执行其后的代码。