Python装饰器：让你的代码更优雅、更强大

最新推荐文章于 2024-09-27 10:11:28 发布

夜雪一千

最新推荐文章于 2024-09-27 10:11:28 发布

阅读量280

点赞数 4

文章标签： python

本文链接：https://blog.csdn.net/2201_75821470/article/details/141398172

版权

在Python编程中，装饰器（Decorators）是一种强大且优雅的特性，它允许我们在不修改原有函数或类定义的情况下，给函数或方法添加新的功能。装饰器本质上是函数，它们可以接收一个函数作为参数并返回一个新的函数。通过这种方式，我们可以在不改变原有函数签名的情况下，给函数“穿上”一层新的逻辑。本文将带你深入了解Python装饰器的基本概念、用法以及高级应用。

一、装饰器的基本概念

1.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函数的行为，使其在调用前后分别打印了一些信息。

1.2 装饰器语法糖

@符号是Python中用于装饰器的语法糖，它让装饰器的使用变得更加简洁和直观。上面的代码等同于：

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

say_hello = my_decorator(say_hello)

但显然，使用@符号的方式更加Pythonic。

二、装饰器的进阶用法

2.1 带参数的装饰器

如果原函数需要参数，那么装饰器中的wrapper函数也需要相应地接收这些参数，并将它们传递给原函数。

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 greet(name):
    print(f"Hello, {name}!")

greet("Alice")

2.2 装饰器带参数

有时候，我们可能希望装饰器本身也能接收参数。这时，我们可以定义一个外部函数来接收这些参数，并返回一个装饰器函数。

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

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

greet("Bob")