python - 为什么函数内的 globals() 不能更新交互式 shell 中的变量？

Question

今天给自己写了一个“别名导入函数”，因为需要写一个脚本对不同的python文件做变量值检查。

# filename: zen_basic.py

import importlib
def from_module_import_alias(module_name, var_name, alias):
    """ equal to from module import a as b """
    agent = importlib.import_module(module_name)
    globals()[alias] = vars(agent)[var_name]

奇怪的是，如果我启动 Python 交互式 shell，我无法使用此函数导入内容。但是通过在函数之外使用它的内容，它可以工作。

>>> from zen_basic import *
>>> module_name = 'autor'
>>> var_name = 'food'
>>> alias = 'fd'
>>> from_module_import_alias(moduele_name, var_name, alias)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
NameError: name 'moduele_name' is not defined
>>> from_module_import_alias(module_name, var_name, alias)
>>> fd
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
NameError: name 'fd' is not defined
>>> agent = importlib.import_module(module_name)
>>> globals()[alias] = vars(agent)[var_name]
>>> fd
'cake'
>>>

之后我又做了 3 个实验:

1. • 执行python -i test.py
   • 在交互式 shell 中导入 zen_basic
   • 调用 from_module_import_alias 函数，失败。
2. • 执行 python -i zen_basic.py
   • 调用from_module_import_alias函数，成功。
3. • 将import zen_basic 和调用from_module_import_alias 函数 的代码添加到test.py
   • 执行python -i test.py，成功

在Python交互shell中直接使用from_module_import_alias函数失败是什么原因？

Answer 1

您正在更新错误的全局变量。您正在更新 zen_basic 模块的全局变量，而不是 __main__ 模块(脚本或交互式解释器的 namespace )。 globals() 函数始终返回定义代码的模块的全局变量，而不是调用函数的模块的全局变量。

您必须检索调用框架的全局变量。请注意，很少建议修改调用框架的全局变量，但如果必须，则可以使用 sys._getframe() function 检索调用框架。 :

import sys

def from_module_import_alias(module_name, var_name, alias):
    """ equal to from module import a as b """
    agent = importlib.import_module(module_name)
    calling_globals = sys._getframe(1).f_globals
    calling_globals[alias] = vars(agent)[var_name]

在您的实验中，python -i zen_basic.py 运行 zen_basic 作为主脚本入口，因此 globals() 引用 __main__ 模块。