动态向Python模块添加函数

Question

我们的框架需要在某些函数周围加上一些看起来很麻烦的代码：

def prefix_myname_suffix(obj):
    def actual():
        print 'hello world'
    obj.register(actual)
    return obj

我想这可能可以通过一个装饰器来简化：

@register
def myname():
    print 'hello world'

但是，这实际上有点棘手，主要是因为框架在模块级别寻找特定的函数名称模式。

我在装饰器中尝试了以下方法，但没有成功：

current_module = __import__(__name__)
new_name = prefix + func.__name__ + suffix
# method A
current_module[new_name] = func
# method B
func.__name__ = new_name
current_module += func

任何帮助都会很感激！

Answer 1

看起来解决你问题的方法是让被装饰的函数表现得像原来的函数。

你可以试试使用Twisted里的一个函数 mergeFunctionMetadata，可以在这里找到： twisted/python/util.py

这个函数可以让你的装饰函数像原来的函数一样工作，希望这样框架能识别到它。

Answer 2

使用下面的任意一个：

current_module.new_name = func

或者

setattr(current_module, new_name, func)