包装器 function 从文件中读取 arguments 然后传递到 python3 中的子例程

Question

我有一个运行多个子例程的包装器 function。 而是定义包装器 function 以获取将传递给每个子例程的数百个输入，我希望包装器 function 读取包含所有 ZDBC11CAA5BDA99F77E6FB4DABD882E7 的输入文件传递给每个子例程

例如，给定一个名为eg_input.py的输入文件：

## example of argument file

a = [1]
b = [2]
c = [3]
d = [4]
e = [5]
f = [6]

我想要一个看起来像这样的包装器 function：

def wrap_fun(argument_file):
    
    ## read in the argument file
    exec(open(argument_file).read())


    ## run first sub routine
    sub_fun1(fun1_parma1=a, fun1_param2=b, fun1_param3=c)

    ## run second sub routine
    sub_fun2(fun2_param1=d, fun2_param2=e, fun2_param3=f)

    ## run third sub routine
    ## note some arguments are shared between subroutines 
    sub_fun2(fun3_param1=a, fun3_param2=f)
    
    ## etc...

    return()

这样我就可以运行如下脚本：

wrap_fun(eg_input.py)

但是，当我运行与上述类似的代码时，子程序无法使用 arguments。

你会注意到我在上面的例子中使用了exec() function 但我并不坚持使用它，即如果有一个pythonic方法来完成这个我更喜欢那个解决方案。

我对这个问题的动机是能够轻松生成各种输入文件（同样每个输入文件包含 100 个参数）并观察更改每个参数如何影响wrap_fun()的结果。

谢谢！

Answer 1

您可以在exec中为全局变量提供字典。 在执行的代码中定义的任何全局变量都到了那里。 现在，当您调用函数时，请从该字典中读取值。 由于 class 实例已经有一个字典，您可以将包装器编写为 class 以更新文件中的 class 实例，然后您可以编写任意数量的实例方法来运行您的代码。

# functions to call
def sub_fun1(fun1_param1="not"):
    print("fun", fun1_param1)

def sub_fun2(fun2_param1="not"):
    print("too", fun2_param1)

# option 1: use a dict
def wrap_fun(argument_file):
    ns = {}
    exec(open(argument_file).read(), ns)
    print(ns.keys())
    sub_fun1(fun1_param1=ns["a"])

# option 2: use instance dict
class FunRunner:

    def __init__(self, argument_file):
        exec(open(argument_file).read(), self.__dict__)

    def have_some_fun(self):
        sub_fun1(fun1_param1=self.a)

    def have_more_fun(self):
        sub_fun2(fun2_param1=self.b)

# write test file
fn = "my_test_args"
open(fn,"w").write("""a = [1]
b = [2]
c = [3]
d = [4]
e = [5]
f = [6]""")

# option1 
wrap_fun(fn)

# option2 one shot
FunRunner(fn).have_some_fun()

# option2 multiple
fun = FunRunner(fn)
fun.have_some_fun()
fun.have_more_fun()