динамическое добавление функций в модуль Python
-
06-07-2019 - |
Вопрос
Наш фреймворк требует обертывания определенных функций в какой-то уродливый шаблонный код:
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)
Другие советы
Похоже, решение вашей проблемы состояло бы в том, чтобы заставить оформленную функцию действовать как исходную функцию.
Попробуйте использовать эту функцию mergeFunctionMetadata
из Twisted, найдено здесь:twisted/python/util.py
Это заставляет вашу оформленную функцию действовать как оригинальную, и, надеюсь, фреймворк подхватит ее.