text 接收可选参数的装饰器

Posted 2021-05-02

>>> def optional_arguments_decorator(real_decorator):
...     def decorator(func=None, **kwargs):
            ''' 接收可选参数的装饰器 '''
...         def decorated(func):
                ''' 普通装饰器 '''
...             def wrapper(*a, **kw):
...                 return real_decorator(func, a, kw, **kwargs)
...             return wrapper
...         if func is None:
...             return decorated  # 返回普通装饰器
...         return decorated(func)  # 返回被普通装饰器装饰后的函数
...     return decorator
...
>>> @optional_arguments_decorator
... def decorate(func, args, kwargs, prefix='Decorated'):
...     return '%s: %s' % (prefix, func(*args, **kwargs))
...
>>> @decorate
... def test(a, b):
... return a + b
...
>>> test(13, 17)
'Decorated: 30'
>>> test = decorate(test, prefix='Decorated again')
>>> test(13, 17)
'Decorated again: Decorated: 30'