Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

dynamically adding functions to a Python module

Our framework requires wrapping certain functions in some ugly boilerplate code:

def prefix_myname_suffix(obj):
    def actual():
        print 'hello world'
    obj.register(actual)
    return obj

I figured this might be simplified with a decorator:

@register
def myname():
    print 'hello world'

However, that turned out to be rather tricky, mainly because the framework looks for a certain pattern of function names at module level.

I've tried the following within the decorator, to no avail:

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

Any help would be appreciated!

like image 665
AnC Avatar asked Oct 25 '09 16:10

AnC


1 Answers

use either

current_module.new_name = func

or

setattr(current_module, new_name, func)
like image 163
Oren S Avatar answered Oct 14 '22 21:10

Oren S