导入模块:__ main__ vs import作为模块

Question

作为前言,我想我可能已经想出如何使这个代码工作(基于导入后更改模块变量),但我的问题是关于为什么会发生以下行为,所以我可以理解将来不做什么.

我有三个文件.第一个是mod1.py:

# mod1.py

import mod2

var1A = None

def func1A():
    global var1
    var1 = 'A'
    mod2.func2()

def func1B():
    global var1
    print var1

if __name__ == '__main__':
    func1A()

接下来我有mod2.py:

# mod2.py

import mod1

def func2():
    mod1.func1B()

最后我有driver.py:

# driver.py

import mod1

if __name__ == '__main__':
    mod1.func1A()

如果我执行命令,python mod1.py则输出为None.基于我上面引用的链接,似乎mod1.py导入__main__和mod1.py导入之间存在一些区别mod2.py.因此,我创造了driver.py.如果我执行命令,python driver.py那么我得到预期的输出:A.我有点看到差异,但我并没有真正看到它的机制或原因.这是怎么发生的？似乎违反直觉的是,同一模块将存在两次.如果我执行python mod1.py,是否可以访问__main__版本mod1.py中的变量而不是导入的版本中的变量mod2.py？

Answer 1

该__name__变量始终包含模块的名称,除非文件已作为脚本加载到解释器中.然后将该变量设置为字符串'__main__'.

毕竟,脚本然后作为整个程序的主文件运行,其他一切都是由该主文件直接或间接导入的模块.通过测试__name__变量,您可以检测文件是作为模块导入还是直接运行.

在内部,模块被赋予一个命名空间字典,该字典存储为每个模块的元数据的一部分sys.modules.主文件,即执行的脚本,存储在与...相同的结构中'__main__'.

但是当您将文件作为模块导入时,python首先sys.modules查看该模块是否已经导入过.因此,import mod1意味着我们先来看看在sys.modules为mod1模块.如果mod1还没有,它将创建一个带有命名空间的新模块结构.

因此,如果你们都mod1.py作为主文件运行,然后将其作为python模块导入,它将获得两个命名空间条目sys.modules.一个'__main__',然后是'mod1'.这两个名称空间完全独立.您的全球var1存储在sys.modules['__main__'],但func1B期待中sys.modules['mod1']的var1,它在哪里None.

但是当你使用时python driver.py,它将成为程序driver.py的'__main__'主文件,并且mod1只会导入到sys.modules['mod1']结构中一次.这一次,func1A存储var1在sys.modules['mod1']结构中,这就是所func1B发现的.