python: jak sprawdzić, czy plik został wykonany jako import, czy główny skrypt?

Question

Piszę plik Pythona mylib.py

chciałbym mylib.py coś zrobić w oparciu o sys.argv jeśli jest wykonywane jako skrypt. Ale jeśli jest importowany z jakiegoś innego skryptu, nie chcę tego robić.

Jak sprawdzić, czy mój plik python jest importowany lub jest to skrypt główny?

(widziałem jak to zrobić wcześniej, ale zapomniałem.)

Answer 1

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

Oto documentation on __main__.

Zwykle kod ten znajduje się na dole modułu, a jednym z typowych sposobów na utrzymanie kodu w czystości jest utworzenie funkcji main(), która wykonuje całą pracę i wywołuje tę funkcję tylko w warunku.

Answer 2

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

Pakiety mogą również zawierać moduł __main__, który jest wykonywany, gdy robisz python -m foo (lub wykonać zipfile zawierający pakiet).

Answer 3

Używając (wprowadzanie sprawozdań, które mają być wykonywane tylko wtedy, gdy moduł pracuje jako main, nie importowane)

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

Generalnie różne Sprawozdanie musi być umieszczony w tym „jeśli” bloku jak moduł specyficzny doctest wezwanie lub druk statements.The rzeczą jest domyślnie (gdy działa jako główny) do „__name__” zmienna jest ustawiona na „__main__”, a inaczej (jeśli importowane) zmienna __name__ otrzyma inną wartość, najprawdopodobniej nazwę modułu.