python 用装饰器给函数增加参数

代码语言：javascript复制

from functools import wraps
 import inspect
def optional_debug(func):
 if 'debug' in inspect.getargspec(func).args:
 raise TypeError('debug argument already defined')

代码语言：javascript复制

@wraps(func)
def wrapper(*args, debug=False, **kwargs):
    if debug:
        print('Calling', func.__name__)
    return func(*args, **kwargs)

sig = inspect.signature(func)
parms = list(sig.parameters.values())
parms.append(inspect.Parameter('debug',
            inspect.Parameter.KEYWORD_ONLY,
            default=False))
wrapper.__signature__ = sig.replace(parameters=parms)
return wrapper

python 装饰器

0 人点赞