一文详解如何创建自己的Python装饰器

如何创建自己的Python装饰器

装饰器是 Python 中非常强大的功能之一。Python 装饰器可以在不修改函数的源代码的情况下，动态地修改函数的行为。以下是如何创建自己的 Python 装饰器的详细攻略。

创建装饰器的基本语法

Python 的装饰器实际上是一个函数，它可以接收一个其它函数作为参数并返回一个新的、修改过的函数。

def my_decorator(func):
    def wrapper():
        # 在调用原始函数之前进行额外操作
        print("在调用函数之前进行操作")
        # 调用原始函数
        func()
        # 在调用原始函数之后进行额外操作
        print("在调用函数之后进行操作")

    return wrapper

考虑下面的示例：

def say_hello():
    print("Hello")

say_hello = my_decorator(say_hello)

say_hello()

输出结果将是：

在调用函数之前进行操作
Hello
在调用函数之后进行操作

按照 Python 的惯例，我们可以使用 @ 符号来应用装饰器。

应用装饰器的基本语法

应用装饰器类似于将函数作为参数传递给装饰器，语法如下所示：

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

say_hello()

输出结果将是：

在调用函数之前进行操作
Hello
在调用函数之后进行操作

与前面示例相同，但现在已经使用了 @ 符号来应用装饰器。

装饰器示例1：计时器

以下示例演示了如何创建一个计时器装饰器，用于测量函数的运行时间。

import time

def timer_decorator(func):
    def wrapper():
        start_time = time.time()
        func()
        end_time = time.time()
        print(f"{func.__name__} 函数的运行时间是：{end_time - start_time} 秒")

    return wrapper

@timer_decorator
def my_func():
    time.sleep(2)

my_func()

输出结果将是：

my_func 函数的运行时间是：2.0002264976501465 秒

装饰器示例2：登录验证

以下示例演示了如何创建一个登录验证装饰器，用于验证用户是否已登录。

def login_required(func):
    def wrapper(is_logged_in):
        if is_logged_in:
            func(is_logged_in=is_logged_in)
        else:
            print("请登录以使用此功能")

    return wrapper

@login_required
def protected_function(is_logged_in):
    print("已验证登录用户")

protected_function(True)

输出结果将是：

已验证登录用户

如果 protected_function() 函数的参数 is_logged_in 参数为 False，则输出：

请登录以使用此功能

以上就是创建自己的 Python 装饰器的攻略。通过合理地应用装饰器，可以帮助开发者大大增强 Python 的灵活性和可扩展性。