【python】if __name__ == 'main': 的作用和原理


       第一种解释

      在大多数编排得好一点的脚本或者程序里面都有这段if __name__ == 'main': ,虽然一直知道他的作用,但是一直比较模糊,收集资料详细理解之后与打架分享。

       1、这段代码的功能

      一个python的文件有两种使用的方法,第一是直接作为脚本执行,第二是import到其他的python脚本中被调用(模块重用)执行。因此if __name__ == 'main': 的作用就是控制这两种情况执行代码的过程,在if __name__ == 'main': 下的代码只有在第一种情况下(即文件作为脚本直接执行)才会被执行,而import到其他脚本中是不会被执行的。

      举个例子,下面在test.py中写入如下代码:

 print "I'm the first."

           if __name__=="__main__":

                   print "I'm the second."

  并直接执行test.py,结果如下图,可以成功print两行字符串。即,if __name__=="__main__": 语句之前和之后的代码都被执行

                                                      【python】if __name__ == 'main': 的作用和原理


  然后在同一文件夹新建名称为import_test.py的脚本,只输入如代码: import test  执行import_test.py脚本,输出结果如下:    

 【python】if __name__ == 'main': 的作用和原理

     只输出了第一行字符串。即,if __name__=="__main__": 之前的语句被执行,之后的没有被执行

      2、运行的原理

     每个python模块(python文件,也就是此处的test.py和import_test.py)都包含内置的变量__name__,当运行模块被执行的时候,__name__等于文件名(包含了后缀.py);如果import到其他模块中,则__name__等于模块名称(不包含后缀.py)。而“__main__”等于当前执行文件的名称(包含了后缀.py)。进而当模块被直接执行时,__name__ == 'main'结果为真。

      同样举例说明,我们在test.py脚本的if __name__=="__main__":之前加入print __name__,即将__name__打印出来。文件内容和结果如下,

                                                   【python】if __name__ == 'main': 的作用和原理

       可以看出,此时变量__name__的值为"__main__";
       再执行import_test.py,模块内容和执行结果如下:

 【python】if __name__ == 'main': 的作用和原理

      此时,test.py中的__name__变量值为test,不满足__name__=="__main__"的条件,因此,无法执行其后的代码。




第二种解释

 Python中,一个.py文件就是一个模块,一般情况下,模块的名字就是文件名(不包括扩展名.py)。

 全局变量__name__存放的就是模块的名字。

 特殊情况就是,当一个模块作为脚本执行时或者在交互式环境中,如IpythonPython自带的shell等直接运行代码,__name__的值不再是模块名,而是__main____main__是顶层代码执行作用域的名字。

 if __name__ == "__main__":

     # execute only if run as a script

     main()

      也就是说,通过__name__的值,我们可以判断出该模块是作为脚本正在执行还是被其他模块导入,而根据这个判断,我们就可以选择性地执行代码。将只有模块作为脚本执行时的代码放入到

 if __name__ == '__main__':

 条件中,比如执行单元测试等。

 而当模块被其他模块通过import等语句导入时,Python解释器会首先检查模块注册表sys.modules中是否有该模块,如果有,则直接利用存在的模块对象的名字空间中的变量;如果没有,创建一个空的模块对象,将其添加到sys.modules中,在以该模块名为名的名字空间下执行该模块,除了if __name__ == '__main__': 条件中的代码,而被执行代码中定义的变量在该名字空间下都是可访问的。

 



第三种解释

 写小模块(比如内含几个工具函数,几个常用的类)的时候加上这条方便调试,不需要打开第二个py脚本,再把它import,再balabala,直接在最底下写测试的几行代码就行。