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