导入模块:uuu main_uuuvs import as modu

2024-09-28 21:53:41 发布

您现在位置:Python中文网/ 问答频道 /正文

首先,我想我可能已经弄清楚了如何使代码工作(基于Changing module variables after import),但我的问题是,为什么会发生以下行为,以便我能够理解将来不应该做什么。

我有三份文件。第一个是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__导入与mod2.pymod1.py导入之间存在一些区别。因此,我创建了driver.py。如果我执行命令python driver.py,那么我将获得预期的输出:A。我有点明白其中的区别,但我并没有真正明白其中的机制和原因。为什么会这样?同一个模块可能存在两次,这似乎有悖常理。如果我执行python mod1.py,是否可以访问mod1.py版本中的变量,而不是mod2.py导入的版本中的变量?


Tags: namepyimportnoneifmaindefdriver
2条回答

关于可选地将模块用作主脚本的实用解决方案-支持一致的交叉导入:

解决方案1:

例如,在Python的pdb模块中,当执行为__main__(在末尾)时,如何通过导入自身将其作为脚本运行:

#! /usr/bin/env python
"""A Python debugger."""
# (See pdb.doc for documentation.)
import sys
import linecache

...

# When invoked as main program, invoke the debugger on a script
if __name__ == '__main__':
    import pdb
    pdb.main()

只是我建议将__main__启动重新组织到脚本的开头,如下所示:

#! /usr/bin/env python
"""A Python debugger."""
# When invoked as main program, invoke the debugger on a script
import sys
if __name__ == '__main__':        
    ##assert os.path.splitext(os.path.basename(__file__))[0] == 'pdb'
    import pdb
    pdb.main()
    sys.exit(0)

import linecache
...

这样模块体就不会被执行两次——这是“昂贵的”,不受欢迎的,有时是至关重要的。

解决方案2:

在较少见的情况下,需要公开实际的脚本模块__main__,甚至直接作为实际的模块别名(mod1):

# mod1.py    
import mod2

...

if __name__ == '__main__':
    # use main script directly as cross-importable module 
    _mod = sys.modules['mod1'] = sys.modules[__name__]
    ##_modname = os.path.splitext(os.path.basename(os.path.realpath(__file__)))[0]
    ##_mod = sys.modules[_modname] = sys.modules[__name__]
    func1A()

已知缺点:

  • reload(_mod)失败
  • pickle类需要额外的映射来取消绑定(find_global..)

变量__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']结构中一次。这一次,func1Avar1存储在sys.modules['mod1']结构中,这就是func1B将找到的。

相关问题 更多 >