help with flexible decorators

[james_027]

Hi,

I want to write a flexible decorators to edit a function that may have
1 or more arguments...

def enhance(func):
def new(x):
#do something ...
return func(x)
return new

@enhance
def method_a(x):
#do something ...

While the enhance decorator work with functions of 1 argument, how do
I make it to work with more than one arguments.

Thanks
james

 

[Lawrence Oluyede]

While the enhance decorator work with functions of 1 argument, how do
I make it to work with more than one arguments.

Using *args. Something like this:

def enhance(f):
def _new(*args):
return f(*args) + 1
return _new

@enhance
def f(*args):
return sum(args)

In [22]: f(6)
Out[22]: 7

In [23]: f(6, 4)
Out[23]: 11

In [24]: f(6, 4, 10)
Out[24]: 21

 

[Bruno Desthuilliers]

james_027 a écrit :

Hi,

I want to write a flexible decorators to edit a function that may have
1 or more arguments...

def enhance(func):
def new(x):
#do something ...
return func(x)
return new

@enhance
def method_a(x):
#do something ...

While the enhance decorator work with functions of 1 argument, how do
I make it to work with more than one arguments.

Use *args (for positional args) and **kw (for keyword args):

def enhance(func):
def new(*args, **kw):
#do something ...
return func(*args, **kw)
return new

@enhance
def method_a(x):
#do something ...


HTH