如何在模块中动态创建 python 函数？

Question

我写了一个函数def_function，它在运行时动态定义了另一个函数。

ma​​in.py

#!/usr/bin/python3

def def_function(name):
  lines = list()
  lines.append('global {}\n'.format(name))
  lines.append('def {}():\n'.format(name))
  lines.append('  print(\'{}() has been called!\')\n'.format(name))
  code = ''.join(lines)
  function = compile(code, '', 'exec')
  exec(function)

def_function('foo')
foo()

运行 ma​​in.py 给出预期的输出：

foo() 已被调用！

现在我想将def_function 的定义移动到module.py 并从main.py 导入。

module.py

def def_function(name):
  lines = list()
  lines.append('global {}\n'.format(name))
  lines.append('def {}():\n'.format(name))
  lines.append('  print(\'{}() has been called!\')\n'.format(name))
  code = ''.join(lines)
  function = compile(code, '', 'exec')
  exec(function)

ma​​in.py

#!/usr/bin/python3

from module import def_function

def_function('foo')
foo()

这会导致以下错误：

Traceback (most recent call last):
  File "./main.py", line 6, in <module>
    foo()
NameError: name 'foo' is not defined

我已经在 SO 上搜索了我的问题并阅读了各种问题，但我找不到解决方案。请帮帮我。

Answer 1

您在module.py 范围内定义foo。不要忘记导入module.py。

致电foo()，使用module.foo()。

import module

module.def_function('foo')
module.foo()

可以帮助你https://realpython.com/python-scope-legb-rule/

附加

在exec(function) 之后添加这个。同时导入sys。

setattr(sys.modules['__main__'], name, globals()[name])

Setattr 函数将值分配给对象的属性或创建新属性。 https://docs.python.org/3/library/functions.html#setattr

module.py

import sys


def def_function(name):
    lines = list()
    lines.append('global {}\n'.format(name))
    lines.append('def {}():\n'.format(name))
    lines.append('  print(\'{}() has been called!\')\n'.format(name))
    code = ''.join(lines)
    function = compile(code, '', 'exec')
    exec(function)
    setattr(sys.modules['__main__'], name, globals()[name])

ma​​in.py

from module import def_function

def_function('foo')
foo()

Answer 2

你的例子可以这样写，有一个嵌套的函数定义，没有compile或exec：

def def_function(name):
    def f():
        print(name, "has been called")
    return f

# ...

foo = def_function("foo")

但是，您必须退回它。