Python装饰器：给你的函数加点“佐料”-CSDN博客

本文链接：https://blog.csdn.net/qq_37806690/article/details/137979843

本文介绍了装饰器如何像‘佐料’一样为函数添加额外功能，包括装饰器的工作原理、实例演示、基本概念、装饰器参数、类装饰器以及常用装饰器如日志记录、性能分析等。通过本文，读者将掌握如何在不修改函数源码的情况下增强其功能。

摘要由CSDN通过智能技术生成

文章目录

装饰器：给你的函数加点“佐料”

想象一下，你有一个做菜的函数（就像 Python 中的函数），比如做一份简单的煎蛋。你想在煎蛋前后做些额外的事情，例如：

煎蛋前: 检查一下冰箱里有没有鸡蛋，如果没有就去超市买。
煎蛋后: 撒点胡椒粉和盐，让味道更好。

当然，你可以在煎蛋函数里面直接写这些额外的步骤，但这样会让函数变得很复杂。装饰器就像一种特殊的 “佐料”，可以让你在不改变煎蛋函数本身的情况下，为它添加这些额外的功能。

装饰器的工作原理

准备 “佐料盒” (装饰器函数): 首先，你需要定义一个装饰器函数，就像准备一个装佐料的盒子。这个盒子会接收你的煎蛋函数作为 “食材”。
在 “佐料盒” 里做些准备: 在装饰器函数里面，你可以定义一些在煎蛋前后要做的事情，比如检查鸡蛋和撒调料。
将 “佐料” 添加到煎蛋上: 当你使用装饰器时 (用 @ 符号)，就像把煎蛋放进 “佐料盒” 里，让它沾上你准备好的 “佐料”，然后返回一个新的、功能更强大的 “煎蛋” (函数)。

装饰器的例子

假设你有一个简单的 make_omelette() 函数，用来做煎蛋：

def make_omelette():
    print("煎蛋中...")
    print("煎蛋完成！")

现在，你想在煎蛋前后添加检查鸡蛋和撒调料的功能。你可以使用装饰器来实现：

def check_eggs(func):
    def wrapper():
        print("检查冰箱里有没有鸡蛋...")
        func()  # 这里调用了原来的煎蛋函数
        print("撒上胡椒粉和盐")
    return wrapper

@check_eggs  # 使用装饰器，就像把煎蛋放进 "佐料盒"
def make_omelette():
    print("煎蛋中...")
    print("煎蛋完成！")

make_omelette()  # 现在煎蛋函数就包含了检查鸡蛋和撒调料的功能

输出：

检查冰箱里有没有鸡蛋...
煎蛋中...
煎蛋完成！
撒上胡椒粉和盐

装饰器 (Decorator) 允许你修改函数或类的行为，而无需更改其原始代码。装饰器本质上是一个函数，它接受另一个函数作为参数，并返回一个新的函数。

基本概念

函数是一等公民: 在 Python 中，函数可以像其他对象一样传递、赋值和返回。
闭包: 闭包是指一个内部函数可以访问其外部函数作用域中的变量。
语法糖: 装饰器语法 @decorator_name 只是 func = decorator_name(func) 的简写形式。

创建装饰器

def my_decorator(func):
    def wrapper(*args, **kwargs):
        # 在执行 func() 之前做一些事情
        print("Before function call")
        result = func(*args, **kwargs)
        # 在执行 func() 之后做一些事情
        print("After function call")
        return result
    return wrapper

@my_decorator
def say_hello(name):
    print(f"Hello, {name}!")

say_hello("World")

输出：

Before function call
Hello, World!
After function call

装饰器参数

装饰器也可以接受参数，例如：

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

@repeat(3)
def greet(name):
    print(f"Hello, {name}!")

greet("Alice")

输出：

Hello, Alice!
Hello, Alice!
Hello, Alice!

类装饰器

装饰器也可以是类，例如：

class LogCalls:
    def __init__(self, func):
        self.func = func

    def __call__(self, *args, **kwargs):
        print(f"Calling function: {self.func.__name__}")
        return self.func(*args, **kwargs)

@LogCalls
def say_hi():
    print("Hi!")

say_hi()

输出：