測試 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()