【Python】用Python实现自己的装饰器库：从基础到进阶

蒙娜丽宁

于 2024-12-01 17:07:09 发布

阅读量926

点赞数 7

CC 4.0 BY-SA版权

分类专栏： Python杂谈文章标签： python 开发语言

本文链接：https://blog.csdn.net/nokiaguy/article/details/144171889

《Python OpenCV从菜鸟到高手》带你进入图像处理与计算机视觉的大门！

装饰器是Python中一项强大的功能，可用于简化代码、提高复用性和增强可读性。在本文中，我们将深入探讨如何设计和实现一套通用的装饰器库，涵盖从基础到进阶的用法。文章包括装饰器的核心原理、常见场景的实现（如性能监控、权限校验、日志记录等），以及如何组织装饰器代码以便于维护和扩展。通过丰富的代码示例和详细解释，读者将学会如何编写、优化和测试自己的装饰器库，为项目开发注入强大的生产力工具。文章包含详细代码和中文注释，适合希望深入掌握装饰器的开发者阅读。

1. 装饰器的基础概念

装饰器本质上是一个高阶函数，它接受一个函数作为输入，并返回一个函数作为输出。它常被用于在不修改原函数代码的情况下，增强函数的功能。

示例代码：一个简单的装饰器

# 一个简单的装饰器
def simple_decorator(func):
    def wrapper():
        print("装饰器开始作用")
        func()
        print("装饰器作用结束")
    return wrapper

# 使用装饰器
@simple_decorator
def my_function():
    print("这是原始函数")

# 调用
my_function()

运行结果：

装饰器开始作用
这是原始函数
装饰器作用结束

工作原理图解

@simple_decorator 会将 my_function 替换为 simple_decorator(my_function)。
simple_decorator 返回一个 wrapper 函数，最终调用 wrapper() 时间接执行了 my_function。

2. 构建简单装饰器

让我们从几个常见的简单装饰器开始，这些装饰器可以在日常开发中频繁使用。

2.1 打印函数执行时间的装饰器

import time

# 计算函数执行时间的装饰器
def timer(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        print(f"函数 {
     
     func.__name__} 执行时间: {
     
     end_time - start_time:.4f} 秒")
        return result
    return wrapper