这是做什么的,为什么要包括该if声明?
if
if __name__ == "__main__": print("Hello, World!")
它是样板代码,可防止用户在无意中意外调用脚本。以下是脚本中省略守卫时的一些常见问题:
import my_script_without_a_name_eq_main_guard
为了更好地理解这为什么以及如何重要,我们需要退后一步来了解 Python 如何初始化脚本以及它如何与其模块导入机制进行交互。
每当 Python 解释器读取源文件时,它会做两件事:
__name__
让我们看看它是如何工作的,以及它与您关于__name__我们在 Python 脚本中经常看到的检查的问题有何关系。
让我们使用一个稍微不同的代码示例来探索导入和脚本是如何工作的。假设以下内容位于名为foo.py.
foo.py
# Suppose this is foo.py. print("before import") import math print("before function_a") def function_a(): print("Function A") print("before function_b") def function_b(): print("Function B {}".format(math.sqrt(100))) print("before __name__ guard") if __name__ == '__main__': function_a() function_b() print("after __name__ guard")
当 Python 解释器读取一个源文件时,它首先定义了一些特殊的变量。在这种情况下,我们关心__name__变量。
当你的模块是主程序时
如果您将模块(源文件)作为主程序运行,例如
python foo.py
解释器会将硬编码的字符串分配给"__main__"变量__name__,即
"__main__"
# It's as if the interpreter inserts this at the top # of your module when run as the main program. __name__ = "__main__"
当您的模块被另一个模块导入时
另一方面,假设某个其他模块是主程序,它会导入您的模块。这意味着在主程序中或在主程序导入的其他模块中存在这样的语句:
# Suppose this is in some other main program. import foo
解释器将搜索您的foo.py文件(以及搜索一些其他变体),并且在执行该模块之前,它会将"foo"导入语句中的名称分配给__name__变量,即
"foo"
# It's as if the interpreter inserts this at the top # of your module when it's imported from another module. __name__ = "foo"
设置特殊变量后,解释器执行模块中的所有代码,一次一条语句。您可能希望在代码示例旁边打开另一个窗口,以便您可以按照此说明进行操作。
总是
"before import"
math
import math
__import__
# Find and load a module given its string name, "math", # then assign it to a local variable called math. math = __import__("math")
"before function_a"
def
function_a
"before function_b"
function_b
"before __name__ guard"
仅当您的模块是主程序时
"Function A"
"Function B 10.0"
仅当您的模块被另一个模块导入时
"after __name__ guard"
概括
总而言之,以下是两种情况下打印的内容:
# What gets printed if foo is the main program before import before function_a before function_b before __name__ guard Function A Function B 10.0 after __name__ guard # What gets printed if foo is imported as a regular module before import before function_a before function_b before __name__ guard after __name__ guard
你可能自然想知道为什么有人会想要这个。好吧,有时您想编写一个.py文件,该文件既可以被其他程序和/或模块用作模块,也可以作为主程序本身运行。例子:
.py
除了这些示例之外,在 Python 中运行脚本只是设置一些魔术变量并导入脚本,这很优雅。“运行”脚本是导入脚本模块的副作用。
foo2.py
python foo2.py
# Suppose this is foo2.py. import os, sys; sys.path.insert(0, os.path.dirname(__file__)) # needed for some interpreters def function_a(): print("a1") from foo2 import function_b print("a2") function_b() print("a3") def function_b(): print("b") print("t1") if __name__ == "__main__": print("m1") function_a() print("m2") print("t2")
foo3.py
# Suppose this is foo3.py. import os, sys; sys.path.insert(0, os.path.dirname(__file__)) # needed for some interpreters def function_a(): print("a1") from foo3 import function_b print("a2") function_b() print("a3") def function_b(): print("b") print("t1") print("m1") function_a() print("m2") print("t2")
# Suppose this is in foo4.py __name__ = "__main__" def bar(): print("bar") print("before __name__ guard") if __name__ == "__main__": bar() print("after __name__ guard")