python: làm cách nào để biết liệu tệp có được thực thi khi nhập so với tập lệnh chính không?

Question

Tôi đang viết một tệp python mylib.py

Tôi muốn mylib.py thực hiện điều gì đó dựa trên sys.argv nếu nó được thực thi dưới dạng tập lệnh. Nhưng nếu nó được nhập từ một số tập lệnh khác, tôi không muốn nó làm điều đó.

Làm cách nào để biết tệp python của tôi đang được nhập hay là tập lệnh chính?

(Tôi đã nhìn thấy làm thế nào để làm điều này trước đây, nhưng tôi quên.)

Answer 1

if __name__ == '__main__': 
    # this was run as a main script 

Đây là documentation on __main__.

Thông thường mã này được đặt ở cuối mô-đun và cách thông thường để giữ mã của bạn là tạo một hàm main() thực hiện tất cả công việc và chỉ gọi hàm đó bên trong điều kiện.

Answer 2

if __name__ == '__main__': 
    # goes here only when module is being executed directly 

gói cũng có thể chứa __main__ module, được thực hiện khi bạn làm python -m foo (hoặc thực hiện zipfile chứa gói).

Answer 3

Bằng cách sử dụng (đặt những điều khoản mà bạn muốn được thực hiện chỉ khi các module đang chạy như main, không được nhập)

if __name__ == "__main__": 
     # this was run as a main script 

tuyên bố chung khác nhau có phải được đặt trong khối này 'nếu' như thế nào, module cụ thể doctest cuộc gọi hoặc in điều statements.The là theo mặc định (khi chạy là chính) của __name__ 'biến được thiết lập để "__main__", và nếu không (nếu nhập khẩu) biến số __name__ sẽ nhận được một giá trị khác, có thể là tên của mô-đun.