Python中如何实现装饰器？-Python教程-PHP中文网

Python中如何实现装饰器？

穿越時空

发布： 2025-05-01 12:54:01

原创

292人浏览过

装饰器在python中用于修改或增强函数行为而不改变原函数。实现装饰器的步骤包括：1. 定义装饰器函数，返回包装函数；2. 使用@语法糖应用装饰器；3. 使用functools.wraps保留原函数元数据；4. 实现接受参数的装饰器工厂；5. 考虑使用惰性装饰器延迟执行。

Python中如何实现装饰器？

装饰器在Python中是一个非常强大且灵活的特性，用来在不改变原函数的情况下，修改或增强其行为。它们通常用于日志记录、计时、权限检查等场景。让我带你深入了解一下如何实现装饰器，以及在使用过程中可能遇到的一些问题和最佳实践。

首先，我们来看看基本的装饰器实现方式：

def my_decorator(func):
    def wrapper(*args, **kwargs):
        print("Something is happening before the function is called.")
        result = func(*args, **kwargs)
        print("Something is happening after the function is called.")
        return result
    return wrapper

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

say_hello()

登录后复制

这个例子中，my_decorator 是一个装饰器，它接受一个函数 func 作为参数，并返回一个新的函数 wrapper。wrapper 在调用 func 之前和之后执行一些操作。@my_decorator 语法糖使得 say_hello 函数在定义时就被装饰了。

立即学习“Python免费学习笔记（深入）”；

但在实际使用中，我们可能会遇到一些问题，比如装饰器会改变函数的名称和文档字符串，这会影响代码的可读性和调试：

print(say_hello.__name__)  # 输出: wrapper
print(say_hello.__doc__)   # 输出: None

登录后复制

为了解决这个问题，我们可以使用 functools.wraps 来保留原函数的元数据：

import functools

def my_decorator(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        print("Something is happening before the function is called.")
        result = func(*args, **kwargs)
        print("Something is happening after the function is called.")
        return result
    return wrapper

@my_decorator
def say_hello():
    """This function says hello."""
    print("Hello!")

print(say_hello.__name__)  # 输出: say_hello
print(say_hello.__doc__)   # 输出: This function says hello.

登录后复制

装饰器也可以接受参数，这使得它们更加灵活。例如，我们可以创建一个计时装饰器，允许用户指定计时的单位：

import time
import functools

def timer(unit='seconds'):
    def decorator(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            start_time = time.time()
            result = func(*args, **kwargs)
            end_time = time.time()
            duration = end_time - start_time
            if unit == 'milliseconds':
                duration *= 1000
            print(f"{func.__name__} took {duration:.2f} {unit}")
            return result
        return wrapper
    return decorator

@timer(unit='milliseconds')
def slow_function():
    time.sleep(1)
    print("Function completed")

slow_function()

登录后复制

这个例子展示了如何创建一个接受参数的装饰器工厂 timer，它返回一个装饰器 decorator，然后这个装饰器再返回一个包装函数 wrapper。

在使用装饰器时，还需要注意一些潜在的陷阱。例如，装饰器会在导入模块时立即执行，这可能会导致一些意想不到的问题：

def register(func):
    print(f"Registering {func.__name__}")
    return func

@register
def my_function():
    pass

登录后复制

当你导入包含 my_function 的模块时，register 装饰器会立即执行并打印 "Registering my_function"。如果你希望延迟装饰器的执行，可以使用惰性装饰器：

from functools import partial

def lazy_decorator(decorator):
    def wrapper(func):
        def lazy_func(*args, **kwargs):
            if not hasattr(func, '_decorated'):
                func._decorated = decorator(func)
            return func._decorated(*args, **kwargs)
        return lazy_func
    return wrapper

@lazy_decorator
def register(func):
    print(f"Registering {func.__name__}")
    return func

@register
def my_function():
    pass

my_function()  # 只有在调用时才会执行装饰器

登录后复制

最后，分享一些使用装饰器的最佳实践：