views:

164

answers:

2

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!

+4  A: 

use either

current_module.new_name = func

or

setattr(current_module, new_name, func)
Oren S
Duh, of course - what was I thinking... Thanks!
AnC
+1  A: 

It seems the solution to your problem would be to make the decorated function act as the original function.

Try using the function mergeFunctionMetadata from Twisted, found here: twisted/python/util.py

It makes your decorated function act as the original, hopefully making the framework pick it up.

truppo
Thanks, that looks interesting - will have to look into how best to apply this.
AnC