动态添加函数到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

使用其中之一

current_module.new_name = func

要么

setattr(current_module, new_name, func)