首页 > 其他 > 详细

编写为包装函数添加参数的装饰器

时间:2020-02-04 23:11:27      阅读:76      评论:0      收藏:0      [点我收藏+]

编写一个装饰器,在包装函数的调用签名中添加一个额外的参数。 但是,添加的参数不会干扰该函数的现有调用约定。

额外的参数可以通过关键字参数注入到调用签名中,如下:

from functools import wraps


def optional_debug(func):
    if 'debug' in inspect.getfullargspec(func).args:
        raise TypeError('debug argument already defined')

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


@optional_debug
def spam(a, b, c):
    print(a, b, c)

>>> spam(1, 2, 3)
1 2 3
>>> spam(1, 2, 3, debug=True)
Calling spam
1 2 3

若被包裹的函数已经定义了额外增加的参数,则程序会崩溃。

聪明的人会发现,使用装饰器后,新的函数的签名并没有包含debug参数,如下:

import inspect


@optional_debug
def add(x, y):
    return x + y

>>> print(inspect.signature(add))
(x, y)

这种bug可以通过以下方式修复:

from functools import wraps
import inspect


def optional_debug(func):
    if 'debug' in inspect.getfullargspec(func).args:
        raise TypeError('debug argument already defined')

    @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


@optional_debug
def add(x, y):
    return x + y

>>> print(inspect.signature(add))
(x, y, *, debug=False)

编写为包装函数添加参数的装饰器

原文:https://www.cnblogs.com/jeffrey-yang/p/12261448.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!