9.2 函数装饰器与类装饰器：让你的代码更具魔力

欢迎来到我的博客，很高兴能够在这里和您见面！欢迎订阅相关专栏：
工💗重💗hao💗：野老杂谈
⭐️ 全网最全IT互联网公司面试宝典：收集整理全网各大IT互联网公司技术、项目、HR面试真题.
⭐️ AIGC时代的创新与未来：详细讲解AIGC的概念、核心技术、应用领域等内容。
⭐️ 全流程数据技术实战指南：全面讲解从数据采集到数据可视化的整个过程，掌握构建现代化数据平台和数据仓库的核心技术和方法。
⭐️ 构建全面的数据指标体系：通过深入的理论解析、详细的实操步骤和丰富的案例分析，为读者提供系统化的指导，帮助他们构建和应用数据指标体系，提升数据驱动的决策水平。
⭐️《遇见Python：初识、了解与热恋》 ：涵盖了Python学习的基础知识、进阶技巧和实际应用案例，帮助读者从零开始逐步掌握Python的各个方面，并最终能够进行项目开发和解决实际问题。

摘要

装饰器是 Python 中的一种神奇工具，能够在不改变原有代码的基础上，为函数或类添加额外的功能。本文将深入探讨函数装饰器与类装饰器的概念和用法，结合实际案例和幽默故事，帮助读者轻松理解并灵活运用这些编程“魔法”。通过这篇文章，你将学会如何编写自己的装饰器，让代码更加优雅高效。

标签： Python、函数装饰器、类装饰器、代码优化、编程技巧

---

什么是装饰器？

函数装饰器的定义

装饰器，就像一位精明的魔法师，能够悄悄地为你的代码加点料，不需要你改动任何核心代码。说白了，装饰器就是一个返回函数的函数，可以用来修改或扩展函数的行为。

def simple_decorator(func):
    def wrapper():
        print("调用前，准备点燃魔法！")
        func()
        print("调用后，魔法释放完毕！")
    return wrapper

在上面的代码中，simple_decorator 是一个装饰器，它接受一个函数 func 作为参数，并返回一个新的函数 wrapper。这个 wrapper 函数在执行原函数前后添加了一些额外的操作。

故事：魔法师与他的魔法咒语

想象一下，你是一位魔法师，每次施法前，你都会默念一段神秘的咒语，让你的魔法变得更加强大。装饰器就像是这些咒语，帮助你的函数变得更具魔力。

使用函数装饰器

让我们看看如何在函数中使用装饰器：

@simple_decorator
def greet():
    print("Hello, World!")

greet()

输出结果：

调用前，准备点燃魔法！
Hello, World!
调用后，魔法释放完毕！

你会发现，在函数 greet 前面加上 @simple_decorator 后，原来的 greet 函数被“装饰”了，执行时会在打印 “Hello, World!” 之前和之后多了两句额外的打印。

更复杂的装饰器：带参数的装饰器

定制你的魔法

如果你希望装饰器更加灵活，比如根据传入的参数定制不同的行为，该怎么办呢？别担心，带参数的装饰器就是为这种情况量身定做的。

def decorator_with_args(greeting):
    def simple_decorator(func):
        def wrapper():
            print(f"{greeting}，准备开始魔法！")
            func()
            print(f"{greeting}，魔法结束了！")
        return wrapper
    return simple_decorator

这个装饰器和之前的有些类似，但它可以接受一个参数 greeting，根据传入的值来调整输出内容。

@decorator_with_args("你好")
def greet():
    print("Hello, World!")

greet()

输出结果：

你好，准备开始魔法！
Hello, World!
你好，魔法结束了！

故事：不同语言的咒语

就像魔法师在不同国家施法时会根据当地的语言调整咒语一样，带参数的装饰器可以让你为不同的函数定制不同的行为。

类装饰器：另一种魔法

类装饰器的定义

函数装饰器是函数处理函数，而类装饰器是类处理函数。这有点像你不仅仅是给你的魔法棒加点魔法，而是给你整个魔法师的装备加上一层魔力外壳。

class SimpleDecorator:
    def __init__(self, func):
        self.func = func
    
    def __call__(self):
        print("类装饰器调用前")
        self.func()
        print("类装饰器调用后")

在这里，SimpleDecorator 是一个类，它的 __init__ 方法接受一个函数作为参数，并将其存储在实例中。当装饰器作用于一个函数时，类的 __call__ 方法会被调用，这个方法定义了函数的行为。

使用类装饰器

@SimpleDecorator
def greet():
    print("Hello, World!")

greet()

输出结果：

类装饰器调用前
Hello, World!
类装饰器调用后

故事：魔法防护装备

假设你作为一名魔法师，在战斗中不仅仅需要一根魔法棒，还需要一整套装备。这些装备由不同的元素组成，为你提供全方位的保护和增强能力。类装饰器就像是这些魔法装备，它们可以为你的函数或类提供更加复杂和强大的功能。

实际应用：装饰器的魔法应用场景

日志记录

在开发中，日志记录是一个非常常见的需求。你可以用装饰器来自动记录函数的调用情况，而不必手动添加重复的日志代码。

import time

def log_execution_time(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

@log_execution_time
def slow_function():
    time.sleep(2)
    print("执行完成")

slow_function()

输出结果：

执行完成
slow_function 执行时间: 2.0021 秒

权限验证

另一个常见的场景是权限验证，你可以使用装饰器来检查用户是否有权限执行某个操作。

def requires_permission(permission):
    def decorator(func):
        def wrapper(user, *args, **kwargs):
            if user.has_permission(permission):
                return func(user, *args, **kwargs)
            else:
                print(f"用户 {user.name} 无权限执行 {func.__name__}")
        return wrapper
    return decorator

class User:
    def __init__(self, name, permissions):
        self.name = name
        self.permissions = permissions
    
    def has_permission(self, permission):
        return permission in self.permissions

@requires_permission("admin")
def delete_user(user):
    print(f"用户 {user.name} 被删除")

admin_user = User("Admin", ["admin"])
guest_user = User("Guest", [])

delete_user(admin_user)  # 输出: 用户 Admin 被删除
delete_user(guest_user)  # 输出: 用户 Guest 无权限执行 delete_user

故事：魔法安全机制

你是一位魔法学院的院长，拥有通往神秘魔法图书馆的钥匙。只有特定的魔法师才能进入图书馆，装饰器就像是图书馆的门禁系统，检查每位访客是否拥有进入的权限。

多重装饰器：让魔法叠加

给函数加上多重魔法

你可以为一个函数添加多个装饰器，就像给自己叠加多个魔法护盾，获得更强大的保护。

def decorator_one(func):
    def wrapper():
        print("装饰器一开始")
        func()
        print("装饰器一结束")
    return wrapper

def decorator_two(func):
    def wrapper():
        print("装饰器二开始")
        func()
        print("装饰器二结束")
    return wrapper

@decorator_one
@decorator_two
def greet():
    print("Hello, World!")

greet()

输出结果：

装饰器一开始
装饰器二开始
Hello, World!
装饰器二结束
装饰器一结束

故事：双重魔法盾

你是一名强大的魔法师，不仅拥有火焰护盾，还拥有冰霜护盾。当敌人攻击你时，这两种魔法盾叠加在一起，给你提供了双重防护。多重装饰器就像这些护盾，为你的函数提供多层保护。

装饰器的注意事项

保持函数签名

装饰器有时会覆盖原函数的元数据，比如函数名和文档字符串。为了避免这种情况，你可以使用 functools.wraps。

import functools

def simple_decorator(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        print("装饰器开始")
        result = func(*args, **kwargs)
        print("装饰器结束")
        return result
    return wrapper

@simple_decorator
def greet():
    """打印欢迎信息。"""
    print("Hello, World!")

print(greet.__name__)  # 输出: greet
print(greet.__doc__)  # 输出: 打印欢迎信息。

调试难度

装饰器可能会增加

调试的难度，尤其是当装饰器嵌套过多时。为了简化调试，可以在开发过程中暂时禁用某些装饰器，或者使用调试工具来追踪执行流程。

结论：灵活运用装饰器

通过本文的学习，你已经掌握了函数装饰器和类装饰器的基本概念、使用方法以及实际应用场景。装饰器不仅可以让你的代码更加简洁优雅，还能为你带来无限的可能性。希望你能将这些“魔法”运用到你的编程实践中，为你的 Python 代码注入更多的魔力！