Python 的 `if __name__ == '__main__':` 语句有什么作用？

if __name__ == "__main__": 会做什么？

__name__ 保护是保护用户的常见 Python 习惯用法当他们无意时意外调用脚本。如果您将脚本导入到另一个脚本中或将其序列化为 pickle，则忽略此防护可能会导致意外行为。

它是如何工作的

当解释器读取时一个源文件，它做了两件事：

• 设置特殊变量，例如__name__
• 执行文件中找到的代码

如果将模块作为主程序运行，则 __name__ 设置为“__main__”。否则，当您导入模块时，__name__ 将设置为模块的名称。

__name__ 防护的工作方式如下：

• 如果 __name__ == '__main__' (main程序）：

  • 执行guard body内的代码（通常函数)。

• 如果 __name__ != '__main__' (导入模块):

  • 跳过守卫内的代码body.

代码示例

让我们检查以下代码：

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")

• 导入：数学导入时不带
• 主程序：如果代码作为主程序运行，它将在 __name__ 保护中打印“Function A”和“Function B 10.0”。
• 导入的模块：如果代码作为常规模块导入，则守卫将被跳过，函数不会被执行。

为什么使用它？

这个习惯用法允许你编写可以被执行的 .py 文件既可以用作独立模块，也可以用作作为主程序运行的脚本。其有用性的一些示例：

• 具有演示模式的库：具有用于单元测试或运行演示的脚本模式的库。
• 单元测试: 测试框架可能会将 .py 文件作为模块导入，需要 __name__ 防护来阻止脚本
• API暴露：作为独立程序运行时为高级用户提供API的模块。
• 多功能代码：它很优雅通过导入其模块来运行脚本，并且 __name__ 防护有助于此。

到总而言之， if __name__ == '__main__' 保护可以省略，但建议用于多功能性、错误保护，并防止导入或序列化代码时出现意外行为。

以上是Python 的 `if __name__ == '__main__':` 语句有什么作用？的详细内容。更多信息请关注PHP中文网其他相关文章！