从文本文件中读取python函数并将其分配给变量

Question

我有一个包含Python函数的文本文件，如下所示：

a.txt

def func():
    var = 5
    return var

然后，我用Python脚本读取了此文件：

b.py

python_file = open("a.txt").read()

现在，我想将a.txt文件的函数分配给变量，而不必担心函数名称并执行它。 我尝试过这样的事情：

python_file = open("a.txt").read()
b = exec(python_file)
b()

但这没有用，我也尝试了execfile 。

Answer 1

exec完字符串后，您可以直接调用func ，因为它已添加到当前名称空间中：

>>> exec("""def func():
    var = 5  # note that the semicolons are redundant and unpythonic
    return var""")
>>> func()
5

根据其文档， exec实际上并不返回任何内容，因此分配例如foo = exec(...)毫无意义。

---

要查看正在执行的代码中本地定义的名称，请将一个空字典传递给exec作为locals参数：

>>> ns = {}
>>> exec("""def func():
    var = 5
    return var""", globals(), ns)
>>> ns
{'func': <function func at 0x0315F540>}

然后，您可以像通常那样分配函数并调用它：

>>> b, = ns.values()  # this will only work if only one name was defined
>>> b()
5

Answer 2

在提供我的解决方案之前，我强烈警告您不要这样做，除非您确定a.txt中没有恶意代码。

我的解决方案使用execfile函数加载文本文件并返回第一个对象（可以是变量或函数）：

def load_function(filename):
    """ Assume that filename contains only 1 function """
    global_var = dict()
    execfile(filename, global_var)
    del global_var['__builtins__']
    return next(global_var.itervalues())

# Use it
myfunction = load_function('a.txt')
print myfunction()

更新资料

为了更加谨慎，请像以下所示修改返回行，以使其跳过变量（但是，不能跳过类声明）。

    return next(f for f in global_var.itervalues() if callable(f))

更新2

感谢johnsharpe指出Python 3中没有execfile 。这是一个修改的解决方案，改用exec 。 这次，应在“本地”范围内找到该功能。

def load_function(filename):
    """ Assume that filename contains only 1 function """
    with open(filename) as f:
        file_contents = f.read()
        global_var = dict()
        local_var = dict()
        exec file_contents in global_var, local_var
        return next(f for f in local_var.itervalues() if callable(f))

# Use it
myfunction = load_function('a.txt')
print myfunction()