python: cómo saber si el archivo se ejecutó como importación frente a script principal?

Question

Estoy escribiendo un archivo de Python mylib.py

me gustaría mylib.py para hacer algo basado en sys.argv si está siendo ejecutado como un script. Pero si se importa desde algún otro script, no quiero que haga eso.

¿Cómo puedo saber si mi archivo python se está importando o si es un script principal?

(he visto cómo hacer esto antes, pero se me olvidó.)

Answer 1

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

Aquí es el documentation on __main__.

Normalmente este código se coloca en la parte inferior de un módulo, y una forma común de mantener su código limpio es crear una función main() que hace todo el trabajo, y solo llama a esa función dentro del condicional.

Answer 2

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

Los paquetes también pueden contener __main__ módulo, que se ejecuta cuando se hace python -m foo (o ejecutar archivo zip que contiene el paquete).

Answer 3

Mediante el uso de (la colocación de las declaraciones que desea ser ejecutado sólo cuando el módulo se está ejecutando como main, no importadas)

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

declaraciones Generalmente diferentes tienen tiene que ser colocado en este bloque 'si' como, específico del módulo de doctest llamada o de impresión cosa statements.The es por defecto (cuando se ejecuta como principal) del '__name__' variable se establece en "__main__", y de otra manera (si importados) la variable __name__ obtendrá un valor diferente, probablemente el nombre del módulo.