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,似乎函数功能又偏离了预期,求教