将命令行参数传递给runpy

Question

我有两个文件，其中一个有我关心的副作用，它发生在if __name__ == "__main__"保护中：

# a.py
d = {}
if __name__ == "__main__":
    d['arg'] = 'hello'

第二个文件导入第一个文件(使用runpy)并打印字典：

# b.py
import runpy
m = runpy.run_module('a', run_name='__main__')
print(m['d'])  # {'arg': 'hello'}

到目前为止，这是可行的。但现在我想将第一个文件更改为接受命令行参数：

import sys
d = {}
if __name__ == "__main__":
    d['arg'] = process(sys.argv[1])

问题是，process()是由其他人编写的，不在我的控制范围内，但我仍然希望在更新后的字典d被“处理”。

如何在调用runpy之前模拟sys.argv，或者以其他方式将值提供给a.py

Answer 1

能这么简单吗？我几乎是偶然发现了这个解决方案。

$ python3 b.py foo
{'arg': 'foo'}

即使从命令行调用b.py，似乎也会将sys.argv[1]传递给a。

编辑

下面是我的shell中的一个会话示例：

me@desktop$ cat a.py b.py
import sys
d = {}
filename = sys.argv[1]
if __name__ == "__main__":
    d['result'] = filename + ' world'


# usage:
# python3 b.py filename
# 'filename' is passed in to the a module
import runpy
aa = runpy.run_module('a', run_name='__main__')
print(aa['d'])  # {}

me@desktop$ python3 b.py hello
{'result': 'hello world'}