可能很多同学在阅读Python源代码时会发现经常会出现if __name__ == '__main__':这样的代码,那么这样的代码起到什么作用呢?本文将为你深入解析__name__变量的含义和应用场景。
当Python解析器读源代码文件时,会做如下两件事情:
设置特殊变量,如__name__;
执行源代码文件中的所有代码;
现在我们将焦点放到__name__变量上来,看看在Python程序中为什么要使用__name__变量。
让我们先使用一段代码示例来探索导入和脚本的工作方式。假设这些代码位于名为foo.py的文件中。
print("before import") import math print("before functionA") def functionA(): print("Function A") print("before functionB") def functionB(): print("Function B {}".format(math.sqrt(100))) print("before __name__ guard") if __name__ == '__main__': functionA() functionB() print("after __name__ guard")当Python解释器读取源文件时,它首先定义一些特殊变量。在这个案例中,我们关心的是__name__变量。如果将Python脚本文件作为主程序运行,也就是通过下面的命令运行foo.py。
python foo.pyPython解释器会使用下面的代码为__name__变量赋值,也就是说,__name__变量的值是"__main__"。
__name__ = "__main__"另一方面,假设其他模块是主程序,并且它将导入foo.py。这意味着在主程序中会有如下的语句:
import fooPython解释器将搜索foo.py文件(以及搜索其他一些变体),在执行该模块之前,它将从import语句中将名称“ foo”分配给__name__变量,也即是使用下面的代码为__name__变量赋值。
__name__ = "foo"设置__name__变量后,Python解释器通过一次执行一条语句的方式执行模块中的所有代码。您可能想要在代码示例侧面打开另一个窗口,以便您可以按照以下说明进行操作。
现在执行前面的代码,会输出如下内容:
before import before functionA before functionB before __name__ guard Function A Function B 10.0 after __name__ guard如果当前的Python脚本按着主程序的方式执行(使用python命令),那么__name__变量的值就是"__main__"。这样就会调用functionA和functionB函数,从而输出 "Function A" 和 "Function B 10.0"
如果foo.py脚本不是作为主程序运行,而是被另一个程序导入,则__name__变量的值是“ foo”,而不是“ __main__”,在这种情况下,将不会调用functionA和functionB函数。不过在这种情况下,仍然会输出 "after __name__ guard" ,因为这条语句不属于if语句。
现在总结一下:
根据foo.py的运行方式,可能会有两种输出结果。
(1)作为主程序运行,会输出如下结果
before import before functionA before functionB before __name__ guard Function A Function B 10.0 after __name__ guard(2)作为模块被导入,会输出如下结果:
before import before functionA before functionB before __name__ guard after __name__ guard可能有很多同学会有这样的疑问,为什么Python要提供这个功能呢?像C#、Java这样的编程语言并没有这样的功能啊!其实这要从Python脚本的运行机理谈起。如果Python脚本作为主程序执行,这个执行方式与Java类似。不过当导入一个Python模块就不一样了。对于Java语言,导入一个包,也只是导入而已,除非显式调用包中的API,否则单单导入,是不会执行Java代码的。但Python就不一样了,如果使用import语句导入一个模块,其实是先执行被导入模块中的所有代码,然后才会执行当前模块的代码。例如,下面的代码,会先执行foo.py脚本中的代码,然后才会执行print('current module')
import foo print('current module')执行这段代码,会输出如下的内容:
before import before functionA before functionB before __name__ guard after __name__ guard current module如果一个Python脚本,同时即可以作为主程序执行,也可以作为模块被导入,这就要求在模块被导入时不执行作为主程序执行时的代码,所以如果是在主程序中执行的代码,应该使用__main__变量进行判断。
关于__name__的一些疑问
有的同学问,在脚本文件中可以有多个__name__校验代码块吗?其实通常只有一个__name__校验代码块,但Python解析器并不会阻止你编写多个__name__校验代码块吗。
下面再给大家2段代码,看看输出结果会是什么:
# foo2.py def functionA(): print("a1") import foo2 print("a2") functionB() print("a3") def functionB(): print("b") print("t1") if __name__ == "__main__": print("m1") functionA() print("m2") print("t2")在这段代码中导入了模块本身(foo2.py),执行代码,会输出如下结果:
t1 m1 a1 t1 t2 a2 b a3 m2 t2大家可以分析一下这段代码的执行过程。下面将if __name__ == "__main__":去掉,看看会发生什么。
# foo3.py def functionA(): print("a1") import foo3 print("a2") functionB() print("a3") def functionB(): print("b") print("t1") print("m1") functionA() print("m2") print("t2")执行这段代码,会引起死循环吗?其实是不会的,因为Python解析器有缓存,如果一个模块在当前模块中已经被导入了,当第二次导入时,将不会再次执行被导入模块的代码,而是直接使用缓存中的内容,所以import foo3只会导致foo3.py文件中的代码作为导入模块的方式被执行一次。所以执行这段代码,输出结果如下:
t1 m1 a1 t1 m1 a1 a2 b a3 m2 t2 a2 b a3 m2 t2大家可以自己分析一下程序的执行过程。
推荐阅读 点击标题可跳转
1、连Python产生器(Generator)的原理都解释不了,还敢说Python用了5年?
2、牛掰了!鸿蒙与Android完美融合,将鸿蒙设备当Android设备用
3、【鸿蒙学院】鸿蒙App开发直播学员提问与回答
4、【鸿蒙学院】鸿蒙IDE:下载、安装DevEco Studio
5、 Python高效编程之88条军规(2):你真的会格式化字符串吗?
6、像极客一样提取Android的Root权限
关注「极客起源」公众号,加星标,不错过精彩技术干货
蒙娜丽宁 认证博客专家 公众号:极客起源 达芬奇 UM 更多精彩内容请关注微信公众号:「极客起源」,东北大学计算机专业硕士。UnityMarvel创始人,企业内训讲师、IT畅销书作者,学院讲师,拥有多年软件开发经验和培训经验。主要著作包括《Python从菜鸟到高手》、《Python爬虫技术:深入理解原理、技术与开发》,培训企业学员数千人。制作在线课程数千小时。更多视频课程,请访问我的B站:https://space.bilibili.com/477001733