装饰器(decorators)是 Python 中的一种高级功能,它允许你动态地修改函数或类的行为。
装饰器是一种函数,它接受一个函数作为参数,并返回一个新的函数或修改原来的函数。
装饰器的语法使用 @xxx 来应用在函数或方法上。
基本语法
函数实现返回字符串的长度,装饰器在函数调用前后打印一句话
# 自定义函数名,声明装饰器decorator_new,参数是一个函数名,这里任意写一个参数名
def decorator_new(fun_get_len):# wrapper 是内部函数,它是实际会被调用的新函数,它包裹了原始函数的调用,并在其前后增加了额外的行为。def wrapper(str):print('开始前')# 形参调用,最后执行时调用的是目标函数,这里没有实现过程len = fun_get_len(str)print('结束后')return lenreturn wrapper# 注解使用装饰器,标在要被改变的函数上
@decorator_new
def get_len_new(str):# 这里是原函数的实现过程,装饰器里面调用的就是这个print('在装饰器中执行吗?')len = 0for i in str:len = len + 1return len print('调用原方法')
print(get_len_new('123'))
print('装饰器结束')
输出
调用原方法
开始前
在装饰器中执行吗?
结束后
3
装饰器结束
带参数的装饰器
带参数的装饰器,这里的参数是给装饰器用的,在修改原函数实现过程中使用,而不是指原函数参数
函数实现打印参数,装饰器传入参数表示打印n次
# 自定义装饰器名,带参数的装饰器,参数n是给装饰器用,用在改造原函数上
def fun_arg(n):# 自定义decorator_arg,自定义原函数名funcdef decorator_arg(func):# str是原函数的参数,不是装饰器的参数def wrapper(str):# n是装饰器的参数for i in range(n):func(str)return wrapperreturn decorator_arg# 传入装饰器的参数
@fun_arg(3)
def print_str(str):print(str)# 传入原函数的参数,并调用
print_str('hello')
输出
hello
hello
hello