Python中的if __name__ == '__main__'

在具体说明if __name__ == '__main__'功能前，先从一个简单的实例直观上感受一下。

# const.py

PI = 3.14


def train():
    print("PI:", PI)

train()

# area.py

from const import PI


def calc_round_area(radius):
    return PI * (radius ** 2)


def calculate():
    print("round area: ", calc_round_area(2))

calculate()

现在运行area.py文件，看结果：

 

可以看见，const.py中的train()也被运行了，实际上我们是不希望它被运行，只是想把const.py中 PI 变量导入到 area.py。现在把 const.py 改一下：

PI = 3.14


def train():
    print("PI:", PI)


if __name__ == "__main__":
    train()

运行area.py文件发现没有执行const.py中train()。

1. if __name__ == '__main__'功能

丛上述实例可以发现，运行const.py文件时if __name__=="__main__": 语句之前与之后代码都被执行，但作为area.py导入文件时就不会执行if __name__=="__main__": 语句之后代码。

实际上if __name__=="__main__":就相当于是 Python 模拟的程序入口。由于模块之间相互引用，不同模块可能都有这样的定义，而入口程序只能有一个，选中哪个入口程序取决于 __name__ 的值。

可以做一下测试：

# test.py
print("hahaha")
print(__name__)

if __name__ == '__main__':
    print("I'm test.py")

可以发现，此时变量__name__的值为"__main__"，即打印“I'm test.py”。如果运行其他文件，通过运行的文件调用本文件，则不会打印该语句，因为程序入口不对，该语句不执行。

举个栗子：

# receiver.py
print("receiver_haha")


def test():
    print("test can be called!")


def receiver_print():
    print("I'm receiver.py")


if __name__ == '__main__':
    receiver_print()

# caller.py
import receiver

print("caller_haha")


def test():
    print("caller_test can be called!")


def caller_print():
    print("I'm caller.py")


if __name__ == '__main__':
    caller_print()
    test()

运行caller.py，结果如下：

可以发现receiver.py中if __name__=="__main__"后语句没有执行，因为此时运行的是caller.py，而找不到进去receiver.py的入口！使用这种方法可以控制代码的执行！

 

2. __name__

2.1 __name__是内置变量，可用于表示当前模块的名字（见上）。

2.2 __name__是内置变量，可用于反映一个包的结构。

 

推荐阅读：

Python中的_, __, __xx__ 区别