当前位置：首页 > 文章列表 > 文章 > python教程 > Python装饰器从入门到精通，原理+实战一网打尽！

Python装饰器从入门到精通，原理+实战一网打尽！

2025-06-21 18:16:35 0浏览收藏

Python装饰器是一种强大的语法糖，它允许你在不修改函数源代码的前提下，动态地增加函数的功能，例如日志记录、权限验证、缓存和重试机制。本文将深入解析Python装饰器的原理，从定义装饰器函数、内部包装函数到使用@语法应用，一步步揭示其工作方式。装饰器本质上是接收函数作为参数并返回新函数的高阶函数，通过包装原始函数，在函数执行前后添加额外操作。此外，本文还将介绍如何使用functools.wraps保留元数据，以及调试装饰器的技巧，助你彻底掌握这一高级Python概念，显著提升代码的简洁性和可维护性，让你的Python技能更上一层楼。

装饰器是一种语法糖，用于在不修改函数代码的情况下增加功能。1.定义装饰器函数，接收函数作为参数并返回新函数；2.在装饰器内部定义包装函数，执行原始函数及额外操作；3.返回包装函数；4.使用@语法应用装饰器。例如，通过@my_decorator装饰say_hello函数，实现在其执行前后打印信息。装饰器可接受参数，如使用三层嵌套实现函数执行次数控制。常见用途包括日志记录、权限验证、缓存和重试机制。调试时可用functools.wraps保留元数据、插入print语句或使用调试器单步执行。掌握装饰器能显著提升代码简洁性和可维护性。

Python中的装饰器是如何工作的装饰器在Python中有哪些常见用途

装饰器本质上是一种语法糖，它允许你在不修改函数代码的前提下，增加函数的功能。你可以把它想象成给函数穿上了一件“外衣”，这件“外衣”可以在函数执行前后做一些事情，比如记录日志、验证权限、缓存结果等等。

装饰器是一种高级Python概念，理解起来可能需要一些时间，但一旦掌握，你会发现它在代码复用和简化方面非常强大。

解决方案

装饰器的工作方式可以概括为以下几个步骤：

定义装饰器函数： 这个函数接收一个函数作为参数，并返回一个新的函数（通常是一个闭包）。
在装饰器函数内部定义一个包装函数： 这个包装函数会调用原始函数，并在调用前后执行一些额外的操作。
返回包装函数： 装饰器函数返回这个包装函数。
使用@语法应用装饰器： 在需要装饰的函数上方使用@装饰器函数名来应用装饰器。

让我们看一个简单的例子：

def my_decorator(func):
    def wrapper():
        print("在函数执行之前做一些事情")
        func()
        print("在函数执行之后做一些事情")
    return wrapper

@my_decorator
def say_hello():
    print("Hello!")

say_hello()

在这个例子中，my_decorator 是一个装饰器函数，它接收 say_hello 函数作为参数，并返回一个新的函数 wrapper。wrapper 函数在调用 say_hello 函数前后分别打印一些信息。使用 @my_decorator 语法将 my_decorator 应用于 say_hello 函数，实际上等价于 say_hello = my_decorator(say_hello)。

运行这段代码，你会看到以下输出：

在函数执行之前做一些事情
Hello!
在函数执行之后做一些事情

这就是装饰器的基本工作原理。它通过包装原始函数，实现了在不修改原始函数代码的情况下，增加函数功能的目的。

装饰器可以接受参数吗？

当然可以。如果装饰器需要接受参数，我们需要再嵌套一层函数。例如：

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

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

greet("Alice")

在这个例子中，repeat 装饰器接受一个参数 num_times，并使用这个参数来控制 greet 函数的执行次数。注意 wrapper 函数使用了 *args 和 **kwargs 来接收任意数量的位置参数和关键字参数，这样可以确保装饰器可以应用于任何函数。

装饰器在Python中有哪些常见用途

装饰器的用途非常广泛，以下是一些常见的应用场景：

日志记录： 你可以使用装饰器来记录函数的调用信息，例如函数名、参数、执行时间等等。这对于调试和性能分析非常有用。

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__} took {end_time - start_time:.4f} seconds to execute")
        return result
    return wrapper

@log_execution_time
def my_function():
    time.sleep(1)

my_function()

权限验证： 你可以使用装饰器来验证用户是否有权限访问某个函数。例如，你可以检查用户是否已登录，或者是否具有特定的角色。

def requires_login(func):
    def wrapper(*args, **kwargs):
        # 假设有一个函数 is_logged_in() 用来检查用户是否已登录
        if not is_logged_in():
            return "You need to be logged in to access this function."
        return func(*args, **kwargs)
    return wrapper

@requires_login
def my_protected_function():
    return "Access granted!"

print(my_protected_function())

缓存： 你可以使用装饰器来缓存函数的计算结果，避免重复计算。这对于计算量大的函数非常有用。

import functools

def memoize(func):
    cache = {}
    @functools.wraps(func)
    def wrapper(*args):
        if args not in cache:
            cache[args] = func(*args)
        return cache[args]
    return wrapper

@memoize
def fibonacci(n):
    if n < 2:
        return n
    return fibonacci(n-1) + fibonacci(n-2)

print(fibonacci(10))

functools.wraps 是一个装饰器，它可以保留原始函数的元数据，例如函数名、文档字符串等等。这对于调试和代码维护非常重要。

重试机制： 你可以使用装饰器来实现函数的自动重试机制，当函数执行失败时，自动重试几次。

import time

def retry(num_attempts):
    def my_decorator(func):
        def wrapper(*args, **kwargs):
            for i in range(num_attempts):
                try:
                    return func(*args, **kwargs)
                except Exception as e:
                    print(f"Attempt {i+1} failed: {e}")
                    time.sleep(1) # 等待1秒后重试
            print(f"Function {func.__name__} failed after {num_attempts} attempts.")
        return wrapper
    return my_decorator

@retry(num_attempts=3)
def unreliable_function():
    # 模拟一个可能失败的函数
    import random
    if random.random() < 0.5:
        raise Exception("Something went wrong!")
    return "Success!"

print(unreliable_function())

如何调试装饰器

调试装饰器可能会比较棘手，因为装饰器会隐藏原始函数的调用栈。以下是一些调试装饰器的技巧：

使用functools.wraps： 如前所述，functools.wraps 可以保留原始函数的元数据，这使得调试更加容易。
使用print语句： 在装饰器函数和包装函数中插入 print 语句，可以帮助你了解代码的执行流程。
使用调试器： 你可以使用 Python 调试器 (例如 pdb) 来单步执行装饰器代码，并查看变量的值。
```
import pdb

def my_decorator(func):
    def wrapper(*args, **kwargs):
        pdb.set_trace()  # 设置断点
        result = func(*args, **kwargs)
        return result
    return wrapper

@my_decorator
def my_function(x, y):
    return x + y

my_function(1, 2)
```
运行这段代码，当程序执行到 pdb.set_trace() 语句时，调试器会暂停执行，你可以使用调试器命令来查看变量的值，单步执行代码等等。