__name__ == '__main__'到底是什么意思
相信许多人初学 Python 时,常会在例子程序中看到如下语句:
1 if __name__ == '__main__': 2 foo() # 或其他语句
第 1 行的__name__ == '__main__'到底是什么意思呢?
首先,如果你永远都只执行一个 xxx.py,而不去 import它的话,那你完全不需要了解这是什么。例如你写了一个程序,文件名叫 hello.py
1 print("====this is hello.py====") 2 def foo(): 3 print("hello") 4 print("call foo() which is defined here") 5 foo() 6 print("====this is hello.py END ====")
然后你总是直接执行它,就像这样
1 $ python3 hello.py 2 ====this is hello.py==== 3 call foo() which is defined here 4 hello 5 ====this is hello.py END ====
这完全没有问题。
但是,当别人想引用你的foo()函数时,就有问题了。
例如别人写了一个文件,名字是 world.py
1 # world.py 2 print("====this is world.py====") 3 from hello import foo 4 5 print("call foo() which is defined in other file") 6 foo() 7 print("====this is world.py END ====")
运行结果如下:
1 $ python3 world.py 2 ====this is world.py==== 3 ====this is hello.py==== 4 call foo() which is defined here 5 hello 6 ====this is hello.py END ==== 7 call foo() which is defined in other file 8 hello 9 ====this is world.py END ====
可以看到,解释器是逐行解释源码的,当执行到源码的第 3 行时,也就是 hello.py 被引用时,hello.py 的每一行都会被解释器读取并执行,执行效果就是结果中的3~6行,然后解释器执行源码的5~7行。
如果你不想要结果中的3~6行,该怎么做呢?
Python 解释器执行代码时,有一些内建、隐含的变量,__name__就是其中之一,其意义是“模块名称”。
如果该模块是被引用,那么__name__的值会是此模块的名称;如果该模块是直接被执行,那么__name__的值是__main__。
或许你还不明白,那我们来做个实验。
1 # hello.py 2 print("====this is hello.py====") 3 print(__name__) 4 def foo(): 5 print("Ha") 6 print("call foo() which is defined here") 7 foo() 8 print("====this is hello.py END ====")
请注意上面第3行
1 # world.py 2 print("====this is world.py====") 3 print(__name__) 4 from hello import foo 5 print("call foo() which is defined in other file") 6 foo() 7 print("====this is world.py END ====")
同样,请注意上面第3行
我们看一下运行结果。
对于第一个:
1 $ python3 hello.py 2 ====this is hello.py==== 3 __main__ 4 call foo() which is defined here 5 Ha 6 ====this is hello.py END ====
从结果的第3行可以看出,对于直接运行的模块,其模块名是 __main__
对于第二个:
1 $ python3 world.py 2 ====this is world.py==== 3 __main__ # 因为 world.py 被直接执行,所以这里的值是 __main__ 4 ====this is hello.py==== 5 hello # 因为 hello.py 被引用,所以这里的值是 hello 6 call foo() which is defined here 7 Ha 8 ====this is hello.py END ==== 9 call foo() which is defined in other file 10 Ha 11 ====this is world.py END ====
注意到第5行了吗?这里的“hello”正是 hello.py 被引用时的模块名称。
由此可见,__name__的值在模块被直接执行时与被引用时是不同的。
回到上面的问题:当一个模块被引用时,如何不执行该模块的语句?答案就是靠判断__name__是否等于 __main__。当一个模块被直接执行时,其__name__必然等于__main__;当一个模块被引用时,其__name__必然等于文件名(不含.py);
总结
之所以常看见这样的写法,是因为该程序可能有“单独执行”(例如执行一些单元测试)与“被引用”两种情况,鉴于这两种情况中__name__的值是不同的:当一个模块被直接执行时,其__name__必然等于__main__;当一个模块被引用时,其__name__必然等于文件名(不含.py)。所以利用判断__name__ == '__main__'的真假就可以将这两种情况区分出来。