Question

我有一个模块，我需要导入并更改该模块的导入实例中的特定变量值，然后执行它。

请注意由于遗留原因，我无法对导入的模块进行任何更改。

以下是我要做的事情：

说我要导入的模块，a.py，看起来像

var1 = 1
var2 = 2

if __name__ == '__main__':
    print var1+var2

现在我在调用者脚本b.py中导入它。我正在尝试导入它，更改变量var1的值并使用runpy将其作为主程序运行，如下所示。

import runpy
import a

a.var1 = 2
result = runpy._run_module_as_main(a.__name__)

但是这只会将结果打印为3，而不是预期的4。

任何其他方法来实现这一点（除了使用runpy）而不改变a.py中的任何内容？只要我没有在导入的模块中进行更改，就可以打开任何第三方模块。

我正在使用python 2.6

Answer 1

以下是我在评论中提出的具体实现：

#File temp.py
var1 = 1
var2 = 2

if __name__ == '__main__':
    print var1+var2

我通过另一个文件run.py调用它。关键是要有另一个dict类似的类，它可以抵御模块在导入/运行它时可能会改变的变量的任何变化。该字典包含您想要更改其值的变量。

#File: run.py
class UpdatedDict(dict):
    def __setitem__(self, key, value):
        if key != "var1":
            super(UpdatedDict, self).__setitem__(key, value)

u = UpdatedDict({"var1": 10, '__name__': '__main__'})
exec open("temp.py").read() in u  #u here is the globals() for the module.

<强>输出

~/Temp$ python run.py
12  #Yay!

注意：这是一个快速而又脏的实现，只是为了向您展示可以执行此操作的方式。我建议通过runpy并使这段代码看起来更健壮。它现在的方式，它是 hackish

的正确定义

如何导入模块并更改模块内的变量值并执行它

1 个答案: