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

python的装饰器

news 来源：原创 2025/5/8 1:36:34

Python 装饰器是一种特殊的函数，它可以接受一个函数作为输入，并返回一个新的函数，用于增强或修改原函数的功能
装饰器在不修改原函数代码的前提下，为函数添加额外的功能，比如日志记录、性能测试、权限验证等

语法

装饰器的语法使用 @ 符号，它通常放在被装饰函数的定义之前：

def my_decorator(func):
    def wrapper():
        print("在函数执行之前做一些事情")
        func()
        print("在函数执行之后做一些事情")
    return wrapper

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

say_hello()

my_decorator 是一个装饰器函数，say_hello 是被装饰的函数
@my_decorator 这一行代码相当于执行了 say_hello = my_decorator(say_hello)

原理

装饰器本质上是一个高阶函数，即一个可以接受函数作为参数并返回函数的函数。当使用 @ 语法应用装饰器时，Python 会自动将被装饰的函数作为参数传递给装饰器函数，并将装饰器函数返回的新函数赋值给原来的函数名

my_decorator 接受 say_hello 函数作为参数，返回一个新的函数 wrapper
当调用 say_hello() 时，实际上调用的是 wrapper()，在 wrapper 函数内部，会先执行一些额外的操作，然后调用原函数 func()（即 say_hello），最后再执行一些其他操作

处理带参数的函数

如果被装饰的函数带有参数，装饰器中的 wrapper 函数也需要能够接受这些参数。可以使用 *args 和 **kwargs ：

def my_decorator(func):
    def wrapper(*args, **kwargs):
        print("在函数执行之前做一些事情")
        result = func(*args, **kwargs)
        print("在函数执行之后做一些事情")
        return result
    return wrapper

@my_decorator
def add(a, b):
    return a + b

result = add(3, 5)
print(result)

wrapper 函数使用 *args 和 **kwargs 来接受任意数量的位置参数和关键字参数，并将它们传递给原函数 func

带参数的装饰器

装饰器本身也可以接受参数。这需要在原有的装饰器基础上再嵌套一层函数：

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

@repeat(3)
def say_hi():
    print("Hi!")

say_hi()