一、基本概念
装饰器:给函数增加一些额外的功能,但是又不改变函数原来的调用,我们叫做装饰器Decorator,Decorator就是一个返回函数的高阶函数
二、使用场景
基础场景:按钮index = 0的时候进行发图片,index = 1的时候进行发发文字,基本功能实现
#功能函数
def ftp():
print("发图片")
def fwz():
print("发文字")
# 业务逻辑
buttonIndex = 1
if buttonIndex == 0:
ftp()
else:
fwz()
现在有个额外的功能需求,就是在发送之前判断用户是否登录?
要解决这个需求,我们考虑有三种方式解决:
1、直接在代码里面增加一个校验的函数,然后在业务逻辑的时候调用,缺点:就是改变原来的业务逻辑
def checkIsLogin():
print("登录验证......")
def ftp():
print("发图片")
def fwz():
print("发文字")
# 业务逻辑
buttonIndex = 1
if buttonIndex == 0:
checkIsLogin()
ftp()
else:
checkIsLogin()
fwz()
2、增加函数闭包调用,缺点:代码太多,想是否有简便的方法
# 功能函数
def checkIsLogin(func): # 闭包的使用
def inner():
print("登录验证......")
func()
return inner
def ftp():
print("发图片")
def fwz():
print("发文字")
ftp = checkIsLogin(ftp)
fwz = checkIsLogin(fwz)
#业务逻辑
buttonIndex = 1
if buttonIndex == 0:
ftp()
else:
fwz()
3、采用装饰器的语法糖实现
# 功能函数
def checkIsLogin(func): # 闭包的使用
def inner():
print("登录验证......")
func()
return inner
@checkIsLogin # ftp = checkIsLogin(ftp)就相当于@checkIsLogin
def ftp():
print("发图片")
@checkIsLogin # fwz = checkIsLogin(fwz) 就相当于@checkIsLogin
def fwz():
print("发文字")
#业务逻辑
buttonIndex = 1
if buttonIndex == 0:
ftp()
else:
fwz()
三、装饰的常用操作
注意:写出装饰器语法糖的时候就已经执行装饰函数,并返回内部函数啦
1、装饰器装饰有参数的函数
def zsq(func):
def inner(*args,**kwargs):
print("*" * 30)
func(*args,**kwargs)
return inner
@zsq
def pnum(num):
print('打印参数---',num)
@zsq
def pnum2(num1,num2,num3):
print('打印参数---',num1,num2,num3)
pnum2(1122,33,num3=4)
函数执行过程:
@zsq装饰pnum函数之后是指向inner 函数的地址,当执行inner函数的时候,传参数就是innner接受参数 执行func 就是执行pnum函数
2、装饰器自身有参数
因为装饰函数默认都会传入一个函数参数,如果想要让装饰器自身有参数,需要在外层增加一个函数,主要保证回来的是一个装饰器就可以
def getzsq(c):
def zsq(func):
def innner(*args, **kwargs):
print("装饰器")
print(c * 30)
res = func(*args, **kwargs)
return res
return innner
return zsq
@getzsq("===") #主要保证回来的是一个装饰器就可以
def pnum(num):
print(num)
pnum(3)
3、装饰器有返回值
def computeSum(func):
def inner(*args,**kwargs):
print("检测参数")
print(args,kwargs)
res = func(*args,**kwargs)
return res
return inner
@computeSum
def pnum(num1,num2,num3):
return num1 + num2 + num3
@computeSum
def pnum2(num):
print("没有返回值",num)
resut = pnum(1,2,num3=3)
print(resut)
pnum2(2)
函数执行过程:@computeSum函数装饰的时候返回inner函数的地址pnum = computeSum(pnum) ,执行pnum函数 就是执行就是inner函数,执行inner里面的func 才是真正执行pnum()函数,返回值在返回给外界
4、装饰器的叠加
从函数的头部开始 向上装饰 执行 从代码的行数 从上到下
def print_line(func):
print("------print_line-----")
def inner():
print("-----------")
func()
print("-----------")
return inner
def print_star(func):
print("*****print_star******")
def inner():
print("***********")
func()
print("***********")
return inner
@print_line #print_content = print_line(print_content)
@print_star # print_content = print_star(print_content)
def print_content():
print("刘子彬笔记")
print_content()
#执行结果
#*****print_star******
# ------print_line-----
# -----------
# ***********
# 刘子彬笔记
# ***********
# -----------
如果需要了解更多Python知识,请查看
Python的基础知识之生成器
Python的基础知识之装饰器
Python的基础知识之迭代器
最后赠言
学无止境,学习Python的伙伴可以多多交流