Python 理解Main函数
在很多python脚本中在最后的部分会执行一个判断语句if name == “main:",之后还可能会有一些执行语句。那添加这个判断的目的何在?
先来看一个例子
#hello.py
def foo():
str="function"
print(str);
if __name__=="__main__":
print("main")
foo()
# 在Cmd中运行结果
# C:\work\python\divepy>python hello.py
# main
# function
# 在Python Shell中运行结果
# >>> import hello
# >>> hello.foo()
# function
# >>> hello.__name__
# 'hello'
# >>>
可以发现这个内置属性__name__自动的发生了变化。这是由于当你以单个文件运行时,name__便是__main,当你以模块导入使用时,这个属性便是这个模块的名字。
在python编译器读取源文件的时候会执行它找到的所有代码,而在执行之前会根据当前运行的模块是否为主程序而定义变量__name__的值为__main__还是模块名。因此,该判断语句为真的时候,说明当前运行的脚本为主程序,而非主程序所引用的一个模块。这在当你想要运行一些只有在将模块当做程序运行时而非当做模块引用时才执行的命令,只要将它们放到if name == “main:“判断语句之后就可以了。
# file one.py
def func():
print("func() in one.py")
print("top-level in one.py")
if __name__ == "__main__":
print("one.py is being run directly")
else:
print("one.py is being imported into another module")
# file two.py
import one # start executing one.py
print("top-level in two.py")
one.func()
if __name__ == "__main__":
print("two.py is being run directly")
else:
print("two.py is being imported into another module")
# 当运行python one.py,输出:
# top-level in one.py
# one.py is being run directly
# 当运行python two.py,输出:
# top-level in one.py
# one.py is being imported into another module
# top-level in two.py
# func() in one.py
# two.py is being run directly