文章目录


每当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__​​​虽然它长得很奇怪,前面和后面都有一个下划线​​_​​​,但是记得变量命名的规则嘛?下划线​​_​​​是可以出现在变量首字母的。所以,​​__name__​​仍然是一个变量,只不过,是解释器自己定义的。

2.1 作为主程序运行

当我们在命令行中使用​​python foo.py​​,或者直接在ide(比如pycharm)图形界面里点运行foo.py,那么这时候,foo.py就是作为主程序运行的。

此时:
Python解释器会直接给​​​_name_​​​变量赋值为"​​_main_​​"

2.2 作为导入的模块运行

如果有另一个程序,叫​​main.py​​,它里面的代码是这样的。

import

那么如果我们在命令行中使用​​python main.py​​​,则​​main.py​​作为主程序运行,而foo.py就是导入的模块。

此时:
Python解释器会令​​​__name__ = "foo"​

2.3 执行foo.py文件中的代码

如果使用主程序运行​​!python foo.py​​​:
输出如下:

import
before functionA
before functionB
before __name__ guard
Function A
Function B 10.0

如果使用主程序运行​​import foo​​​:
输出如下:

before import

可以明显看到,当使用主程序运行​​import foo​​​时,没有执行下面语句的内容,因为此时​​__name__ = 'foo'​

if __name__ == '__main__':
functionA()
functionB()

为什么这样工作?

有时我们想编写一个.py文件,该文件既可以被其他程序和模块导入,也可以作为主程序运行。
例子如下:

  • 这个文件是一个库,可以被其他文件导入。但是我们希望可以在其中运行一些单元测试或演示。
  • 这个文件仅用作主程序,但具有一些单元测试的功能,一些测试框架(类似​​unittest​​和​​doctest​​)需要导入这个​​.py​​文件来测试。我们不希望,它只是因为被导入为模块,就直接运行整个脚本。
  • 这个模块主要用作主程序,但它也为高级用户提供了程序员友好的​​API​​。

所以,其实就是有的时候希望他在被导入的时候运行一些代码,有的时候希望他作为主程序的时候运行另一些代码。所以需要进行判断。


​参考链接​

微信公众号

欢迎大家关注我的个人公众号,现阶段主要总结Robomaster相关的计算机视觉知识:Qt,C++,CMake,OpenCV等等
公众号名称:三丰杂货铺

`if __name__ == “__main__“:`是什么意思___name__