python装饰器详解

 

在 Python 中，装饰器（decorator）是一种高级功能，用于修改或增强函数或方法的行为。装饰器本质上是一个函数，它接收另一个函数作为参数，并返回一个新的函数或修改后的函数。装饰器常用于代码复用、日志记录、性能测试等场景。

 

基本概念

1. 装饰器的基本语法

装饰器的基本结构如下：

 

def my_decorator(func):

    def wrapper():

        # 在函数调用之前执行的代码

        print("Something is happening before the function is called.")

        func()

        # 在函数调用之后执行的代码

        print("Something is happening after the function is called.")

    return wrapper

 

@my_decorator

def say_hello():

    print("Hello!")

 

say_hello()

在这个例子中，@my_decorator 是一个装饰器的应用，它会在调用 say_hello 函数之前和之后分别打印一些内容。

 

2. 装饰器的工作原理

装饰器的工作原理可以分为以下几个步骤：

 

定义装饰器函数：装饰器是一个函数，它接收一个函数作为参数，并返回一个新的函数。

应用装饰器：使用 @decorator_name 的语法，将装饰器应用到目标函数上。这个过程实际上是将目标函数传递给装饰器函数，并用装饰器函数返回的新函数替代原来的函数。

调用装饰后的函数：装饰器返回的函数可以在原函数之前或之后执行额外的代码。

常见的装饰器模式

1. 带参数的装饰器

如果需要让装饰器接受参数，可以将装饰器再封装一层：

 

python

Copy Code

def repeat(num_times):

    def decorator_repeat(func):

        def wrapper(*args, **kwargs):

            for _ in range(num_times):

                func(*args, **kwargs)

        return wrapper

    return decorator_repeat

 

@repeat(num_times=3)

def greet(name):

    print(f"Hello {name}")

 

greet("Alice")

在这个例子中，repeat 装饰器接受一个参数 num_times，表示要重复调用被装饰的函数的次数。

 

2. 装饰器的类实现

装饰器不仅可以用函数实现，还可以用类实现：

 

python

Copy Code

class MyDecorator:

    def __init__(self, func):

        self.func = func

 

    def __call__(self, *args, **kwargs):

        print("Something is happening before the function is called.")

        result = self.func(*args, **kwargs)

        print("Something is happening after the function is called.")

        return result

 

@MyDecorator

def say_hello():

    print("Hello!")

 

say_hello()

在这个例子中，MyDecorator 是一个装饰器类，它的 __call__ 方法定义了装饰器的行为。

 

内置装饰器

Python 标准库中也提供了一些内置装饰器，例如：

 

@staticmethod：将方法定义为静态方法。

@classmethod：将方法定义为类方法。

@property：将方法定义为属性。

装饰器的实际应用

装饰器在实际开发中非常有用，例如：

 

日志记录：在函数调用时自动记录日志。

权限验证：检查用户是否有权限执行某些操作。

性能测试：测量函数执行的时间。

缓存：缓存函数的结果以提高性能。

总之，装饰器是 Python 中一个强大的工具，能够帮助开发者以更优雅和模块化的方式扩展和修改函数的行为。