动态地向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)