Python中的装饰器-CSDN博客

Python中的装饰器

什么是Python中的装饰器
理解装饰器的工作原理
装饰器的高级应用
总结

什么是Python中的装饰器

在Python编程中，装饰器是一种非常有用的特性，它允许我们在不修改原有函数代码的情况下，给函数增加新的功能。装饰器本质上是一个函数，它接收另一个函数作为参数，并返回一个新的函数，这个新函数通常会在执行原函数之前或之后添加一些额外的操作。

理解装饰器的工作原理

装饰器的工作原理可以通过以下步骤来理解：

定义装饰器函数：首先，你需要定义一个装饰器函数，它接收一个函数作为参数。
定义包装函数：在装饰器函数内部，你需要定义一个包装函数（通常称为wrapper），这个函数将调用原始函数，并可以在调用前后添加额外的逻辑。
返回包装函数：装饰器函数最后返回这个包装函数。
使用装饰器：在需要增强的函数定义前使用@符号加上装饰器名称，这样Python解释器会自动将这个函数作为参数传递给装饰器，并用返回的新函数替换原函数。

from functools import wraps

def my_decorator(func):
	@wraps(func)
	def wrapper(*args, **kwargs):
	    # 在这里添加额外的功能
		print("Before calling the function")
		result = func(*args, **kwargs)
		print("After calling the function")
		return result
	return wrapper

@my_decorator
def my_function():
    print("Hello, this is my function.")
    
my_function()

输出结果：

Before calling the function
Hello, this is my function.
After calling the function

装饰器的高级应用

装饰器的应用不仅限于简单的前后操作，它们还可以：

带参数：装饰器可以接收外部参数，这样你可以在装饰器内部根据这些参数来改变装饰器的行为。
类装饰器：除了函数装饰器，你还可以使用类来创建装饰器，这在某些情况下可以使代码更加清晰和易于管理。
保留原函数信息：使用functools.wraps可以保留被装饰函数的元信息，如文档字符串和函数名。

from functools import wraps

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

@repeat(3)
def say_hello():
	print("Hello!")

say_hello()