测试 main 以避免意外的代码执行
最好在执行代码之前测试调用程序的 __name__
变量。
import sys
def main():
# Your code starts here
# Don't forget to provide a return code
return 0
if __name__ == "__main__":
sys.exit(main())
使用此模式可确保你的代码仅在你预期时执行; 例如,当你显式运行文件时:
python my_program.py
但是,如果你决定在另一个程序中创建文件(例如,如果你将其作为库的一部分编写),则会带来好处。然后你可以知道你的文件,而 __main__
陷阱将确保没有意外执行代码:
# A new program file
import my_program # main() is not run
# But you can run main() explicitly if you really want it to run:
my_program.main()