if __name__ == "__main__": 做啥?
Posted
技术标签:
【中文标题】if __name__ == "__main__": 做啥?【英文标题】:What does if __name__ == "__main__": do?if __name__ == "__main__": 做什么? 【发布时间】:2021-12-07 07:34:41 【问题描述】:给定以下代码,if __name__ == "__main__":
做了什么?
# Threading example
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))
【问题讨论】:
仅作记录——什么是“main”:docs.python.org/3/reference/… 和什么是“name”:docs.python.org/3/reference/… 【参考方案1】:简答
它是样板代码,可防止用户在无意中意外调用脚本。以下是脚本中省略守卫时的一些常见问题:
如果您在另一个脚本中导入无保护脚本(例如import my_script_without_a_name_eq_main_guard
),那么第二个脚本将触发第一个运行在导入时并使用第二个脚本的命令行参数。这几乎总是一个错误。
如果您在无保护脚本中有一个自定义类并将其保存到一个 pickle 文件中,那么在另一个脚本中将其取消腌制将触发无保护脚本的导入,与上一个项目符号中概述的问题相同。
长答案
为了更好地理解为什么以及如何重要,我们需要退后一步来了解 Python 如何初始化脚本以及它如何与其模块导入机制交互。
每当 Python 解释器读取源文件时,它都会做两件事:
它设置了一些特殊的变量,比如__name__
,然后
它执行文件中的所有代码。
让我们看看它是如何工作的,以及它与您对我们在 Python 脚本中经常看到的__name__
检查的问题有何关联。
代码示例
让我们使用一个稍微不同的代码示例来探索导入和脚本的工作原理。假设以下内容位于名为 foo.py
的文件中。
# Suppose this is 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__
变量。
当你的模块是主程序时
如果您将模块(源文件)作为主程序运行,例如
python foo.py
解释器会将硬编码字符串"__main__"
分配给__name__
变量,即
# 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__
变量,即
# 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
模块并将其分配给一个名为math
的变量。这相当于将import math
替换为以下内容(注意__import__
是Python 中的一个低级函数,它接受一个字符串并触发实际导入):
# Find and load a module given its string name, "math",
# then assign it to a local variable called math.
math = __import__("math")
它打印字符串"before functionA"
。
它执行def
块,创建一个函数对象,然后将该函数对象分配给一个名为functionA
的变量。
它打印字符串"before functionB"
。
它执行第二个def
块,创建另一个函数对象,然后将其分配给一个名为functionB
的变量。
它打印字符串"before __name__ guard"
。
仅当您的模块是主程序时
-
如果您的模块是主程序,那么它会看到
__name__
确实设置为"__main__"
,并调用这两个函数,打印字符串"Function A"
和"Function B 10.0"
。
仅当您的模块被其他人导入时
-
(instead) 如果你的模块不是主程序而是被另一个程序导入,那么
__name__
将是"foo"
,而不是"__main__"
,它会跳过正文if
声明。
总是
-
在这两种情况下都会打印字符串
"after __name__ guard"
。
总结
总之,以下是两种情况下会打印的内容:
# What gets printed if foo is the main program
before import
before functionA
before functionB
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 functionA
before functionB
before __name__ guard
after __name__ guard
为什么会这样?
您可能自然会想知道为什么有人会想要这个。好吧,有时您想编写一个.py
文件,该文件既可以被其他程序和/或模块用作模块,也可以作为主程序本身运行。例子:
您的模块是一个库,但您希望有一个脚本模式,它可以运行一些单元测试或演示。
您的模块仅用作主程序,但它具有一些单元测试,并且测试框架通过导入 .py
文件(如您的脚本)并运行特殊的测试功能来工作。您不希望它仅仅因为它正在导入模块而尝试运行脚本。
您的模块主要用作主程序,但它也为高级用户提供了对程序员友好的 API。
除了这些示例之外,在 Python 中运行脚本只是设置一些魔术变量并导入脚本,这很优雅。 “运行”脚本是导入脚本模块的副作用。
发人深省
问题:我可以有多个__name__
检查块吗?答:这样做很奇怪,但语言不会阻止你。
假设以下内容位于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 functionA():
print("a1")
from foo2 import functionB
print("a2")
functionB()
print("a3")
def functionB():
print("b")
print("t1")
if __name__ == "__main__":
print("m1")
functionA()
print("m2")
print("t2")
现在,弄清楚如果您删除 __name__
签入 foo3.py
会发生什么:
# Suppose this is foo3.py.
import os, sys; sys.path.insert(0, os.path.dirname(__file__)) # needed for some interpreters
def functionA():
print("a1")
from foo3 import functionB
print("a2")
functionB()
print("a3")
def functionB():
print("b")
print("t1")
print("m1")
functionA()
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")
【讨论】:
出于好奇:如果我在 python 脚本中运行subprocess.run('foo_bar.py')
会发生什么?我想foo_bar
将以__name__ = '__main__'
开头,就像我在cmd 中手动提示foo_bar.py
一样。是这样吗?考虑到@MrFooz 的回答,这样做应该没有任何问题,并且一次拥有尽可能多的“主要”模块。即使更改 __name__
值或让多个独立创建的实例(或由 subprocess
相互创建的实例)相互交互,对于 Python 来说也应该照常进行。我错过了什么吗?
@hajef 你对subprocess.run
的工作方式是正确的。也就是说,在脚本之间共享代码通常更好的方法是创建模块并让脚本调用共享模块,而不是作为脚本相互调用。很难调试subprocess.run
调用,因为大多数调试器不会跨越进程边界,它会增加重要的系统开销来创建和销毁额外的进程等。
我对思考部分中的 foo2.py 示例有疑问。 from foo2.py import functionB 有什么作用?在我看来,它只是从 functionB 导入 foo2.py
multiprocessing
是可能导入您的代码的模块之一,尤其是在 Windows 上需要进行此测试。
非常小的一点,但我相信python实际上是从import语句中确定导入模块的__name__
,而不是从文件名中剥离“.py”。因为 python 标识符区分大小写但文件名可能不区分大小写(例如在 windows 上),所以文件名中不一定有足够的信息来确定正确的 python 模块名称。【参考方案2】:
当您的脚本通过将其作为命令传递给 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__"
。
【讨论】:
那么,if __name__ == "__main__":
基本上会检查您是否正在运行您的 python 脚本本身,而不是导入它或其他什么?【参考方案3】:
__name__
变量(恕我直言)的最简单解释如下:
创建以下文件。
# a.py
import b
和
# b.py
print "Hello World from %s!" % __name__
if __name__ == '__main__':
print "Hello World again from %s!" % __name__
运行它们会得到这个输出:
$ python a.py
Hello World from b!
如你所见,当一个模块被导入时,Python 将这个模块中的globals()['__name__']
设置为模块的名称。此外,导入后,模块中的所有代码都在运行。由于if
语句的计算结果为False
,这部分不会被执行。
$ python b.py
Hello World from __main__!
Hello World again from __main__!
如你所见,当一个文件被执行时,Python 将这个文件中的globals()['__name__']
设置为"__main__"
。这一次,if
语句的计算结果为 True
并且正在运行。
【讨论】:
【参考方案4】:
if __name__ == "__main__":
有什么作用?
概述基础知识:
作为程序入口点的模块中的全局变量__name__
是'__main__'
。否则,它就是您导入模块的名称。
因此,if
块下的代码只有在模块是程序的入口点时才会运行。
它允许模块中的代码可以被其他模块导入,而不需要在导入时执行下面的代码块。
我们为什么需要这个?
开发和测试您的代码
假设您正在编写一个旨在用作模块的 Python 脚本:
def do_important():
"""This function does something very important"""
您可以通过将这个函数调用添加到底部来测试模块:
do_important()
并使用以下内容(在命令提示符下)运行它:
~$ python important.py
问题
但是,如果您想将模块导入另一个脚本:
import important
在导入时,do_important
函数将被调用,因此您可能会在底部注释掉您的函数调用 do_important()
。
# do_important() # I must remember to uncomment to execute this!
然后你必须记住你是否已经注释掉了你的测试函数调用。而且这种额外的复杂性意味着您可能会忘记,从而使您的开发过程更加麻烦。
更好的方法
__name__
变量指向 Python 解释器当前所在的命名空间。
在导入的模块中,它是该模块的名称。
但在主模块(或交互式 Python 会话,即解释器的 Read、Eval、Print Loop 或 REPL)内,您正在运行其 "__main__"
中的所有内容。
所以如果你在执行前检查:
if __name__ == "__main__":
do_important()
通过上述方式,您的代码只会在您将其作为主要模块运行(或有意从另一个脚本中调用它)时执行。
更好的方法
不过,有一种 Pythonic 方法可以改进这一点。
如果我们想从模块外部运行这个业务流程怎么办?
如果我们将开发和测试时想要执行的代码放入这样的函数中,然后立即检查'__main__'
:
def main():
"""business logic for when running this module as the primary one!"""
setup()
foo = do_important()
bar = do_even_more_important(foo)
for baz in bar:
do_super_important(baz)
teardown()
# Here's our payoff idiom!
if __name__ == '__main__':
main()
我们现在有一个用于模块末尾的最终函数,如果我们将该模块作为主模块运行,它将运行该函数。
它将允许在不运行main
函数的情况下将模块及其函数和类导入其他脚本,并且还允许在从不同的@987654339 运行时调用模块(及其函数和类) @模块,即
import important
important.main()
This idiom can also be found in the Python documentation in an explanation of the __main__
module. 文字说明:
此模块表示(否则为匿名的)范围,其中 解释器的主程序执行——命令从 标准输入,来自脚本文件,或来自交互式提示。它 在这个环境中,惯用的“条件脚本”节 导致脚本运行:
if __name__ == '__main__': main()
【讨论】:
显然有 Twitch 频道的那个给出了最好的答案 :)【参考方案5】:if __name__ == "__main__"
是当脚本从(比如)命令行使用python myscript.py
之类的命令运行时运行的部分。
【讨论】:
为什么即使没有if __name__ == "__main__"
,文件helloworld.py
中只有print("hello world")
也可以使用命令python helloworld.py
运行?
当您运行python helloworld.py
时,它将运行整个脚本文件(无论您是否指定if __name__ == "__main__"
)。当您从不同的脚本中导入helloworld.py
时,只有在执行上有所不同。在这种情况下,if __name__ == "__main__"
代码块根本不会执行。【参考方案6】:
if __name__ == "__main__":
是做什么的?
__name__
是一个存在于所有命名空间中的全局变量(在 Python 中,全局实际上是指 module level 上的)。它通常是模块的名称(str
类型)。
然而,作为唯一的特殊情况,在你运行的任何 Python 进程中,如在 mycode.py 中:
python mycode.py
原本匿名的全局命名空间被分配了'__main__'
的值给它的__name__
。
因此,包括the final lines
if __name__ == '__main__':
main()
在 mycode.py 脚本的末尾,
当它是由 Python 进程运行的主要入口点模块时,
将导致您的脚本唯一定义的main
函数运行。
使用此构造的另一个好处:您还可以将代码作为模块导入另一个脚本中,然后在您的程序决定时运行 main 函数:
import mycode
# ... any amount of other code
mycode.main()
【讨论】:
【参考方案7】:这里对所讨论代码的机制有很多不同的看法,即“如何”,但对我来说,在我理解“为什么”之前,这一切都没有意义。这对新程序员应该特别有帮助。
获取文件“ab.py”:
def a():
print('A function in ab file');
a()
还有第二个文件“xy.py”:
import ab
def main():
print('main function: this is where the action is')
def x():
print ('peripheral task: might be useful in other projects')
x()
if __name__ == "__main__":
main()
这段代码实际上在做什么?
当你执行xy.py
,你import ab
。 import 语句在导入时立即运行模块,因此ab
的操作在xy
的其余部分之前执行。完成ab
后,继续使用xy
。
解释器跟踪使用__name__
运行的脚本。当您运行脚本时(无论您将其命名为什么),解释器都会将其称为 "__main__"
,使其成为运行外部脚本后返回的主脚本或“主”脚本。
从此"__main__"
脚本调用的任何其他脚本都将其文件名指定为其__name__
(例如__name__ == "ab.py"
)。因此,if __name__ == "__main__":
行是解释器的测试,以确定它是否正在解释/解析最初执行的“主”脚本,或者它是否暂时窥视另一个(外部)脚本。这让程序员可以灵活地让脚本在直接执行和外部调用时表现不同。
让我们逐步浏览上面的代码以了解发生了什么,首先关注未缩进的行以及它们在脚本中出现的顺序。记住这个函数——或者def
——块在被调用之前不会自己做任何事情。如果口译员自言自语,可能会说什么:
__name__
变量中调用它"__main__"
。
使用__name__ == "ab.py"
导入和打开文件。
哦,一个函数。我会记住的。
好的,函数a()
;我刚刚学到了。打印“ab 文件中的函数”。
文件结束;回"__main__"
!
哦,一个函数。我会记住的。
另一个。
函数x()
;好的,打印“周边任务:可能对其他项目有用”。
这是什么? if
声明。嗯,条件已经满足了(变量__name__
已经设置为"__main__"
),所以我进入main()
函数并打印'main function: this is where the action is'。
下面两行的意思是:“如果这是"__main__"
或'home' 脚本,请执行名为main()
的函数”。这就是为什么您会在顶部看到一个 def main():
块,其中包含脚本功能的主要流程。
为什么要实现这个?
还记得我之前说过的关于 import 语句的内容吗?当您导入一个模块时,它不仅会“识别”它并等待进一步的指令 - 它实际上会运行脚本中包含的所有可执行操作。因此,将脚本的内容放入 main()
函数可以有效地隔离它,将其隔离,以便在被另一个脚本导入时不会立即运行。
同样,会有例外,但通常的做法是main()
通常不会被外部调用。所以你可能想知道一件事:如果我们不调用main()
,我们为什么要调用脚本呢?这是因为许多人使用独立的函数来构建他们的脚本,这些函数被构建为独立于文件中的其余代码运行。然后在脚本正文的其他地方调用它们。这让我想到了这一点:
但是没有它代码也可以工作
是的,没错。这些单独的函数可以从不包含在main()
函数中的内嵌脚本中调用。如果您习惯于(就像我一样,在我编程的早期学习阶段)构建完全符合您需要的内联脚本,并且如果您再次需要该操作,您将尝试再次弄清楚...... . 好吧,你不习惯代码的这种内部结构,因为它构建起来更复杂,而且阅读起来也不那么直观。
但这是一个可能无法在外部调用其函数的脚本,因为如果这样做,它将立即开始计算和分配变量。如果您尝试重用一个函数,那么您的新脚本与旧脚本的相关性很可能会导致变量冲突。
在拆分独立函数时,您可以通过将它们调用到另一个脚本中来重用以前的工作。例如,“example.py”可能会导入“xy.py”并调用x()
,利用“xy.py”中的“x”函数。 (也许它是将给定文本字符串的第三个单词大写;从数字列表创建一个 NumPy 数组并将它们平方;或者去除 3D 表面的趋势。可能性是无限的。)
(顺便说一句,this question 包含@kindall 的答案,最终帮助我理解了 - 为什么,而不是如何。不幸的是,它被标记为 this one 的副本,我认为这是一个错误。 )
【讨论】:
【参考方案8】:当我们的模块(M.py
)中有某些语句时,我们希望在它作为 main(未导入)运行时执行,我们可以将这些语句(测试用例,打印语句)放在这个 @ 下987654322@块。
默认情况下(当模块作为主模块运行时,未导入)__name__
变量设置为 "__main__"
,当它被导入时,__name__
变量将获得不同的值,很可能是名称模块 ('M'
)。
这有助于一起运行模块的不同变体,并分离它们的特定输入和输出语句以及是否有任何测试用例。
简而言之,使用这个 'if __name__ == "main"
' 块来防止在导入模块时运行(某些)代码。
【讨论】:
+1 有时,像这个答案中提到的那样获得一个简短的单行是很好的:“简而言之,使用这个'if name ==“main”'阻止以防止在导入模块时运行(某些)代码。”。这很有用。谢谢!【参考方案9】:简单地说,__name__
是为每个脚本定义的变量,用于定义脚本是作为主模块运行还是作为导入模块运行。
如果我们有两个脚本;
#script1.py
print "Script 1's name: ".format(__name__)
和
#script2.py
import script1
print "Script 2's name: ".format(__name__)
执行script1的输出是
Script 1's name: __main__
执行script2的输出是:
Script1's name is script1
Script 2's name: __main__
如您所见,__name__
告诉我们哪个代码是“主”模块。
这很棒,因为您可以只编写代码而不必担心像 C/C++ 中的结构问题,如果文件没有实现“main”函数,那么它就无法编译为可执行文件,如果确实如此,然后它不能用作库。
假设您编写了一个出色的 Python 脚本,并实现了一大堆可用于其他目的的函数。如果我想使用它们,我可以只导入您的脚本并使用它们而不执行您的程序(假设您的代码仅在 if __name__ == "__main__":
上下文中执行)。而在 C/C++ 中,您必须将这些部分分成一个单独的模块,然后包含该文件。想象一下下面的情况;
箭头是导入链接。对于三个模块,每个模块都试图包含以前的模块代码,有六个文件(九个,计算实现文件)和五个链接。这使得很难将其他代码包含到 C 项目中,除非它被专门编译为库。现在为 Python 想象一下:
您编写一个模块,如果有人想使用您的代码,他们只需导入它,__name__
变量可以帮助将程序的可执行部分与库部分分开。
【讨论】:
C/C++ 图解错误:3 次相同的单元名(file1)。【参考方案10】:if __name__ == '__main__':
下的代码仅在模块作为脚本调用时才会执行。
例如,考虑以下模块my_test_module.py
:
# my_test_module.py
print('This is going to be printed out, no matter what')
if __name__ == '__main__':
print('This is going to be printed out, only if user invokes the module as a script')
第一种可能性:在另一个模块中导入my_test_module.py
# main.py
import my_test_module
if __name__ == '__main__':
print('Hello from main.py')
现在如果你调用main.py
:
python main.py
>> 'This is going to be printed out, no matter what'
>> 'Hello from main.py'
请注意,只执行my_test_module
中的***print()
语句。
第二种可能性:将my_test_module.py
作为脚本调用
现在,如果您将 my_test_module.py
作为 Python 脚本运行,则两个 print()
语句都将被执行:
python my_test_module.py
>>> 'This is going to be printed out, no matter what'
>>> 'This is going to be printed out, only if user invokes the module as a script'
更全面的解释可以阅读What does if __name__ == '__main__'
do in Python。
【讨论】:
【参考方案11】:让我们以更抽象的方式来看答案:
假设我们在x.py
中有这段代码:
...
<Block A>
if __name__ == '__main__':
<Block B>
...
当我们运行 x.py
时运行块 A 和 B。
但是当我们运行另一个模块时,只运行块 A(而不是 B),例如 y.py
,其中导入了 x.py
并从那里运行代码(就像 x.py
中的函数时一样)从y.py
调用)。
【讨论】:
我无法编辑帖子(如果需要更改,最少 6 个字符)。第 14 行有 'x.y' 而不是 'x.py'。【参考方案12】:简而言之,你需要知道几点:
import a
操作实际上运行了所有可以在a.py
中运行的内容,即a.py
中的每一行
由于第 1 点,您可能不希望在导入时所有内容都在 a.py
中运行
为了解决第 2 点中的问题,Python 允许您使用条件检查
__name__
是所有.py
模块中的隐式变量:
a.py
为import
ed 时,a.py
模块的__name__
的值设置为其文件名“a
”
当a.py
直接使用“python a.py
”运行时,__name__
的值设置为字符串__main__
-
基于Python如何为每个模块设置变量
__name__
的机制,你知道如何实现第3点吗?答案很简单,对吧?使用 if 条件:if __name__ == "__main__": // do A
python a.py
将运行部件// do A
和import a
将跳过部分// do A
-
您甚至可以根据您的功能需要输入 if
__name__ == "a"
,但很少这样做
Python 的特别之处在于第 4 点!其余的只是基本逻辑。
我一直在阅读此页面上的所有答案。我想说,如果你知道这个东西,你肯定会明白那些答案,否则你还是一头雾水。
【讨论】:
是的,理解第 1 点至关重要。由此,对这种机制的需求变得清晰。 钉了它,仍然对上面的答案感到完全困惑,但现在它非常清晰! 这是迄今为止最好的“可以理解”的答案。 这是最适合初学者的答案。不敢相信我不得不滚动页面的一半只是为了找到一个人在没有所有技术术语的情况下用实际术语清楚明确地解释这一点。 ⭐【参考方案13】:当您以交互方式运行 Python 时,本地变量 __name__
被赋值为 __main__
。同样,当您从命令行执行 Python 模块时,而不是将其导入另一个模块,它的__name__
属性被分配一个值__main__
,而不是模块的实际名称。通过这种方式,模块可以查看它们自己的__name__
值以确定它们是如何被使用的,无论是作为对另一个程序的支持还是作为从命令行执行的主应用程序。因此,以下习语在 Python 模块中很常见:
if __name__ == '__main__':
# Do something appropriate here, like calling a
# main() function defined elsewhere in this module.
main()
else:
# Do nothing. This module has been imported by another
# module that wants to make use of the functions,
# classes and other useful bits it has defined.
【讨论】:
【参考方案14】:考虑:
if __name__ == "__main__":
main()
它检查 Python 脚本的 __name__
属性是否为 "__main__"
。换句话说,如果程序本身被执行,属性将是__main__
,所以程序会被执行(这里是main()
函数)。
但是,如果您的 Python 脚本被模块使用,则if
语句之外的任何代码都将被执行,因此if \__name__ == "\__main__"
仅用于检查程序是否用作模块,因此决定是否运行代码。
【讨论】:
【参考方案15】:在解释任何关于if __name__ == '__main__'
的内容之前,了解__name__
是什么以及它的作用很重要。
什么是
__name__
?
__name__
是一个DunderAlias - 可以被认为是一个全局变量(可从模块访问),其工作方式类似于global
。
它是由type(__name__)
(产生<class 'str'>
)指示的字符串(如上所述的全局),并且是Python 3 和Python 2 版本的内置标准。
地点:
它不仅可以在脚本中使用,还可以在解释器和模块/包中找到。
口译员:
>>> print(__name__)
__main__
>>>
脚本:
test_file.py:
print(__name__)
导致__main__
模块或包:
somefile.py:
def somefunction():
print(__name__)
test_file.py:
import somefile
somefile.somefunction()
导致somefile
请注意,当在包或模块中使用时,__name__
采用文件名。没有给出实际模块或包路径的路径,但有自己的 DunderAlias __file__
,允许这样做。
你应该看到,__name__
,它是主文件(或程序)将总是返回__main__
,如果它是一个模块/包,或者任何东西运行其他一些 Python 脚本,将返回它所来自的文件的名称。
练习:
作为变量意味着它的值可以被覆盖(“可以”并不意味着“应该”),覆盖__name__
的值将导致缺乏可读性。因此,无论出于何种原因,都不要这样做。如果您需要一个变量,请定义一个新变量。
始终假定__name__
的值为__main__
或文件名。再次更改此默认值会导致更多混淆,认为它会做得很好,从而导致进一步的问题。
例子:
>>> __name__ = 'Horrify' # Change default from __main__
>>> if __name__ == 'Horrify': print(__name__)
...
>>> else: print('Not Horrify')
...
Horrify
>>>
在脚本中包含if __name__ == '__main__'
通常被认为是一种很好的做法。
现在回答
if __name__ == '__main__'
:
现在我们知道__name__
的行为变得更加清晰了:
if
是一个流控制语句,如果给定值为真,则包含将执行的代码块。我们已经看到__name__
可以采取
__main__
或从中导入它的文件名。
这意味着如果 __name__
等于 __main__
则该文件必须是主文件并且必须实际运行(或者它是解释器),而不是导入到脚本中的模块或包。
如果确实__name__
确实采用了__main__
的值,那么该代码块中的任何内容都将执行。
这告诉我们,如果运行的文件是主文件(或者您直接从解释器运行),那么必须执行该条件。如果它是一个包那么它不应该,并且值不会是__main__
。
模块:
__name__
也可以用在modules中来定义一个module的名字
变体:
还可以使用__name__
做其他不太常见但有用的事情,我将在这里展示一些:
仅当文件是模块或包时执行:
if __name__ != '__main__':
# Do some useful things
如果文件是主文件,则运行一个条件,如果不是,则运行另一个条件:
if __name__ == '__main__':
# Execute something
else:
# Do some useful things
您还可以使用它为包和模块提供可运行的帮助功能/实用程序,而无需精心使用库。
它还允许模块作为主脚本从命令行运行,这也非常有用。
【讨论】:
【参考方案16】:我认为最好深入浅出地打破答案:
__name__
:Python 中的每个模块都有一个特殊的属性,称为__name__
。
它是一个返回模块名称的内置变量。
__main__
:与其他编程语言一样,Python 也有一个执行入口点,即 main。 '__main__'
是***代码执行的范围的名称。基本上你有两种使用 Python 模块的方法:直接作为脚本运行它,或者导入它。当模块作为脚本运行时,其__name__
设置为__main__
。
因此,当模块作为主程序运行时,__name__
属性的值设置为__main__
。否则,__name__
的值将设置为包含模块的名称。
【讨论】:
【参考方案17】:当从命令行调用 Python 文件时,它是特殊的。这通常用于调用“main()”函数或执行其他适当的启动代码,例如命令行参数处理。
它可以用多种方式编写。另一个是:
def some_function_for_instance_main():
dosomething()
__name__ == '__main__' and some_function_for_instance_main()
我并不是说您应该在生产代码中使用它,但它可以说明if __name__ == '__main__'
没有什么“神奇”。
这只是在 Python 文件中调用 main 函数的约定。
【讨论】:
我会考虑这种糟糕的形式,因为您 1) 依赖副作用和 2) 滥用and
。 and
用于检查两个布尔语句是否都为真。由于您对and
的结果不感兴趣,因此if
语句更清楚地传达了您的意图。
撇开利用布尔运算符的短路行为作为流控制机制是否是不好的风格的问题不谈,更大的问题是这 没有回答问题全部.
@jpmc26 任何具有 Perl 或 javascript 背景的人都对这个习惯用法非常满意,使用 and
作为控制语句。我对此没有任何问题。另一个类似的习惯用法是使用or
设置默认值。例如,x = input("what is your name? ") or "Nameless Person"
。
@JohnHenckel 这不是 Perl 或 JavaScript。这不是 Python 习语。在 Python 中的布尔语句中间使用具有副作用的函数被认为是不好的形式。特别是在这种情况下,在这里使用and
绝对没有任何好处;该函数甚至不返回值。它只是让代码不那么明显。
@jpmc26 我正在努力寻找与您一致的权威来源。这是在某处提到的吗?例如,在 PEP8 中是否说我们应该避免使用 and
进行控制,或者使用 or
分配默认值?我试图用谷歌搜索它,但我找不到任何东西。【参考方案18】:
系统(Python 解释器)为源文件(模块)提供了许多变量。您可以随时获取它们的值,因此,让我们关注 __name__ 变量/属性:
当 Python 加载源代码文件时,它会执行其中找到的所有代码。 (请注意,它不会调用文件中定义的所有方法和函数,但会定义它们。)
不过,在解释器执行源代码文件之前,它会为该文件定义一些特殊变量; __name__ 是 Python 为每个源代码文件自动定义的特殊变量之一。
如果 Python 将此源代码文件作为主程序(即您运行的文件)加载,那么它将为该文件设置特殊的 __name__ 变量,使其具有值 "__main__" 。
如果这是从另一个模块导入的,__name__ 将设置为该模块的名称。
所以,在你的部分例子中:
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))
表示代码块:
lock = thread.allocate_lock()
thread.start_new_thread(myfunction, ("Thread #: 1", 2, lock))
thread.start_new_thread(myfunction, ("Thread #: 2", 2, lock))
只有当你直接运行模块时才会执行;如果另一个模块正在调用/导入代码块,则该代码块将不会执行,因为在该特定实例中 __name__ 的值将不等于“main”。
希望这会有所帮助。
【讨论】:
您好,您是少数解决多线程方面问题的人之一。我可以问你这个,如果我有“main”之外的代码并且没有封装在函数中会发生什么?这段代码会被从 main 启动的每个新线程一次又一次地执行吗?【参考方案19】:if __name__ == "__main__":
基本上是***脚本环境,它指定了解释器('我有最高优先级先执行')。
'__main__'
是***代码执行的范围的名称。当从标准输入、脚本或交互式提示中读取时,模块的 __name__
设置为等于 '__main__'
。
if __name__ == "__main__":
# Execute only if run as a script
main()
【讨论】:
【参考方案20】:考虑:
print __name__
上面的输出是__main__
。
if __name__ == "__main__":
print "direct method"
以上陈述为真,打印出“直接方法”。假设如果他们在另一个类中导入这个类,它不会打印 "direct method",因为在导入时,它会设置 __name__ equal to "first model name"
。
【讨论】:
【参考方案21】:您可以将文件用作脚本以及可导入模块。
fibo.py(一个名为fibo
的模块)
# Other modules can IMPORT this MODULE to use the function fib
def fib(n): # write Fibonacci series up to n
a, b = 0, 1
while b < n:
print(b, end=' ')
a, b = b, a+b
print()
# This allows the file to be used as a SCRIPT
if __name__ == "__main__":
import sys
fib(int(sys.argv[1]))
参考:https://docs.python.org/3.5/tutorial/modules.html
【讨论】:
【参考方案22】:原因
if __name__ == "__main__":
main()
主要是为了避免having code directly imported 引起的import lock 问题。如果您的文件被直接调用,您希望 main()
运行(这是 __name__ == "__main__"
的情况),但如果您的代码是导入的,那么导入器必须从真正的主模块输入您的代码以避免导入锁定问题。
副作用是您会自动登录支持多个入口点的方法。您可以使用main()
作为入口点来运行您的程序,但您不必。虽然setup.py
需要main()
,但其他工具使用备用入口点。例如,要将文件作为gunicorn
进程运行,您需要定义app()
函数而不是main()
。与setup.py
一样,gunicorn
导入您的代码,因此您不希望它在导入时执行任何操作(因为导入锁定问题)。
【讨论】:
很好了解导入锁。您能否解释一下签署一种[...]更多部分的方法? @Wolf:当然。我添加了一些关于多入口点方法的句子。【参考方案23】:Python 中的每个模块都有一个名为__name__
的属性。直接运行模块时__name__
属性的值为__main__
,如python my_module.py
。否则(就像你说import my_module
)__name__
的值是模块的名称。
小例子简单解释一下。
脚本 test.py
apple = 42
def hello_world():
print("I am inside hello_world")
if __name__ == "__main__":
print("Value of __name__ is: ", __name__)
print("Going to call hello_world")
hello_world()
我们可以直接这样执行
python test.py
输出
Value of __name__ is: __main__
Going to call hello_world
I am inside hello_world
现在假设我们从另一个脚本调用上述脚本:
脚本external_calling.py
import test
print(test.apple)
test.hello_world()
print(test.__name__)
当你执行这个时,
python external_calling.py
输出
42
I am inside hello_world
test
因此,上面的内容是不言自明的,当您从另一个脚本调用 test 时,test.py
中的 if 循环 __name__
将不会执行。
【讨论】:
【参考方案24】:此答案适用于学习 Python 的 Java 程序员。 每个 Java 文件通常都包含一个公共类。您可以通过两种方式使用该类:
从其他文件调用类。您只需在调用程序中导入即可。
单独运行该类,用于测试目的。
对于后一种情况,该类应包含一个公共静态 void main() 方法。在 Python 中,此目的由全局定义的标签 '__main__'
提供。
【讨论】:
【参考方案25】:如果这个.py文件被其他.py文件导入,“if语句”下的代码将不会被执行。
如果这个 .py 由 python this_py.py
在 shell 下运行,或者在 Windows 中双击。 “if语句”下的代码将被执行。
它通常是为测试而编写的。
【讨论】:
【参考方案26】:如果 Python 解释器正在运行特定模块,则 __name__
全局变量将具有值 "__main__"
:
def a():
print("a")
def b():
print("b")
if __name__ == "__main__":
print ("you can see me")
a()
else:
print ("You can't see me")
b()
当你运行这个脚本时,它会打印出you can see me。
一个
如果你导入这个文件,说 A 到文件 B,然后执行文件 B,那么文件 A 中的if __name__ == "__main__"
变为 false,所以它打印 You can't see me。
b
【讨论】:
【参考方案27】:如果 name == 'main':
我们是否经常查看__name__ == '__main__':
。
它检查一个模块是否正在被导入。
换句话说,if
块中的代码只有在代码直接运行时才会被执行。这里directly
表示not imported
。
让我们看看它使用打印模块名称的简单代码做了什么:
# test.py
def test():
print('test module name=%s' %(__name__))
if __name__ == '__main__':
print('call test()')
test()
如果我们直接通过python test.py
运行代码,模块名称为__main__
:
call test()
test module name=__main__
【讨论】:
【参考方案28】:简单来说:
您在if __name__ == "__main__":
下看到的代码只会在您的 python 文件作为“python example1.py”执行时才会被调用。
但是,如果您希望将您的 python 文件“example1.py”作为模块导入以使用另一个 python 文件,例如“example2.py”,if __name__ == "__main__":
下的代码将不会运行或生效。
【讨论】:
【参考方案29】:所有答案几乎都解释了该功能。但我将提供一个它的用法示例,这可能有助于进一步理清这一概念。
假设您有两个 Python 文件,a.py 和 b.py。现在,a.py 导入 b.py。我们运行 a.py 文件,其中首先执行“import b.py”代码。在剩下的 a.py 代码运行之前,b.py 文件中的代码必须运行完整。
在 b.py 代码中,有一些代码是该文件 b.py 独有的,我们不希望导入 b.py 文件的任何其他文件(除了 b.py 文件)运行它。
这就是这行代码检查的内容。如果它是运行代码的主文件(即 b.py),在这种情况下它不是(a.py 是运行的主文件),那么只有代码被执行。
【讨论】:
【参考方案30】:如果您是初学者,您现在可能唯一需要的答案是对于一个简单的脚本来说,这段代码是不必要的。仅当您希望能够 import
您的脚本(或 unpickle
等;请参阅此处的其他答案以了解其他一些非初学者场景)时,它才有用。
稍微详细一点,假设您有一个简单的脚本fib.py
(改编自this answer):
# XXX FIXME: useless (see below)
if __name__ == "__main__":
n = int(input('Write a number: '))
a, b = 0, 1
while b < n:
a, b = b, a+b
print('Fibonacci number %i: %i' % (n, b))
现在,如果您只是运行python fib.py
,它就可以正常工作。但是__name__
在这种情况下总是"__main__"
,所以这个条件实际上是不必要的。脚本可以简化为
n = int(input('Write a number: '))
a, b = 0, 1
while b < n:
a, b = b, a+b
print('Fibonacci number %i: %i' % (n, b))
现在,你不能import fib
使用新版本,但如果你一开始不打算这样做,这个版本实际上更好,因为它更简单、更清晰。
如果你确实希望能够import fib
,那么第一个版本也是没用的,因为有用的代码在你import
这个文件时不会运行的部分(在这种情况下,__name__
将不是 "__main__"
)。在这种情况下,正确的设计是重构代码,以便在您拥有import
ed 之后,您可以在想要运行的时候运行该功能。
def main():
n = int(input('Write a number: '))
a, b = 0, 1
while b < n:
a, b = b, a+b
print('Fibonacci number %i: %i' % (n, b))
if __name__ == "__main__":
main()
现在,如果您import fib
,则不会执行对main()
的调用;但是当你运行python fib.py
时,它会。
实际上,更好的设计仍然是将可重用部分(实际计算)与用户可见的输入/输出隔离开来:
def fibn(n: int) -> int:
a, b = 0, 1
while b < n:
a, b = b, a+b
return b
def main() -> None:
n = int(input('Write a number: '))
print('Fibonacci number %i: %i' % (n, fibn(n)))
if __name__ == "__main__":
main()
现在,您可以from fib import fibn
并从执行此import
的代码中调用fibn()
函数。
(我调用函数fibn()
只是为了让这个例子更清楚。在现实生活中,你可能会调用它fib()
并做from fib import fib
。)
同样,如果你想重用它,你可以import
并调用main
函数。
回到问题中的代码,我同样会将代码从if
移到一个函数中,以便调用者可以根据需要调用该函数。
def main():
lock = thread.allocate_lock()
thread.start_new_thread(myfunction, ("Thread #: 1", 2, lock))
thread.start_new_thread(myfunction, ("Thread #: 2", 2, lock))
if __name__ == "__main__":
main()
这改变了lock
变量的范围;如果周围的代码需要访问它,您需要将其设为global
(或者,也许更好,将main
重构为return lock
,并让调用者在其自己的局部变量中捕获该值)。
【讨论】:
我保持示例代码简单以避免分心;在现实生活中,您需要处理input
返回的不是数字的东西,当然,可能应该优化现实生活中的斐波那契函数以避免每次调用时重新计算序列的开头。
n: int
和 -> int
类型注解需要 Python >= 3.6
斐波那契数列的排序规则并非一成不变。通常的约定是第一个数字是 0,但在数学上通常是 F(0)
。您可能会将简单脚本的结果视为一个错误,并相应地对其进行修复。以上是关于if __name__ == "__main__": 做啥?的主要内容,如果未能解决你的问题,请参考以下文章
if __name__ == "__main__": 做啥?
unittest: if __name__ == "__main__" 理解
python if __name__ == "__main__"
if __name__ == "__main__" 语法错误