深入理解Python装饰器：丰富函数功能的强大工具

最新推荐文章于 2023-12-23 17:53:57 发布

昊昊该干饭了

最新推荐文章于 2023-12-23 17:53:57 发布

阅读量2k

点赞数 22

分类专栏： python 算法文章标签： python 开发语言个人开发

本文链接：https://blog.csdn.net/qq_52213943/article/details/135061631

版权

python 同时被 2 个专栏收录

92 篇文章 11 订阅

订阅专栏

算法

58 篇文章 1 订阅

订阅专栏

导语：装饰器是Python非常强大的功能之一，它们允许程序员修改或增强已有函数或方法的行为，而无需更改其本身的代码，这篇文章，让我们一起来看看（我自己这里理解的也不是很透彻）

----------更正---------

写完这篇文章，敲了不少示例，我感觉我透彻了

装饰器基础

装饰器简介

在Python中，装饰器是一种特殊的函数，它允许修改或增强其他函数或方法的功能，而不需要更改其自身代码。装饰器可以看作是对函数进行包装的包装器，它们在不修改原始函数的情况下，提供了一种灵活的方式来添加新功能。

创建简单的装饰器

装饰器本身是一个函数，它接收一个函数作为参数，并返回一个新的函数。使用装饰器的常见方式是使用@语法，称为语法糖。

示例：一个简单的装饰器

def simple_decorator(func):
    def wrapper():
        print("函数前========")
        func()
        print("函数后========")
    return wrapper

@simple_decorator
def say_hello():
    print("Hello world!")

say_hello()

输出结果：

在这个例子中，simple_decorator是一个装饰器，它在被装饰的函数say_hello执行前后添加了额外的打印语句。

装饰器的工作原理

当你使用@simple_decorator时，Python会执行以下步骤：

将say_hello函数作为参数传递给simple_decorator函数。
simple_decorator函数返回wrapper函数。
say_hello现在指向wrapper函数。

当我调用say_hello()时，实际上是在调用wrapper()函数。

装饰器的使用场景

日志记录：自动记录函数的调用细节。
性能测试：测量和报告函数的执行时间。
访问控制和认证：在函数执行前进行权限校验。
缓存和备忘：存储函数的结果以避免重复计算。

装饰器进阶

带参数的装饰器

在某些情况下，你可能需要让装饰器接受额外的参数。这可以通过在装饰器外层再添加一个函数来实现。

示例：带参数的装饰器

def repeat(num_times):
    def decorator_repeat(func):
        def wrapper(*args, **kwargs):
            for _ in range(num_times):
                value = func(*args, **kwargs)
            return value
        return wrapper
    return decorator_repeat


@repeat(num_times=3)
def greet(name):
    print(f"Hello {name}")


greet("World")

输出结果：

在这个示例中，repeat是一个接受参数的装饰器工厂函数，它返回一个装饰器。这个装饰器随后应用于greet函数。

注意！

为了使装饰器更通用，能够适用于任意参数的函数，通常会在内部函数wrapper中使用*args和**kwargs。

示例：通用装饰器

def debug(func):
    def wrapper(*args, **kwargs):
        args_repr = [repr(a) for a in args]
        kwargs_repr = [f"{k}={v!r}" for k, v in kwargs.items()]
        signature = ", ".join(args_repr + kwargs_repr)
        print(f"===》 {func.__name__}({signature})")
        value = func(*args, **kwargs)
        print(f"{func.__name__!r} 返回的是： {value!r}")
        return value
    return wrapper

@debug
def make_greeting(name, age=None):
    greeting = f"{name}!"
    if age:
        greeting += f" 你今年 {age} 岁了"
    return greeting

print(make_greeting("小明", age=25))

返回结果：

在这个例子中，debug装饰器打印被装饰函数的调用细节，包括函数名称、传递的参数和返回值。

装饰器的嵌套

装饰器可以嵌套使用，这意味着你可以在一个函数上应用多个装饰器，从而将多个功能叠加到该函数上。

示例：装饰器嵌套

#上面省略

@debug
@repeat(num_times=2)
def say_hello(name):
    greeting = f"Hello {name}"
    print(greeting)
    return greeting

say_hello("Alice")

在这个例子中，say_hello函数同时被debug和repeat装饰器装饰。装饰器的应用顺序是从最接近函数定义的装饰器开始向外应用

装饰器的实际应用

装饰器不仅是一个理论概念，它们在实际编程中有着广泛的应用。通过具体的例子，我们可以更好地理解装饰器如何在各种情况下提供优雅的解决方案。

日志记录装饰器

在应用程序中，跟踪函数的调用细节对于调试和性能分析非常重要。装饰器可以用来自动记录函数的调用和返回信息。

示例：日志记录装饰器

在这个例子中，log装饰器记录了函数名、参数和返回值。

import logging

def log(func):
    def wrapper(*args, **kwargs):
        logging.info(f"Running {func.__name__} with arguments {args} and {kwargs}")
        output = func(*args, **kwargs)
        logging.info(f"{func.__name__} returned {output}")
        return output
    return wrapper

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

性能测试装饰器

测量函数的执行时间是性能分析中的常见需求。装饰器可以被用来自动化这个过程。

示例：性能测试装饰器

在这个例子中，timeit装饰器测量并打印了函数的执行时间。

import time

def timeit(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        print(f"{func.__name__} took {end_time - start_time} seconds to run.")
        return result
    return wrapper

@timeit
def do_some_heavy_lifting():
    time.sleep(2)
    return "Task completed"

权限校验和认证

装饰器可以用于在函数执行前进行权限检查，这在Web开发和API设计中尤其有用。

示例：权限校验装饰器

在这个例子中，authorized装饰器检查用户是否具有执行特定函数的权限。

def authorized(roles):
    def decorator(func):
        def wrapper(*args, **kwargs):
            if user.role not in roles:
                raise Exception("Unauthorized")
            return func(*args, **kwargs)
        return wrapper
    return decorator

@authorized(["admin", "user"])
def update_database():
    print("Database updated")