динамическое добавление функций в модуль Python - PullRequest
10 голосов
/ 25 октября 2009

Наша структура требует включения определенных функций в некрасивый шаблонный код:

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

Любая помощь будет оценена!

Ответы [ 2 ]

15 голосов
/ 25 октября 2009

используйте либо

current_module.new_name = func

или

setattr(current_module, new_name, func)
0 голосов
/ 25 октября 2009

Похоже, что решением вашей проблемы было бы заставить декорированную функцию действовать как оригинальная функция.

Попробуйте использовать функцию mergeFunctionMetadata от Twisted, найденную здесь: витой / питон / util.py

Это делает вашу декорированную функцию похожей на оригинал, и мы надеемся, что фреймворк ее подхватит.

...