python - 修改Python导入脚本的命名空间

Question

我希望我的模块中有一个函数能够访问和更改导入它的脚本的本地命名空间。这将启用如下功能:

>>> import foo
>>> foo.set_a_to_three()
>>> a
3
>>>

这在 Python 中可行吗？

Answer 1

@omz 慷慨地提供了答案在 Slack 团队中:

import inspect
def set_a_to_three():
    f = inspect.currentframe().f_back
    f.f_globals['a'] = 3

这提供了优于 __main__ 解决方案的优势，它可以在多个级别的导入中工作，例如，如果 a 导入 b 导入我的module, foo, foo 可以修改 b 的全局变量，而不仅仅是 a 的(我认为)

不过，如果我没理解错的话，修改本地命名空间比较复杂。正如其他人指出的那样:

It breaks when you call that function from within another function. Then the next namespace up the stack is a fake dict of locals, which you cannot write to. Well, you can, but writes are ignored.

如果有更可靠的解决方案，将不胜感激。