欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

idata中的装饰器wrapper如何继承原函数名?(python )

程序员文章站 2022-07-08 14:03:29
下面是python ddt模块的一段源码,请教下idata中的装饰器wrapper如何继承原函数名? def data(*values): ""...

下面是python ddt模块的一段源码,请教下idata中的装饰器wrapper如何继承原函数名?

def data(*values):
    """
    Method decorator to add to your test methods.

    Should be added to methods of instances of ``unittest.TestCase``.

    """
    return idata(values)


def idata(iterable):
    """
    Method decorator to add to your test methods.

    Should be added to methods of instances of ``unittest.TestCase``.

    """
    def wrapper(func):
        setattr(func, DATA_ATTR, iterable)
        return func
    return wrapper #返回的函数名为wrapper,与func传入不一致

按一般写法,wrapper之前应该还有def decorator(func),然后通过@wraps(func)保留原函数名,可是文中参数func直接在wrapper中,该如何处理呢?

已尝试直接添加@wraps(func)会提示func未定义,改写为一般形态decorator,似乎函数功能又偏离了预期,求教