首页 > 后端开发 > Python教程 > `functools.wraps` 如何在 Python 装饰器中保留函数元数据?

`functools.wraps` 如何在 Python 装饰器中保留函数元数据?

Mary-Kate Olsen
发布: 2024-11-30 05:52:22
原创
669 人浏览过

How Does `functools.wraps` Preserve Function Metadata in Python Decorators?

揭开 functools.wraps 的神秘面纱:维护函数元数据

装饰器在扩展 Python 函数的功能方面发挥着至关重要的作用。然而,它们通常以丢失重要的函数元数据为代价,例如名称、文档字符串和参数列表。幸运的是,functools.wraps 作为救世主出现,保留了这些宝贵的信息。

应用装饰器时,原始函数被包装函数替换,导致某些关键细节被掩盖。例如,考虑以下示例:

def logged(func):
    def with_logging(*args, **kwargs):
        print(func.__name__ + " was called")
        return func(*args, **kwargs)
    return with_logging

@logged
def f(x):
    """does some math"""
    return x + x * x
登录后复制

如果执行 print(f.__name__),它将产生“with_logging”,并且 f.__doc__ 将返回一个空字符串。出现这种结果是因为包装函数 with_logging 已经取代了 f.

functools.wraps 通过包装装饰器函数并确保装饰函数的元数据保持完整来解决此问题。通过将 @wraps(func) 应用于内部函数,如下所示,我们恢复了所需的行为:

from functools import wraps
def logged(func):
    @wraps(func)
    def with_logging(*args, **kwargs):
        print(func.__name__ + " was called")
        return func(*args, **kwargs)
    return with_logging

@logged
def f(x):
    """does some math"""
    return x + x * x

print(f.__name__)  # prints 'f'
print(f.__doc__)   # prints 'does some math'
登录后复制

在这个修改后的示例中,f.__name__ 显示“f”,而 f.__doc__ 正确反映原始函数的文档字符串。元数据保存对于维护函数的身份、文档和自省功能至关重要。

因此,请记住 functools.wraps 是装饰器工具包中的重要工具,可防止信息丢失并确保您的装饰函数保留其功能预期属性。

以上是`functools.wraps` 如何在 Python 装饰器中保留函数元数据?的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
作者最新文章
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板