Python测试“ __main__”以避免意外的代码执行

示例

__name__在执行代码之前,最好测试调用程序的变量。

import sys

def main():
    # 您的代码从这里开始

    # 不要忘记提供返回码
    return 0

if __name__ == "__main__":
    sys.exit(main())

使用此模式可确保仅在您期望的时间执行代码。例如,当您明确运行文件时:

python my_program.py

但是,如果您决定import在另一个程序中使用文件(例如,如果您将其编写为库的一部分),则会带来好处。然后import,您就可以使用文件了,__main__陷阱将确保不会意外执行任何代码:

# 一个新的程序文件
import my_program        # main() is not run

# But you can run main() explicitly if you really want it to run:
my_program.main()