装饰器是 Python 中一个常用的高级编程技巧,可以动态地修改类或函数的功能。装饰器本质上是一个 Python 函数,可以将一个函数或类作为参数传递给装饰器,然后装饰器返回一个修改后的函数或类。
1、装饰器的基本语法
@decorator_function
def my_function():
# function code
其中,@decorator_function 表示使用 decorator_function 装饰 my_function() 函数。装饰器可以有多个,多个装饰器会按照从上到下的顺序执行。
2、装饰器函数的特点
- 接受一个函数作为参数。
- 返回一个函数对象。
- 内部定义了一个函数,该函数可以执行被修饰的函数并进行一些操作。
3、示例
以下是一个装饰器的示例,用于输出函数执行时间:
import time
def timer(func):
def wrapper(*args, **kwargs):
start_time = time.time()
result = func(*args, **kwargs)
end_time = time.time()
print("Execution time: ", end_time - start_time, "s")
return result
return wrapper
@timer
def my_function():
time.sleep(1)
print("Hello world!")
my_function()
该装饰器函数 timer
接受一个函数作为参数,然后返回一个新的函数 wrapper
。在 wrapper
函数中,记录函数执行的开始和结束时间,并计算时间差,最后输出执行时间。通过@timer
语法,我们可以将timer
装饰器应用到 my_function
函数上。
当运行my_function()
函数时,会先调用 wrapper
函数,然后再执行原来的 my_function
函数。在 wrapper
函数中,我们可以在执行原来函数的前后进行一些额外的操作,比如记录日志、统计执行时间等。
装饰器是 Python 中非常强大和灵活的特性,可以用于很多场景,比如实现权限控制、缓存数据、调试代码等。