Python装饰器(Decorator)函数如何保留被装饰函数的name和docstring

2023-12-14 11:01:11

问题描述

在利用 Python 书写装饰器模式时,Python对设计模式中的装饰器模式具有天然的支持。但要确保装饰器保留被装饰函数的name和docstring时,需要使用包import functools中的@functools.wraps 1

举例说明

使用如下代码:

# testDecorators.py
def float_args_and_return(function):
    def wrapper(*args, **kwargs):
        args = [float(arg) for arg in args]
        return float(function(*args, **kwargs))
    return wrapper

@float_args_and_return
def mean(first, second, *rest):
    """
    Caculate the means
    """
    numbers = (first, second) + rest 
    return sum(numbers) / len(numbers)

if __name__ == "__main__":
    print(help(mean))

运行结果为:
未保留被装饰函数的name和docstring
可以看出,被装饰函数mean__name__和docstring都没被保留。若将上述代码改为:

# testDecorators.py
import functools 
def float_args_and_return(function):
    @functools.wraps(function)
    def wrapper(*args, **kwargs):
        args = [float(arg) for arg in args]
        return float(function(*args, **kwargs))
    return wrapper

@float_args_and_return
def mean(first, second, *rest):
    """
    Caculate the means
    """
    numbers = (first, second) + rest 
    return sum(numbers) / len(numbers)

if __name__ == "__main__":
    print(help(mean))

则运行结果为:
保留被装饰函数的name和docstring
可以看出,使用@functools.wraps ,实现了保留被装饰函数的__name__docstring功能。


  1. Mark Summerfield. Python in Practice: Create Better Programs Using Concurrrency, Libraries, and Patterns. Addison-Wesley, 2014. ??

文章来源:https://blog.csdn.net/RobertChenGuangzhi/article/details/134987870
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。