在Python中定义main()

Question

我经常在Python程序中看到以下代码

if __name__ == '__main__':
    main()

我在谷歌代码上参加一个Python课程，上面说这是标准的模板代码。

我真的需要在我所有的脚本中都写这样的代码吗？

这段代码能给我的程序带来什么功能呢？

Answer 1

在导入文件的时候，这个很有用。你可以把这个Python文件当作一个独立的程序来运行，或者把里面的一些部分导入到其他程序中使用。

Answer 2

不，你不一定要这样做，但对于单元测试之类的事情，这样做非常有价值。

你可以在每个Python文件里创建一个main函数，这样如果你直接运行这个文件，__name__就会被设置为"__main__"，然后就会对相关的代码进行一系列测试。

如果你只是从另一个程序中正常导入这个文件，那就不会发生这种情况，因为这时__name__会被设置为一个不同的值。