Python杂谈 | (8) Python中的__name__=='__main__'的作用

原文地址

目录

 

1. 作用

2. 总结

---

1. 作用

有句话经典的概括了__name__=='__main__'的作用，"Make a script both importable and executable',意思就是说让你写的脚本模块(.py)既可以导入到别的模块中使用，另外该模块也可以自己执行。

举例说明，先写一个模块module.py:

#module.py
def main():
    print("we are in %s"%__name__)

if __name__ == '__main__':
    main()

这个函数定义了一个main函数，我们执行一个当前的module.py,结果打印的是"we are in __main__",说明我们的if语句中的内容被执行了，调用了main()。

但是如果我们从另外一个模块anothermodule.py中导入该模块，并调用一次main()函数会发生什么呢？

#anothermodule.py
from module import main
main()

执行结果是："we are in module"。 没有打印"we are in __main__"，也就是说module.py模块中if语句中的内容没有被执行。这样既可以让模块文件自己运行，也可以被其他模块引入，而且函数不会被执行两次。

2. 总结

如果我们直接执行某个.py文件时，该文件中的__name__=='__main__'是True，但是我们如果从另一个.py文件中通过import或from...import...导入该文件时，此时__name__的值就是我们导入的py文件的名字，而不是__main__。

这个功能还有一个用处：调试代码的时候，在”if __name__ == '__main__'“中加入一些我们的调试代码，我们可以让外部模块调用的时候不执行我们的调试代码，但是如果我们想排查问题的时候，直接执行该模块文件，调试代码能够正常运行！