1.引言
在Python当中,如果代码写得规范一些,通常会写上一句if __name__ == "__main__":作为程序的入口.
但似乎没有这么一句代码,程序也能正常运行。
这句代码多余吗?原理又在哪里?本篇博文对此进行总结说明。
2.2个模块正常调用
我们知道,当我们把模块A中的代码在模块B中进行import A时,只要B模块代码运行到该import语句,模块A的代码会被执行。
# @File : test2.py
class A:
a = 100
print('你好,我是模块A……')
print(a)
运行结果:
你好,我是模块A……
100
# @File : test4.py
from test2 import A
class B:
b = 200
print('你好,我是模块B……')
print(b)
运行结果:
你好,我是模块A……
100
你好,我是模块B……
200
3.2个模块通过if __name__ == "__main__":进行调用
如果在模块A中,我们有部分的代码不想在被导入到B时直接被运行,但在直接运行A时可直接运行,那该怎么做呢?
那就可以用到if __name__ == "__main__":这行代码了,我们对上面用到的A模块代码进行修改,B保持原样:
# @File : test2.py
class A:
a = 100
print('你好,我是模块A……')
if __name__ == "__main__":
print(a)
运行结果:
你好,我是模块A……
100
# @File : test4.py
from test2 import A
class B:
b = 200
print('你好,我是模块B……')
print(b)
运行结果:
你好,我是模块A……
你好,我是模块B……
200
看到了吗,A模块中的a的值就没有再被输出了。
所以,当你要导入某个模块,但又不想改模块的部分代码被直接执行,那就可以这一部分代码放在if __name__ == "__main__":内部。
4.__name__ == "__main__"到底是什么关系
__name__是Python的内置变量,用于指代当前模块。
我们修改上面用到的A模块和B模块,在模块中分别输出模块的名称:
# @File : test2.py
class A:
print('你好,我是模块A……')
print('模块A中__name__的值:{}'.format(__name__))
运行结果:
你好,我是模块A……
模块A中__name__的值:__main__
# @File : test4.py
from test2 import A
class B:
print('你好,我是模块B……')
print('模块B中__name__的值:{}'.format(__name__))
运行结果:
你好,我是模块A……
模块A中__name__的值:test2
你好,我是模块B……
模块B中__name__的值:__main__
当哪个模块被直接执行时,该模块“__name__”的值就是“__main__”。
当被导入另一模块时,“__name__”的值就是该模块的真实名称。
5.总结
所以,如果当前模块A时直接执行,运行“if __name__ == "__main__":语句时,__name__的值就是__main__,结果是True,if __name__ == "__main__":下面的代码块就会被执行。
但是被引入时模块B时,__name__的值就是A,结果是False,if __name__ == "__main__":下面的代码块不会被执行。