当前位置：首页 > news >正文

Python 装饰器：函数的函数，代码的艺术

news 2025/7/13 12:19:25

引言

在Python中，装饰器是一种强大的功能，允许程序员在不修改原函数源码的情况下增强或修改函数行为。装饰器本质上是一个接收函数作为参数的高阶函数，并返回一个新的函数或修改原函数的行为。这种机制极大地提高了代码的复用性、可读性和模块化程度。

装饰器基础

装饰器的基本语法如下：

def decorator_function(original_function):def wrapper(*args, **kwargs):# 执行一些前置操作print("Before calling the original function.")# 调用原函数result = original_function(*args, **kwargs)# 执行一些后置操作print("After calling the original function.")# 返回原函数的结果return resultreturn wrapper

使用装饰器

要在函数上应用装饰器，只需在函数定义前加上 @decorator_name 即可：

@decorator_function
def target_function(arg1, arg2):print(f"Inside target function with args: {arg1}, {arg2}")

这相当于：

def target_function(arg1, arg2):print(f"Inside target_function with args: {arg1}, {arg2}")target_function = decorator_function(target_function)

带参数的装饰器

装饰器不仅可以作用于函数，还可以带有参数，这使得装饰器本身具有更多的灵活性。下面是一个带参数的装饰器示例，它使函数重复执行指定次数：

def repeat(n):def decorator(func):def wrapper(*args, **kwargs):for _ in range(n):result = func(*args, **kwargs)return resultreturn wrapperreturn decorator@repeat(3)
def greet(name):print(f"Hello, {name}!")greet("Alice")  # 输出 Hello, Alice! 三次

类装饰器

类装饰器是一个包含 __call__ 方法的类。当装饰器被应用时，__call__ 方法被调用，它接收原函数作为参数，并返回新的函数。

class DecoratorClass:def __init__(self, func):self.func = funcdef __call__(self, *args, **kwargs):print("Before calling the original function.")result = self.func(*args, **kwargs)print("After calling the original function.")return result@DecoratorClass
def my_function():print("Inside my_function.")my_function()  # 输出 Before, Inside, After