在函数的代码对象上使用Python exec时如何获取返回值？

Question

出于测试目的,我想直接执行另一个函数内部定义的函数.

我可以通过父函数的代码(func_code)来获取子函数的代码对象,但是当我执行它时,我得不到返回值.

有没有办法从exec的代码中获取返回值？

Answer 1

几年后，但以下片段帮助了我：

the_code = '''
a = 1
b = 2
return_me = a + b
'''

loc = {}
exec(the_code, globals(), loc)
return_workaround = loc['return_me']
print(return_workaround)  # 3

exec()本身不返回任何内容，但您可以传递 a dict，其中在执行后存储了所有局部变量。通过访问它，您可以获得类似回报的东西。

我希望它可以帮助某人。

Answer 2

是的,您需要在exec声明中进行分配:

>>> def foo():
...     return 5
...
>>> exec("a = foo()")
>>> a
5

这可能与您的案例无关,因为它用于受控测试,但要小心使用exec用户定义的输入.

Answer 3

使用eval()代替exec()，它返回结果

Answer 4

虽然这是人类见过的最丑陋的野兽，但您可以通过在 exec 调用中使用全局变量来做到这一点：

def my_exec(code):
    exec('global i; i = %s' % code)
    global i
    return i

这是滥用全局变量来跨越边界获取数据。

>>> my_exec('1 + 2')
3

毋庸置疑，您永远不应该允许任何用户输入用于此功能的输入，因为它会带来极大的安全风险。

Answer 5

像这样的事情可以工作：

def outer():
    def inner(i):
        return i + 10


for f in outer.func_code.co_consts:
    if getattr(f, 'co_name', None) == 'inner':

        inner = type(outer)(f, globals())

        # can also use `types` module for readability:
        # inner = types.FunctionType(f, globals())

        print inner(42) # 52

这个想法是从内部函数中提取代码对象并基于它创建一个新函数。

当内部函数可以包含自由变量时，需要进行额外的工作。您还必须提取它们并传递给最后一个参数 ( closure) 中的函数构造函数。