if __name__ == "__main__": 的作用


if __name__ == "__main__":的作用

当Python解释器读取源文件时,它会执行其中的所有代码。

在执行代码之前,它将定义一些特殊变量。例如,如果Python解释器将该模块(源文件)作为主程序运行,则它将特殊__name__变量设置为具有值"__main__"。如果从另一个模块导入此文件,__name__则将其设置为模块的名称。

实例

import time, thread

def myfunction(string, sleeptime, lock, *args):
    while True:
        lock.acquire()
        time.sleep(sleeptime)
        lock.release()
        time.sleep(sleeptime)

if __name__ == "__main__":
    lock = thread.allocate_lock()
    thread.start_new_thread(myfunction, ("Thread #: 1", 2, lock))
    thread.start_new_thread(myfunction, ("Thread #: 2", 2, lock))

在你的脚本的情况下,让我们假设它作为主要功能执行,例如你说的话

python threading_example.py

在命令行上。设置特殊变量后,它将执行import语句并加载这些模块。然后它将评估def块,创建一个函数对象并创建一个myfunction指向函数对象的变量。然后它将读取该if语句并看到它name确实相等"__main__",因此它将执行那里显示的块。

这样做的一个原因是,有时您可以编写一个.py可以直接执行的模块(文件)。或者,它也可以导入并在另一个模块中使用。通过执行主检查,您可以仅在希望将模块作为程序运行时执行该代码,而在有人只想导入模块并自行调用函数时不执行该代码。

另外的解释

通过将脚本作为命令传递给Python解释器来运行脚本时,

python myscript.py

处于缩进级别0的所有代码都将被执行。已定义的函数和类已定义,但它们的代码都不会运行。与其他语言不同,没有main()自动运行的main()功能- 该功能隐含在顶层的所有代码中。

在这种情况下,顶级代码是一个if块。 __name__是一个内置变量,它计算当前模块的名称。但是,如果正在直接运行模块(myscript.py如上所述),则将name其设置为字符串"__main__"。因此,您可以通过测试来测试脚本是直接运行还是由其他内容导入

if __name__ == "__main__":
    ...

如果您的脚本被导入到另一个模块中,它的各种函数和类定义将被导入并且它的顶级代码将被执行,但是if上面条款的then-body中的代码将不会被运行,因为条件是没见过。作为基本示例,请考虑以下两个脚本:

# file one.py
def func():
    print("func() in one.py")

print("top-level in one.py")

if __name__ == "__main__":
    print("one.py is being run directly")
else:
    print("one.py is being imported into another module")
# file two.py
import one

print("top-level in two.py")
one.func()

if __name__ == "__main__":
    print("two.py is being run directly")
else:
    print("two.py is being imported into another module")

现在,如果你调用解释器

python one.py

输出将是

top-level in one.py
one.py is being run directly

如果你two.py改为:

python two.py

你得到

top-level in one.py
one.py is being imported into another module
top-level in two.py
func() in one.py
two.py is being run directly

因此,当模块one加载时,它__name__等于"one"而不是"__main__"