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