动态地向Python模块添加函数


问题内容

我们的框架要求将某些功能包装在一些难看的样板代码中:

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

任何帮助,将不胜感激!


问题答案:

使用之一

current_module.new_name = func

要么

setattr(current_module, new_name, func)