如何使用变量 arguments / exec() 返回另一个 function 中的定义的子函数？ (Python 3.x)

Question

我有一个foo function，它返回一个值，然后将其用作bar function 中的输入。 简而言之， baz = bar(foo(•), •) 。

现在，因为我想用一个保持不变的bar来测试多个foo迭代，所以我创建了make_baz function 以便根据迭代返回baz 。 （即baz_i = bar(foo_i(•), •)等）

arguments 和关键字并不总是相同的，我不能像使用scipy.optimize.curve_fit那样简单地传递字典。

我找到了一种理论上做我想做的事情的方法（见下文），但似乎我无法在另一个 function 中定义一个带有exec的 function （它在全球范围内按预期工作）。

def foo_1(x, arg1, arg2, kwarg1=None):
    y = "some function of x"
    return y

def foo_2(x, arg1, kwarg1=None, kwarg2=None):
    y = "some other function of x"
    return y

def bar(y, bar_arg):
    z = "function of y"
    return z

def make_baz(foo):
    args = list(inspect.signature(foo).parameters)
    args_str = ", ".join(args)
    kwargs_str = ", ".join([f"{arg}={arg}" for arg in args])
    baz = f"def baz({args_str}, bar_arg):\n" \
          f"    y = {foo.__name__}({kwargs_str})\n" \
          f"    return bar(y=y, bar_arg=bar_arg)\n"
    exec(baz)
    return baz

print(make_baz(foo_1))
print(make_baz(foo_2))

问题：它返回字符串而不是实际函数。

"""
def baz(x, arg1, arg2, kwarg1, bar_arg):
    y = foo(x=x, arg1=arg1, arg2=arg2, kwarg1=kwarg1)
    return bar(y=y, bar_arg=bar_arg)
"""
"""
def baz(x, arg1, kwarg1, kwarg2, bar_arg):
    y = foo(x=x, arg1=arg1, kwarg1=kwarg1, kwarg2=kwarg2)
    return bar(y=y, bar_arg=bar_arg)
"""

问题：您有任何解决方法/解决方案吗？

Answer 1

该解决方案已在 Python 2 和 Python 3 中的 exec function 的行为中找到，并将其定义loc = locals() exec的参数：（）

def make_baz(foo):
    args = list(inspect.signature(foo).parameters)
    args_str = ", ".join(args)
    kwargs_str = ", ".join([f"{arg}={arg}" for arg in args])
    baz = f"def baz({args_str}, bar_arg):\n" \
          f"    y = foo({kwargs_str})\n" \
          f"    return bar(y=y, bar_arg=bar_arg)\n"
    loc = locals()
    exec(baz, globals(), loc)
    return loc["baz"]

编辑：为了让exec识别bar ，我们必须添加globals()作为参数。